Package ai.djl.mxnet.jna
Interface MxnetLibrary
- All Superinterfaces:
com.sun.jna.Library
public interface MxnetLibrary
extends com.sun.jna.Library
-
Nested Class Summary
Nested ClassesModifier and TypeInterfaceDescriptionstatic interfacestatic interfacestatic enumstatic interfacestatic interfacestatic interfacestatic enumstatic interfacestatic interfacestatic interfacestatic interfacestatic interfacestatic interfacestatic interfacestatic enumstatic interfacestatic interfacestatic interfacestatic interfacestatic interfacestatic interfacestatic interfacestatic interfacestatic interfaceNested classes/interfaces inherited from interface com.sun.jna.Library
com.sun.jna.Library.Handler -
Field Summary
Fields inherited from interface com.sun.jna.Library
OPTION_ALLOW_OBJECTS, OPTION_CALLING_CONVENTION, OPTION_CLASSLOADER, OPTION_FUNCTION_MAPPER, OPTION_INVOCATION_MAPPER, OPTION_OPEN_FLAGS, OPTION_STRING_ENCODING, OPTION_STRUCTURE_ALIGNMENT, OPTION_SYMBOL_PROVIDER, OPTION_TYPE_MAPPER -
Method Summary
Modifier and TypeMethodDescriptionintMXAggregateProfileStatsPrint(String[] out_str, int reset) intMXAggregateProfileStatsPrintEx(String[] out_str, int reset, int format, int sort_by, int ascending) intMXAutogradBackward(int num_output, ai.djl.mxnet.jna.PointerArray output_handles, com.sun.jna.ptr.PointerByReference ograd_handles, int retain_graph) intMXAutogradBackwardEx(int num_output, ai.djl.mxnet.jna.PointerArray output_handles, ai.djl.mxnet.jna.PointerArray ograd_handles, int num_variables, com.sun.jna.ptr.PointerByReference var_handles, int retain_graph, int create_graph, int is_train, com.sun.jna.ptr.PointerByReference grad_handles, com.sun.jna.ptr.PointerByReference grad_stypes) intMXAutogradComputeGradient(int num_output, com.sun.jna.ptr.PointerByReference output_handles) intMXAutogradGetSymbol(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) intintintMXAutogradMarkVariables(int num_var, com.sun.jna.ptr.PointerByReference var_handles, IntBuffer reqs_array, com.sun.jna.ptr.PointerByReference grad_handles) intMXAutogradSetIsRecording(int is_recording, IntBuffer prev) intMXAutogradSetIsTraining(int is_training, IntBuffer prev) intMXCachedOpRegisterOpHook(com.sun.jna.Pointer handle, MxnetLibrary.CachedOpMonitorCallback callback, byte monitor_all) intMXCreateCachedOp(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) intMXCreateCachedOpEx(com.sun.jna.Pointer handle, int num_flags, String[] keys, String[] vals, com.sun.jna.ptr.PointerByReference out) intMXCreateCachedOpEX(com.sun.jna.Pointer handle, int num_flags, String[] keys, String[] vals, com.sun.jna.ptr.PointerByReference out, byte thread_safe) intMXCustomFunctionRecord(int num_inputs, com.sun.jna.ptr.PointerByReference inputs, int num_outputs, com.sun.jna.ptr.PointerByReference outputs, MXCallbackList.ByReference callbacks) intMXCustomOpRegister(String op_type, MxnetLibrary.CustomOpPropCreator creator) intMXDataIterBeforeFirst(com.sun.jna.Pointer handle) intMXDataIterCreateIter(com.sun.jna.Pointer handle, int num_param, String[] keys, String[] vals, com.sun.jna.ptr.PointerByReference out) intMXDataIterFree(com.sun.jna.Pointer handle) intMXDataIterGetData(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) intMXDataIterGetIndex(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out_index, LongBuffer out_size) intMXDataIterGetIterInfo(com.sun.jna.Pointer creator, String[] name, String[] description, IntBuffer num_args, com.sun.jna.ptr.PointerByReference arg_names, com.sun.jna.ptr.PointerByReference arg_type_infos, com.sun.jna.ptr.PointerByReference arg_descriptions) intMXDataIterGetLabel(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) intMXDataIterGetPadNum(com.sun.jna.Pointer handle, IntBuffer pad) intMXDataIterNext(com.sun.jna.Pointer handle, IntBuffer out) intMXDumpProcessProfile(int finished, int profile_process, com.sun.jna.Pointer kvStoreHandle) intMXDumpProfile(int finished) intMXEnginePushAsync(MxnetLibrary.EngineAsyncFunc async_func, com.sun.jna.Pointer func_param, MxnetLibrary.EngineFuncParamDeleter deleter, com.sun.jna.Pointer ctx_handle, com.sun.jna.Pointer const_vars_handle, int num_const_vars, com.sun.jna.Pointer mutable_vars_handle, int num_mutable_vars, com.sun.jna.Pointer prop_handle, int priority, String opr_name, byte wait) intMXEnginePushAsyncND(MxnetLibrary.EngineAsyncFunc async_func, com.sun.jna.Pointer func_param, MxnetLibrary.EngineFuncParamDeleter deleter, com.sun.jna.Pointer ctx_handle, com.sun.jna.ptr.PointerByReference const_nds_handle, int num_const_nds, com.sun.jna.ptr.PointerByReference mutable_nds_handle, int num_mutable_nds, com.sun.jna.Pointer prop_handle, int priority, String opr_name, byte wait) intMXEnginePushSync(MxnetLibrary.EngineSyncFunc sync_func, com.sun.jna.Pointer func_param, MxnetLibrary.EngineFuncParamDeleter deleter, com.sun.jna.Pointer ctx_handle, com.sun.jna.Pointer const_vars_handle, int num_const_vars, com.sun.jna.Pointer mutable_vars_handle, int num_mutable_vars, com.sun.jna.Pointer prop_handle, int priority, String opr_name) intMXEnginePushSyncND(MxnetLibrary.EngineSyncFunc sync_func, com.sun.jna.Pointer func_param, MxnetLibrary.EngineFuncParamDeleter deleter, com.sun.jna.Pointer ctx_handle, com.sun.jna.ptr.PointerByReference const_nds_handle, int num_const_nds, com.sun.jna.ptr.PointerByReference mutable_nds_handle, int num_mutable_nds, com.sun.jna.Pointer prop_handle, int priority, String opr_name) intMXEngineSetBulkSize(int bulk_size, IntBuffer prev_bulk_size) intMXExecutorBackward(com.sun.jna.Pointer handle, int len, com.sun.jna.ptr.PointerByReference head_grads) intMXExecutorBackwardEx(com.sun.jna.Pointer handle, int len, com.sun.jna.ptr.PointerByReference head_grads, int is_train) intMXExecutorBind(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int len, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grad_store, IntBuffer grad_req_type, int aux_states_len, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.ptr.PointerByReference out) intMXExecutorBindEX(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int num_map_keys, String[] map_keys, int[] map_dev_types, int[] map_dev_ids, int len, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grad_store, IntBuffer grad_req_type, int aux_states_len, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec, com.sun.jna.ptr.PointerByReference out) intMXExecutorBindX(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int num_map_keys, String[] map_keys, int[] map_dev_types, int[] map_dev_ids, int len, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grad_store, IntBuffer grad_req_type, int aux_states_len, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.ptr.PointerByReference out) intMXExecutorForward(com.sun.jna.Pointer handle, int is_train) intMXExecutorFree(com.sun.jna.Pointer handle) intMXExecutorGetOptimizedSymbol(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) intMXExecutorOutputs(com.sun.jna.Pointer handle, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out) intMXExecutorPrint(com.sun.jna.Pointer handle, String[] out_str) intMXExecutorReshape(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, int num_map_keys, String[] map_keys, int[] map_dev_types, int[] map_dev_ids, int num_provided_arg_shapes, String[] provided_arg_shape_names, int[] provided_arg_shape_data, int[] provided_arg_shape_idx, IntBuffer num_in_args, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grads, IntBuffer num_aux_states, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec, com.sun.jna.ptr.PointerByReference out) intMXExecutorReshapeEx(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, int num_map_keys, String[] map_keys, int[] map_dev_types, int[] map_dev_ids, int num_provided_arg_shapes, String[] provided_arg_shape_names, int[] provided_arg_shape_data, int[] provided_arg_shape_idx, IntBuffer num_in_args, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grads, IntBuffer num_aux_states, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec, com.sun.jna.ptr.PointerByReference out) intMXExecutorSetMonitorCallback(com.sun.jna.Pointer handle, MxnetLibrary.ExecutorMonitorCallback callback, com.sun.jna.Pointer callback_handle) intMXExecutorSetMonitorCallbackEX(com.sun.jna.Pointer handle, MxnetLibrary.ExecutorMonitorCallback callback, com.sun.jna.Pointer callback_handle, byte monitor_all) intMXExecutorSimpleBind(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int num_g2c_keys, String[] g2c_keys, int[] g2c_dev_types, int[] g2c_dev_ids, int provided_grad_req_list_len, String[] provided_grad_req_names, String[] provided_grad_req_types, int num_provided_arg_shapes, String[] provided_arg_shape_names, int[] provided_arg_shape_data, int[] provided_arg_shape_idx, int num_provided_arg_dtypes, String[] provided_arg_dtype_names, int[] provided_arg_dtypes, int num_provided_arg_stypes, String[] provided_arg_stype_names, int[] provided_arg_stypes, int num_shared_arg_names, String[] shared_arg_name_list, IntBuffer shared_buffer_len, String[] shared_buffer_name_list, com.sun.jna.ptr.PointerByReference shared_buffer_handle_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_name_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_handle_list, IntBuffer num_in_args, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grads, IntBuffer num_aux_states, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec_handle, com.sun.jna.ptr.PointerByReference out) intMXExecutorSimpleBindEx(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int num_g2c_keys, String[] g2c_keys, int[] g2c_dev_types, int[] g2c_dev_ids, int provided_grad_req_list_len, String[] provided_grad_req_names, String[] provided_grad_req_types, int num_provided_arg_shapes, String[] provided_arg_shape_names, int[] provided_arg_shape_data, int[] provided_arg_shape_idx, int num_provided_arg_dtypes, String[] provided_arg_dtype_names, int[] provided_arg_dtypes, int num_provided_arg_stypes, String[] provided_arg_stype_names, int[] provided_arg_stypes, int num_shared_arg_names, String[] shared_arg_name_list, IntBuffer shared_buffer_len, String[] shared_buffer_name_list, com.sun.jna.ptr.PointerByReference shared_buffer_handle_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_name_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_handle_list, IntBuffer num_in_args, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grads, IntBuffer num_aux_states, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec_handle, com.sun.jna.ptr.PointerByReference out) intMXExecutorSimpleBindEx64(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int num_g2c_keys, String[] g2c_keys, int[] g2c_dev_types, int[] g2c_dev_ids, int provided_grad_req_list_len, String[] provided_grad_req_names, String[] provided_grad_req_types, int num_provided_arg_shapes, String[] provided_arg_shape_names, long[] provided_arg_shape_data, int[] provided_arg_shape_idx, int num_provided_arg_dtypes, String[] provided_arg_dtype_names, int[] provided_arg_dtypes, int num_provided_arg_stypes, String[] provided_arg_stype_names, int[] provided_arg_stypes, int num_shared_arg_names, String[] shared_arg_name_list, IntBuffer shared_buffer_len, String[] shared_buffer_name_list, com.sun.jna.ptr.PointerByReference shared_buffer_handle_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_name_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_handle_list, IntBuffer num_in_args, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grads, IntBuffer num_aux_states, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec_handle, com.sun.jna.ptr.PointerByReference out) intMXFreeCachedOp(com.sun.jna.Pointer handle) intMXFuncDescribe(com.sun.jna.Pointer fun, IntBuffer num_use_vars, IntBuffer num_scalars, IntBuffer num_mutate_vars, IntBuffer type_mask) intMXFuncGetInfo(com.sun.jna.Pointer fun, String[] name, String[] description, IntBuffer num_args, com.sun.jna.ptr.PointerByReference arg_names, com.sun.jna.ptr.PointerByReference arg_type_infos, com.sun.jna.ptr.PointerByReference arg_descriptions, String[] return_type) intMXFuncInvoke(com.sun.jna.Pointer fun, com.sun.jna.ptr.PointerByReference use_vars, FloatBuffer scalar_args, com.sun.jna.ptr.PointerByReference mutate_vars) intMXFuncInvokeEx(com.sun.jna.Pointer fun, com.sun.jna.ptr.PointerByReference use_vars, FloatBuffer scalar_args, com.sun.jna.ptr.PointerByReference mutate_vars, int num_params, com.sun.jna.ptr.PointerByReference param_keys, com.sun.jna.ptr.PointerByReference param_vals) intMXGenAtomicSymbolFromSymbol(com.sun.jna.Pointer sym_handle, com.sun.jna.ptr.PointerByReference ret_sym_handle) intMXGenBackendSubgraph(com.sun.jna.Pointer sym_handle, String backend, com.sun.jna.ptr.PointerByReference ret_sym_handle) intMXGetFunction(String name, com.sun.jna.ptr.PointerByReference out) intMXGetGPUCount(IntBuffer out) intMXGetGPUMemoryInformation(int dev, IntBuffer free_mem, IntBuffer total_mem) intMXGetGPUMemoryInformation64(int dev, LongBuffer free_mem, LongBuffer total_mem) intMXGetVersion(IntBuffer out) intMXImperativeInvoke(com.sun.jna.Pointer creator, int num_inputs, ai.djl.mxnet.jna.PointerArray inputs, IntBuffer num_outputs, com.sun.jna.ptr.PointerByReference outputs, int num_params, String[] param_keys, String[] param_vals) intMXImperativeInvokeEx(com.sun.jna.Pointer creator, int num_inputs, ai.djl.mxnet.jna.PointerArray inputs, IntBuffer num_outputs, com.sun.jna.ptr.PointerByReference outputs, int num_params, ai.djl.mxnet.jna.StringArray param_keys, ai.djl.mxnet.jna.StringArray param_vals, com.sun.jna.ptr.PointerByReference out_stypes) intMXInitPSEnv(int num_vars, String[] keys, String[] vals) intMXInvokeCachedOp(com.sun.jna.Pointer handle, int num_inputs, com.sun.jna.Pointer inputs, IntBuffer num_outputs, com.sun.jna.ptr.PointerByReference outputs) intMXInvokeCachedOpEx(com.sun.jna.Pointer handle, int num_inputs, com.sun.jna.Pointer inputs, IntBuffer num_outputs, com.sun.jna.ptr.PointerByReference outputs, com.sun.jna.ptr.PointerByReference out_stypes) intMXIsNumpyShape(IntBuffer curr) intMXKVStoreBarrier(com.sun.jna.Pointer handle) intMXKVStoreBroadcast(com.sun.jna.Pointer handle, int vnum, int[] vkeys, int onum, int[] okeys, com.sun.jna.ptr.PointerByReference vals, com.sun.jna.ptr.PointerByReference outs, int priority) intMXKVStoreBroadcastEx(com.sun.jna.Pointer handle, int vnum, String[] vkeys, int onum, String[] okeys, com.sun.jna.ptr.PointerByReference vals, com.sun.jna.ptr.PointerByReference outs, int priority) intMXKVStoreCreate(String type, com.sun.jna.ptr.PointerByReference out) intMXKVStoreFree(com.sun.jna.Pointer handle) intMXKVStoreGetGroupSize(com.sun.jna.Pointer handle, IntBuffer ret) intMXKVStoreGetNumDeadNode(com.sun.jna.Pointer handle, int node_id, IntBuffer number, int timeout_sec) intMXKVStoreGetRank(com.sun.jna.Pointer handle, IntBuffer ret) intMXKVStoreGetType(com.sun.jna.Pointer handle, String[] type) intMXKVStoreInit(com.sun.jna.Pointer handle, int num, int[] keys, ai.djl.mxnet.jna.PointerArray vals) intMXKVStoreInitEx(com.sun.jna.Pointer handle, int num, String[] keys, ai.djl.mxnet.jna.PointerArray vals) intintintintMXKVStorePull(com.sun.jna.Pointer handle, int num, int[] keys, ai.djl.mxnet.jna.PointerArray vals, int priority) intMXKVStorePullEx(com.sun.jna.Pointer handle, int num, String[] keys, ai.djl.mxnet.jna.PointerArray vals, int priority) intMXKVStorePullRowSparse(com.sun.jna.Pointer handle, int num, int[] keys, com.sun.jna.ptr.PointerByReference vals, com.sun.jna.ptr.PointerByReference row_ids, int priority) intMXKVStorePullRowSparseEx(com.sun.jna.Pointer handle, int num, String[] keys, com.sun.jna.ptr.PointerByReference vals, com.sun.jna.ptr.PointerByReference row_ids, int priority) intMXKVStorePullWithSparse(com.sun.jna.Pointer handle, int num, int[] keys, com.sun.jna.ptr.PointerByReference vals, int priority, byte ignore_sparse) intMXKVStorePullWithSparseEx(com.sun.jna.Pointer handle, int num, String[] keys, com.sun.jna.ptr.PointerByReference vals, int priority, byte ignore_sparse) intMXKVStorePush(com.sun.jna.Pointer handle, int num, int[] keys, ai.djl.mxnet.jna.PointerArray vals, int priority) intMXKVStorePushEx(com.sun.jna.Pointer handle, int num, String[] keys, ai.djl.mxnet.jna.PointerArray vals, int priority) intMXKVStorePushPull(com.sun.jna.Pointer handle, int vnum, int[] vkeys, int onum, int[] okeys, com.sun.jna.ptr.PointerByReference vals, com.sun.jna.ptr.PointerByReference outs, int priority) intMXKVStorePushPullEx(com.sun.jna.Pointer handle, int vnum, String[] vkeys, int onum, String[] okeys, ai.djl.mxnet.jna.PointerArray vals, ai.djl.mxnet.jna.PointerArray outs, int priority) intMXKVStoreRunServer(com.sun.jna.Pointer handle, MxnetLibrary.MXKVStoreServerController controller, com.sun.jna.Pointer controller_handle) intMXKVStoreSendCommmandToServers(com.sun.jna.Pointer handle, int cmd_id, String cmd_body) intMXKVStoreSetBarrierBeforeExit(com.sun.jna.Pointer handle, int barrier_before_exit) intMXKVStoreSetGradientCompression(com.sun.jna.Pointer handle, int num_params, String[] keys, String[] vals) intMXKVStoreSetUpdater(com.sun.jna.Pointer handle, MxnetLibrary.MXKVStoreUpdater updater, com.sun.jna.Pointer updater_handle) intMXKVStoreSetUpdaterEx(com.sun.jna.Pointer handle, MxnetLibrary.MXKVStoreUpdater updater, MxnetLibrary.MXKVStoreStrUpdater str_updater, com.sun.jna.Pointer updater_handle) intMXLibInfoFeatures(com.sun.jna.ptr.PointerByReference libFeature, NativeSizeByReference size) intMXListAllOpNames(IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_array) intMXListDataIters(IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_array) intMXListFunctions(IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_array) intintMXLoadTVMConfig(com.sun.jna.ptr.PointerByReference config) intMXLoadTVMOp(String libpath) intMXNDArrayAt(com.sun.jna.Pointer handle, int idx, com.sun.jna.ptr.PointerByReference out) intMXNDArrayAt64(com.sun.jna.Pointer handle, long idx, com.sun.jna.ptr.PointerByReference out) intMXNDArrayCallDLPackDeleter(com.sun.jna.Pointer dlpack) intMXNDArrayCreate(int[] shape, int ndim, int dev_type, int dev_id, int delay_alloc, com.sun.jna.ptr.PointerByReference out) intMXNDArrayCreateEx(int[] shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, com.sun.jna.ptr.PointerByReference out) intMXNDArrayCreateEx64(long[] shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, com.sun.jna.ptr.PointerByReference out) intMXNDArrayCreateFromSharedMem(int shared_pid, int shared_id, int[] shape, int ndim, int dtype, com.sun.jna.ptr.PointerByReference out) intMXNDArrayCreateFromSharedMemEx(int shared_pid, int shared_id, int[] shape, int ndim, int dtype, com.sun.jna.ptr.PointerByReference out) intMXNDArrayCreateNone(com.sun.jna.ptr.PointerByReference out) intMXNDArrayCreateSparseEx(int storage_type, int[] shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, int num_aux, IntBuffer aux_type, IntBuffer aux_ndims, int[] aux_shape, com.sun.jna.ptr.PointerByReference out) intMXNDArrayCreateSparseEx64(int storage_type, long[] shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, int num_aux, IntBuffer aux_type, IntBuffer aux_ndims, long[] aux_shape, com.sun.jna.ptr.PointerByReference out) intMXNDArrayDetach(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) intMXNDArrayFree(com.sun.jna.Pointer handle) intMXNDArrayFromDLPack(com.sun.jna.Pointer dlpack, com.sun.jna.ptr.PointerByReference out_handle) intMXNDArrayFromDLPackEx(com.sun.jna.Pointer dlpack, byte transient_handle, com.sun.jna.ptr.PointerByReference out_handle) intMXNDArrayGetAuxNDArray(com.sun.jna.Pointer handle, int i, com.sun.jna.ptr.PointerByReference out) intMXNDArrayGetAuxNDArray64(com.sun.jna.Pointer handle, long i, com.sun.jna.ptr.PointerByReference out) intMXNDArrayGetAuxType(com.sun.jna.Pointer handle, int i, IntBuffer out_type) intMXNDArrayGetAuxType64(com.sun.jna.Pointer handle, long i, IntBuffer out_type) intMXNDArrayGetContext(com.sun.jna.Pointer handle, IntBuffer out_dev_type, IntBuffer out_dev_id) intMXNDArrayGetData(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out_pdata) intMXNDArrayGetDataNDArray(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) intMXNDArrayGetDType(com.sun.jna.Pointer handle, IntBuffer out_dtype) intMXNDArrayGetGrad(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) intMXNDArrayGetGradState(com.sun.jna.Pointer handle, IntBuffer out) intMXNDArrayGetShape(com.sun.jna.Pointer handle, IntBuffer out_dim, com.sun.jna.ptr.PointerByReference out_pdata) intMXNDArrayGetShapeEx(com.sun.jna.Pointer handle, IntBuffer out_dim, com.sun.jna.ptr.PointerByReference out_pdata) intMXNDArrayGetShapeEx64(com.sun.jna.Pointer handle, IntBuffer out_dim, com.sun.jna.ptr.PointerByReference out_pdata) intMXNDArrayGetSharedMemHandle(com.sun.jna.Pointer handle, IntBuffer shared_pid, IntBuffer shared_id) intMXNDArrayGetStorageType(com.sun.jna.Pointer handle, IntBuffer out_storage_type) intMXNDArrayLoad(String fname, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_arr, IntBuffer out_name_size, com.sun.jna.ptr.PointerByReference out_names) intMXNDArrayLoadFromBuffer(com.sun.jna.Pointer ndarray_buffer, NativeSize size, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_arr, IntBuffer out_name_size, com.sun.jna.ptr.PointerByReference out_names) intMXNDArrayLoadFromRawBytes(com.sun.jna.Pointer buf, NativeSize size, com.sun.jna.ptr.PointerByReference out) intMXNDArrayReshape(com.sun.jna.Pointer handle, int ndim, IntBuffer dims, com.sun.jna.ptr.PointerByReference out) intMXNDArrayReshape64(com.sun.jna.Pointer handle, int ndim, LongBuffer dims, byte reverse, com.sun.jna.ptr.PointerByReference out) intMXNDArraySave(String fname, int num_args, ai.djl.mxnet.jna.PointerArray args, String[] keys) intMXNDArraySaveRawBytes(com.sun.jna.Pointer handle, NativeSizeByReference out_size, com.sun.jna.ptr.PointerByReference out_buf) intMXNDArraySetGradState(com.sun.jna.Pointer handle, int state) intMXNDArraySlice(com.sun.jna.Pointer handle, int slice_begin, int slice_end, com.sun.jna.ptr.PointerByReference out) intMXNDArraySlice64(com.sun.jna.Pointer handle, long slice_begin, long slice_end, com.sun.jna.ptr.PointerByReference out) intMXNDArraySyncCheckFormat(com.sun.jna.Pointer handle, byte full_check) intMXNDArraySyncCopyFromCPU(com.sun.jna.Pointer handle, com.sun.jna.Pointer data, NativeSize size) intMXNDArraySyncCopyFromNDArray(com.sun.jna.Pointer handle_dst, com.sun.jna.Pointer handle_src, int i) intMXNDArraySyncCopyToCPU(com.sun.jna.Pointer handle, com.sun.jna.Pointer data, NativeSize size) intMXNDArrayToDLPack(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out_dlpack) intintMXNDArrayWaitToRead(com.sun.jna.Pointer handle) intMXNDArrayWaitToWrite(com.sun.jna.Pointer handle) intintMXOptimizeForBackend(com.sun.jna.Pointer sym_handle, String backend_name, int dev_type, com.sun.jna.ptr.PointerByReference ret_sym_handle, int args_len, com.sun.jna.ptr.PointerByReference in_args_handle, int aux_len, com.sun.jna.ptr.PointerByReference in_aux_handle, int num_options, String[] keys, String[] vals, IntBuffer new_args_cnt, com.sun.jna.ptr.PointerByReference new_args_handle, com.sun.jna.ptr.PointerByReference new_arg_names_handle, IntBuffer new_aux_cnt, com.sun.jna.ptr.PointerByReference new_aux_handle, com.sun.jna.ptr.PointerByReference new_aux_names_handle) intMXProcessProfilePause(int paused, int profile_process, com.sun.jna.Pointer kvStoreHandle) intMXProfileAdjustCounter(com.sun.jna.Pointer counter_handle, long value) intMXProfileCreateCounter(com.sun.jna.Pointer domain, String counter_name, com.sun.jna.ptr.PointerByReference out) intMXProfileCreateDomain(String domain, com.sun.jna.ptr.PointerByReference out) intMXProfileCreateEvent(String event_name, com.sun.jna.ptr.PointerByReference out) intMXProfileCreateFrame(com.sun.jna.Pointer domain, String frame_name, com.sun.jna.ptr.PointerByReference out) intMXProfileCreateTask(com.sun.jna.Pointer domain, String task_name, com.sun.jna.ptr.PointerByReference out) intMXProfileDestroyHandle(com.sun.jna.Pointer frame_handle) intMXProfileDurationStart(com.sun.jna.Pointer duration_handle) intMXProfileDurationStop(com.sun.jna.Pointer duration_handle) intMXProfilePause(int paused) intMXProfileSetCounter(com.sun.jna.Pointer counter_handle, long value) intMXProfileSetMarker(com.sun.jna.Pointer domain, String instant_marker_name, String scope) intMXQuantizeSymbol(com.sun.jna.Pointer sym_handle, com.sun.jna.ptr.PointerByReference ret_sym_handle, int[] dev_type, int num_excluded_sym_names, String[] excluded_sym_names, int num_excluded_op_names, String[] excluded_op_names, int num_offline, String[] offline_params, String quantized_dtype, byte calib_quantize, String quantize_mode, String quantize_granularity, IntBuffer out_num_calib_names, com.sun.jna.ptr.PointerByReference out_calib_names) intMXRandomSeed(int seed) intMXRandomSeedContext(int seed, int dev_type, int dev_id) intMXRecordIOReaderCreate(String uri, com.sun.jna.ptr.PointerByReference out) intMXRecordIOReaderFree(com.sun.jna.Pointer handle) intMXRecordIOReaderReadRecord(com.sun.jna.Pointer handle, String buf, NativeSizeByReference size) intMXRecordIOReaderSeek(com.sun.jna.Pointer handle, NativeSize pos) intMXRecordIOReaderTell(com.sun.jna.Pointer handle, NativeSizeByReference pos) intMXRecordIOWriterCreate(String uri, com.sun.jna.ptr.PointerByReference out) intMXRecordIOWriterFree(com.sun.jna.Pointer handle) intMXRecordIOWriterTell(com.sun.jna.Pointer handle, NativeSizeByReference pos) intMXRecordIOWriterWriteRecord(com.sun.jna.Pointer handle, String buf, NativeSize size) intMXReducePrecisionSymbol(com.sun.jna.Pointer sym_handle, com.sun.jna.ptr.PointerByReference ret_sym_handle, int num_args, int[] arg_type_data, int num_ind_ptr, int[] ind_ptr, int[] target_dtype, int cast_optional_params, int num_target_dtype_op_names, int num_fp32_op_names, int num_widest_dtype_op_names, int num_conditional_fp32_op_names, int num_excluded_symbols, int num_model_params, String[] target_dtype_op_names, String[] fp32_op_names, String[] widest_dtype_op_names, String[] conditional_fp32_op_names, String[] excluded_symbols, String[] conditional_param_names, String[] conditional_param_vals, String[] model_param_names, String[] arg_names) intMXRtcCreate(ByteBuffer name, int num_input, int num_output, com.sun.jna.ptr.PointerByReference input_names, com.sun.jna.ptr.PointerByReference output_names, com.sun.jna.ptr.PointerByReference inputs, com.sun.jna.ptr.PointerByReference outputs, ByteBuffer kernel, com.sun.jna.ptr.PointerByReference out) intMXRtcCudaKernelCall(com.sun.jna.Pointer handle, int dev_id, com.sun.jna.ptr.PointerByReference args, int grid_dim_x, int grid_dim_y, int grid_dim_z, int block_dim_x, int block_dim_y, int block_dim_z, int shared_mem) intMXRtcCudaKernelCreate(com.sun.jna.Pointer handle, String name, int num_args, IntBuffer is_ndarray, IntBuffer is_const, IntBuffer arg_types, com.sun.jna.ptr.PointerByReference out) intMXRtcCudaKernelFree(com.sun.jna.Pointer handle) intMXRtcCudaModuleCreate(String source, int num_options, String[] options, int num_exports, String[] exports, com.sun.jna.ptr.PointerByReference out) intMXRtcCudaModuleFree(com.sun.jna.Pointer handle) intMXRtcFree(com.sun.jna.Pointer handle) intMXRtcPush(com.sun.jna.Pointer handle, int num_input, int num_output, com.sun.jna.ptr.PointerByReference inputs, com.sun.jna.ptr.PointerByReference outputs, int gridDimX, int gridDimY, int gridDimZ, int blockDimX, int blockDimY, int blockDimZ) intMXSetCalibTableToQuantizedSymbol(com.sun.jna.Pointer qsym_handle, int num_layers, String[] layer_names, FloatBuffer low_quantiles, FloatBuffer high_quantiles, com.sun.jna.ptr.PointerByReference ret_sym_handle) intMXSetIsNumpyShape(int is_np_shape, IntBuffer prev) intMXSetNumOMPThreads(int thread_num) intMXSetProcessProfilerConfig(int num_params, String[] keys, String[] vals, com.sun.jna.Pointer kvstoreHandle) intMXSetProcessProfilerState(int state, int profile_process, com.sun.jna.Pointer kvStoreHandle) intMXSetProfilerConfig(int num_params, String[] keys, String[] vals) intMXSetProfilerState(int state) intMXShallowCopyNDArray(com.sun.jna.Pointer src, com.sun.jna.ptr.PointerByReference out) intMXShallowCopySymbol(com.sun.jna.Pointer src, com.sun.jna.ptr.PointerByReference out) intMXStorageEmptyCache(int dev_type, int dev_id) intMXSymbolCompose(com.sun.jna.Pointer sym, String name, int num_args, String[] keys, com.sun.jna.ptr.PointerByReference args) intMXSymbolCopy(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) intMXSymbolCreateAtomicSymbol(com.sun.jna.Pointer creator, int num_param, String[] keys, String[] vals, com.sun.jna.ptr.PointerByReference out) intMXSymbolCreateFromFile(String fname, com.sun.jna.ptr.PointerByReference out) intMXSymbolCreateFromJSON(String json, com.sun.jna.ptr.PointerByReference out) intMXSymbolCreateGroup(int num_symbols, com.sun.jna.ptr.PointerByReference symbols, com.sun.jna.ptr.PointerByReference out) intMXSymbolCreateVariable(String name, com.sun.jna.ptr.PointerByReference out) intMXSymbolCutSubgraph(com.sun.jna.Pointer sym, com.sun.jna.ptr.PointerByReference inputs, IntBuffer input_size) intMXSymbolFree(com.sun.jna.Pointer symbol) intMXSymbolGetAtomicSymbolInfo(com.sun.jna.Pointer creator, String[] name, String[] description, IntBuffer num_args, com.sun.jna.ptr.PointerByReference arg_names, com.sun.jna.ptr.PointerByReference arg_type_infos, com.sun.jna.ptr.PointerByReference arg_descriptions, String[] key_var_num_args, String[] return_type) intMXSymbolGetAtomicSymbolName(com.sun.jna.Pointer creator, String[] name) intMXSymbolGetAttr(com.sun.jna.Pointer symbol, String key, String[] out, IntBuffer success) intMXSymbolGetChildren(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) intMXSymbolGetInputSymbols(com.sun.jna.Pointer sym, com.sun.jna.ptr.PointerByReference inputs, IntBuffer input_size) intMXSymbolGetInternals(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) intMXSymbolGetName(com.sun.jna.Pointer symbol, String[] out, IntBuffer success) intMXSymbolGetNumOutputs(com.sun.jna.Pointer symbol, IntBuffer output_count) intMXSymbolGetOutput(com.sun.jna.Pointer symbol, int index, com.sun.jna.ptr.PointerByReference out) intMXSymbolGrad(com.sun.jna.Pointer sym, int num_wrt, String[] wrt, com.sun.jna.ptr.PointerByReference out) intMXSymbolInferShape(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_ind_ptr, int[] arg_shape_data, IntBuffer in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, IntBuffer out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, IntBuffer aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) intMXSymbolInferShapeEx(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_ind_ptr, int[] arg_shape_data, IntBuffer in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, IntBuffer out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, IntBuffer aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) intMXSymbolInferShapeEx64(com.sun.jna.Pointer sym, int num_args, String[] keys, long[] arg_ind_ptr, long[] arg_shape_data, NativeSizeByReference in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, NativeSizeByReference out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, NativeSizeByReference aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) intMXSymbolInferShapePartial(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_ind_ptr, int[] arg_shape_data, IntBuffer in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, IntBuffer out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, IntBuffer aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) intMXSymbolInferShapePartialEx(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_ind_ptr, int[] arg_shape_data, IntBuffer in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, IntBuffer out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, IntBuffer aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) intMXSymbolInferShapePartialEx64(com.sun.jna.Pointer sym, int num_args, String[] keys, long[] arg_ind_ptr, long[] arg_shape_data, NativeSizeByReference in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, NativeSizeByReference out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, NativeSizeByReference aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) intMXSymbolInferType(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_type_data, IntBuffer in_type_size, com.sun.jna.ptr.PointerByReference in_type_data, IntBuffer out_type_size, com.sun.jna.ptr.PointerByReference out_type_data, IntBuffer aux_type_size, com.sun.jna.ptr.PointerByReference aux_type_data, IntBuffer complete) intMXSymbolInferTypePartial(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_type_data, IntBuffer in_type_size, com.sun.jna.ptr.PointerByReference in_type_data, IntBuffer out_type_size, com.sun.jna.ptr.PointerByReference out_type_data, IntBuffer aux_type_size, com.sun.jna.ptr.PointerByReference aux_type_data, IntBuffer complete) intMXSymbolListArguments(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_str_array) intMXSymbolListAtomicSymbolCreators(IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_array) intMXSymbolListAttr(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out) intMXSymbolListAttrShallow(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out) intMXSymbolListAuxiliaryStates(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_str_array) intMXSymbolListOutputs(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_str_array) intMXSymbolPrint(com.sun.jna.Pointer symbol, String[] out_str) intMXSymbolRemoveAmpCast(com.sun.jna.Pointer sym_handle, com.sun.jna.ptr.PointerByReference ret_sym_handle) intMXSymbolSaveToFile(com.sun.jna.Pointer symbol, String fname) intMXSymbolSaveToJSON(com.sun.jna.Pointer symbol, String[] out_json) intMXSymbolSetAttr(com.sun.jna.Pointer symbol, String key, String value) intNNAddControlDeps(com.sun.jna.Pointer handle, com.sun.jna.Pointer src_dep) voidNNAPISetLastError(String msg) intNNGetOpHandle(String op_name, com.sun.jna.ptr.PointerByReference op_out) intNNGetOpInfo(com.sun.jna.Pointer op, String[] real_name, String[] description, IntBuffer num_doc_args, com.sun.jna.ptr.PointerByReference arg_names, com.sun.jna.ptr.PointerByReference arg_type_infos, com.sun.jna.ptr.PointerByReference arg_descriptions, String[] return_type) intNNGraphApplyPasses(com.sun.jna.Pointer src, int num_pass, String[] pass_names, com.sun.jna.ptr.PointerByReference dst) intNNGraphCreate(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference graph) intNNGraphFree(com.sun.jna.Pointer handle) intNNGraphGetJSONAttr(com.sun.jna.Pointer handle, String key, String[] json_out, IntBuffer success) intNNGraphGetSymbol(com.sun.jna.Pointer graph, com.sun.jna.ptr.PointerByReference symbol) intNNGraphSetJSONAttr(com.sun.jna.Pointer handle, String key, String json_value) intNNGraphSetNodeEntryListAttr_(com.sun.jna.Pointer handle, String key, com.sun.jna.Pointer list) intNNListAllOpNames(IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_array) intNNListUniqueOps(IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_array) intNNSymbolCompose(com.sun.jna.Pointer sym, String name, int num_args, String[] keys, com.sun.jna.ptr.PointerByReference args) intNNSymbolCopy(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) intNNSymbolCreateAtomicSymbol(com.sun.jna.Pointer op, int num_param, String[] keys, String[] vals, com.sun.jna.ptr.PointerByReference out) intNNSymbolCreateGroup(int num_symbols, com.sun.jna.ptr.PointerByReference symbols, com.sun.jna.ptr.PointerByReference out) intNNSymbolCreateVariable(String name, com.sun.jna.ptr.PointerByReference out) intNNSymbolFree(com.sun.jna.Pointer symbol) intNNSymbolGetAttr(com.sun.jna.Pointer symbol, String key, String[] out, IntBuffer success) intNNSymbolGetChildren(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) intNNSymbolGetInternals(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) intNNSymbolGetNumOutputs(com.sun.jna.Pointer symbol, IntBuffer output_count) intNNSymbolGetOutput(com.sun.jna.Pointer symbol, int index, com.sun.jna.ptr.PointerByReference out) intNNSymbolListAttrs(com.sun.jna.Pointer symbol, int recursive_option, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out) intNNSymbolListInputNames(com.sun.jna.Pointer symbol, int option, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_str_array) intNNSymbolListInputVariables(com.sun.jna.Pointer symbol, int option, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_sym_array) intNNSymbolListOutputNames(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_str_array) intNNSymbolPrint(com.sun.jna.Pointer symbol, String[] out_str) intNNSymbolSetAttrs(com.sun.jna.Pointer symbol, int num_param, String[] keys, String[] values)
-
Method Details
-
MXGetLastError
String MXGetLastError() -
MXLoadLib
-
MXLibInfoFeatures
-
MXRandomSeed
int MXRandomSeed(int seed) -
MXRandomSeedContext
int MXRandomSeedContext(int seed, int dev_type, int dev_id) -
MXNotifyShutdown
int MXNotifyShutdown() -
MXSetProcessProfilerConfig
-
MXSetProfilerConfig
-
MXSetProcessProfilerState
int MXSetProcessProfilerState(int state, int profile_process, com.sun.jna.Pointer kvStoreHandle) -
MXSetProfilerState
int MXSetProfilerState(int state) -
MXDumpProcessProfile
int MXDumpProcessProfile(int finished, int profile_process, com.sun.jna.Pointer kvStoreHandle) -
MXDumpProfile
int MXDumpProfile(int finished) -
MXAggregateProfileStatsPrint
-
MXAggregateProfileStatsPrintEx
int MXAggregateProfileStatsPrintEx(String[] out_str, int reset, int format, int sort_by, int ascending) -
MXProcessProfilePause
int MXProcessProfilePause(int paused, int profile_process, com.sun.jna.Pointer kvStoreHandle) -
MXProfilePause
int MXProfilePause(int paused) -
MXProfileCreateDomain
-
MXProfileCreateTask
int MXProfileCreateTask(com.sun.jna.Pointer domain, String task_name, com.sun.jna.ptr.PointerByReference out) -
MXProfileCreateFrame
int MXProfileCreateFrame(com.sun.jna.Pointer domain, String frame_name, com.sun.jna.ptr.PointerByReference out) -
MXProfileCreateEvent
-
MXProfileCreateCounter
int MXProfileCreateCounter(com.sun.jna.Pointer domain, String counter_name, com.sun.jna.ptr.PointerByReference out) -
MXProfileDestroyHandle
int MXProfileDestroyHandle(com.sun.jna.Pointer frame_handle) -
MXProfileDurationStart
int MXProfileDurationStart(com.sun.jna.Pointer duration_handle) -
MXProfileDurationStop
int MXProfileDurationStop(com.sun.jna.Pointer duration_handle) -
MXProfileSetCounter
int MXProfileSetCounter(com.sun.jna.Pointer counter_handle, long value) -
MXProfileAdjustCounter
int MXProfileAdjustCounter(com.sun.jna.Pointer counter_handle, long value) -
MXProfileSetMarker
-
MXSetNumOMPThreads
int MXSetNumOMPThreads(int thread_num) -
MXEngineSetBulkSize
-
MXGetGPUCount
-
MXGetGPUMemoryInformation
-
MXGetGPUMemoryInformation64
-
MXGetVersion
-
MXLoadTVMOp
-
MXLoadTVMConfig
int MXLoadTVMConfig(com.sun.jna.ptr.PointerByReference config) -
MXNDArrayCreateNone
int MXNDArrayCreateNone(com.sun.jna.ptr.PointerByReference out) -
MXNDArrayCreate
int MXNDArrayCreate(int[] shape, int ndim, int dev_type, int dev_id, int delay_alloc, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayCreateEx
int MXNDArrayCreateEx(int[] shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayCreateEx64
int MXNDArrayCreateEx64(long[] shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayCreateSparseEx
-
MXNDArrayCreateSparseEx64
-
MXNDArrayLoadFromRawBytes
int MXNDArrayLoadFromRawBytes(com.sun.jna.Pointer buf, NativeSize size, com.sun.jna.ptr.PointerByReference out) -
MXNDArraySaveRawBytes
int MXNDArraySaveRawBytes(com.sun.jna.Pointer handle, NativeSizeByReference out_size, com.sun.jna.ptr.PointerByReference out_buf) -
MXNDArraySave
-
MXNDArrayLoad
-
MXNDArrayLoadFromBuffer
int MXNDArrayLoadFromBuffer(com.sun.jna.Pointer ndarray_buffer, NativeSize size, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_arr, IntBuffer out_name_size, com.sun.jna.ptr.PointerByReference out_names) -
MXNDArraySyncCopyFromCPU
-
MXNDArraySyncCopyToCPU
-
MXNDArraySyncCopyFromNDArray
int MXNDArraySyncCopyFromNDArray(com.sun.jna.Pointer handle_dst, com.sun.jna.Pointer handle_src, int i) -
MXNDArraySyncCheckFormat
int MXNDArraySyncCheckFormat(com.sun.jna.Pointer handle, byte full_check) -
MXNDArrayWaitToRead
int MXNDArrayWaitToRead(com.sun.jna.Pointer handle) -
MXNDArrayWaitToWrite
int MXNDArrayWaitToWrite(com.sun.jna.Pointer handle) -
MXNDArrayWaitAll
int MXNDArrayWaitAll() -
MXNDArrayFree
int MXNDArrayFree(com.sun.jna.Pointer handle) -
MXNDArraySlice
int MXNDArraySlice(com.sun.jna.Pointer handle, int slice_begin, int slice_end, com.sun.jna.ptr.PointerByReference out) -
MXNDArraySlice64
int MXNDArraySlice64(com.sun.jna.Pointer handle, long slice_begin, long slice_end, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayAt
int MXNDArrayAt(com.sun.jna.Pointer handle, int idx, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayAt64
int MXNDArrayAt64(com.sun.jna.Pointer handle, long idx, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayGetStorageType
-
MXNDArrayReshape
int MXNDArrayReshape(com.sun.jna.Pointer handle, int ndim, IntBuffer dims, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayReshape64
int MXNDArrayReshape64(com.sun.jna.Pointer handle, int ndim, LongBuffer dims, byte reverse, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayGetShape
int MXNDArrayGetShape(com.sun.jna.Pointer handle, IntBuffer out_dim, com.sun.jna.ptr.PointerByReference out_pdata) -
MXNDArrayGetShapeEx
int MXNDArrayGetShapeEx(com.sun.jna.Pointer handle, IntBuffer out_dim, com.sun.jna.ptr.PointerByReference out_pdata) -
MXNDArrayGetShapeEx64
int MXNDArrayGetShapeEx64(com.sun.jna.Pointer handle, IntBuffer out_dim, com.sun.jna.ptr.PointerByReference out_pdata) -
MXNDArrayGetData
int MXNDArrayGetData(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out_pdata) -
MXNDArrayToDLPack
int MXNDArrayToDLPack(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out_dlpack) -
MXNDArrayFromDLPack
int MXNDArrayFromDLPack(com.sun.jna.Pointer dlpack, com.sun.jna.ptr.PointerByReference out_handle) -
MXNDArrayFromDLPackEx
int MXNDArrayFromDLPackEx(com.sun.jna.Pointer dlpack, byte transient_handle, com.sun.jna.ptr.PointerByReference out_handle) -
MXNDArrayCallDLPackDeleter
int MXNDArrayCallDLPackDeleter(com.sun.jna.Pointer dlpack) -
MXNDArrayGetDType
-
MXNDArrayGetAuxType
-
MXNDArrayGetAuxType64
-
MXNDArrayGetAuxNDArray
int MXNDArrayGetAuxNDArray(com.sun.jna.Pointer handle, int i, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayGetAuxNDArray64
int MXNDArrayGetAuxNDArray64(com.sun.jna.Pointer handle, long i, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayGetDataNDArray
int MXNDArrayGetDataNDArray(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayGetContext
-
MXNDArrayGetGrad
int MXNDArrayGetGrad(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) -
MXNDArrayDetach
int MXNDArrayDetach(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) -
MXNDArraySetGradState
int MXNDArraySetGradState(com.sun.jna.Pointer handle, int state) -
MXNDArrayGetGradState
-
MXListFunctions
-
MXGetFunction
-
MXFuncGetInfo
-
MXFuncDescribe
-
MXFuncInvoke
int MXFuncInvoke(com.sun.jna.Pointer fun, com.sun.jna.ptr.PointerByReference use_vars, FloatBuffer scalar_args, com.sun.jna.ptr.PointerByReference mutate_vars) -
MXFuncInvokeEx
int MXFuncInvokeEx(com.sun.jna.Pointer fun, com.sun.jna.ptr.PointerByReference use_vars, FloatBuffer scalar_args, com.sun.jna.ptr.PointerByReference mutate_vars, int num_params, com.sun.jna.ptr.PointerByReference param_keys, com.sun.jna.ptr.PointerByReference param_vals) -
MXImperativeInvoke
-
MXImperativeInvokeEx
int MXImperativeInvokeEx(com.sun.jna.Pointer creator, int num_inputs, ai.djl.mxnet.jna.PointerArray inputs, IntBuffer num_outputs, com.sun.jna.ptr.PointerByReference outputs, int num_params, ai.djl.mxnet.jna.StringArray param_keys, ai.djl.mxnet.jna.StringArray param_vals, com.sun.jna.ptr.PointerByReference out_stypes) -
MXAutogradSetIsRecording
-
MXAutogradSetIsTraining
-
MXAutogradIsRecording
-
MXAutogradIsTraining
-
MXIsNumpyShape
-
MXSetIsNumpyShape
-
MXAutogradMarkVariables
int MXAutogradMarkVariables(int num_var, com.sun.jna.ptr.PointerByReference var_handles, IntBuffer reqs_array, com.sun.jna.ptr.PointerByReference grad_handles) -
MXAutogradComputeGradient
int MXAutogradComputeGradient(int num_output, com.sun.jna.ptr.PointerByReference output_handles) -
MXAutogradBackward
int MXAutogradBackward(int num_output, ai.djl.mxnet.jna.PointerArray output_handles, com.sun.jna.ptr.PointerByReference ograd_handles, int retain_graph) -
MXAutogradBackwardEx
int MXAutogradBackwardEx(int num_output, ai.djl.mxnet.jna.PointerArray output_handles, ai.djl.mxnet.jna.PointerArray ograd_handles, int num_variables, com.sun.jna.ptr.PointerByReference var_handles, int retain_graph, int create_graph, int is_train, com.sun.jna.ptr.PointerByReference grad_handles, com.sun.jna.ptr.PointerByReference grad_stypes) -
MXAutogradGetSymbol
int MXAutogradGetSymbol(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) -
MXCreateCachedOp
int MXCreateCachedOp(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) -
MXCreateCachedOpEx
-
MXCreateCachedOpEX
-
MXFreeCachedOp
int MXFreeCachedOp(com.sun.jna.Pointer handle) -
MXInvokeCachedOp
int MXInvokeCachedOp(com.sun.jna.Pointer handle, int num_inputs, com.sun.jna.Pointer inputs, IntBuffer num_outputs, com.sun.jna.ptr.PointerByReference outputs) -
MXInvokeCachedOpEx
int MXInvokeCachedOpEx(com.sun.jna.Pointer handle, int num_inputs, com.sun.jna.Pointer inputs, IntBuffer num_outputs, com.sun.jna.ptr.PointerByReference outputs, com.sun.jna.ptr.PointerByReference out_stypes) -
MXCachedOpRegisterOpHook
int MXCachedOpRegisterOpHook(com.sun.jna.Pointer handle, MxnetLibrary.CachedOpMonitorCallback callback, byte monitor_all) -
MXListAllOpNames
-
MXSymbolListAtomicSymbolCreators
int MXSymbolListAtomicSymbolCreators(IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_array) -
MXSymbolGetAtomicSymbolName
-
MXSymbolGetInputSymbols
int MXSymbolGetInputSymbols(com.sun.jna.Pointer sym, com.sun.jna.ptr.PointerByReference inputs, IntBuffer input_size) -
MXSymbolCutSubgraph
int MXSymbolCutSubgraph(com.sun.jna.Pointer sym, com.sun.jna.ptr.PointerByReference inputs, IntBuffer input_size) -
MXSymbolGetAtomicSymbolInfo
int MXSymbolGetAtomicSymbolInfo(com.sun.jna.Pointer creator, String[] name, String[] description, IntBuffer num_args, com.sun.jna.ptr.PointerByReference arg_names, com.sun.jna.ptr.PointerByReference arg_type_infos, com.sun.jna.ptr.PointerByReference arg_descriptions, String[] key_var_num_args, String[] return_type) -
MXSymbolCreateAtomicSymbol
-
MXSymbolCreateVariable
-
MXSymbolCreateGroup
int MXSymbolCreateGroup(int num_symbols, com.sun.jna.ptr.PointerByReference symbols, com.sun.jna.ptr.PointerByReference out) -
MXSymbolCreateFromFile
-
MXSymbolCreateFromJSON
-
MXSymbolRemoveAmpCast
int MXSymbolRemoveAmpCast(com.sun.jna.Pointer sym_handle, com.sun.jna.ptr.PointerByReference ret_sym_handle) -
MXSymbolSaveToFile
-
MXSymbolSaveToJSON
-
MXSymbolFree
int MXSymbolFree(com.sun.jna.Pointer symbol) -
MXSymbolCopy
int MXSymbolCopy(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) -
MXSymbolPrint
-
MXSymbolGetName
-
MXSymbolGetAttr
-
MXSymbolSetAttr
-
MXSymbolListAttr
int MXSymbolListAttr(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out) -
MXSymbolListAttrShallow
int MXSymbolListAttrShallow(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out) -
MXSymbolListArguments
int MXSymbolListArguments(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_str_array) -
MXSymbolListOutputs
int MXSymbolListOutputs(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_str_array) -
MXSymbolGetNumOutputs
-
MXSymbolGetInternals
int MXSymbolGetInternals(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) -
MXSymbolGetChildren
int MXSymbolGetChildren(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) -
MXSymbolGetOutput
int MXSymbolGetOutput(com.sun.jna.Pointer symbol, int index, com.sun.jna.ptr.PointerByReference out) -
MXSymbolListAuxiliaryStates
int MXSymbolListAuxiliaryStates(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_str_array) -
MXSymbolCompose
-
MXSymbolGrad
int MXSymbolGrad(com.sun.jna.Pointer sym, int num_wrt, String[] wrt, com.sun.jna.ptr.PointerByReference out) -
MXSymbolInferShape
int MXSymbolInferShape(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_ind_ptr, int[] arg_shape_data, IntBuffer in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, IntBuffer out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, IntBuffer aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) -
MXSymbolInferShapeEx
int MXSymbolInferShapeEx(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_ind_ptr, int[] arg_shape_data, IntBuffer in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, IntBuffer out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, IntBuffer aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) -
MXSymbolInferShapeEx64
int MXSymbolInferShapeEx64(com.sun.jna.Pointer sym, int num_args, String[] keys, long[] arg_ind_ptr, long[] arg_shape_data, NativeSizeByReference in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, NativeSizeByReference out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, NativeSizeByReference aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) -
MXSymbolInferShapePartial
int MXSymbolInferShapePartial(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_ind_ptr, int[] arg_shape_data, IntBuffer in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, IntBuffer out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, IntBuffer aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) -
MXSymbolInferShapePartialEx
int MXSymbolInferShapePartialEx(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_ind_ptr, int[] arg_shape_data, IntBuffer in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, IntBuffer out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, IntBuffer aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) -
MXSymbolInferShapePartialEx64
int MXSymbolInferShapePartialEx64(com.sun.jna.Pointer sym, int num_args, String[] keys, long[] arg_ind_ptr, long[] arg_shape_data, NativeSizeByReference in_shape_size, com.sun.jna.ptr.PointerByReference in_shape_ndim, com.sun.jna.ptr.PointerByReference in_shape_data, NativeSizeByReference out_shape_size, com.sun.jna.ptr.PointerByReference out_shape_ndim, com.sun.jna.ptr.PointerByReference out_shape_data, NativeSizeByReference aux_shape_size, com.sun.jna.ptr.PointerByReference aux_shape_ndim, com.sun.jna.ptr.PointerByReference aux_shape_data, IntBuffer complete) -
MXSymbolInferType
int MXSymbolInferType(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_type_data, IntBuffer in_type_size, com.sun.jna.ptr.PointerByReference in_type_data, IntBuffer out_type_size, com.sun.jna.ptr.PointerByReference out_type_data, IntBuffer aux_type_size, com.sun.jna.ptr.PointerByReference aux_type_data, IntBuffer complete) -
MXSymbolInferTypePartial
int MXSymbolInferTypePartial(com.sun.jna.Pointer sym, int num_args, String[] keys, int[] arg_type_data, IntBuffer in_type_size, com.sun.jna.ptr.PointerByReference in_type_data, IntBuffer out_type_size, com.sun.jna.ptr.PointerByReference out_type_data, IntBuffer aux_type_size, com.sun.jna.ptr.PointerByReference aux_type_data, IntBuffer complete) -
MXQuantizeSymbol
int MXQuantizeSymbol(com.sun.jna.Pointer sym_handle, com.sun.jna.ptr.PointerByReference ret_sym_handle, int[] dev_type, int num_excluded_sym_names, String[] excluded_sym_names, int num_excluded_op_names, String[] excluded_op_names, int num_offline, String[] offline_params, String quantized_dtype, byte calib_quantize, String quantize_mode, String quantize_granularity, IntBuffer out_num_calib_names, com.sun.jna.ptr.PointerByReference out_calib_names) -
MXReducePrecisionSymbol
int MXReducePrecisionSymbol(com.sun.jna.Pointer sym_handle, com.sun.jna.ptr.PointerByReference ret_sym_handle, int num_args, int[] arg_type_data, int num_ind_ptr, int[] ind_ptr, int[] target_dtype, int cast_optional_params, int num_target_dtype_op_names, int num_fp32_op_names, int num_widest_dtype_op_names, int num_conditional_fp32_op_names, int num_excluded_symbols, int num_model_params, String[] target_dtype_op_names, String[] fp32_op_names, String[] widest_dtype_op_names, String[] conditional_fp32_op_names, String[] excluded_symbols, String[] conditional_param_names, String[] conditional_param_vals, String[] model_param_names, String[] arg_names) -
MXSetCalibTableToQuantizedSymbol
int MXSetCalibTableToQuantizedSymbol(com.sun.jna.Pointer qsym_handle, int num_layers, String[] layer_names, FloatBuffer low_quantiles, FloatBuffer high_quantiles, com.sun.jna.ptr.PointerByReference ret_sym_handle) -
MXGenBackendSubgraph
int MXGenBackendSubgraph(com.sun.jna.Pointer sym_handle, String backend, com.sun.jna.ptr.PointerByReference ret_sym_handle) -
MXGenAtomicSymbolFromSymbol
int MXGenAtomicSymbolFromSymbol(com.sun.jna.Pointer sym_handle, com.sun.jna.ptr.PointerByReference ret_sym_handle) -
MXOptimizeForBackend
int MXOptimizeForBackend(com.sun.jna.Pointer sym_handle, String backend_name, int dev_type, com.sun.jna.ptr.PointerByReference ret_sym_handle, int args_len, com.sun.jna.ptr.PointerByReference in_args_handle, int aux_len, com.sun.jna.ptr.PointerByReference in_aux_handle, int num_options, String[] keys, String[] vals, IntBuffer new_args_cnt, com.sun.jna.ptr.PointerByReference new_args_handle, com.sun.jna.ptr.PointerByReference new_arg_names_handle, IntBuffer new_aux_cnt, com.sun.jna.ptr.PointerByReference new_aux_handle, com.sun.jna.ptr.PointerByReference new_aux_names_handle) -
MXExecutorFree
int MXExecutorFree(com.sun.jna.Pointer handle) -
MXExecutorPrint
-
MXExecutorForward
int MXExecutorForward(com.sun.jna.Pointer handle, int is_train) -
MXExecutorBackward
int MXExecutorBackward(com.sun.jna.Pointer handle, int len, com.sun.jna.ptr.PointerByReference head_grads) -
MXExecutorBackwardEx
int MXExecutorBackwardEx(com.sun.jna.Pointer handle, int len, com.sun.jna.ptr.PointerByReference head_grads, int is_train) -
MXExecutorOutputs
int MXExecutorOutputs(com.sun.jna.Pointer handle, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out) -
MXExecutorBind
int MXExecutorBind(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int len, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grad_store, IntBuffer grad_req_type, int aux_states_len, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.ptr.PointerByReference out) -
MXExecutorBindX
int MXExecutorBindX(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int num_map_keys, String[] map_keys, int[] map_dev_types, int[] map_dev_ids, int len, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grad_store, IntBuffer grad_req_type, int aux_states_len, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.ptr.PointerByReference out) -
MXExecutorBindEX
int MXExecutorBindEX(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int num_map_keys, String[] map_keys, int[] map_dev_types, int[] map_dev_ids, int len, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grad_store, IntBuffer grad_req_type, int aux_states_len, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec, com.sun.jna.ptr.PointerByReference out) -
MXExecutorSimpleBind
int MXExecutorSimpleBind(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int num_g2c_keys, String[] g2c_keys, int[] g2c_dev_types, int[] g2c_dev_ids, int provided_grad_req_list_len, String[] provided_grad_req_names, String[] provided_grad_req_types, int num_provided_arg_shapes, String[] provided_arg_shape_names, int[] provided_arg_shape_data, int[] provided_arg_shape_idx, int num_provided_arg_dtypes, String[] provided_arg_dtype_names, int[] provided_arg_dtypes, int num_provided_arg_stypes, String[] provided_arg_stype_names, int[] provided_arg_stypes, int num_shared_arg_names, String[] shared_arg_name_list, IntBuffer shared_buffer_len, String[] shared_buffer_name_list, com.sun.jna.ptr.PointerByReference shared_buffer_handle_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_name_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_handle_list, IntBuffer num_in_args, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grads, IntBuffer num_aux_states, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec_handle, com.sun.jna.ptr.PointerByReference out) -
MXExecutorSimpleBindEx
int MXExecutorSimpleBindEx(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int num_g2c_keys, String[] g2c_keys, int[] g2c_dev_types, int[] g2c_dev_ids, int provided_grad_req_list_len, String[] provided_grad_req_names, String[] provided_grad_req_types, int num_provided_arg_shapes, String[] provided_arg_shape_names, int[] provided_arg_shape_data, int[] provided_arg_shape_idx, int num_provided_arg_dtypes, String[] provided_arg_dtype_names, int[] provided_arg_dtypes, int num_provided_arg_stypes, String[] provided_arg_stype_names, int[] provided_arg_stypes, int num_shared_arg_names, String[] shared_arg_name_list, IntBuffer shared_buffer_len, String[] shared_buffer_name_list, com.sun.jna.ptr.PointerByReference shared_buffer_handle_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_name_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_handle_list, IntBuffer num_in_args, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grads, IntBuffer num_aux_states, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec_handle, com.sun.jna.ptr.PointerByReference out) -
MXExecutorSimpleBindEx64
int MXExecutorSimpleBindEx64(com.sun.jna.Pointer symbol_handle, int dev_type, int dev_id, int num_g2c_keys, String[] g2c_keys, int[] g2c_dev_types, int[] g2c_dev_ids, int provided_grad_req_list_len, String[] provided_grad_req_names, String[] provided_grad_req_types, int num_provided_arg_shapes, String[] provided_arg_shape_names, long[] provided_arg_shape_data, int[] provided_arg_shape_idx, int num_provided_arg_dtypes, String[] provided_arg_dtype_names, int[] provided_arg_dtypes, int num_provided_arg_stypes, String[] provided_arg_stype_names, int[] provided_arg_stypes, int num_shared_arg_names, String[] shared_arg_name_list, IntBuffer shared_buffer_len, String[] shared_buffer_name_list, com.sun.jna.ptr.PointerByReference shared_buffer_handle_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_name_list, com.sun.jna.ptr.PointerByReference updated_shared_buffer_handle_list, IntBuffer num_in_args, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grads, IntBuffer num_aux_states, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec_handle, com.sun.jna.ptr.PointerByReference out) -
MXExecutorReshape
int MXExecutorReshape(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, int num_map_keys, String[] map_keys, int[] map_dev_types, int[] map_dev_ids, int num_provided_arg_shapes, String[] provided_arg_shape_names, int[] provided_arg_shape_data, int[] provided_arg_shape_idx, IntBuffer num_in_args, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grads, IntBuffer num_aux_states, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec, com.sun.jna.ptr.PointerByReference out) -
MXExecutorReshapeEx
int MXExecutorReshapeEx(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, int num_map_keys, String[] map_keys, int[] map_dev_types, int[] map_dev_ids, int num_provided_arg_shapes, String[] provided_arg_shape_names, int[] provided_arg_shape_data, int[] provided_arg_shape_idx, IntBuffer num_in_args, com.sun.jna.ptr.PointerByReference in_args, com.sun.jna.ptr.PointerByReference arg_grads, IntBuffer num_aux_states, com.sun.jna.ptr.PointerByReference aux_states, com.sun.jna.Pointer shared_exec, com.sun.jna.ptr.PointerByReference out) -
MXExecutorGetOptimizedSymbol
int MXExecutorGetOptimizedSymbol(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) -
MXExecutorSetMonitorCallback
int MXExecutorSetMonitorCallback(com.sun.jna.Pointer handle, MxnetLibrary.ExecutorMonitorCallback callback, com.sun.jna.Pointer callback_handle) -
MXExecutorSetMonitorCallbackEX
int MXExecutorSetMonitorCallbackEX(com.sun.jna.Pointer handle, MxnetLibrary.ExecutorMonitorCallback callback, com.sun.jna.Pointer callback_handle, byte monitor_all) -
MXListDataIters
-
MXDataIterCreateIter
-
MXDataIterGetIterInfo
-
MXDataIterFree
int MXDataIterFree(com.sun.jna.Pointer handle) -
MXDataIterNext
-
MXDataIterBeforeFirst
int MXDataIterBeforeFirst(com.sun.jna.Pointer handle) -
MXDataIterGetData
int MXDataIterGetData(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) -
MXDataIterGetIndex
int MXDataIterGetIndex(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out_index, LongBuffer out_size) -
MXDataIterGetPadNum
-
MXDataIterGetLabel
int MXDataIterGetLabel(com.sun.jna.Pointer handle, com.sun.jna.ptr.PointerByReference out) -
MXInitPSEnv
-
MXKVStoreCreate
-
MXKVStoreSetGradientCompression
-
MXKVStoreFree
int MXKVStoreFree(com.sun.jna.Pointer handle) -
MXKVStoreInit
int MXKVStoreInit(com.sun.jna.Pointer handle, int num, int[] keys, ai.djl.mxnet.jna.PointerArray vals) -
MXKVStoreInitEx
int MXKVStoreInitEx(com.sun.jna.Pointer handle, int num, String[] keys, ai.djl.mxnet.jna.PointerArray vals) -
MXKVStorePush
int MXKVStorePush(com.sun.jna.Pointer handle, int num, int[] keys, ai.djl.mxnet.jna.PointerArray vals, int priority) -
MXKVStorePushEx
int MXKVStorePushEx(com.sun.jna.Pointer handle, int num, String[] keys, ai.djl.mxnet.jna.PointerArray vals, int priority) -
MXKVStorePullWithSparse
int MXKVStorePullWithSparse(com.sun.jna.Pointer handle, int num, int[] keys, com.sun.jna.ptr.PointerByReference vals, int priority, byte ignore_sparse) -
MXKVStorePullWithSparseEx
int MXKVStorePullWithSparseEx(com.sun.jna.Pointer handle, int num, String[] keys, com.sun.jna.ptr.PointerByReference vals, int priority, byte ignore_sparse) -
MXKVStorePull
int MXKVStorePull(com.sun.jna.Pointer handle, int num, int[] keys, ai.djl.mxnet.jna.PointerArray vals, int priority) -
MXKVStorePullEx
int MXKVStorePullEx(com.sun.jna.Pointer handle, int num, String[] keys, ai.djl.mxnet.jna.PointerArray vals, int priority) -
MXKVStorePullRowSparse
int MXKVStorePullRowSparse(com.sun.jna.Pointer handle, int num, int[] keys, com.sun.jna.ptr.PointerByReference vals, com.sun.jna.ptr.PointerByReference row_ids, int priority) -
MXKVStorePullRowSparseEx
int MXKVStorePullRowSparseEx(com.sun.jna.Pointer handle, int num, String[] keys, com.sun.jna.ptr.PointerByReference vals, com.sun.jna.ptr.PointerByReference row_ids, int priority) -
MXKVStoreBroadcast
int MXKVStoreBroadcast(com.sun.jna.Pointer handle, int vnum, int[] vkeys, int onum, int[] okeys, com.sun.jna.ptr.PointerByReference vals, com.sun.jna.ptr.PointerByReference outs, int priority) -
MXKVStoreBroadcastEx
-
MXKVStorePushPull
int MXKVStorePushPull(com.sun.jna.Pointer handle, int vnum, int[] vkeys, int onum, int[] okeys, com.sun.jna.ptr.PointerByReference vals, com.sun.jna.ptr.PointerByReference outs, int priority) -
MXKVStorePushPullEx
-
MXKVStoreSetUpdater
int MXKVStoreSetUpdater(com.sun.jna.Pointer handle, MxnetLibrary.MXKVStoreUpdater updater, com.sun.jna.Pointer updater_handle) -
MXKVStoreSetUpdaterEx
int MXKVStoreSetUpdaterEx(com.sun.jna.Pointer handle, MxnetLibrary.MXKVStoreUpdater updater, MxnetLibrary.MXKVStoreStrUpdater str_updater, com.sun.jna.Pointer updater_handle) -
MXKVStoreGetType
-
MXKVStoreGetRank
-
MXKVStoreGetGroupSize
-
MXKVStoreIsWorkerNode
-
MXKVStoreIsServerNode
-
MXKVStoreIsSchedulerNode
-
MXKVStoreBarrier
int MXKVStoreBarrier(com.sun.jna.Pointer handle) -
MXKVStoreSetBarrierBeforeExit
int MXKVStoreSetBarrierBeforeExit(com.sun.jna.Pointer handle, int barrier_before_exit) -
MXKVStoreRunServer
int MXKVStoreRunServer(com.sun.jna.Pointer handle, MxnetLibrary.MXKVStoreServerController controller, com.sun.jna.Pointer controller_handle) -
MXKVStoreSendCommmandToServers
-
MXKVStoreGetNumDeadNode
int MXKVStoreGetNumDeadNode(com.sun.jna.Pointer handle, int node_id, IntBuffer number, int timeout_sec) -
MXRecordIOWriterCreate
-
MXRecordIOWriterFree
int MXRecordIOWriterFree(com.sun.jna.Pointer handle) -
MXRecordIOWriterWriteRecord
-
MXRecordIOWriterTell
-
MXRecordIOReaderCreate
-
MXRecordIOReaderFree
int MXRecordIOReaderFree(com.sun.jna.Pointer handle) -
MXRecordIOReaderReadRecord
-
MXRecordIOReaderSeek
-
MXRecordIOReaderTell
-
MXRtcCreate
int MXRtcCreate(ByteBuffer name, int num_input, int num_output, com.sun.jna.ptr.PointerByReference input_names, com.sun.jna.ptr.PointerByReference output_names, com.sun.jna.ptr.PointerByReference inputs, com.sun.jna.ptr.PointerByReference outputs, ByteBuffer kernel, com.sun.jna.ptr.PointerByReference out) -
MXRtcPush
int MXRtcPush(com.sun.jna.Pointer handle, int num_input, int num_output, com.sun.jna.ptr.PointerByReference inputs, com.sun.jna.ptr.PointerByReference outputs, int gridDimX, int gridDimY, int gridDimZ, int blockDimX, int blockDimY, int blockDimZ) -
MXRtcFree
int MXRtcFree(com.sun.jna.Pointer handle) -
MXCustomOpRegister
-
MXCustomFunctionRecord
int MXCustomFunctionRecord(int num_inputs, com.sun.jna.ptr.PointerByReference inputs, int num_outputs, com.sun.jna.ptr.PointerByReference outputs, MXCallbackList.ByReference callbacks) -
MXRtcCudaModuleCreate
-
MXRtcCudaModuleFree
int MXRtcCudaModuleFree(com.sun.jna.Pointer handle) -
MXRtcCudaKernelCreate
-
MXRtcCudaKernelFree
int MXRtcCudaKernelFree(com.sun.jna.Pointer handle) -
MXRtcCudaKernelCall
int MXRtcCudaKernelCall(com.sun.jna.Pointer handle, int dev_id, com.sun.jna.ptr.PointerByReference args, int grid_dim_x, int grid_dim_y, int grid_dim_z, int block_dim_x, int block_dim_y, int block_dim_z, int shared_mem) -
MXStorageEmptyCache
int MXStorageEmptyCache(int dev_type, int dev_id) -
MXEnginePushAsync
int MXEnginePushAsync(MxnetLibrary.EngineAsyncFunc async_func, com.sun.jna.Pointer func_param, MxnetLibrary.EngineFuncParamDeleter deleter, com.sun.jna.Pointer ctx_handle, com.sun.jna.Pointer const_vars_handle, int num_const_vars, com.sun.jna.Pointer mutable_vars_handle, int num_mutable_vars, com.sun.jna.Pointer prop_handle, int priority, String opr_name, byte wait) -
MXEnginePushSync
int MXEnginePushSync(MxnetLibrary.EngineSyncFunc sync_func, com.sun.jna.Pointer func_param, MxnetLibrary.EngineFuncParamDeleter deleter, com.sun.jna.Pointer ctx_handle, com.sun.jna.Pointer const_vars_handle, int num_const_vars, com.sun.jna.Pointer mutable_vars_handle, int num_mutable_vars, com.sun.jna.Pointer prop_handle, int priority, String opr_name) -
MXShallowCopyNDArray
int MXShallowCopyNDArray(com.sun.jna.Pointer src, com.sun.jna.ptr.PointerByReference out) -
MXShallowCopySymbol
int MXShallowCopySymbol(com.sun.jna.Pointer src, com.sun.jna.ptr.PointerByReference out) -
MXEnginePushAsyncND
int MXEnginePushAsyncND(MxnetLibrary.EngineAsyncFunc async_func, com.sun.jna.Pointer func_param, MxnetLibrary.EngineFuncParamDeleter deleter, com.sun.jna.Pointer ctx_handle, com.sun.jna.ptr.PointerByReference const_nds_handle, int num_const_nds, com.sun.jna.ptr.PointerByReference mutable_nds_handle, int num_mutable_nds, com.sun.jna.Pointer prop_handle, int priority, String opr_name, byte wait) -
MXEnginePushSyncND
int MXEnginePushSyncND(MxnetLibrary.EngineSyncFunc sync_func, com.sun.jna.Pointer func_param, MxnetLibrary.EngineFuncParamDeleter deleter, com.sun.jna.Pointer ctx_handle, com.sun.jna.ptr.PointerByReference const_nds_handle, int num_const_nds, com.sun.jna.ptr.PointerByReference mutable_nds_handle, int num_mutable_nds, com.sun.jna.Pointer prop_handle, int priority, String opr_name) -
NNAPISetLastError
-
NNGetLastError
String NNGetLastError() -
NNListAllOpNames
-
NNGetOpHandle
-
NNListUniqueOps
-
NNGetOpInfo
-
NNSymbolCreateAtomicSymbol
-
NNSymbolCreateVariable
-
NNSymbolCreateGroup
int NNSymbolCreateGroup(int num_symbols, com.sun.jna.ptr.PointerByReference symbols, com.sun.jna.ptr.PointerByReference out) -
NNAddControlDeps
int NNAddControlDeps(com.sun.jna.Pointer handle, com.sun.jna.Pointer src_dep) -
NNSymbolFree
int NNSymbolFree(com.sun.jna.Pointer symbol) -
NNSymbolCopy
int NNSymbolCopy(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) -
NNSymbolPrint
-
NNSymbolGetAttr
-
NNSymbolSetAttrs
-
NNSymbolListAttrs
int NNSymbolListAttrs(com.sun.jna.Pointer symbol, int recursive_option, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out) -
NNSymbolListInputVariables
int NNSymbolListInputVariables(com.sun.jna.Pointer symbol, int option, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_sym_array) -
NNSymbolListInputNames
int NNSymbolListInputNames(com.sun.jna.Pointer symbol, int option, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_str_array) -
NNSymbolListOutputNames
int NNSymbolListOutputNames(com.sun.jna.Pointer symbol, IntBuffer out_size, com.sun.jna.ptr.PointerByReference out_str_array) -
NNSymbolGetNumOutputs
-
NNSymbolGetInternals
int NNSymbolGetInternals(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) -
NNSymbolGetChildren
int NNSymbolGetChildren(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference out) -
NNSymbolGetOutput
int NNSymbolGetOutput(com.sun.jna.Pointer symbol, int index, com.sun.jna.ptr.PointerByReference out) -
NNSymbolCompose
-
NNGraphCreate
int NNGraphCreate(com.sun.jna.Pointer symbol, com.sun.jna.ptr.PointerByReference graph) -
NNGraphFree
int NNGraphFree(com.sun.jna.Pointer handle) -
NNGraphGetSymbol
int NNGraphGetSymbol(com.sun.jna.Pointer graph, com.sun.jna.ptr.PointerByReference symbol) -
NNGraphSetJSONAttr
-
NNGraphGetJSONAttr
-
NNGraphSetNodeEntryListAttr_
-
NNGraphApplyPasses
int NNGraphApplyPasses(com.sun.jna.Pointer src, int num_pass, String[] pass_names, com.sun.jna.ptr.PointerByReference dst)
-