public interface MxnetLibrary
extends com.sun.jna.Library
| Modifier and Type | Method and Description |
|---|---|
int |
MXAggregateProfileStatsPrint(java.lang.String[] out_str,
int reset) |
int |
MXAggregateProfileStatsPrintEx(java.lang.String[] out_str,
int reset,
int format,
int sort_by,
int ascending) |
int |
MXAutogradBackward(int num_output,
PointerArray output_handles,
com.sun.jna.ptr.PointerByReference ograd_handles,
int retain_graph) |
int |
MXAutogradBackwardEx(int num_output,
PointerArray output_handles,
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) |
int |
MXAutogradComputeGradient(int num_output,
com.sun.jna.ptr.PointerByReference output_handles) |
int |
MXAutogradGetSymbol(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXAutogradIsRecording(java.nio.ByteBuffer curr) |
int |
MXAutogradIsTraining(java.nio.ByteBuffer curr) |
int |
MXAutogradMarkVariables(int num_var,
com.sun.jna.ptr.PointerByReference var_handles,
java.nio.IntBuffer reqs_array,
com.sun.jna.ptr.PointerByReference grad_handles) |
int |
MXAutogradSetIsRecording(int is_recording,
java.nio.IntBuffer prev) |
int |
MXAutogradSetIsTraining(int is_training,
java.nio.IntBuffer prev) |
int |
MXCachedOpRegisterOpHook(com.sun.jna.Pointer handle,
MxnetLibrary.CachedOpMonitorCallback callback,
byte monitor_all) |
int |
MXCreateCachedOp(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXCreateCachedOpEx(com.sun.jna.Pointer handle,
int num_flags,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.ptr.PointerByReference out) |
int |
MXCreateCachedOpEX(com.sun.jna.Pointer handle,
int num_flags,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.ptr.PointerByReference out,
byte thread_safe) |
int |
MXCustomFunctionRecord(int num_inputs,
com.sun.jna.ptr.PointerByReference inputs,
int num_outputs,
com.sun.jna.ptr.PointerByReference outputs,
MXCallbackList.ByReference callbacks) |
int |
MXCustomOpRegister(java.lang.String op_type,
MxnetLibrary.CustomOpPropCreator creator) |
int |
MXDataIterBeforeFirst(com.sun.jna.Pointer handle) |
int |
MXDataIterCreateIter(com.sun.jna.Pointer handle,
int num_param,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.ptr.PointerByReference out) |
int |
MXDataIterFree(com.sun.jna.Pointer handle) |
int |
MXDataIterGetData(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXDataIterGetIndex(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out_index,
java.nio.LongBuffer out_size) |
int |
MXDataIterGetIterInfo(com.sun.jna.Pointer creator,
java.lang.String[] name,
java.lang.String[] description,
java.nio.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) |
int |
MXDataIterGetLabel(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXDataIterGetPadNum(com.sun.jna.Pointer handle,
java.nio.IntBuffer pad) |
int |
MXDataIterNext(com.sun.jna.Pointer handle,
java.nio.IntBuffer out) |
int |
MXDumpProcessProfile(int finished,
int profile_process,
com.sun.jna.Pointer kvStoreHandle) |
int |
MXDumpProfile(int finished) |
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,
java.lang.String opr_name,
byte wait) |
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,
java.lang.String opr_name,
byte wait) |
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,
java.lang.String opr_name) |
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,
java.lang.String opr_name) |
int |
MXEngineSetBulkSize(int bulk_size,
java.nio.IntBuffer prev_bulk_size) |
int |
MXExecutorBackward(com.sun.jna.Pointer handle,
int len,
com.sun.jna.ptr.PointerByReference head_grads) |
int |
MXExecutorBackwardEx(com.sun.jna.Pointer handle,
int len,
com.sun.jna.ptr.PointerByReference head_grads,
int is_train) |
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,
java.nio.IntBuffer grad_req_type,
int aux_states_len,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.ptr.PointerByReference out) |
int |
MXExecutorBindEX(com.sun.jna.Pointer symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
java.lang.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,
java.nio.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) |
int |
MXExecutorBindX(com.sun.jna.Pointer symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
java.lang.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,
java.nio.IntBuffer grad_req_type,
int aux_states_len,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.ptr.PointerByReference out) |
int |
MXExecutorForward(com.sun.jna.Pointer handle,
int is_train) |
int |
MXExecutorFree(com.sun.jna.Pointer handle) |
int |
MXExecutorGetOptimizedSymbol(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXExecutorOutputs(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out) |
int |
MXExecutorPrint(com.sun.jna.Pointer handle,
java.lang.String[] out_str) |
int |
MXExecutorReshape(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
java.lang.String[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int num_provided_arg_shapes,
java.lang.String[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
java.nio.IntBuffer num_in_args,
com.sun.jna.ptr.PointerByReference in_args,
com.sun.jna.ptr.PointerByReference arg_grads,
java.nio.IntBuffer num_aux_states,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.Pointer shared_exec,
com.sun.jna.ptr.PointerByReference out) |
int |
MXExecutorReshapeEx(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
java.lang.String[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int num_provided_arg_shapes,
java.lang.String[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
java.nio.IntBuffer num_in_args,
com.sun.jna.ptr.PointerByReference in_args,
com.sun.jna.ptr.PointerByReference arg_grads,
java.nio.IntBuffer num_aux_states,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.Pointer shared_exec,
com.sun.jna.ptr.PointerByReference out) |
int |
MXExecutorSetMonitorCallback(com.sun.jna.Pointer handle,
MxnetLibrary.ExecutorMonitorCallback callback,
com.sun.jna.Pointer callback_handle) |
int |
MXExecutorSetMonitorCallbackEX(com.sun.jna.Pointer handle,
MxnetLibrary.ExecutorMonitorCallback callback,
com.sun.jna.Pointer callback_handle,
byte monitor_all) |
int |
MXExecutorSimpleBind(com.sun.jna.Pointer symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
java.lang.String[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
java.lang.String[] provided_grad_req_names,
java.lang.String[] provided_grad_req_types,
int num_provided_arg_shapes,
java.lang.String[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
java.lang.String[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
java.lang.String[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
java.lang.String[] shared_arg_name_list,
java.nio.IntBuffer shared_buffer_len,
java.lang.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,
java.nio.IntBuffer num_in_args,
com.sun.jna.ptr.PointerByReference in_args,
com.sun.jna.ptr.PointerByReference arg_grads,
java.nio.IntBuffer num_aux_states,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.Pointer shared_exec_handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXExecutorSimpleBindEx(com.sun.jna.Pointer symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
java.lang.String[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
java.lang.String[] provided_grad_req_names,
java.lang.String[] provided_grad_req_types,
int num_provided_arg_shapes,
java.lang.String[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
java.lang.String[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
java.lang.String[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
java.lang.String[] shared_arg_name_list,
java.nio.IntBuffer shared_buffer_len,
java.lang.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,
java.nio.IntBuffer num_in_args,
com.sun.jna.ptr.PointerByReference in_args,
com.sun.jna.ptr.PointerByReference arg_grads,
java.nio.IntBuffer num_aux_states,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.Pointer shared_exec_handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXExecutorSimpleBindEx64(com.sun.jna.Pointer symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
java.lang.String[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
java.lang.String[] provided_grad_req_names,
java.lang.String[] provided_grad_req_types,
int num_provided_arg_shapes,
java.lang.String[] provided_arg_shape_names,
long[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
java.lang.String[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
java.lang.String[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
java.lang.String[] shared_arg_name_list,
java.nio.IntBuffer shared_buffer_len,
java.lang.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,
java.nio.IntBuffer num_in_args,
com.sun.jna.ptr.PointerByReference in_args,
com.sun.jna.ptr.PointerByReference arg_grads,
java.nio.IntBuffer num_aux_states,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.Pointer shared_exec_handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXFreeCachedOp(com.sun.jna.Pointer handle) |
int |
MXFuncDescribe(com.sun.jna.Pointer fun,
java.nio.IntBuffer num_use_vars,
java.nio.IntBuffer num_scalars,
java.nio.IntBuffer num_mutate_vars,
java.nio.IntBuffer type_mask) |
int |
MXFuncGetInfo(com.sun.jna.Pointer fun,
java.lang.String[] name,
java.lang.String[] description,
java.nio.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,
java.lang.String[] return_type) |
int |
MXFuncInvoke(com.sun.jna.Pointer fun,
com.sun.jna.ptr.PointerByReference use_vars,
java.nio.FloatBuffer scalar_args,
com.sun.jna.ptr.PointerByReference mutate_vars) |
int |
MXFuncInvokeEx(com.sun.jna.Pointer fun,
com.sun.jna.ptr.PointerByReference use_vars,
java.nio.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) |
int |
MXGenAtomicSymbolFromSymbol(com.sun.jna.Pointer sym_handle,
com.sun.jna.ptr.PointerByReference ret_sym_handle) |
int |
MXGenBackendSubgraph(com.sun.jna.Pointer sym_handle,
java.lang.String backend,
com.sun.jna.ptr.PointerByReference ret_sym_handle) |
int |
MXGetFunction(java.lang.String name,
com.sun.jna.ptr.PointerByReference out) |
int |
MXGetGPUCount(java.nio.IntBuffer out) |
int |
MXGetGPUMemoryInformation(int dev,
java.nio.IntBuffer free_mem,
java.nio.IntBuffer total_mem) |
int |
MXGetGPUMemoryInformation64(int dev,
java.nio.LongBuffer free_mem,
java.nio.LongBuffer total_mem) |
java.lang.String |
MXGetLastError() |
int |
MXGetVersion(java.nio.IntBuffer out) |
int |
MXImperativeInvoke(com.sun.jna.Pointer creator,
int num_inputs,
PointerArray inputs,
java.nio.IntBuffer num_outputs,
com.sun.jna.ptr.PointerByReference outputs,
int num_params,
java.lang.String[] param_keys,
java.lang.String[] param_vals) |
int |
MXImperativeInvokeEx(com.sun.jna.Pointer creator,
int num_inputs,
PointerArray inputs,
java.nio.IntBuffer num_outputs,
com.sun.jna.ptr.PointerByReference outputs,
int num_params,
java.lang.String[] param_keys,
java.lang.String[] param_vals,
com.sun.jna.ptr.PointerByReference out_stypes) |
int |
MXInitPSEnv(int num_vars,
java.lang.String[] keys,
java.lang.String[] vals) |
int |
MXInvokeCachedOp(com.sun.jna.Pointer handle,
int num_inputs,
com.sun.jna.Pointer inputs,
java.nio.IntBuffer num_outputs,
com.sun.jna.ptr.PointerByReference outputs) |
int |
MXInvokeCachedOpEx(com.sun.jna.Pointer handle,
int num_inputs,
com.sun.jna.Pointer inputs,
java.nio.IntBuffer num_outputs,
com.sun.jna.ptr.PointerByReference outputs,
com.sun.jna.ptr.PointerByReference out_stypes) |
int |
MXIsNumpyShape(java.nio.IntBuffer curr) |
int |
MXKVStoreBarrier(com.sun.jna.Pointer handle) |
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) |
int |
MXKVStoreBroadcastEx(com.sun.jna.Pointer handle,
int vnum,
java.lang.String[] vkeys,
int onum,
java.lang.String[] okeys,
com.sun.jna.ptr.PointerByReference vals,
com.sun.jna.ptr.PointerByReference outs,
int priority) |
int |
MXKVStoreCreate(java.lang.String type,
com.sun.jna.ptr.PointerByReference out) |
int |
MXKVStoreFree(com.sun.jna.Pointer handle) |
int |
MXKVStoreGetGroupSize(com.sun.jna.Pointer handle,
java.nio.IntBuffer ret) |
int |
MXKVStoreGetNumDeadNode(com.sun.jna.Pointer handle,
int node_id,
java.nio.IntBuffer number,
int timeout_sec) |
int |
MXKVStoreGetRank(com.sun.jna.Pointer handle,
java.nio.IntBuffer ret) |
int |
MXKVStoreGetType(com.sun.jna.Pointer handle,
java.lang.String[] type) |
int |
MXKVStoreInit(com.sun.jna.Pointer handle,
int num,
int[] keys,
PointerArray vals) |
int |
MXKVStoreInitEx(com.sun.jna.Pointer handle,
int num,
java.lang.String[] keys,
PointerArray vals) |
int |
MXKVStoreIsSchedulerNode(java.nio.IntBuffer ret) |
int |
MXKVStoreIsServerNode(java.nio.IntBuffer ret) |
int |
MXKVStoreIsWorkerNode(java.nio.IntBuffer ret) |
int |
MXKVStorePull(com.sun.jna.Pointer handle,
int num,
int[] keys,
PointerArray vals,
int priority) |
int |
MXKVStorePullEx(com.sun.jna.Pointer handle,
int num,
java.lang.String[] keys,
PointerArray vals,
int priority) |
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) |
int |
MXKVStorePullRowSparseEx(com.sun.jna.Pointer handle,
int num,
java.lang.String[] keys,
com.sun.jna.ptr.PointerByReference vals,
com.sun.jna.ptr.PointerByReference row_ids,
int priority) |
int |
MXKVStorePullWithSparse(com.sun.jna.Pointer handle,
int num,
int[] keys,
com.sun.jna.ptr.PointerByReference vals,
int priority,
byte ignore_sparse) |
int |
MXKVStorePullWithSparseEx(com.sun.jna.Pointer handle,
int num,
java.lang.String[] keys,
com.sun.jna.ptr.PointerByReference vals,
int priority,
byte ignore_sparse) |
int |
MXKVStorePush(com.sun.jna.Pointer handle,
int num,
int[] keys,
PointerArray vals,
int priority) |
int |
MXKVStorePushEx(com.sun.jna.Pointer handle,
int num,
java.lang.String[] keys,
PointerArray vals,
int priority) |
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) |
int |
MXKVStorePushPullEx(com.sun.jna.Pointer handle,
int vnum,
java.lang.String[] vkeys,
int onum,
java.lang.String[] okeys,
com.sun.jna.ptr.PointerByReference vals,
com.sun.jna.ptr.PointerByReference outs,
int priority) |
int |
MXKVStoreRunServer(com.sun.jna.Pointer handle,
MxnetLibrary.MXKVStoreServerController controller,
com.sun.jna.Pointer controller_handle) |
int |
MXKVStoreSendCommmandToServers(com.sun.jna.Pointer handle,
int cmd_id,
java.lang.String cmd_body) |
int |
MXKVStoreSetBarrierBeforeExit(com.sun.jna.Pointer handle,
int barrier_before_exit) |
int |
MXKVStoreSetGradientCompression(com.sun.jna.Pointer handle,
int num_params,
java.lang.String[] keys,
java.lang.String[] vals) |
int |
MXKVStoreSetUpdater(com.sun.jna.Pointer handle,
MxnetLibrary.MXKVStoreUpdater updater,
com.sun.jna.Pointer updater_handle) |
int |
MXKVStoreSetUpdaterEx(com.sun.jna.Pointer handle,
MxnetLibrary.MXKVStoreUpdater updater,
MxnetLibrary.MXKVStoreStrUpdater str_updater,
com.sun.jna.Pointer updater_handle) |
int |
MXLibInfoFeatures(com.sun.jna.ptr.PointerByReference libFeature,
NativeSizeByReference size) |
int |
MXListAllOpNames(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array) |
int |
MXListDataIters(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array) |
int |
MXListFunctions(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array) |
int |
MXLoadLib(java.lang.String path) |
int |
MXLoadTVMConfig(com.sun.jna.ptr.PointerByReference config) |
int |
MXLoadTVMOp(java.lang.String libpath) |
int |
MXNDArrayAt(com.sun.jna.Pointer handle,
int idx,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayAt64(com.sun.jna.Pointer handle,
long idx,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayCallDLPackDeleter(com.sun.jna.Pointer dlpack) |
int |
MXNDArrayCreate(int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayCreateEx(int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayCreateEx64(long[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayCreateFromSharedMem(int shared_pid,
int shared_id,
int[] shape,
int ndim,
int dtype,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayCreateFromSharedMemEx(int shared_pid,
int shared_id,
int[] shape,
int ndim,
int dtype,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayCreateNone(com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayCreateSparseEx(int storage_type,
int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
int num_aux,
java.nio.IntBuffer aux_type,
java.nio.IntBuffer aux_ndims,
int[] aux_shape,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayCreateSparseEx64(int storage_type,
long[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
int num_aux,
java.nio.IntBuffer aux_type,
java.nio.IntBuffer aux_ndims,
long[] aux_shape,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayDetach(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayFree(com.sun.jna.Pointer handle) |
int |
MXNDArrayFromDLPack(com.sun.jna.Pointer dlpack,
com.sun.jna.ptr.PointerByReference out_handle) |
int |
MXNDArrayFromDLPackEx(com.sun.jna.Pointer dlpack,
byte transient_handle,
com.sun.jna.ptr.PointerByReference out_handle) |
int |
MXNDArrayGetAuxNDArray(com.sun.jna.Pointer handle,
int i,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayGetAuxNDArray64(com.sun.jna.Pointer handle,
long i,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayGetAuxType(com.sun.jna.Pointer handle,
int i,
java.nio.IntBuffer out_type) |
int |
MXNDArrayGetAuxType64(com.sun.jna.Pointer handle,
long i,
java.nio.IntBuffer out_type) |
int |
MXNDArrayGetContext(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_dev_type,
java.nio.IntBuffer out_dev_id) |
int |
MXNDArrayGetData(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out_pdata) |
int |
MXNDArrayGetDataNDArray(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayGetDType(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_dtype) |
int |
MXNDArrayGetGrad(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayGetGradState(com.sun.jna.Pointer handle,
java.nio.IntBuffer out) |
int |
MXNDArrayGetShape(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_dim,
com.sun.jna.ptr.PointerByReference out_pdata) |
int |
MXNDArrayGetShapeEx(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_dim,
com.sun.jna.ptr.PointerByReference out_pdata) |
int |
MXNDArrayGetShapeEx64(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_dim,
com.sun.jna.ptr.PointerByReference out_pdata) |
int |
MXNDArrayGetSharedMemHandle(com.sun.jna.Pointer handle,
java.nio.IntBuffer shared_pid,
java.nio.IntBuffer shared_id) |
int |
MXNDArrayGetStorageType(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_storage_type) |
int |
MXNDArrayLoad(java.lang.String fname,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_arr,
java.nio.IntBuffer out_name_size,
com.sun.jna.ptr.PointerByReference out_names) |
int |
MXNDArrayLoadFromBuffer(com.sun.jna.Pointer ndarray_buffer,
NativeSize size,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_arr,
java.nio.IntBuffer out_name_size,
com.sun.jna.ptr.PointerByReference out_names) |
int |
MXNDArrayLoadFromRawBytes(com.sun.jna.Pointer buf,
NativeSize size,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayReshape(com.sun.jna.Pointer handle,
int ndim,
java.nio.IntBuffer dims,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArrayReshape64(com.sun.jna.Pointer handle,
int ndim,
java.nio.LongBuffer dims,
byte reverse,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArraySave(java.lang.String fname,
int num_args,
PointerArray args,
java.lang.String[] keys) |
int |
MXNDArraySaveRawBytes(com.sun.jna.Pointer handle,
NativeSizeByReference out_size,
com.sun.jna.ptr.PointerByReference out_buf) |
int |
MXNDArraySetGradState(com.sun.jna.Pointer handle,
int state) |
int |
MXNDArraySlice(com.sun.jna.Pointer handle,
int slice_begin,
int slice_end,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArraySlice64(com.sun.jna.Pointer handle,
long slice_begin,
long slice_end,
com.sun.jna.ptr.PointerByReference out) |
int |
MXNDArraySyncCheckFormat(com.sun.jna.Pointer handle,
byte full_check) |
int |
MXNDArraySyncCopyFromCPU(com.sun.jna.Pointer handle,
com.sun.jna.Pointer data,
NativeSize size) |
int |
MXNDArraySyncCopyFromNDArray(com.sun.jna.Pointer handle_dst,
com.sun.jna.Pointer handle_src,
int i) |
int |
MXNDArraySyncCopyToCPU(com.sun.jna.Pointer handle,
com.sun.jna.Pointer data,
NativeSize size) |
int |
MXNDArrayToDLPack(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out_dlpack) |
int |
MXNDArrayWaitAll() |
int |
MXNDArrayWaitToRead(com.sun.jna.Pointer handle) |
int |
MXNDArrayWaitToWrite(com.sun.jna.Pointer handle) |
int |
MXNotifyShutdown() |
int |
MXOptimizeForBackend(com.sun.jna.Pointer sym_handle,
java.lang.String backend_name,
int dev_type,
com.sun.jna.ptr.PointerByReference ret_sym_handle,
int len,
com.sun.jna.ptr.PointerByReference in_args_handle,
int num_options,
java.lang.String[] keys,
java.lang.String[] vals) |
int |
MXProcessProfilePause(int paused,
int profile_process,
com.sun.jna.Pointer kvStoreHandle) |
int |
MXProfileAdjustCounter(com.sun.jna.Pointer counter_handle,
long value) |
int |
MXProfileCreateCounter(com.sun.jna.Pointer domain,
java.lang.String counter_name,
com.sun.jna.ptr.PointerByReference out) |
int |
MXProfileCreateDomain(java.lang.String domain,
com.sun.jna.ptr.PointerByReference out) |
int |
MXProfileCreateEvent(java.lang.String event_name,
com.sun.jna.ptr.PointerByReference out) |
int |
MXProfileCreateFrame(com.sun.jna.Pointer domain,
java.lang.String frame_name,
com.sun.jna.ptr.PointerByReference out) |
int |
MXProfileCreateTask(com.sun.jna.Pointer domain,
java.lang.String task_name,
com.sun.jna.ptr.PointerByReference out) |
int |
MXProfileDestroyHandle(com.sun.jna.Pointer frame_handle) |
int |
MXProfileDurationStart(com.sun.jna.Pointer duration_handle) |
int |
MXProfileDurationStop(com.sun.jna.Pointer duration_handle) |
int |
MXProfilePause(int paused) |
int |
MXProfileSetCounter(com.sun.jna.Pointer counter_handle,
long value) |
int |
MXProfileSetMarker(com.sun.jna.Pointer domain,
java.lang.String instant_marker_name,
java.lang.String scope) |
int |
MXQuantizeSymbol(com.sun.jna.Pointer sym_handle,
com.sun.jna.ptr.PointerByReference ret_sym_handle,
int[] dev_type,
int num_excluded_sym_names,
java.lang.String[] excluded_sym_names,
int num_excluded_op_names,
java.lang.String[] excluded_op_names,
int num_offline,
java.lang.String[] offline_params,
java.lang.String quantized_dtype,
byte calib_quantize,
java.lang.String quantize_mode,
java.lang.String quantize_granularity,
java.nio.IntBuffer out_num_calib_names,
com.sun.jna.ptr.PointerByReference out_calib_names) |
int |
MXRandomSeed(int seed) |
int |
MXRandomSeedContext(int seed,
int dev_type,
int dev_id) |
int |
MXRecordIOReaderCreate(java.lang.String uri,
com.sun.jna.ptr.PointerByReference out) |
int |
MXRecordIOReaderFree(com.sun.jna.Pointer handle) |
int |
MXRecordIOReaderReadRecord(com.sun.jna.Pointer handle,
java.lang.String buf,
NativeSizeByReference size) |
int |
MXRecordIOReaderSeek(com.sun.jna.Pointer handle,
NativeSize pos) |
int |
MXRecordIOReaderTell(com.sun.jna.Pointer handle,
NativeSizeByReference pos) |
int |
MXRecordIOWriterCreate(java.lang.String uri,
com.sun.jna.ptr.PointerByReference out) |
int |
MXRecordIOWriterFree(com.sun.jna.Pointer handle) |
int |
MXRecordIOWriterTell(com.sun.jna.Pointer handle,
NativeSizeByReference pos) |
int |
MXRecordIOWriterWriteRecord(com.sun.jna.Pointer handle,
java.lang.String buf,
NativeSize size) |
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,
java.lang.String[] target_dtype_op_names,
java.lang.String[] fp32_op_names,
java.lang.String[] widest_dtype_op_names,
java.lang.String[] conditional_fp32_op_names,
java.lang.String[] excluded_symbols,
java.lang.String[] conditional_param_names,
java.lang.String[] conditional_param_vals,
java.lang.String[] model_param_names,
java.lang.String[] arg_names) |
int |
MXRtcCreate(java.nio.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,
java.nio.ByteBuffer kernel,
com.sun.jna.ptr.PointerByReference out) |
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) |
int |
MXRtcCudaKernelCreate(com.sun.jna.Pointer handle,
java.lang.String name,
int num_args,
java.nio.IntBuffer is_ndarray,
java.nio.IntBuffer is_const,
java.nio.IntBuffer arg_types,
com.sun.jna.ptr.PointerByReference out) |
int |
MXRtcCudaKernelFree(com.sun.jna.Pointer handle) |
int |
MXRtcCudaModuleCreate(java.lang.String source,
int num_options,
java.lang.String[] options,
int num_exports,
java.lang.String[] exports,
com.sun.jna.ptr.PointerByReference out) |
int |
MXRtcCudaModuleFree(com.sun.jna.Pointer handle) |
int |
MXRtcFree(com.sun.jna.Pointer handle) |
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) |
int |
MXSetCalibTableToQuantizedSymbol(com.sun.jna.Pointer qsym_handle,
int num_layers,
java.lang.String[] layer_names,
java.nio.FloatBuffer low_quantiles,
java.nio.FloatBuffer high_quantiles,
com.sun.jna.ptr.PointerByReference ret_sym_handle) |
int |
MXSetIsNumpyShape(int is_np_shape,
java.nio.IntBuffer prev) |
int |
MXSetNumOMPThreads(int thread_num) |
int |
MXSetProcessProfilerConfig(int num_params,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.Pointer kvstoreHandle) |
int |
MXSetProcessProfilerState(int state,
int profile_process,
com.sun.jna.Pointer kvStoreHandle) |
int |
MXSetProfilerConfig(int num_params,
java.lang.String[] keys,
java.lang.String[] vals) |
int |
MXSetProfilerState(int state) |
int |
MXShallowCopyNDArray(com.sun.jna.Pointer src,
com.sun.jna.ptr.PointerByReference out) |
int |
MXShallowCopySymbol(com.sun.jna.Pointer src,
com.sun.jna.ptr.PointerByReference out) |
int |
MXStorageEmptyCache(int dev_type,
int dev_id) |
int |
MXSymbolCompose(com.sun.jna.Pointer sym,
java.lang.String name,
int num_args,
java.lang.String[] keys,
com.sun.jna.ptr.PointerByReference args) |
int |
MXSymbolCopy(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolCreateAtomicSymbol(com.sun.jna.Pointer creator,
int num_param,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolCreateFromFile(java.lang.String fname,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolCreateFromJSON(java.lang.String json,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolCreateGroup(int num_symbols,
com.sun.jna.ptr.PointerByReference symbols,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolCreateVariable(java.lang.String name,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolCutSubgraph(com.sun.jna.Pointer sym,
com.sun.jna.ptr.PointerByReference inputs,
java.nio.IntBuffer input_size) |
int |
MXSymbolFree(com.sun.jna.Pointer symbol) |
int |
MXSymbolGetAtomicSymbolInfo(com.sun.jna.Pointer creator,
java.lang.String[] name,
java.lang.String[] description,
java.nio.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,
java.lang.String[] key_var_num_args,
java.lang.String[] return_type) |
int |
MXSymbolGetAtomicSymbolName(com.sun.jna.Pointer creator,
java.lang.String[] name) |
int |
MXSymbolGetAttr(com.sun.jna.Pointer symbol,
java.lang.String key,
java.lang.String[] out,
java.nio.IntBuffer success) |
int |
MXSymbolGetChildren(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolGetInputSymbols(com.sun.jna.Pointer sym,
com.sun.jna.ptr.PointerByReference inputs,
java.nio.IntBuffer input_size) |
int |
MXSymbolGetInternals(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolGetName(com.sun.jna.Pointer symbol,
java.lang.String[] out,
java.nio.IntBuffer success) |
int |
MXSymbolGetNumOutputs(com.sun.jna.Pointer symbol,
java.nio.IntBuffer output_count) |
int |
MXSymbolGetOutput(com.sun.jna.Pointer symbol,
int index,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolGrad(com.sun.jna.Pointer sym,
int num_wrt,
java.lang.String[] wrt,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolInferShape(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
java.nio.IntBuffer in_shape_size,
com.sun.jna.ptr.PointerByReference in_shape_ndim,
com.sun.jna.ptr.PointerByReference in_shape_data,
java.nio.IntBuffer out_shape_size,
com.sun.jna.ptr.PointerByReference out_shape_ndim,
com.sun.jna.ptr.PointerByReference out_shape_data,
java.nio.IntBuffer aux_shape_size,
com.sun.jna.ptr.PointerByReference aux_shape_ndim,
com.sun.jna.ptr.PointerByReference aux_shape_data,
java.nio.IntBuffer complete) |
int |
MXSymbolInferShapeEx(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
java.nio.IntBuffer in_shape_size,
com.sun.jna.ptr.PointerByReference in_shape_ndim,
com.sun.jna.ptr.PointerByReference in_shape_data,
java.nio.IntBuffer out_shape_size,
com.sun.jna.ptr.PointerByReference out_shape_ndim,
com.sun.jna.ptr.PointerByReference out_shape_data,
java.nio.IntBuffer aux_shape_size,
com.sun.jna.ptr.PointerByReference aux_shape_ndim,
com.sun.jna.ptr.PointerByReference aux_shape_data,
java.nio.IntBuffer complete) |
int |
MXSymbolInferShapeEx64(com.sun.jna.Pointer sym,
int num_args,
java.lang.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,
java.nio.IntBuffer complete) |
int |
MXSymbolInferShapePartial(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
java.nio.IntBuffer in_shape_size,
com.sun.jna.ptr.PointerByReference in_shape_ndim,
com.sun.jna.ptr.PointerByReference in_shape_data,
java.nio.IntBuffer out_shape_size,
com.sun.jna.ptr.PointerByReference out_shape_ndim,
com.sun.jna.ptr.PointerByReference out_shape_data,
java.nio.IntBuffer aux_shape_size,
com.sun.jna.ptr.PointerByReference aux_shape_ndim,
com.sun.jna.ptr.PointerByReference aux_shape_data,
java.nio.IntBuffer complete) |
int |
MXSymbolInferShapePartialEx(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
java.nio.IntBuffer in_shape_size,
com.sun.jna.ptr.PointerByReference in_shape_ndim,
com.sun.jna.ptr.PointerByReference in_shape_data,
java.nio.IntBuffer out_shape_size,
com.sun.jna.ptr.PointerByReference out_shape_ndim,
com.sun.jna.ptr.PointerByReference out_shape_data,
java.nio.IntBuffer aux_shape_size,
com.sun.jna.ptr.PointerByReference aux_shape_ndim,
com.sun.jna.ptr.PointerByReference aux_shape_data,
java.nio.IntBuffer complete) |
int |
MXSymbolInferShapePartialEx64(com.sun.jna.Pointer sym,
int num_args,
java.lang.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,
java.nio.IntBuffer complete) |
int |
MXSymbolInferType(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_type_data,
java.nio.IntBuffer in_type_size,
com.sun.jna.ptr.PointerByReference in_type_data,
java.nio.IntBuffer out_type_size,
com.sun.jna.ptr.PointerByReference out_type_data,
java.nio.IntBuffer aux_type_size,
com.sun.jna.ptr.PointerByReference aux_type_data,
java.nio.IntBuffer complete) |
int |
MXSymbolInferTypePartial(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_type_data,
java.nio.IntBuffer in_type_size,
com.sun.jna.ptr.PointerByReference in_type_data,
java.nio.IntBuffer out_type_size,
com.sun.jna.ptr.PointerByReference out_type_data,
java.nio.IntBuffer aux_type_size,
com.sun.jna.ptr.PointerByReference aux_type_data,
java.nio.IntBuffer complete) |
int |
MXSymbolListArguments(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_str_array) |
int |
MXSymbolListAtomicSymbolCreators(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array) |
int |
MXSymbolListAttr(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolListAttrShallow(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out) |
int |
MXSymbolListAuxiliaryStates(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_str_array) |
int |
MXSymbolListOutputs(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_str_array) |
int |
MXSymbolPrint(com.sun.jna.Pointer symbol,
java.lang.String[] out_str) |
int |
MXSymbolRemoveAmpCast(com.sun.jna.Pointer sym_handle,
com.sun.jna.ptr.PointerByReference ret_sym_handle) |
int |
MXSymbolSaveToFile(com.sun.jna.Pointer symbol,
java.lang.String fname) |
int |
MXSymbolSaveToJSON(com.sun.jna.Pointer symbol,
java.lang.String[] out_json) |
int |
MXSymbolSetAttr(com.sun.jna.Pointer symbol,
java.lang.String key,
java.lang.String value) |
int |
NNAddControlDeps(com.sun.jna.Pointer handle,
com.sun.jna.Pointer src_dep) |
void |
NNAPISetLastError(java.lang.String msg) |
java.lang.String |
NNGetLastError() |
int |
NNGetOpHandle(java.lang.String op_name,
com.sun.jna.ptr.PointerByReference op_out) |
int |
NNGetOpInfo(com.sun.jna.Pointer op,
java.lang.String[] real_name,
java.lang.String[] description,
java.nio.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,
java.lang.String[] return_type) |
int |
NNGraphApplyPasses(com.sun.jna.Pointer src,
int num_pass,
java.lang.String[] pass_names,
com.sun.jna.ptr.PointerByReference dst) |
int |
NNGraphCreate(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference graph) |
int |
NNGraphFree(com.sun.jna.Pointer handle) |
int |
NNGraphGetJSONAttr(com.sun.jna.Pointer handle,
java.lang.String key,
java.lang.String[] json_out,
java.nio.IntBuffer success) |
int |
NNGraphGetSymbol(com.sun.jna.Pointer graph,
com.sun.jna.ptr.PointerByReference symbol) |
int |
NNGraphSetJSONAttr(com.sun.jna.Pointer handle,
java.lang.String key,
java.lang.String json_value) |
int |
NNGraphSetNodeEntryListAttr_(com.sun.jna.Pointer handle,
java.lang.String key,
com.sun.jna.Pointer list) |
int |
NNListAllOpNames(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array) |
int |
NNListUniqueOps(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array) |
int |
NNSymbolCompose(com.sun.jna.Pointer sym,
java.lang.String name,
int num_args,
java.lang.String[] keys,
com.sun.jna.ptr.PointerByReference args) |
int |
NNSymbolCopy(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out) |
int |
NNSymbolCreateAtomicSymbol(com.sun.jna.Pointer op,
int num_param,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.ptr.PointerByReference out) |
int |
NNSymbolCreateGroup(int num_symbols,
com.sun.jna.ptr.PointerByReference symbols,
com.sun.jna.ptr.PointerByReference out) |
int |
NNSymbolCreateVariable(java.lang.String name,
com.sun.jna.ptr.PointerByReference out) |
int |
NNSymbolFree(com.sun.jna.Pointer symbol) |
int |
NNSymbolGetAttr(com.sun.jna.Pointer symbol,
java.lang.String key,
java.lang.String[] out,
java.nio.IntBuffer success) |
int |
NNSymbolGetChildren(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out) |
int |
NNSymbolGetInternals(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out) |
int |
NNSymbolGetNumOutputs(com.sun.jna.Pointer symbol,
java.nio.IntBuffer output_count) |
int |
NNSymbolGetOutput(com.sun.jna.Pointer symbol,
int index,
com.sun.jna.ptr.PointerByReference out) |
int |
NNSymbolListAttrs(com.sun.jna.Pointer symbol,
int recursive_option,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out) |
int |
NNSymbolListInputNames(com.sun.jna.Pointer symbol,
int option,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_str_array) |
int |
NNSymbolListInputVariables(com.sun.jna.Pointer symbol,
int option,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_sym_array) |
int |
NNSymbolListOutputNames(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_str_array) |
int |
NNSymbolPrint(com.sun.jna.Pointer symbol,
java.lang.String[] out_str) |
int |
NNSymbolSetAttrs(com.sun.jna.Pointer symbol,
int num_param,
java.lang.String[] keys,
java.lang.String[] values) |
java.lang.String MXGetLastError()
int MXLoadLib(java.lang.String path)
int MXLibInfoFeatures(com.sun.jna.ptr.PointerByReference libFeature,
NativeSizeByReference size)
int MXRandomSeed(int seed)
int MXRandomSeedContext(int seed,
int dev_type,
int dev_id)
int MXNotifyShutdown()
int MXSetProcessProfilerConfig(int num_params,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.Pointer kvstoreHandle)
int MXSetProfilerConfig(int num_params,
java.lang.String[] keys,
java.lang.String[] vals)
int MXSetProcessProfilerState(int state,
int profile_process,
com.sun.jna.Pointer kvStoreHandle)
int MXSetProfilerState(int state)
int MXDumpProcessProfile(int finished,
int profile_process,
com.sun.jna.Pointer kvStoreHandle)
int MXDumpProfile(int finished)
int MXAggregateProfileStatsPrint(java.lang.String[] out_str,
int reset)
int MXAggregateProfileStatsPrintEx(java.lang.String[] out_str,
int reset,
int format,
int sort_by,
int ascending)
int MXProcessProfilePause(int paused,
int profile_process,
com.sun.jna.Pointer kvStoreHandle)
int MXProfilePause(int paused)
int MXProfileCreateDomain(java.lang.String domain,
com.sun.jna.ptr.PointerByReference out)
int MXProfileCreateTask(com.sun.jna.Pointer domain,
java.lang.String task_name,
com.sun.jna.ptr.PointerByReference out)
int MXProfileCreateFrame(com.sun.jna.Pointer domain,
java.lang.String frame_name,
com.sun.jna.ptr.PointerByReference out)
int MXProfileCreateEvent(java.lang.String event_name,
com.sun.jna.ptr.PointerByReference out)
int MXProfileCreateCounter(com.sun.jna.Pointer domain,
java.lang.String counter_name,
com.sun.jna.ptr.PointerByReference out)
int MXProfileDestroyHandle(com.sun.jna.Pointer frame_handle)
int MXProfileDurationStart(com.sun.jna.Pointer duration_handle)
int MXProfileDurationStop(com.sun.jna.Pointer duration_handle)
int MXProfileSetCounter(com.sun.jna.Pointer counter_handle,
long value)
int MXProfileAdjustCounter(com.sun.jna.Pointer counter_handle,
long value)
int MXProfileSetMarker(com.sun.jna.Pointer domain,
java.lang.String instant_marker_name,
java.lang.String scope)
int MXSetNumOMPThreads(int thread_num)
int MXEngineSetBulkSize(int bulk_size,
java.nio.IntBuffer prev_bulk_size)
int MXGetGPUCount(java.nio.IntBuffer out)
int MXGetGPUMemoryInformation(int dev,
java.nio.IntBuffer free_mem,
java.nio.IntBuffer total_mem)
int MXGetGPUMemoryInformation64(int dev,
java.nio.LongBuffer free_mem,
java.nio.LongBuffer total_mem)
int MXGetVersion(java.nio.IntBuffer out)
int MXLoadTVMOp(java.lang.String libpath)
int MXLoadTVMConfig(com.sun.jna.ptr.PointerByReference config)
int MXNDArrayCreateNone(com.sun.jna.ptr.PointerByReference out)
int MXNDArrayCreate(int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayCreateEx(int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayCreateEx64(long[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayCreateSparseEx(int storage_type,
int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
int num_aux,
java.nio.IntBuffer aux_type,
java.nio.IntBuffer aux_ndims,
int[] aux_shape,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayCreateSparseEx64(int storage_type,
long[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
int num_aux,
java.nio.IntBuffer aux_type,
java.nio.IntBuffer aux_ndims,
long[] aux_shape,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayLoadFromRawBytes(com.sun.jna.Pointer buf,
NativeSize size,
com.sun.jna.ptr.PointerByReference out)
int MXNDArraySaveRawBytes(com.sun.jna.Pointer handle,
NativeSizeByReference out_size,
com.sun.jna.ptr.PointerByReference out_buf)
int MXNDArraySave(java.lang.String fname,
int num_args,
PointerArray args,
java.lang.String[] keys)
int MXNDArrayLoad(java.lang.String fname,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_arr,
java.nio.IntBuffer out_name_size,
com.sun.jna.ptr.PointerByReference out_names)
int MXNDArrayLoadFromBuffer(com.sun.jna.Pointer ndarray_buffer,
NativeSize size,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_arr,
java.nio.IntBuffer out_name_size,
com.sun.jna.ptr.PointerByReference out_names)
int MXNDArraySyncCopyFromCPU(com.sun.jna.Pointer handle,
com.sun.jna.Pointer data,
NativeSize size)
int MXNDArraySyncCopyToCPU(com.sun.jna.Pointer handle,
com.sun.jna.Pointer data,
NativeSize size)
int MXNDArraySyncCopyFromNDArray(com.sun.jna.Pointer handle_dst,
com.sun.jna.Pointer handle_src,
int i)
int MXNDArraySyncCheckFormat(com.sun.jna.Pointer handle,
byte full_check)
int MXNDArrayWaitToRead(com.sun.jna.Pointer handle)
int MXNDArrayWaitToWrite(com.sun.jna.Pointer handle)
int MXNDArrayWaitAll()
int MXNDArrayFree(com.sun.jna.Pointer handle)
int MXNDArraySlice(com.sun.jna.Pointer handle,
int slice_begin,
int slice_end,
com.sun.jna.ptr.PointerByReference out)
int MXNDArraySlice64(com.sun.jna.Pointer handle,
long slice_begin,
long slice_end,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayAt(com.sun.jna.Pointer handle,
int idx,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayAt64(com.sun.jna.Pointer handle,
long idx,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayGetStorageType(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_storage_type)
int MXNDArrayReshape(com.sun.jna.Pointer handle,
int ndim,
java.nio.IntBuffer dims,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayReshape64(com.sun.jna.Pointer handle,
int ndim,
java.nio.LongBuffer dims,
byte reverse,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayGetShape(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_dim,
com.sun.jna.ptr.PointerByReference out_pdata)
int MXNDArrayGetShapeEx(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_dim,
com.sun.jna.ptr.PointerByReference out_pdata)
int MXNDArrayGetShapeEx64(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_dim,
com.sun.jna.ptr.PointerByReference out_pdata)
int MXNDArrayGetData(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out_pdata)
int MXNDArrayToDLPack(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out_dlpack)
int MXNDArrayFromDLPack(com.sun.jna.Pointer dlpack,
com.sun.jna.ptr.PointerByReference out_handle)
int MXNDArrayFromDLPackEx(com.sun.jna.Pointer dlpack,
byte transient_handle,
com.sun.jna.ptr.PointerByReference out_handle)
int MXNDArrayCallDLPackDeleter(com.sun.jna.Pointer dlpack)
int MXNDArrayGetDType(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_dtype)
int MXNDArrayGetAuxType(com.sun.jna.Pointer handle,
int i,
java.nio.IntBuffer out_type)
int MXNDArrayGetAuxType64(com.sun.jna.Pointer handle,
long i,
java.nio.IntBuffer out_type)
int MXNDArrayGetAuxNDArray(com.sun.jna.Pointer handle,
int i,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayGetAuxNDArray64(com.sun.jna.Pointer handle,
long i,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayGetDataNDArray(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayGetContext(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_dev_type,
java.nio.IntBuffer out_dev_id)
int MXNDArrayGetGrad(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out)
int MXNDArrayDetach(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out)
int MXNDArraySetGradState(com.sun.jna.Pointer handle,
int state)
int MXNDArrayGetGradState(com.sun.jna.Pointer handle,
java.nio.IntBuffer out)
int MXListFunctions(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array)
int MXGetFunction(java.lang.String name,
com.sun.jna.ptr.PointerByReference out)
int MXFuncGetInfo(com.sun.jna.Pointer fun,
java.lang.String[] name,
java.lang.String[] description,
java.nio.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,
java.lang.String[] return_type)
int MXFuncDescribe(com.sun.jna.Pointer fun,
java.nio.IntBuffer num_use_vars,
java.nio.IntBuffer num_scalars,
java.nio.IntBuffer num_mutate_vars,
java.nio.IntBuffer type_mask)
int MXFuncInvoke(com.sun.jna.Pointer fun,
com.sun.jna.ptr.PointerByReference use_vars,
java.nio.FloatBuffer scalar_args,
com.sun.jna.ptr.PointerByReference mutate_vars)
int MXFuncInvokeEx(com.sun.jna.Pointer fun,
com.sun.jna.ptr.PointerByReference use_vars,
java.nio.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)
int MXImperativeInvoke(com.sun.jna.Pointer creator,
int num_inputs,
PointerArray inputs,
java.nio.IntBuffer num_outputs,
com.sun.jna.ptr.PointerByReference outputs,
int num_params,
java.lang.String[] param_keys,
java.lang.String[] param_vals)
int MXImperativeInvokeEx(com.sun.jna.Pointer creator,
int num_inputs,
PointerArray inputs,
java.nio.IntBuffer num_outputs,
com.sun.jna.ptr.PointerByReference outputs,
int num_params,
java.lang.String[] param_keys,
java.lang.String[] param_vals,
com.sun.jna.ptr.PointerByReference out_stypes)
int MXAutogradSetIsRecording(int is_recording,
java.nio.IntBuffer prev)
int MXAutogradSetIsTraining(int is_training,
java.nio.IntBuffer prev)
int MXAutogradIsRecording(java.nio.ByteBuffer curr)
int MXAutogradIsTraining(java.nio.ByteBuffer curr)
int MXIsNumpyShape(java.nio.IntBuffer curr)
int MXSetIsNumpyShape(int is_np_shape,
java.nio.IntBuffer prev)
int MXAutogradMarkVariables(int num_var,
com.sun.jna.ptr.PointerByReference var_handles,
java.nio.IntBuffer reqs_array,
com.sun.jna.ptr.PointerByReference grad_handles)
int MXAutogradComputeGradient(int num_output,
com.sun.jna.ptr.PointerByReference output_handles)
int MXAutogradBackward(int num_output,
PointerArray output_handles,
com.sun.jna.ptr.PointerByReference ograd_handles,
int retain_graph)
int MXAutogradBackwardEx(int num_output,
PointerArray output_handles,
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)
int MXAutogradGetSymbol(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out)
int MXCreateCachedOp(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out)
int MXCreateCachedOpEx(com.sun.jna.Pointer handle,
int num_flags,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.ptr.PointerByReference out)
int MXCreateCachedOpEX(com.sun.jna.Pointer handle,
int num_flags,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.ptr.PointerByReference out,
byte thread_safe)
int MXFreeCachedOp(com.sun.jna.Pointer handle)
int MXInvokeCachedOp(com.sun.jna.Pointer handle,
int num_inputs,
com.sun.jna.Pointer inputs,
java.nio.IntBuffer num_outputs,
com.sun.jna.ptr.PointerByReference outputs)
int MXInvokeCachedOpEx(com.sun.jna.Pointer handle,
int num_inputs,
com.sun.jna.Pointer inputs,
java.nio.IntBuffer num_outputs,
com.sun.jna.ptr.PointerByReference outputs,
com.sun.jna.ptr.PointerByReference out_stypes)
int MXCachedOpRegisterOpHook(com.sun.jna.Pointer handle,
MxnetLibrary.CachedOpMonitorCallback callback,
byte monitor_all)
int MXListAllOpNames(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array)
int MXSymbolListAtomicSymbolCreators(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array)
int MXSymbolGetAtomicSymbolName(com.sun.jna.Pointer creator,
java.lang.String[] name)
int MXSymbolGetInputSymbols(com.sun.jna.Pointer sym,
com.sun.jna.ptr.PointerByReference inputs,
java.nio.IntBuffer input_size)
int MXSymbolCutSubgraph(com.sun.jna.Pointer sym,
com.sun.jna.ptr.PointerByReference inputs,
java.nio.IntBuffer input_size)
int MXSymbolGetAtomicSymbolInfo(com.sun.jna.Pointer creator,
java.lang.String[] name,
java.lang.String[] description,
java.nio.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,
java.lang.String[] key_var_num_args,
java.lang.String[] return_type)
int MXSymbolCreateAtomicSymbol(com.sun.jna.Pointer creator,
int num_param,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolCreateVariable(java.lang.String name,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolCreateGroup(int num_symbols,
com.sun.jna.ptr.PointerByReference symbols,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolCreateFromFile(java.lang.String fname,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolCreateFromJSON(java.lang.String json,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolRemoveAmpCast(com.sun.jna.Pointer sym_handle,
com.sun.jna.ptr.PointerByReference ret_sym_handle)
int MXSymbolSaveToFile(com.sun.jna.Pointer symbol,
java.lang.String fname)
int MXSymbolSaveToJSON(com.sun.jna.Pointer symbol,
java.lang.String[] out_json)
int MXSymbolFree(com.sun.jna.Pointer symbol)
int MXSymbolCopy(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolPrint(com.sun.jna.Pointer symbol,
java.lang.String[] out_str)
int MXSymbolGetName(com.sun.jna.Pointer symbol,
java.lang.String[] out,
java.nio.IntBuffer success)
int MXSymbolGetAttr(com.sun.jna.Pointer symbol,
java.lang.String key,
java.lang.String[] out,
java.nio.IntBuffer success)
int MXSymbolSetAttr(com.sun.jna.Pointer symbol,
java.lang.String key,
java.lang.String value)
int MXSymbolListAttr(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolListAttrShallow(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolListArguments(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_str_array)
int MXSymbolListOutputs(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_str_array)
int MXSymbolGetNumOutputs(com.sun.jna.Pointer symbol,
java.nio.IntBuffer output_count)
int MXSymbolGetInternals(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolGetChildren(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolGetOutput(com.sun.jna.Pointer symbol,
int index,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolListAuxiliaryStates(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_str_array)
int MXSymbolCompose(com.sun.jna.Pointer sym,
java.lang.String name,
int num_args,
java.lang.String[] keys,
com.sun.jna.ptr.PointerByReference args)
int MXSymbolGrad(com.sun.jna.Pointer sym,
int num_wrt,
java.lang.String[] wrt,
com.sun.jna.ptr.PointerByReference out)
int MXSymbolInferShape(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
java.nio.IntBuffer in_shape_size,
com.sun.jna.ptr.PointerByReference in_shape_ndim,
com.sun.jna.ptr.PointerByReference in_shape_data,
java.nio.IntBuffer out_shape_size,
com.sun.jna.ptr.PointerByReference out_shape_ndim,
com.sun.jna.ptr.PointerByReference out_shape_data,
java.nio.IntBuffer aux_shape_size,
com.sun.jna.ptr.PointerByReference aux_shape_ndim,
com.sun.jna.ptr.PointerByReference aux_shape_data,
java.nio.IntBuffer complete)
int MXSymbolInferShapeEx(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
java.nio.IntBuffer in_shape_size,
com.sun.jna.ptr.PointerByReference in_shape_ndim,
com.sun.jna.ptr.PointerByReference in_shape_data,
java.nio.IntBuffer out_shape_size,
com.sun.jna.ptr.PointerByReference out_shape_ndim,
com.sun.jna.ptr.PointerByReference out_shape_data,
java.nio.IntBuffer aux_shape_size,
com.sun.jna.ptr.PointerByReference aux_shape_ndim,
com.sun.jna.ptr.PointerByReference aux_shape_data,
java.nio.IntBuffer complete)
int MXSymbolInferShapeEx64(com.sun.jna.Pointer sym,
int num_args,
java.lang.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,
java.nio.IntBuffer complete)
int MXSymbolInferShapePartial(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
java.nio.IntBuffer in_shape_size,
com.sun.jna.ptr.PointerByReference in_shape_ndim,
com.sun.jna.ptr.PointerByReference in_shape_data,
java.nio.IntBuffer out_shape_size,
com.sun.jna.ptr.PointerByReference out_shape_ndim,
com.sun.jna.ptr.PointerByReference out_shape_data,
java.nio.IntBuffer aux_shape_size,
com.sun.jna.ptr.PointerByReference aux_shape_ndim,
com.sun.jna.ptr.PointerByReference aux_shape_data,
java.nio.IntBuffer complete)
int MXSymbolInferShapePartialEx(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
java.nio.IntBuffer in_shape_size,
com.sun.jna.ptr.PointerByReference in_shape_ndim,
com.sun.jna.ptr.PointerByReference in_shape_data,
java.nio.IntBuffer out_shape_size,
com.sun.jna.ptr.PointerByReference out_shape_ndim,
com.sun.jna.ptr.PointerByReference out_shape_data,
java.nio.IntBuffer aux_shape_size,
com.sun.jna.ptr.PointerByReference aux_shape_ndim,
com.sun.jna.ptr.PointerByReference aux_shape_data,
java.nio.IntBuffer complete)
int MXSymbolInferShapePartialEx64(com.sun.jna.Pointer sym,
int num_args,
java.lang.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,
java.nio.IntBuffer complete)
int MXSymbolInferType(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_type_data,
java.nio.IntBuffer in_type_size,
com.sun.jna.ptr.PointerByReference in_type_data,
java.nio.IntBuffer out_type_size,
com.sun.jna.ptr.PointerByReference out_type_data,
java.nio.IntBuffer aux_type_size,
com.sun.jna.ptr.PointerByReference aux_type_data,
java.nio.IntBuffer complete)
int MXSymbolInferTypePartial(com.sun.jna.Pointer sym,
int num_args,
java.lang.String[] keys,
int[] arg_type_data,
java.nio.IntBuffer in_type_size,
com.sun.jna.ptr.PointerByReference in_type_data,
java.nio.IntBuffer out_type_size,
com.sun.jna.ptr.PointerByReference out_type_data,
java.nio.IntBuffer aux_type_size,
com.sun.jna.ptr.PointerByReference aux_type_data,
java.nio.IntBuffer complete)
int MXQuantizeSymbol(com.sun.jna.Pointer sym_handle,
com.sun.jna.ptr.PointerByReference ret_sym_handle,
int[] dev_type,
int num_excluded_sym_names,
java.lang.String[] excluded_sym_names,
int num_excluded_op_names,
java.lang.String[] excluded_op_names,
int num_offline,
java.lang.String[] offline_params,
java.lang.String quantized_dtype,
byte calib_quantize,
java.lang.String quantize_mode,
java.lang.String quantize_granularity,
java.nio.IntBuffer out_num_calib_names,
com.sun.jna.ptr.PointerByReference out_calib_names)
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,
java.lang.String[] target_dtype_op_names,
java.lang.String[] fp32_op_names,
java.lang.String[] widest_dtype_op_names,
java.lang.String[] conditional_fp32_op_names,
java.lang.String[] excluded_symbols,
java.lang.String[] conditional_param_names,
java.lang.String[] conditional_param_vals,
java.lang.String[] model_param_names,
java.lang.String[] arg_names)
int MXSetCalibTableToQuantizedSymbol(com.sun.jna.Pointer qsym_handle,
int num_layers,
java.lang.String[] layer_names,
java.nio.FloatBuffer low_quantiles,
java.nio.FloatBuffer high_quantiles,
com.sun.jna.ptr.PointerByReference ret_sym_handle)
int MXGenBackendSubgraph(com.sun.jna.Pointer sym_handle,
java.lang.String backend,
com.sun.jna.ptr.PointerByReference ret_sym_handle)
int MXGenAtomicSymbolFromSymbol(com.sun.jna.Pointer sym_handle,
com.sun.jna.ptr.PointerByReference ret_sym_handle)
int MXOptimizeForBackend(com.sun.jna.Pointer sym_handle,
java.lang.String backend_name,
int dev_type,
com.sun.jna.ptr.PointerByReference ret_sym_handle,
int len,
com.sun.jna.ptr.PointerByReference in_args_handle,
int num_options,
java.lang.String[] keys,
java.lang.String[] vals)
int MXExecutorFree(com.sun.jna.Pointer handle)
int MXExecutorPrint(com.sun.jna.Pointer handle,
java.lang.String[] out_str)
int MXExecutorForward(com.sun.jna.Pointer handle,
int is_train)
int MXExecutorBackward(com.sun.jna.Pointer handle,
int len,
com.sun.jna.ptr.PointerByReference head_grads)
int MXExecutorBackwardEx(com.sun.jna.Pointer handle,
int len,
com.sun.jna.ptr.PointerByReference head_grads,
int is_train)
int MXExecutorOutputs(com.sun.jna.Pointer handle,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out)
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,
java.nio.IntBuffer grad_req_type,
int aux_states_len,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.ptr.PointerByReference out)
int MXExecutorBindX(com.sun.jna.Pointer symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
java.lang.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,
java.nio.IntBuffer grad_req_type,
int aux_states_len,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.ptr.PointerByReference out)
int MXExecutorBindEX(com.sun.jna.Pointer symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
java.lang.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,
java.nio.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)
int MXExecutorSimpleBind(com.sun.jna.Pointer symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
java.lang.String[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
java.lang.String[] provided_grad_req_names,
java.lang.String[] provided_grad_req_types,
int num_provided_arg_shapes,
java.lang.String[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
java.lang.String[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
java.lang.String[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
java.lang.String[] shared_arg_name_list,
java.nio.IntBuffer shared_buffer_len,
java.lang.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,
java.nio.IntBuffer num_in_args,
com.sun.jna.ptr.PointerByReference in_args,
com.sun.jna.ptr.PointerByReference arg_grads,
java.nio.IntBuffer num_aux_states,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.Pointer shared_exec_handle,
com.sun.jna.ptr.PointerByReference out)
int MXExecutorSimpleBindEx(com.sun.jna.Pointer symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
java.lang.String[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
java.lang.String[] provided_grad_req_names,
java.lang.String[] provided_grad_req_types,
int num_provided_arg_shapes,
java.lang.String[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
java.lang.String[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
java.lang.String[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
java.lang.String[] shared_arg_name_list,
java.nio.IntBuffer shared_buffer_len,
java.lang.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,
java.nio.IntBuffer num_in_args,
com.sun.jna.ptr.PointerByReference in_args,
com.sun.jna.ptr.PointerByReference arg_grads,
java.nio.IntBuffer num_aux_states,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.Pointer shared_exec_handle,
com.sun.jna.ptr.PointerByReference out)
int MXExecutorSimpleBindEx64(com.sun.jna.Pointer symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
java.lang.String[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
java.lang.String[] provided_grad_req_names,
java.lang.String[] provided_grad_req_types,
int num_provided_arg_shapes,
java.lang.String[] provided_arg_shape_names,
long[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
java.lang.String[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
java.lang.String[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
java.lang.String[] shared_arg_name_list,
java.nio.IntBuffer shared_buffer_len,
java.lang.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,
java.nio.IntBuffer num_in_args,
com.sun.jna.ptr.PointerByReference in_args,
com.sun.jna.ptr.PointerByReference arg_grads,
java.nio.IntBuffer num_aux_states,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.Pointer shared_exec_handle,
com.sun.jna.ptr.PointerByReference out)
int MXExecutorReshape(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
java.lang.String[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int num_provided_arg_shapes,
java.lang.String[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
java.nio.IntBuffer num_in_args,
com.sun.jna.ptr.PointerByReference in_args,
com.sun.jna.ptr.PointerByReference arg_grads,
java.nio.IntBuffer num_aux_states,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.Pointer shared_exec,
com.sun.jna.ptr.PointerByReference out)
int MXExecutorReshapeEx(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
java.lang.String[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int num_provided_arg_shapes,
java.lang.String[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
java.nio.IntBuffer num_in_args,
com.sun.jna.ptr.PointerByReference in_args,
com.sun.jna.ptr.PointerByReference arg_grads,
java.nio.IntBuffer num_aux_states,
com.sun.jna.ptr.PointerByReference aux_states,
com.sun.jna.Pointer shared_exec,
com.sun.jna.ptr.PointerByReference out)
int MXExecutorGetOptimizedSymbol(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out)
int MXExecutorSetMonitorCallback(com.sun.jna.Pointer handle,
MxnetLibrary.ExecutorMonitorCallback callback,
com.sun.jna.Pointer callback_handle)
int MXExecutorSetMonitorCallbackEX(com.sun.jna.Pointer handle,
MxnetLibrary.ExecutorMonitorCallback callback,
com.sun.jna.Pointer callback_handle,
byte monitor_all)
int MXListDataIters(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array)
int MXDataIterCreateIter(com.sun.jna.Pointer handle,
int num_param,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.ptr.PointerByReference out)
int MXDataIterGetIterInfo(com.sun.jna.Pointer creator,
java.lang.String[] name,
java.lang.String[] description,
java.nio.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)
int MXDataIterFree(com.sun.jna.Pointer handle)
int MXDataIterNext(com.sun.jna.Pointer handle,
java.nio.IntBuffer out)
int MXDataIterBeforeFirst(com.sun.jna.Pointer handle)
int MXDataIterGetData(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out)
int MXDataIterGetIndex(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out_index,
java.nio.LongBuffer out_size)
int MXDataIterGetPadNum(com.sun.jna.Pointer handle,
java.nio.IntBuffer pad)
int MXDataIterGetLabel(com.sun.jna.Pointer handle,
com.sun.jna.ptr.PointerByReference out)
int MXInitPSEnv(int num_vars,
java.lang.String[] keys,
java.lang.String[] vals)
int MXKVStoreCreate(java.lang.String type,
com.sun.jna.ptr.PointerByReference out)
int MXKVStoreSetGradientCompression(com.sun.jna.Pointer handle,
int num_params,
java.lang.String[] keys,
java.lang.String[] vals)
int MXKVStoreFree(com.sun.jna.Pointer handle)
int MXKVStoreInit(com.sun.jna.Pointer handle,
int num,
int[] keys,
PointerArray vals)
int MXKVStoreInitEx(com.sun.jna.Pointer handle,
int num,
java.lang.String[] keys,
PointerArray vals)
int MXKVStorePush(com.sun.jna.Pointer handle,
int num,
int[] keys,
PointerArray vals,
int priority)
int MXKVStorePushEx(com.sun.jna.Pointer handle,
int num,
java.lang.String[] keys,
PointerArray vals,
int priority)
int MXKVStorePullWithSparse(com.sun.jna.Pointer handle,
int num,
int[] keys,
com.sun.jna.ptr.PointerByReference vals,
int priority,
byte ignore_sparse)
int MXKVStorePullWithSparseEx(com.sun.jna.Pointer handle,
int num,
java.lang.String[] keys,
com.sun.jna.ptr.PointerByReference vals,
int priority,
byte ignore_sparse)
int MXKVStorePull(com.sun.jna.Pointer handle,
int num,
int[] keys,
PointerArray vals,
int priority)
int MXKVStorePullEx(com.sun.jna.Pointer handle,
int num,
java.lang.String[] keys,
PointerArray vals,
int priority)
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)
int MXKVStorePullRowSparseEx(com.sun.jna.Pointer handle,
int num,
java.lang.String[] keys,
com.sun.jna.ptr.PointerByReference vals,
com.sun.jna.ptr.PointerByReference row_ids,
int priority)
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)
int MXKVStoreBroadcastEx(com.sun.jna.Pointer handle,
int vnum,
java.lang.String[] vkeys,
int onum,
java.lang.String[] okeys,
com.sun.jna.ptr.PointerByReference vals,
com.sun.jna.ptr.PointerByReference outs,
int priority)
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)
int MXKVStorePushPullEx(com.sun.jna.Pointer handle,
int vnum,
java.lang.String[] vkeys,
int onum,
java.lang.String[] okeys,
com.sun.jna.ptr.PointerByReference vals,
com.sun.jna.ptr.PointerByReference outs,
int priority)
int MXKVStoreSetUpdater(com.sun.jna.Pointer handle,
MxnetLibrary.MXKVStoreUpdater updater,
com.sun.jna.Pointer updater_handle)
int MXKVStoreSetUpdaterEx(com.sun.jna.Pointer handle,
MxnetLibrary.MXKVStoreUpdater updater,
MxnetLibrary.MXKVStoreStrUpdater str_updater,
com.sun.jna.Pointer updater_handle)
int MXKVStoreGetType(com.sun.jna.Pointer handle,
java.lang.String[] type)
int MXKVStoreGetRank(com.sun.jna.Pointer handle,
java.nio.IntBuffer ret)
int MXKVStoreGetGroupSize(com.sun.jna.Pointer handle,
java.nio.IntBuffer ret)
int MXKVStoreIsWorkerNode(java.nio.IntBuffer ret)
int MXKVStoreIsServerNode(java.nio.IntBuffer ret)
int MXKVStoreIsSchedulerNode(java.nio.IntBuffer ret)
int MXKVStoreBarrier(com.sun.jna.Pointer handle)
int MXKVStoreSetBarrierBeforeExit(com.sun.jna.Pointer handle,
int barrier_before_exit)
int MXKVStoreRunServer(com.sun.jna.Pointer handle,
MxnetLibrary.MXKVStoreServerController controller,
com.sun.jna.Pointer controller_handle)
int MXKVStoreSendCommmandToServers(com.sun.jna.Pointer handle,
int cmd_id,
java.lang.String cmd_body)
int MXKVStoreGetNumDeadNode(com.sun.jna.Pointer handle,
int node_id,
java.nio.IntBuffer number,
int timeout_sec)
int MXRecordIOWriterCreate(java.lang.String uri,
com.sun.jna.ptr.PointerByReference out)
int MXRecordIOWriterFree(com.sun.jna.Pointer handle)
int MXRecordIOWriterWriteRecord(com.sun.jna.Pointer handle,
java.lang.String buf,
NativeSize size)
int MXRecordIOWriterTell(com.sun.jna.Pointer handle,
NativeSizeByReference pos)
int MXRecordIOReaderCreate(java.lang.String uri,
com.sun.jna.ptr.PointerByReference out)
int MXRecordIOReaderFree(com.sun.jna.Pointer handle)
int MXRecordIOReaderReadRecord(com.sun.jna.Pointer handle,
java.lang.String buf,
NativeSizeByReference size)
int MXRecordIOReaderSeek(com.sun.jna.Pointer handle,
NativeSize pos)
int MXRecordIOReaderTell(com.sun.jna.Pointer handle,
NativeSizeByReference pos)
int MXRtcCreate(java.nio.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,
java.nio.ByteBuffer kernel,
com.sun.jna.ptr.PointerByReference out)
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)
int MXRtcFree(com.sun.jna.Pointer handle)
int MXCustomOpRegister(java.lang.String op_type,
MxnetLibrary.CustomOpPropCreator creator)
int MXCustomFunctionRecord(int num_inputs,
com.sun.jna.ptr.PointerByReference inputs,
int num_outputs,
com.sun.jna.ptr.PointerByReference outputs,
MXCallbackList.ByReference callbacks)
int MXRtcCudaModuleCreate(java.lang.String source,
int num_options,
java.lang.String[] options,
int num_exports,
java.lang.String[] exports,
com.sun.jna.ptr.PointerByReference out)
int MXRtcCudaModuleFree(com.sun.jna.Pointer handle)
int MXRtcCudaKernelCreate(com.sun.jna.Pointer handle,
java.lang.String name,
int num_args,
java.nio.IntBuffer is_ndarray,
java.nio.IntBuffer is_const,
java.nio.IntBuffer arg_types,
com.sun.jna.ptr.PointerByReference out)
int MXRtcCudaKernelFree(com.sun.jna.Pointer handle)
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)
int MXNDArrayGetSharedMemHandle(com.sun.jna.Pointer handle,
java.nio.IntBuffer shared_pid,
java.nio.IntBuffer shared_id)
int MXNDArrayCreateFromSharedMem(int shared_pid,
int shared_id,
int[] shape,
int ndim,
int dtype,
com.sun.jna.ptr.PointerByReference out)
int MXStorageEmptyCache(int dev_type,
int dev_id)
int MXNDArrayCreateFromSharedMemEx(int shared_pid,
int shared_id,
int[] shape,
int ndim,
int dtype,
com.sun.jna.ptr.PointerByReference out)
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, java.lang.String opr_name, byte wait)
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, java.lang.String opr_name)
int MXShallowCopyNDArray(com.sun.jna.Pointer src,
com.sun.jna.ptr.PointerByReference out)
int MXShallowCopySymbol(com.sun.jna.Pointer src,
com.sun.jna.ptr.PointerByReference out)
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, java.lang.String opr_name, byte wait)
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, java.lang.String opr_name)
void NNAPISetLastError(java.lang.String msg)
java.lang.String NNGetLastError()
int NNListAllOpNames(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array)
int NNGetOpHandle(java.lang.String op_name,
com.sun.jna.ptr.PointerByReference op_out)
int NNListUniqueOps(java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_array)
int NNGetOpInfo(com.sun.jna.Pointer op,
java.lang.String[] real_name,
java.lang.String[] description,
java.nio.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,
java.lang.String[] return_type)
int NNSymbolCreateAtomicSymbol(com.sun.jna.Pointer op,
int num_param,
java.lang.String[] keys,
java.lang.String[] vals,
com.sun.jna.ptr.PointerByReference out)
int NNSymbolCreateVariable(java.lang.String name,
com.sun.jna.ptr.PointerByReference out)
int NNSymbolCreateGroup(int num_symbols,
com.sun.jna.ptr.PointerByReference symbols,
com.sun.jna.ptr.PointerByReference out)
int NNAddControlDeps(com.sun.jna.Pointer handle,
com.sun.jna.Pointer src_dep)
int NNSymbolFree(com.sun.jna.Pointer symbol)
int NNSymbolCopy(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out)
int NNSymbolPrint(com.sun.jna.Pointer symbol,
java.lang.String[] out_str)
int NNSymbolGetAttr(com.sun.jna.Pointer symbol,
java.lang.String key,
java.lang.String[] out,
java.nio.IntBuffer success)
int NNSymbolSetAttrs(com.sun.jna.Pointer symbol,
int num_param,
java.lang.String[] keys,
java.lang.String[] values)
int NNSymbolListAttrs(com.sun.jna.Pointer symbol,
int recursive_option,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out)
int NNSymbolListInputVariables(com.sun.jna.Pointer symbol,
int option,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_sym_array)
int NNSymbolListInputNames(com.sun.jna.Pointer symbol,
int option,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_str_array)
int NNSymbolListOutputNames(com.sun.jna.Pointer symbol,
java.nio.IntBuffer out_size,
com.sun.jna.ptr.PointerByReference out_str_array)
int NNSymbolGetNumOutputs(com.sun.jna.Pointer symbol,
java.nio.IntBuffer output_count)
int NNSymbolGetInternals(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out)
int NNSymbolGetChildren(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference out)
int NNSymbolGetOutput(com.sun.jna.Pointer symbol,
int index,
com.sun.jna.ptr.PointerByReference out)
int NNSymbolCompose(com.sun.jna.Pointer sym,
java.lang.String name,
int num_args,
java.lang.String[] keys,
com.sun.jna.ptr.PointerByReference args)
int NNGraphCreate(com.sun.jna.Pointer symbol,
com.sun.jna.ptr.PointerByReference graph)
int NNGraphFree(com.sun.jna.Pointer handle)
int NNGraphGetSymbol(com.sun.jna.Pointer graph,
com.sun.jna.ptr.PointerByReference symbol)
int NNGraphSetJSONAttr(com.sun.jna.Pointer handle,
java.lang.String key,
java.lang.String json_value)
int NNGraphGetJSONAttr(com.sun.jna.Pointer handle,
java.lang.String key,
java.lang.String[] json_out,
java.nio.IntBuffer success)
int NNGraphSetNodeEntryListAttr_(com.sun.jna.Pointer handle,
java.lang.String key,
com.sun.jna.Pointer list)
int NNGraphApplyPasses(com.sun.jna.Pointer src,
int num_pass,
java.lang.String[] pass_names,
com.sun.jna.ptr.PointerByReference dst)