public class mxnet extends mxnet
Modifier and Type | Field and Description |
---|---|
static int |
kCustomFunctionBackward
enum CustomFunctionCallbacks
|
static int |
kCustomFunctionDelete
enum CustomFunctionCallbacks
|
static int |
kCustomOpBackward
enum CustomOpCallbacks
|
static int |
kCustomOpDelete
enum CustomOpCallbacks
|
static int |
kCustomOpForward
enum CustomOpCallbacks
|
static int |
kCustomOpPropBackwardInferStorageType
enum CustomOpPropCallbacks
|
static int |
kCustomOpPropCreateOperator
enum CustomOpPropCallbacks
|
static int |
kCustomOpPropDeclareBackwardDependency
enum CustomOpPropCallbacks
|
static int |
kCustomOpPropDelete
enum CustomOpPropCallbacks
|
static int |
kCustomOpPropInferShape
enum CustomOpPropCallbacks
|
static int |
kCustomOpPropInferStorageType
enum CustomOpPropCallbacks
|
static int |
kCustomOpPropInferType
enum CustomOpPropCallbacks
|
static int |
kCustomOpPropListArguments
enum CustomOpPropCallbacks
|
static int |
kCustomOpPropListAuxiliaryStates
enum CustomOpPropCallbacks
|
static int |
kCustomOpPropListOutputs
enum CustomOpPropCallbacks
|
Constructor and Description |
---|
mxnet() |
Modifier and Type | Method and Description |
---|---|
static int |
MXAggregateProfileStatsPrint(byte[] out_str,
int reset) |
static int |
MXAggregateProfileStatsPrint(ByteBuffer out_str,
int reset) |
static int |
MXAggregateProfileStatsPrint(BytePointer out_str,
int reset) |
static int |
MXAggregateProfileStatsPrint(PointerPointer out_str,
int reset)
\brief Deprecated, use MXAggregateProfileStatsPrintEx instead.
|
static int |
MXAggregateProfileStatsPrintEx(byte[] out_str,
int reset,
int format,
int sort_by,
int ascending) |
static int |
MXAggregateProfileStatsPrintEx(ByteBuffer out_str,
int reset,
int format,
int sort_by,
int ascending) |
static int |
MXAggregateProfileStatsPrintEx(BytePointer out_str,
int reset,
int format,
int sort_by,
int ascending) |
static int |
MXAggregateProfileStatsPrintEx(PointerPointer out_str,
int reset,
int format,
int sort_by,
int ascending)
\brief Print sorted aggregate stats to the a string
How aggregate stats are stored will not change
|
static int |
MXAutogradBackward(int num_output,
NDArrayHandle output_handles,
NDArrayHandle ograd_handles,
int retain_graph) |
static int |
MXAutogradBackward(int num_output,
PointerPointer output_handles,
PointerPointer ograd_handles,
int retain_graph)
\brief compute the gradient of outputs w.r.t variabels
|
static int |
MXAutogradBackwardEx(int num_output,
NDArrayHandle output_handles,
NDArrayHandle ograd_handles,
int num_variables,
NDArrayHandle var_handles,
int retain_graph,
int create_graph,
int is_train,
PointerPointer grad_handles,
int[] grad_stypes) |
static int |
MXAutogradBackwardEx(int num_output,
NDArrayHandle output_handles,
NDArrayHandle ograd_handles,
int num_variables,
NDArrayHandle var_handles,
int retain_graph,
int create_graph,
int is_train,
PointerPointer grad_handles,
IntBuffer grad_stypes) |
static int |
MXAutogradBackwardEx(int num_output,
NDArrayHandle output_handles,
NDArrayHandle ograd_handles,
int num_variables,
NDArrayHandle var_handles,
int retain_graph,
int create_graph,
int is_train,
PointerPointer grad_handles,
IntPointer grad_stypes) |
static int |
MXAutogradBackwardEx(int num_output,
PointerPointer output_handles,
PointerPointer ograd_handles,
int num_variables,
PointerPointer var_handles,
int retain_graph,
int create_graph,
int is_train,
PointerPointer grad_handles,
int[] grad_stypes) |
static int |
MXAutogradBackwardEx(int num_output,
PointerPointer output_handles,
PointerPointer ograd_handles,
int num_variables,
PointerPointer var_handles,
int retain_graph,
int create_graph,
int is_train,
PointerPointer grad_handles,
IntBuffer grad_stypes) |
static int |
MXAutogradBackwardEx(int num_output,
PointerPointer output_handles,
PointerPointer ograd_handles,
int num_variables,
PointerPointer var_handles,
int retain_graph,
int create_graph,
int is_train,
PointerPointer grad_handles,
IntPointer grad_stypes) |
static int |
MXAutogradBackwardEx(int num_output,
PointerPointer output_handles,
PointerPointer ograd_handles,
int num_variables,
PointerPointer var_handles,
int retain_graph,
int create_graph,
int is_train,
PointerPointer grad_handles,
PointerPointer grad_stypes)
\brief compute the gradient of outputs w.r.t variabels
|
static int |
MXAutogradComputeGradient(int num_output,
NDArrayHandle output_handles) |
static int |
MXAutogradComputeGradient(int num_output,
PointerPointer output_handles)
\brief compute the gradient of outputs w.r.t variabels
|
static int |
MXAutogradGetSymbol(NDArrayHandle handle,
PointerPointer out) |
static int |
MXAutogradGetSymbol(NDArrayHandle handle,
SymbolHandle out) |
static int |
MXAutogradIsRecording(boolean[] curr) |
static int |
MXAutogradIsRecording(BoolPointer curr)
\brief get whether autograd recording is on
|
static int |
MXAutogradIsTraining(boolean[] curr) |
static int |
MXAutogradIsTraining(BoolPointer curr)
\brief get whether training mode is on
|
static int |
MXAutogradMarkVariables(int num_var,
NDArrayHandle var_handles,
int[] reqs_array,
NDArrayHandle grad_handles) |
static int |
MXAutogradMarkVariables(int num_var,
NDArrayHandle var_handles,
IntBuffer reqs_array,
NDArrayHandle grad_handles) |
static int |
MXAutogradMarkVariables(int num_var,
NDArrayHandle var_handles,
IntPointer reqs_array,
NDArrayHandle grad_handles) |
static int |
MXAutogradMarkVariables(int num_var,
PointerPointer var_handles,
int[] reqs_array,
PointerPointer grad_handles) |
static int |
MXAutogradMarkVariables(int num_var,
PointerPointer var_handles,
IntBuffer reqs_array,
PointerPointer grad_handles) |
static int |
MXAutogradMarkVariables(int num_var,
PointerPointer var_handles,
IntPointer reqs_array,
PointerPointer grad_handles)
\brief mark NDArrays as variables to compute gradient for autograd
|
static int |
MXAutogradSetIsRecording(int is_recording,
int[] prev) |
static int |
MXAutogradSetIsRecording(int is_recording,
IntBuffer prev) |
static int |
MXAutogradSetIsRecording(int is_recording,
IntPointer prev)
\brief set whether to record operator for autograd
|
static int |
MXAutogradSetIsTraining(int is_training,
int[] prev) |
static int |
MXAutogradSetIsTraining(int is_training,
IntBuffer prev) |
static int |
MXAutogradSetIsTraining(int is_training,
IntPointer prev)
\brief set whether to record operator for autograd
|
static int |
MXCachedOpRegisterOpHook(NDArrayHandle handle,
CachedOpMonitorCallback callback,
boolean monitor_all)
\brief cached op set monitor callback
|
static int |
MXCreateCachedOp(SymbolHandle handle,
CachedOpHandle out)
\brief create cached operator
|
static int |
MXCreateCachedOpEx(SymbolHandle handle,
int num_flags,
byte[] keys,
byte[] vals,
CachedOpHandle out) |
static int |
MXCreateCachedOpEX(SymbolHandle handle,
int num_flags,
byte[] keys,
byte[] vals,
CachedOpHandle out) |
static int |
MXCreateCachedOpEX(SymbolHandle handle,
int num_flags,
byte[] keys,
byte[] vals,
CachedOpHandle out,
boolean thread_safe) |
static int |
MXCreateCachedOpEx(SymbolHandle handle,
int num_flags,
ByteBuffer keys,
ByteBuffer vals,
CachedOpHandle out) |
static int |
MXCreateCachedOpEX(SymbolHandle handle,
int num_flags,
ByteBuffer keys,
ByteBuffer vals,
CachedOpHandle out) |
static int |
MXCreateCachedOpEX(SymbolHandle handle,
int num_flags,
ByteBuffer keys,
ByteBuffer vals,
CachedOpHandle out,
boolean thread_safe) |
static int |
MXCreateCachedOpEx(SymbolHandle handle,
int num_flags,
BytePointer keys,
BytePointer vals,
CachedOpHandle out) |
static int |
MXCreateCachedOpEX(SymbolHandle handle,
int num_flags,
BytePointer keys,
BytePointer vals,
CachedOpHandle out) |
static int |
MXCreateCachedOpEX(SymbolHandle handle,
int num_flags,
BytePointer keys,
BytePointer vals,
CachedOpHandle out,
boolean thread_safe) |
static int |
MXCreateCachedOpEx(SymbolHandle handle,
int num_flags,
PointerPointer keys,
PointerPointer vals,
CachedOpHandle out)
\brief create cached operator
|
static int |
MXCreateCachedOpEX(SymbolHandle handle,
int num_flags,
PointerPointer keys,
PointerPointer vals,
CachedOpHandle out,
boolean thread_safe)
\brief create cached operator, allows to choose thread_safe version
of cachedop
|
static int |
MXCustomFunctionRecord(int num_inputs,
NDArrayHandle inputs,
int num_outputs,
NDArrayHandle outputs,
MXCallbackList callbacks) |
static int |
MXCustomFunctionRecord(int num_inputs,
PointerPointer inputs,
int num_outputs,
PointerPointer outputs,
MXCallbackList callbacks) |
static int |
MXCustomOpRegister(BytePointer op_type,
CustomOpPropCreator creator) |
static int |
MXCustomOpRegister(String op_type,
CustomOpPropCreator creator) |
static int |
MXDataIterBeforeFirst(DataIterHandle handle)
\brief Call iterator.Reset
|
static int |
MXDataIterCreateIter(DataIterCreator handle,
int num_param,
byte[] keys,
byte[] vals,
DataIterHandle out) |
static int |
MXDataIterCreateIter(DataIterCreator handle,
int num_param,
byte[] keys,
byte[] vals,
PointerPointer out) |
static int |
MXDataIterCreateIter(DataIterCreator handle,
int num_param,
ByteBuffer keys,
ByteBuffer vals,
DataIterHandle out) |
static int |
MXDataIterCreateIter(DataIterCreator handle,
int num_param,
ByteBuffer keys,
ByteBuffer vals,
PointerPointer out) |
static int |
MXDataIterCreateIter(DataIterCreator handle,
int num_param,
BytePointer keys,
BytePointer vals,
DataIterHandle out) |
static int |
MXDataIterCreateIter(DataIterCreator handle,
int num_param,
BytePointer keys,
BytePointer vals,
PointerPointer out) |
static int |
MXDataIterCreateIter(DataIterCreator handle,
int num_param,
PointerPointer keys,
PointerPointer vals,
PointerPointer out)
\brief Init an iterator, init with parameters
the array size of passed in arguments
|
static int |
MXDataIterFree(DataIterHandle handle)
\brief Free the handle to the IO module
|
static int |
MXDataIterGetData(DataIterHandle handle,
NDArrayHandle out) |
static int |
MXDataIterGetData(DataIterHandle handle,
PointerPointer out)
\brief Get the handle to the NDArray of underlying data
|
static int |
MXDataIterGetIndex(DataIterHandle handle,
long[] out_index,
long[] out_size) |
static int |
MXDataIterGetIndex(DataIterHandle handle,
LongBuffer out_index,
LongBuffer out_size) |
static int |
MXDataIterGetIndex(DataIterHandle handle,
LongPointer out_index,
LongPointer out_size) |
static int |
MXDataIterGetIndex(DataIterHandle handle,
PointerPointer out_index,
LongPointer out_size)
\brief Get the image index by array.
|
static int |
MXDataIterGetIterInfo(DataIterCreator creator,
byte[] name,
byte[] description,
int[] num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions) |
static int |
MXDataIterGetIterInfo(DataIterCreator creator,
ByteBuffer name,
ByteBuffer description,
IntBuffer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions) |
static int |
MXDataIterGetIterInfo(DataIterCreator creator,
BytePointer name,
BytePointer description,
IntPointer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions) |
static int |
MXDataIterGetIterInfo(DataIterCreator creator,
PointerPointer name,
PointerPointer description,
IntPointer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions)
\brief Get the detailed information about data iterator.
|
static int |
MXDataIterGetLabel(DataIterHandle handle,
NDArrayHandle out) |
static int |
MXDataIterGetLabel(DataIterHandle handle,
PointerPointer out)
\brief Get the handle to the NDArray of underlying label
|
static int |
MXDataIterGetPadNum(DataIterHandle handle,
int[] pad) |
static int |
MXDataIterGetPadNum(DataIterHandle handle,
IntBuffer pad) |
static int |
MXDataIterGetPadNum(DataIterHandle handle,
IntPointer pad)
\brief Get the padding number in current data batch
|
static int |
MXDataIterNext(DataIterHandle handle,
int[] out) |
static int |
MXDataIterNext(DataIterHandle handle,
IntBuffer out) |
static int |
MXDataIterNext(DataIterHandle handle,
IntPointer out)
\brief Move iterator to next position
|
static int |
MXDumpProcessProfile(int finished,
int profile_process,
KVStoreHandle kvStoreHandle)
\brief Save profile and stop profiler
|
static int |
MXDumpProfile(int finished)
\brief Save profile and stop profiler for worker/current process
|
static int |
MXEnginePushAsync(EngineAsyncFunc async_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
EngineVarHandle const_vars_handle,
int num_const_vars,
EngineVarHandle mutable_vars_handle,
int num_mutable_vars) |
static int |
MXEnginePushAsync(EngineAsyncFunc async_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
EngineVarHandle const_vars_handle,
int num_const_vars,
EngineVarHandle mutable_vars_handle,
int num_mutable_vars,
EngineFnPropertyHandle prop_handle,
int priority,
BytePointer opr_name,
boolean _wait)
\brief Push an asynchronous operation to the engine.
|
static int |
MXEnginePushAsync(EngineAsyncFunc async_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
EngineVarHandle const_vars_handle,
int num_const_vars,
EngineVarHandle mutable_vars_handle,
int num_mutable_vars,
EngineFnPropertyHandle prop_handle,
int priority,
String opr_name,
boolean _wait) |
static int |
MXEnginePushAsyncND(EngineAsyncFunc async_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
NDArrayHandle const_nds_handle,
int num_const_nds,
NDArrayHandle mutable_nds_handle,
int num_mutable_nds) |
static int |
MXEnginePushAsyncND(EngineAsyncFunc async_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
NDArrayHandle const_nds_handle,
int num_const_nds,
NDArrayHandle mutable_nds_handle,
int num_mutable_nds,
EngineFnPropertyHandle prop_handle,
int priority,
String opr_name,
boolean _wait) |
static int |
MXEnginePushAsyncND(EngineAsyncFunc async_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
PointerPointer const_nds_handle,
int num_const_nds,
PointerPointer mutable_nds_handle,
int num_mutable_nds) |
static int |
MXEnginePushAsyncND(EngineAsyncFunc async_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
PointerPointer const_nds_handle,
int num_const_nds,
PointerPointer mutable_nds_handle,
int num_mutable_nds,
EngineFnPropertyHandle prop_handle,
int priority,
BytePointer opr_name,
boolean _wait)
\brief Push an asynchronous operation to the engine.
|
static int |
MXEnginePushSync(EngineSyncFunc sync_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
EngineVarHandle const_vars_handle,
int num_const_vars,
EngineVarHandle mutable_vars_handle,
int num_mutable_vars) |
static int |
MXEnginePushSync(EngineSyncFunc sync_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
EngineVarHandle const_vars_handle,
int num_const_vars,
EngineVarHandle mutable_vars_handle,
int num_mutable_vars,
EngineFnPropertyHandle prop_handle,
int priority,
BytePointer opr_name)
\brief Push a synchronous operation to the engine.
|
static int |
MXEnginePushSync(EngineSyncFunc sync_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
EngineVarHandle const_vars_handle,
int num_const_vars,
EngineVarHandle mutable_vars_handle,
int num_mutable_vars,
EngineFnPropertyHandle prop_handle,
int priority,
String opr_name) |
static int |
MXEnginePushSyncND(EngineSyncFunc sync_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
NDArrayHandle const_nds_handle,
int num_const_nds,
NDArrayHandle mutable_nds_handle,
int num_mutable_nds) |
static int |
MXEnginePushSyncND(EngineSyncFunc sync_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
NDArrayHandle const_nds_handle,
int num_const_nds,
NDArrayHandle mutable_nds_handle,
int num_mutable_nds,
EngineFnPropertyHandle prop_handle,
int priority,
String opr_name) |
static int |
MXEnginePushSyncND(EngineSyncFunc sync_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
PointerPointer const_nds_handle,
int num_const_nds,
PointerPointer mutable_nds_handle,
int num_mutable_nds) |
static int |
MXEnginePushSyncND(EngineSyncFunc sync_func,
Pointer func_param,
EngineFuncParamDeleter deleter,
ContextHandle ctx_handle,
PointerPointer const_nds_handle,
int num_const_nds,
PointerPointer mutable_nds_handle,
int num_mutable_nds,
EngineFnPropertyHandle prop_handle,
int priority,
BytePointer opr_name)
\brief Push a synchronous operation to the engine.
|
static int |
MXEngineSetBulkSize(int bulk_size,
int[] prev_bulk_size) |
static int |
MXEngineSetBulkSize(int bulk_size,
IntBuffer prev_bulk_size) |
static int |
MXEngineSetBulkSize(int bulk_size,
IntPointer prev_bulk_size)
\brief set bulk execution limit
|
static int |
MXExecutorBackward(ExecutorHandle handle,
int len,
NDArrayHandle head_grads) |
static int |
MXExecutorBackward(ExecutorHandle handle,
int len,
PointerPointer head_grads)
\brief Excecutor run backward
|
static int |
MXExecutorBackwardEx(ExecutorHandle handle,
int len,
NDArrayHandle head_grads,
int is_train) |
static int |
MXExecutorBackwardEx(ExecutorHandle handle,
int len,
PointerPointer head_grads,
int is_train)
\brief Excecutor run backward
|
static int |
MXExecutorBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int len,
NDArrayHandle in_args,
NDArrayHandle arg_grad_store,
int[] grad_req_type,
int aux_states_len,
NDArrayHandle aux_states,
ExecutorHandle out) |
static int |
MXExecutorBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int len,
NDArrayHandle in_args,
NDArrayHandle arg_grad_store,
IntBuffer grad_req_type,
int aux_states_len,
NDArrayHandle aux_states,
ExecutorHandle out) |
static int |
MXExecutorBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int len,
NDArrayHandle in_args,
NDArrayHandle arg_grad_store,
IntPointer grad_req_type,
int aux_states_len,
NDArrayHandle aux_states,
ExecutorHandle out) |
static int |
MXExecutorBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
int[] grad_req_type,
int aux_states_len,
PointerPointer aux_states,
PointerPointer out) |
static int |
MXExecutorBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
IntBuffer grad_req_type,
int aux_states_len,
PointerPointer aux_states,
PointerPointer out) |
static int |
MXExecutorBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
IntPointer grad_req_type,
int aux_states_len,
PointerPointer aux_states,
PointerPointer out)
\brief Generate Executor from symbol
|
static int |
MXExecutorBindEX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
byte[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int len,
NDArrayHandle in_args,
NDArrayHandle arg_grad_store,
int[] grad_req_type,
int aux_states_len,
NDArrayHandle aux_states,
ExecutorHandle shared_exec,
ExecutorHandle out) |
static int |
MXExecutorBindEX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
byte[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
int[] grad_req_type,
int aux_states_len,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out) |
static int |
MXExecutorBindEX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
ByteBuffer map_keys,
IntBuffer map_dev_types,
IntBuffer map_dev_ids,
int len,
NDArrayHandle in_args,
NDArrayHandle arg_grad_store,
IntBuffer grad_req_type,
int aux_states_len,
NDArrayHandle aux_states,
ExecutorHandle shared_exec,
ExecutorHandle out) |
static int |
MXExecutorBindEX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
ByteBuffer map_keys,
IntBuffer map_dev_types,
IntBuffer map_dev_ids,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
IntBuffer grad_req_type,
int aux_states_len,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out) |
static int |
MXExecutorBindEX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
BytePointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int len,
NDArrayHandle in_args,
NDArrayHandle arg_grad_store,
IntPointer grad_req_type,
int aux_states_len,
NDArrayHandle aux_states,
ExecutorHandle shared_exec,
ExecutorHandle out) |
static int |
MXExecutorBindEX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
BytePointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
IntPointer grad_req_type,
int aux_states_len,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out) |
static int |
MXExecutorBindEX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
PointerPointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
IntPointer grad_req_type,
int aux_states_len,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out)
\brief Generate Executor from symbol,
This is advanced function, allow specify group2ctx map.
|
static int |
MXExecutorBindX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
byte[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int len,
NDArrayHandle in_args,
NDArrayHandle arg_grad_store,
int[] grad_req_type,
int aux_states_len,
NDArrayHandle aux_states,
ExecutorHandle out) |
static int |
MXExecutorBindX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
byte[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
int[] grad_req_type,
int aux_states_len,
PointerPointer aux_states,
PointerPointer out) |
static int |
MXExecutorBindX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
ByteBuffer map_keys,
IntBuffer map_dev_types,
IntBuffer map_dev_ids,
int len,
NDArrayHandle in_args,
NDArrayHandle arg_grad_store,
IntBuffer grad_req_type,
int aux_states_len,
NDArrayHandle aux_states,
ExecutorHandle out) |
static int |
MXExecutorBindX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
ByteBuffer map_keys,
IntBuffer map_dev_types,
IntBuffer map_dev_ids,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
IntBuffer grad_req_type,
int aux_states_len,
PointerPointer aux_states,
PointerPointer out) |
static int |
MXExecutorBindX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
BytePointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int len,
NDArrayHandle in_args,
NDArrayHandle arg_grad_store,
IntPointer grad_req_type,
int aux_states_len,
NDArrayHandle aux_states,
ExecutorHandle out) |
static int |
MXExecutorBindX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
BytePointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
IntPointer grad_req_type,
int aux_states_len,
PointerPointer aux_states,
PointerPointer out) |
static int |
MXExecutorBindX(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_map_keys,
PointerPointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int len,
PointerPointer in_args,
PointerPointer arg_grad_store,
IntPointer grad_req_type,
int aux_states_len,
PointerPointer aux_states,
PointerPointer out)
\brief Generate Executor from symbol,
This is advanced function, allow specify group2ctx map.
|
static int |
MXExecutorForward(ExecutorHandle handle,
int is_train)
\brief Executor forward method
|
static int |
MXExecutorFree(ExecutorHandle handle)
\brief Delete the executor
|
static int |
MXExecutorGetOptimizedSymbol(ExecutorHandle handle,
PointerPointer out)
\brief get optimized graph from graph executor
|
static int |
MXExecutorGetOptimizedSymbol(ExecutorHandle handle,
SymbolHandle out) |
static int |
MXExecutorOutputs(ExecutorHandle handle,
int[] out_size,
PointerPointer out) |
static int |
MXExecutorOutputs(ExecutorHandle handle,
IntBuffer out_size,
PointerPointer out) |
static int |
MXExecutorOutputs(ExecutorHandle handle,
IntPointer out_size,
PointerPointer out)
\brief Get executor's head NDArray
|
static int |
MXExecutorPrint(ExecutorHandle handle,
byte[] out_str) |
static int |
MXExecutorPrint(ExecutorHandle handle,
ByteBuffer out_str) |
static int |
MXExecutorPrint(ExecutorHandle handle,
BytePointer out_str) |
static int |
MXExecutorPrint(ExecutorHandle handle,
PointerPointer out_str)
\brief Print the content of execution plan, used for debug.
|
static int |
MXExecutorReshape(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
byte[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int num_provided_arg_shapes,
byte[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int[] num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
int[] num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
ExecutorHandle out) |
static int |
MXExecutorReshape(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
byte[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int num_provided_arg_shapes,
byte[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int[] num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
int[] num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out) |
static int |
MXExecutorReshape(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
ByteBuffer map_keys,
IntBuffer map_dev_types,
IntBuffer map_dev_ids,
int num_provided_arg_shapes,
ByteBuffer provided_arg_shape_names,
IntBuffer provided_arg_shape_data,
IntBuffer provided_arg_shape_idx,
IntBuffer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntBuffer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
ExecutorHandle out) |
static int |
MXExecutorReshape(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
ByteBuffer map_keys,
IntBuffer map_dev_types,
IntBuffer map_dev_ids,
int num_provided_arg_shapes,
ByteBuffer provided_arg_shape_names,
IntBuffer provided_arg_shape_data,
IntBuffer provided_arg_shape_idx,
IntBuffer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntBuffer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out) |
static int |
MXExecutorReshape(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
BytePointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int num_provided_arg_shapes,
BytePointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
ExecutorHandle out) |
static int |
MXExecutorReshape(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
BytePointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int num_provided_arg_shapes,
BytePointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out) |
static int |
MXExecutorReshape(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
PointerPointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int num_provided_arg_shapes,
PointerPointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out)
\brief DEPRECATED.
|
static int |
MXExecutorReshapeEx(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
byte[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int num_provided_arg_shapes,
byte[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int[] num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
int[] num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
ExecutorHandle out) |
static int |
MXExecutorReshapeEx(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
byte[] map_keys,
int[] map_dev_types,
int[] map_dev_ids,
int num_provided_arg_shapes,
byte[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int[] num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
int[] num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out) |
static int |
MXExecutorReshapeEx(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
ByteBuffer map_keys,
IntBuffer map_dev_types,
IntBuffer map_dev_ids,
int num_provided_arg_shapes,
ByteBuffer provided_arg_shape_names,
IntBuffer provided_arg_shape_data,
IntBuffer provided_arg_shape_idx,
IntBuffer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntBuffer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
ExecutorHandle out) |
static int |
MXExecutorReshapeEx(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
ByteBuffer map_keys,
IntBuffer map_dev_types,
IntBuffer map_dev_ids,
int num_provided_arg_shapes,
ByteBuffer provided_arg_shape_names,
IntBuffer provided_arg_shape_data,
IntBuffer provided_arg_shape_idx,
IntBuffer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntBuffer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out) |
static int |
MXExecutorReshapeEx(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
BytePointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int num_provided_arg_shapes,
BytePointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
ExecutorHandle out) |
static int |
MXExecutorReshapeEx(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
BytePointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int num_provided_arg_shapes,
BytePointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out) |
static int |
MXExecutorReshapeEx(int partial_shaping,
int allow_up_sizing,
int dev_type,
int dev_id,
int num_map_keys,
PointerPointer map_keys,
IntPointer map_dev_types,
IntPointer map_dev_ids,
int num_provided_arg_shapes,
PointerPointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec,
PointerPointer out)
\brief Return a new executor with the same symbol and shared memory,
but different input/output shapes.
|
static int |
MXExecutorSetMonitorCallback(ExecutorHandle handle,
ExecutorMonitorCallback callback,
Pointer callback_handle)
\brief set a call back to notify the completion of operation
|
static int |
MXExecutorSetMonitorCallbackEX(ExecutorHandle handle,
ExecutorMonitorCallback callback,
Pointer callback_handle,
boolean monitor_all)
\brief set a call back to notify the completion of operation
|
static int |
MXExecutorSimpleBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
byte[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
byte[] provided_grad_req_names,
byte[] provided_grad_req_types,
int num_provided_arg_shapes,
byte[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
byte[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
byte[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
byte[] shared_arg_name_list,
int[] shared_buffer_len,
byte[] shared_buffer_name_list,
NDArrayHandle shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
int[] num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
int[] num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
ExecutorHandle out) |
static int |
MXExecutorSimpleBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
byte[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
byte[] provided_grad_req_names,
byte[] provided_grad_req_types,
int num_provided_arg_shapes,
byte[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
byte[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
byte[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
byte[] shared_arg_name_list,
int[] shared_buffer_len,
byte[] shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
int[] num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
int[] num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXExecutorSimpleBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
ByteBuffer g2c_keys,
IntBuffer g2c_dev_types,
IntBuffer g2c_dev_ids,
int provided_grad_req_list_len,
ByteBuffer provided_grad_req_names,
ByteBuffer provided_grad_req_types,
int num_provided_arg_shapes,
ByteBuffer provided_arg_shape_names,
IntBuffer provided_arg_shape_data,
IntBuffer provided_arg_shape_idx,
int num_provided_arg_dtypes,
ByteBuffer provided_arg_dtype_names,
IntBuffer provided_arg_dtypes,
int num_provided_arg_stypes,
ByteBuffer provided_arg_stype_names,
IntBuffer provided_arg_stypes,
int num_shared_arg_names,
ByteBuffer shared_arg_name_list,
IntBuffer shared_buffer_len,
ByteBuffer shared_buffer_name_list,
NDArrayHandle shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntBuffer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntBuffer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
ExecutorHandle out) |
static int |
MXExecutorSimpleBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
ByteBuffer g2c_keys,
IntBuffer g2c_dev_types,
IntBuffer g2c_dev_ids,
int provided_grad_req_list_len,
ByteBuffer provided_grad_req_names,
ByteBuffer provided_grad_req_types,
int num_provided_arg_shapes,
ByteBuffer provided_arg_shape_names,
IntBuffer provided_arg_shape_data,
IntBuffer provided_arg_shape_idx,
int num_provided_arg_dtypes,
ByteBuffer provided_arg_dtype_names,
IntBuffer provided_arg_dtypes,
int num_provided_arg_stypes,
ByteBuffer provided_arg_stype_names,
IntBuffer provided_arg_stypes,
int num_shared_arg_names,
ByteBuffer shared_arg_name_list,
IntBuffer shared_buffer_len,
ByteBuffer shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntBuffer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntBuffer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXExecutorSimpleBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
BytePointer g2c_keys,
IntPointer g2c_dev_types,
IntPointer g2c_dev_ids,
int provided_grad_req_list_len,
BytePointer provided_grad_req_names,
BytePointer provided_grad_req_types,
int num_provided_arg_shapes,
BytePointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
int num_provided_arg_dtypes,
BytePointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
int num_provided_arg_stypes,
BytePointer provided_arg_stype_names,
IntPointer provided_arg_stypes,
int num_shared_arg_names,
BytePointer shared_arg_name_list,
IntPointer shared_buffer_len,
BytePointer shared_buffer_name_list,
NDArrayHandle shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
ExecutorHandle out) |
static int |
MXExecutorSimpleBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
BytePointer g2c_keys,
IntPointer g2c_dev_types,
IntPointer g2c_dev_ids,
int provided_grad_req_list_len,
BytePointer provided_grad_req_names,
BytePointer provided_grad_req_types,
int num_provided_arg_shapes,
BytePointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
int num_provided_arg_dtypes,
BytePointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
int num_provided_arg_stypes,
BytePointer provided_arg_stype_names,
IntPointer provided_arg_stypes,
int num_shared_arg_names,
BytePointer shared_arg_name_list,
IntPointer shared_buffer_len,
BytePointer shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXExecutorSimpleBind(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
PointerPointer g2c_keys,
IntPointer g2c_dev_types,
IntPointer g2c_dev_ids,
int provided_grad_req_list_len,
PointerPointer provided_grad_req_names,
PointerPointer provided_grad_req_types,
int num_provided_arg_shapes,
PointerPointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
int num_provided_arg_dtypes,
PointerPointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
int num_provided_arg_stypes,
PointerPointer provided_arg_stype_names,
IntPointer provided_arg_stypes,
int num_shared_arg_names,
PointerPointer shared_arg_name_list,
IntPointer shared_buffer_len,
PointerPointer shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out)
\brief DEPRECATED.
|
static int |
MXExecutorSimpleBindEx(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
byte[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
byte[] provided_grad_req_names,
byte[] provided_grad_req_types,
int num_provided_arg_shapes,
byte[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
byte[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
byte[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
byte[] shared_arg_name_list,
int[] shared_buffer_len,
byte[] shared_buffer_name_list,
NDArrayHandle shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
int[] num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
int[] num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
ExecutorHandle out) |
static int |
MXExecutorSimpleBindEx(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
byte[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
byte[] provided_grad_req_names,
byte[] provided_grad_req_types,
int num_provided_arg_shapes,
byte[] provided_arg_shape_names,
int[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
byte[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
byte[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
byte[] shared_arg_name_list,
int[] shared_buffer_len,
byte[] shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
int[] num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
int[] num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXExecutorSimpleBindEx(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
ByteBuffer g2c_keys,
IntBuffer g2c_dev_types,
IntBuffer g2c_dev_ids,
int provided_grad_req_list_len,
ByteBuffer provided_grad_req_names,
ByteBuffer provided_grad_req_types,
int num_provided_arg_shapes,
ByteBuffer provided_arg_shape_names,
IntBuffer provided_arg_shape_data,
IntBuffer provided_arg_shape_idx,
int num_provided_arg_dtypes,
ByteBuffer provided_arg_dtype_names,
IntBuffer provided_arg_dtypes,
int num_provided_arg_stypes,
ByteBuffer provided_arg_stype_names,
IntBuffer provided_arg_stypes,
int num_shared_arg_names,
ByteBuffer shared_arg_name_list,
IntBuffer shared_buffer_len,
ByteBuffer shared_buffer_name_list,
NDArrayHandle shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntBuffer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntBuffer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
ExecutorHandle out) |
static int |
MXExecutorSimpleBindEx(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
ByteBuffer g2c_keys,
IntBuffer g2c_dev_types,
IntBuffer g2c_dev_ids,
int provided_grad_req_list_len,
ByteBuffer provided_grad_req_names,
ByteBuffer provided_grad_req_types,
int num_provided_arg_shapes,
ByteBuffer provided_arg_shape_names,
IntBuffer provided_arg_shape_data,
IntBuffer provided_arg_shape_idx,
int num_provided_arg_dtypes,
ByteBuffer provided_arg_dtype_names,
IntBuffer provided_arg_dtypes,
int num_provided_arg_stypes,
ByteBuffer provided_arg_stype_names,
IntBuffer provided_arg_stypes,
int num_shared_arg_names,
ByteBuffer shared_arg_name_list,
IntBuffer shared_buffer_len,
ByteBuffer shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntBuffer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntBuffer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXExecutorSimpleBindEx(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
BytePointer g2c_keys,
IntPointer g2c_dev_types,
IntPointer g2c_dev_ids,
int provided_grad_req_list_len,
BytePointer provided_grad_req_names,
BytePointer provided_grad_req_types,
int num_provided_arg_shapes,
BytePointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
int num_provided_arg_dtypes,
BytePointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
int num_provided_arg_stypes,
BytePointer provided_arg_stype_names,
IntPointer provided_arg_stypes,
int num_shared_arg_names,
BytePointer shared_arg_name_list,
IntPointer shared_buffer_len,
BytePointer shared_buffer_name_list,
NDArrayHandle shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
ExecutorHandle out) |
static int |
MXExecutorSimpleBindEx(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
BytePointer g2c_keys,
IntPointer g2c_dev_types,
IntPointer g2c_dev_ids,
int provided_grad_req_list_len,
BytePointer provided_grad_req_names,
BytePointer provided_grad_req_types,
int num_provided_arg_shapes,
BytePointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
int num_provided_arg_dtypes,
BytePointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
int num_provided_arg_stypes,
BytePointer provided_arg_stype_names,
IntPointer provided_arg_stypes,
int num_shared_arg_names,
BytePointer shared_arg_name_list,
IntPointer shared_buffer_len,
BytePointer shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXExecutorSimpleBindEx(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
PointerPointer g2c_keys,
IntPointer g2c_dev_types,
IntPointer g2c_dev_ids,
int provided_grad_req_list_len,
PointerPointer provided_grad_req_names,
PointerPointer provided_grad_req_types,
int num_provided_arg_shapes,
PointerPointer provided_arg_shape_names,
IntPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
int num_provided_arg_dtypes,
PointerPointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
int num_provided_arg_stypes,
PointerPointer provided_arg_stype_names,
IntPointer provided_arg_stypes,
int num_shared_arg_names,
PointerPointer shared_arg_name_list,
IntPointer shared_buffer_len,
PointerPointer shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXExecutorSimpleBindEx64(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
byte[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
byte[] provided_grad_req_names,
byte[] provided_grad_req_types,
int num_provided_arg_shapes,
byte[] provided_arg_shape_names,
long[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
byte[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
byte[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
byte[] shared_arg_name_list,
int[] shared_buffer_len,
byte[] shared_buffer_name_list,
NDArrayHandle shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
int[] num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
int[] num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
ExecutorHandle out) |
static int |
MXExecutorSimpleBindEx64(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
byte[] g2c_keys,
int[] g2c_dev_types,
int[] g2c_dev_ids,
int provided_grad_req_list_len,
byte[] provided_grad_req_names,
byte[] provided_grad_req_types,
int num_provided_arg_shapes,
byte[] provided_arg_shape_names,
long[] provided_arg_shape_data,
int[] provided_arg_shape_idx,
int num_provided_arg_dtypes,
byte[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
int num_provided_arg_stypes,
byte[] provided_arg_stype_names,
int[] provided_arg_stypes,
int num_shared_arg_names,
byte[] shared_arg_name_list,
int[] shared_buffer_len,
byte[] shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
int[] num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
int[] num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXExecutorSimpleBindEx64(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
ByteBuffer g2c_keys,
IntBuffer g2c_dev_types,
IntBuffer g2c_dev_ids,
int provided_grad_req_list_len,
ByteBuffer provided_grad_req_names,
ByteBuffer provided_grad_req_types,
int num_provided_arg_shapes,
ByteBuffer provided_arg_shape_names,
LongBuffer provided_arg_shape_data,
IntBuffer provided_arg_shape_idx,
int num_provided_arg_dtypes,
ByteBuffer provided_arg_dtype_names,
IntBuffer provided_arg_dtypes,
int num_provided_arg_stypes,
ByteBuffer provided_arg_stype_names,
IntBuffer provided_arg_stypes,
int num_shared_arg_names,
ByteBuffer shared_arg_name_list,
IntBuffer shared_buffer_len,
ByteBuffer shared_buffer_name_list,
NDArrayHandle shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntBuffer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntBuffer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
ExecutorHandle out) |
static int |
MXExecutorSimpleBindEx64(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
ByteBuffer g2c_keys,
IntBuffer g2c_dev_types,
IntBuffer g2c_dev_ids,
int provided_grad_req_list_len,
ByteBuffer provided_grad_req_names,
ByteBuffer provided_grad_req_types,
int num_provided_arg_shapes,
ByteBuffer provided_arg_shape_names,
LongBuffer provided_arg_shape_data,
IntBuffer provided_arg_shape_idx,
int num_provided_arg_dtypes,
ByteBuffer provided_arg_dtype_names,
IntBuffer provided_arg_dtypes,
int num_provided_arg_stypes,
ByteBuffer provided_arg_stype_names,
IntBuffer provided_arg_stypes,
int num_shared_arg_names,
ByteBuffer shared_arg_name_list,
IntBuffer shared_buffer_len,
ByteBuffer shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntBuffer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntBuffer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXExecutorSimpleBindEx64(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
BytePointer g2c_keys,
IntPointer g2c_dev_types,
IntPointer g2c_dev_ids,
int provided_grad_req_list_len,
BytePointer provided_grad_req_names,
BytePointer provided_grad_req_types,
int num_provided_arg_shapes,
BytePointer provided_arg_shape_names,
LongPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
int num_provided_arg_dtypes,
BytePointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
int num_provided_arg_stypes,
BytePointer provided_arg_stype_names,
IntPointer provided_arg_stypes,
int num_shared_arg_names,
BytePointer shared_arg_name_list,
IntPointer shared_buffer_len,
BytePointer shared_buffer_name_list,
NDArrayHandle shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
ExecutorHandle out) |
static int |
MXExecutorSimpleBindEx64(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
BytePointer g2c_keys,
IntPointer g2c_dev_types,
IntPointer g2c_dev_ids,
int provided_grad_req_list_len,
BytePointer provided_grad_req_names,
BytePointer provided_grad_req_types,
int num_provided_arg_shapes,
BytePointer provided_arg_shape_names,
LongPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
int num_provided_arg_dtypes,
BytePointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
int num_provided_arg_stypes,
BytePointer provided_arg_stype_names,
IntPointer provided_arg_stypes,
int num_shared_arg_names,
BytePointer shared_arg_name_list,
IntPointer shared_buffer_len,
BytePointer shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXExecutorSimpleBindEx64(SymbolHandle symbol_handle,
int dev_type,
int dev_id,
int num_g2c_keys,
PointerPointer g2c_keys,
IntPointer g2c_dev_types,
IntPointer g2c_dev_ids,
int provided_grad_req_list_len,
PointerPointer provided_grad_req_names,
PointerPointer provided_grad_req_types,
int num_provided_arg_shapes,
PointerPointer provided_arg_shape_names,
LongPointer provided_arg_shape_data,
IntPointer provided_arg_shape_idx,
int num_provided_arg_dtypes,
PointerPointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
int num_provided_arg_stypes,
PointerPointer provided_arg_stype_names,
IntPointer provided_arg_stypes,
int num_shared_arg_names,
PointerPointer shared_arg_name_list,
IntPointer shared_buffer_len,
PointerPointer shared_buffer_name_list,
PointerPointer shared_buffer_handle_list,
PointerPointer updated_shared_buffer_name_list,
PointerPointer updated_shared_buffer_handle_list,
IntPointer num_in_args,
PointerPointer in_args,
PointerPointer arg_grads,
IntPointer num_aux_states,
PointerPointer aux_states,
ExecutorHandle shared_exec_handle,
PointerPointer out) |
static int |
MXFreeCachedOp(CachedOpHandle handle)
\brief free cached operator
|
static int |
MXFuncDescribe(FunctionHandle fun,
int[] num_use_vars,
int[] num_scalars,
int[] num_mutate_vars,
int[] type_mask) |
static int |
MXFuncDescribe(FunctionHandle fun,
IntBuffer num_use_vars,
IntBuffer num_scalars,
IntBuffer num_mutate_vars,
IntBuffer type_mask) |
static int |
MXFuncDescribe(FunctionHandle fun,
IntPointer num_use_vars,
IntPointer num_scalars,
IntPointer num_mutate_vars,
IntPointer type_mask)
\brief get the argument requirements of the function
|
static int |
MXFuncGetInfo(FunctionHandle fun,
byte[] name,
byte[] description,
int[] num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions) |
static int |
MXFuncGetInfo(FunctionHandle fun,
byte[] name,
byte[] description,
int[] num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
byte[] return_type) |
static int |
MXFuncGetInfo(FunctionHandle fun,
ByteBuffer name,
ByteBuffer description,
IntBuffer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions) |
static int |
MXFuncGetInfo(FunctionHandle fun,
ByteBuffer name,
ByteBuffer description,
IntBuffer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
ByteBuffer return_type) |
static int |
MXFuncGetInfo(FunctionHandle fun,
BytePointer name,
BytePointer description,
IntPointer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions) |
static int |
MXFuncGetInfo(FunctionHandle fun,
BytePointer name,
BytePointer description,
IntPointer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
BytePointer return_type) |
static int |
MXFuncGetInfo(FunctionHandle fun,
PointerPointer name,
PointerPointer description,
IntPointer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
PointerPointer return_type)
\brief Get the information of the function handle.
|
static int |
MXFuncInvoke(FunctionHandle fun,
NDArrayHandle use_vars,
float[] scalar_args,
NDArrayHandle mutate_vars) |
static int |
MXFuncInvoke(FunctionHandle fun,
NDArrayHandle use_vars,
FloatBuffer scalar_args,
NDArrayHandle mutate_vars) |
static int |
MXFuncInvoke(FunctionHandle fun,
NDArrayHandle use_vars,
FloatPointer scalar_args,
NDArrayHandle mutate_vars) |
static int |
MXFuncInvoke(FunctionHandle fun,
PointerPointer use_vars,
float[] scalar_args,
PointerPointer mutate_vars) |
static int |
MXFuncInvoke(FunctionHandle fun,
PointerPointer use_vars,
FloatBuffer scalar_args,
PointerPointer mutate_vars) |
static int |
MXFuncInvoke(FunctionHandle fun,
PointerPointer use_vars,
FloatPointer scalar_args,
PointerPointer mutate_vars)
\brief invoke a function, the array size of passed in arguments
must match the values in the
|
static int |
MXFuncInvokeEx(FunctionHandle fun,
NDArrayHandle use_vars,
float[] scalar_args,
NDArrayHandle mutate_vars,
int num_params,
byte[] param_keys,
byte[] param_vals) |
static int |
MXFuncInvokeEx(FunctionHandle fun,
NDArrayHandle use_vars,
FloatBuffer scalar_args,
NDArrayHandle mutate_vars,
int num_params,
ByteBuffer param_keys,
ByteBuffer param_vals) |
static int |
MXFuncInvokeEx(FunctionHandle fun,
NDArrayHandle use_vars,
FloatPointer scalar_args,
NDArrayHandle mutate_vars,
int num_params,
BytePointer param_keys,
BytePointer param_vals) |
static int |
MXFuncInvokeEx(FunctionHandle fun,
PointerPointer use_vars,
float[] scalar_args,
PointerPointer mutate_vars,
int num_params,
byte[] param_keys,
byte[] param_vals) |
static int |
MXFuncInvokeEx(FunctionHandle fun,
PointerPointer use_vars,
FloatBuffer scalar_args,
PointerPointer mutate_vars,
int num_params,
ByteBuffer param_keys,
ByteBuffer param_vals) |
static int |
MXFuncInvokeEx(FunctionHandle fun,
PointerPointer use_vars,
FloatPointer scalar_args,
PointerPointer mutate_vars,
int num_params,
BytePointer param_keys,
BytePointer param_vals) |
static int |
MXFuncInvokeEx(FunctionHandle fun,
PointerPointer use_vars,
FloatPointer scalar_args,
PointerPointer mutate_vars,
int num_params,
PointerPointer param_keys,
PointerPointer param_vals)
\brief invoke a function, the array size of passed in arguments
must match the values in the
|
static int |
MXGenAtomicSymbolFromSymbol(SymbolHandle sym_handle,
PointerPointer ret_sym_handle)
\brief Generate atomic symbol (able to be composed) from a source symbol
|
static int |
MXGenAtomicSymbolFromSymbol(SymbolHandle sym_handle,
SymbolHandle ret_sym_handle) |
static int |
MXGenBackendSubgraph(SymbolHandle sym_handle,
BytePointer backend,
PointerPointer ret_sym_handle)
\brief Run subgraph pass based on the backend provided
|
static int |
MXGenBackendSubgraph(SymbolHandle sym_handle,
String backend,
SymbolHandle ret_sym_handle) |
static int |
MXGetFunction(BytePointer name,
PointerPointer out)
\brief get the function handle by name
|
static int |
MXGetFunction(String name,
FunctionHandle out) |
static int |
MXGetGPUCount(int[] out) |
static int |
MXGetGPUCount(IntBuffer out) |
static int |
MXGetGPUCount(IntPointer out)
\brief Get the number of GPUs.
|
static int |
MXGetGPUMemoryInformation(int dev,
int[] free_mem,
int[] total_mem) |
static int |
MXGetGPUMemoryInformation(int dev,
IntBuffer free_mem,
IntBuffer total_mem) |
static int |
MXGetGPUMemoryInformation(int dev,
IntPointer free_mem,
IntPointer total_mem)
\brief get the free and total available memory on a GPU
Note: Deprecated, use MXGetGPUMemoryInformation64 instead.
|
static int |
MXGetGPUMemoryInformation64(int dev,
long[] free_mem,
long[] total_mem) |
static int |
MXGetGPUMemoryInformation64(int dev,
LongBuffer free_mem,
LongBuffer total_mem) |
static int |
MXGetGPUMemoryInformation64(int dev,
LongPointer free_mem,
LongPointer total_mem)
\brief get the free and total available memory on a GPU
|
static BytePointer |
MXGetLastError()
\brief return str message of the last error
all function in this file will return 0 when success
and -1 when an error occured,
MXGetLastError can be called to retrieve the error
this function is threadsafe and can be called by different thread
|
static int |
MXGetVersion(int[] out) |
static int |
MXGetVersion(IntBuffer out) |
static int |
MXGetVersion(IntPointer out)
\brief get the MXNet library version as an integer
|
static int |
MXImperativeInvoke(AtomicSymbolCreator creator,
int num_inputs,
NDArrayHandle inputs,
int[] num_outputs,
PointerPointer outputs,
int num_params,
byte[] param_keys,
byte[] param_vals) |
static int |
MXImperativeInvoke(AtomicSymbolCreator creator,
int num_inputs,
NDArrayHandle inputs,
IntBuffer num_outputs,
PointerPointer outputs,
int num_params,
ByteBuffer param_keys,
ByteBuffer param_vals) |
static int |
MXImperativeInvoke(AtomicSymbolCreator creator,
int num_inputs,
NDArrayHandle inputs,
IntPointer num_outputs,
PointerPointer outputs,
int num_params,
BytePointer param_keys,
BytePointer param_vals) |
static int |
MXImperativeInvoke(AtomicSymbolCreator creator,
int num_inputs,
PointerPointer inputs,
int[] num_outputs,
PointerPointer outputs,
int num_params,
byte[] param_keys,
byte[] param_vals) |
static int |
MXImperativeInvoke(AtomicSymbolCreator creator,
int num_inputs,
PointerPointer inputs,
IntBuffer num_outputs,
PointerPointer outputs,
int num_params,
ByteBuffer param_keys,
ByteBuffer param_vals) |
static int |
MXImperativeInvoke(AtomicSymbolCreator creator,
int num_inputs,
PointerPointer inputs,
IntPointer num_outputs,
PointerPointer outputs,
int num_params,
BytePointer param_keys,
BytePointer param_vals) |
static int |
MXImperativeInvoke(AtomicSymbolCreator creator,
int num_inputs,
PointerPointer inputs,
IntPointer num_outputs,
PointerPointer outputs,
int num_params,
PointerPointer param_keys,
PointerPointer param_vals)
\brief invoke a nnvm op and imperative function
|
static int |
MXImperativeInvokeEx(AtomicSymbolCreator creator,
int num_inputs,
NDArrayHandle inputs,
int[] num_outputs,
PointerPointer outputs,
int num_params,
byte[] param_keys,
byte[] param_vals,
int[] out_stypes) |
static int |
MXImperativeInvokeEx(AtomicSymbolCreator creator,
int num_inputs,
NDArrayHandle inputs,
IntBuffer num_outputs,
PointerPointer outputs,
int num_params,
ByteBuffer param_keys,
ByteBuffer param_vals,
IntBuffer out_stypes) |
static int |
MXImperativeInvokeEx(AtomicSymbolCreator creator,
int num_inputs,
NDArrayHandle inputs,
IntPointer num_outputs,
PointerPointer outputs,
int num_params,
BytePointer param_keys,
BytePointer param_vals,
IntPointer out_stypes) |
static int |
MXImperativeInvokeEx(AtomicSymbolCreator creator,
int num_inputs,
PointerPointer inputs,
int[] num_outputs,
PointerPointer outputs,
int num_params,
byte[] param_keys,
byte[] param_vals,
int[] out_stypes) |
static int |
MXImperativeInvokeEx(AtomicSymbolCreator creator,
int num_inputs,
PointerPointer inputs,
IntBuffer num_outputs,
PointerPointer outputs,
int num_params,
ByteBuffer param_keys,
ByteBuffer param_vals,
IntBuffer out_stypes) |
static int |
MXImperativeInvokeEx(AtomicSymbolCreator creator,
int num_inputs,
PointerPointer inputs,
IntPointer num_outputs,
PointerPointer outputs,
int num_params,
BytePointer param_keys,
BytePointer param_vals,
IntPointer out_stypes) |
static int |
MXImperativeInvokeEx(AtomicSymbolCreator creator,
int num_inputs,
PointerPointer inputs,
IntPointer num_outputs,
PointerPointer outputs,
int num_params,
PointerPointer param_keys,
PointerPointer param_vals,
PointerPointer out_stypes)
\brief invoke a nnvm op and imperative function
|
static int |
MXInitPSEnv(int num_vars,
byte[] keys,
byte[] vals) |
static int |
MXInitPSEnv(int num_vars,
ByteBuffer keys,
ByteBuffer vals) |
static int |
MXInitPSEnv(int num_vars,
BytePointer keys,
BytePointer vals) |
static int |
MXInitPSEnv(int num_vars,
PointerPointer keys,
PointerPointer vals)
\brief Initialized ps-lite environment variables
|
static int |
MXInvokeCachedOp(CachedOpHandle handle,
int num_inputs,
NDArrayHandle inputs,
int[] num_outputs,
PointerPointer outputs) |
static int |
MXInvokeCachedOp(CachedOpHandle handle,
int num_inputs,
NDArrayHandle inputs,
IntBuffer num_outputs,
PointerPointer outputs) |
static int |
MXInvokeCachedOp(CachedOpHandle handle,
int num_inputs,
NDArrayHandle inputs,
IntPointer num_outputs,
PointerPointer outputs) |
static int |
MXInvokeCachedOp(CachedOpHandle handle,
int num_inputs,
PointerPointer inputs,
int[] num_outputs,
PointerPointer outputs) |
static int |
MXInvokeCachedOp(CachedOpHandle handle,
int num_inputs,
PointerPointer inputs,
IntBuffer num_outputs,
PointerPointer outputs) |
static int |
MXInvokeCachedOp(CachedOpHandle handle,
int num_inputs,
PointerPointer inputs,
IntPointer num_outputs,
PointerPointer outputs)
\brief invoke cached operator
|
static int |
MXInvokeCachedOpEx(CachedOpHandle handle,
int num_inputs,
NDArrayHandle inputs,
int[] num_outputs,
PointerPointer outputs,
int[] out_stypes) |
static int |
MXInvokeCachedOpEx(CachedOpHandle handle,
int num_inputs,
NDArrayHandle inputs,
IntBuffer num_outputs,
PointerPointer outputs,
IntBuffer out_stypes) |
static int |
MXInvokeCachedOpEx(CachedOpHandle handle,
int num_inputs,
NDArrayHandle inputs,
IntPointer num_outputs,
PointerPointer outputs,
IntPointer out_stypes) |
static int |
MXInvokeCachedOpEx(CachedOpHandle handle,
int num_inputs,
PointerPointer inputs,
int[] num_outputs,
PointerPointer outputs,
int[] out_stypes) |
static int |
MXInvokeCachedOpEx(CachedOpHandle handle,
int num_inputs,
PointerPointer inputs,
IntBuffer num_outputs,
PointerPointer outputs,
IntBuffer out_stypes) |
static int |
MXInvokeCachedOpEx(CachedOpHandle handle,
int num_inputs,
PointerPointer inputs,
IntPointer num_outputs,
PointerPointer outputs,
IntPointer out_stypes) |
static int |
MXInvokeCachedOpEx(CachedOpHandle handle,
int num_inputs,
PointerPointer inputs,
IntPointer num_outputs,
PointerPointer outputs,
PointerPointer out_stypes)
\brief invoke a cached op
|
static int |
MXIsNumpyShape(int[] curr) |
static int |
MXIsNumpyShape(IntBuffer curr) |
static int |
MXIsNumpyShape(IntPointer curr)
\brief get whether numpy compatibility is on
|
static int |
MXKVStoreBarrier(KVStoreHandle handle)
\brief global barrier among all worker machines
|
static int |
MXKVStoreBroadcast(KVStoreHandle handle,
int vnum,
int[] vkeys,
int onum,
int[] okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStoreBroadcast(KVStoreHandle handle,
int vnum,
int[] vkeys,
int onum,
int[] okeys,
PointerPointer vals,
PointerPointer outs,
int priority) |
static int |
MXKVStoreBroadcast(KVStoreHandle handle,
int vnum,
IntBuffer vkeys,
int onum,
IntBuffer okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStoreBroadcast(KVStoreHandle handle,
int vnum,
IntBuffer vkeys,
int onum,
IntBuffer okeys,
PointerPointer vals,
PointerPointer outs,
int priority) |
static int |
MXKVStoreBroadcast(KVStoreHandle handle,
int vnum,
IntPointer vkeys,
int onum,
IntPointer okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStoreBroadcast(KVStoreHandle handle,
int vnum,
IntPointer vkeys,
int onum,
IntPointer okeys,
PointerPointer vals,
PointerPointer outs,
int priority)
\brief broadcast a list of (key, value) pairs from the kvstore
|
static int |
MXKVStoreBroadcastEx(KVStoreHandle handle,
int vnum,
byte[] vkeys,
int onum,
byte[] okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStoreBroadcastEx(KVStoreHandle handle,
int vnum,
byte[] vkeys,
int onum,
byte[] okeys,
PointerPointer vals,
PointerPointer outs,
int priority) |
static int |
MXKVStoreBroadcastEx(KVStoreHandle handle,
int vnum,
ByteBuffer vkeys,
int onum,
ByteBuffer okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStoreBroadcastEx(KVStoreHandle handle,
int vnum,
ByteBuffer vkeys,
int onum,
ByteBuffer okeys,
PointerPointer vals,
PointerPointer outs,
int priority) |
static int |
MXKVStoreBroadcastEx(KVStoreHandle handle,
int vnum,
BytePointer vkeys,
int onum,
BytePointer okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStoreBroadcastEx(KVStoreHandle handle,
int vnum,
BytePointer vkeys,
int onum,
BytePointer okeys,
PointerPointer vals,
PointerPointer outs,
int priority) |
static int |
MXKVStoreBroadcastEx(KVStoreHandle handle,
int vnum,
PointerPointer vkeys,
int onum,
PointerPointer okeys,
PointerPointer vals,
PointerPointer outs,
int priority)
\brief broadcast a list of (key, value) pairs from the kvstore,
where each key is a string
|
static int |
MXKVStoreCreate(BytePointer type,
PointerPointer out)
\brief Create a kvstore
|
static int |
MXKVStoreCreate(String type,
KVStoreHandle out) |
static int |
MXKVStoreFree(KVStoreHandle handle)
\brief Delete a KVStore handle.
|
static int |
MXKVStoreGetGroupSize(KVStoreHandle handle,
int[] ret) |
static int |
MXKVStoreGetGroupSize(KVStoreHandle handle,
IntBuffer ret) |
static int |
MXKVStoreGetGroupSize(KVStoreHandle handle,
IntPointer ret)
\brief return The number of nodes in this group, which is
- number of workers if if
IsWorkerNode() == true ,
- number of servers if if IsServerNode() == true ,
- 1 if IsSchedulerNode() == true , |
static int |
MXKVStoreGetNumDeadNode(KVStoreHandle handle,
int node_id,
int[] number) |
static int |
MXKVStoreGetNumDeadNode(KVStoreHandle handle,
int node_id,
int[] number,
int timeout_sec) |
static int |
MXKVStoreGetNumDeadNode(KVStoreHandle handle,
int node_id,
IntBuffer number) |
static int |
MXKVStoreGetNumDeadNode(KVStoreHandle handle,
int node_id,
IntBuffer number,
int timeout_sec) |
static int |
MXKVStoreGetNumDeadNode(KVStoreHandle handle,
int node_id,
IntPointer number) |
static int |
MXKVStoreGetNumDeadNode(KVStoreHandle handle,
int node_id,
IntPointer number,
int timeout_sec)
\brief Get the number of ps dead node(s) specified by {node_id}
|
static int |
MXKVStoreGetRank(KVStoreHandle handle,
int[] ret) |
static int |
MXKVStoreGetRank(KVStoreHandle handle,
IntBuffer ret) |
static int |
MXKVStoreGetRank(KVStoreHandle handle,
IntPointer ret)
\brief return The rank of this node in its group, which is in [0, GroupSize).
|
static int |
MXKVStoreGetType(KVStoreHandle handle,
byte[] type) |
static int |
MXKVStoreGetType(KVStoreHandle handle,
ByteBuffer type) |
static int |
MXKVStoreGetType(KVStoreHandle handle,
BytePointer type) |
static int |
MXKVStoreGetType(KVStoreHandle handle,
PointerPointer type)
\brief get the type of the kvstore
|
static int |
MXKVStoreInit(KVStoreHandle handle,
int num,
int[] keys,
NDArrayHandle vals) |
static int |
MXKVStoreInit(KVStoreHandle handle,
int num,
int[] keys,
PointerPointer vals) |
static int |
MXKVStoreInit(KVStoreHandle handle,
int num,
IntBuffer keys,
NDArrayHandle vals) |
static int |
MXKVStoreInit(KVStoreHandle handle,
int num,
IntBuffer keys,
PointerPointer vals) |
static int |
MXKVStoreInit(KVStoreHandle handle,
int num,
IntPointer keys,
NDArrayHandle vals) |
static int |
MXKVStoreInit(KVStoreHandle handle,
int num,
IntPointer keys,
PointerPointer vals)
\brief Init a list of (key,value) pairs in kvstore
|
static int |
MXKVStoreInitEx(KVStoreHandle handle,
int num,
byte[] keys,
NDArrayHandle vals) |
static int |
MXKVStoreInitEx(KVStoreHandle handle,
int num,
byte[] keys,
PointerPointer vals) |
static int |
MXKVStoreInitEx(KVStoreHandle handle,
int num,
ByteBuffer keys,
NDArrayHandle vals) |
static int |
MXKVStoreInitEx(KVStoreHandle handle,
int num,
ByteBuffer keys,
PointerPointer vals) |
static int |
MXKVStoreInitEx(KVStoreHandle handle,
int num,
BytePointer keys,
NDArrayHandle vals) |
static int |
MXKVStoreInitEx(KVStoreHandle handle,
int num,
BytePointer keys,
PointerPointer vals) |
static int |
MXKVStoreInitEx(KVStoreHandle handle,
int num,
PointerPointer keys,
PointerPointer vals)
\brief Init a list of (key,value) pairs in kvstore, where each key is a string
|
static int |
MXKVStoreIsSchedulerNode(int[] ret) |
static int |
MXKVStoreIsSchedulerNode(IntBuffer ret) |
static int |
MXKVStoreIsSchedulerNode(IntPointer ret)
\brief return whether or not this process is a scheduler node.
|
static int |
MXKVStoreIsServerNode(int[] ret) |
static int |
MXKVStoreIsServerNode(IntBuffer ret) |
static int |
MXKVStoreIsServerNode(IntPointer ret)
\brief return whether or not this process is a server node.
|
static int |
MXKVStoreIsWorkerNode(int[] ret) |
static int |
MXKVStoreIsWorkerNode(IntBuffer ret) |
static int |
MXKVStoreIsWorkerNode(IntPointer ret)
\brief return whether or not this process is a worker node.
|
static int |
MXKVStorePull(KVStoreHandle handle,
int num,
int[] keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePull(KVStoreHandle handle,
int num,
int[] keys,
PointerPointer vals,
int priority) |
static int |
MXKVStorePull(KVStoreHandle handle,
int num,
IntBuffer keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePull(KVStoreHandle handle,
int num,
IntBuffer keys,
PointerPointer vals,
int priority) |
static int |
MXKVStorePull(KVStoreHandle handle,
int num,
IntPointer keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePull(KVStoreHandle handle,
int num,
IntPointer keys,
PointerPointer vals,
int priority)
\brief pull a list of (key, value) pairs from the kvstore
|
static int |
MXKVStorePullEx(KVStoreHandle handle,
int num,
byte[] keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePullEx(KVStoreHandle handle,
int num,
byte[] keys,
PointerPointer vals,
int priority) |
static int |
MXKVStorePullEx(KVStoreHandle handle,
int num,
ByteBuffer keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePullEx(KVStoreHandle handle,
int num,
ByteBuffer keys,
PointerPointer vals,
int priority) |
static int |
MXKVStorePullEx(KVStoreHandle handle,
int num,
BytePointer keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePullEx(KVStoreHandle handle,
int num,
BytePointer keys,
PointerPointer vals,
int priority) |
static int |
MXKVStorePullEx(KVStoreHandle handle,
int num,
PointerPointer keys,
PointerPointer vals,
int priority)
\brief pull a list of (key, value) pairs from the kvstore, where each key is a string
|
static int |
MXKVStorePullRowSparse(KVStoreHandle handle,
int num,
int[] keys,
NDArrayHandle vals,
NDArrayHandle row_ids,
int priority) |
static int |
MXKVStorePullRowSparse(KVStoreHandle handle,
int num,
int[] keys,
PointerPointer vals,
PointerPointer row_ids,
int priority) |
static int |
MXKVStorePullRowSparse(KVStoreHandle handle,
int num,
IntBuffer keys,
NDArrayHandle vals,
NDArrayHandle row_ids,
int priority) |
static int |
MXKVStorePullRowSparse(KVStoreHandle handle,
int num,
IntBuffer keys,
PointerPointer vals,
PointerPointer row_ids,
int priority) |
static int |
MXKVStorePullRowSparse(KVStoreHandle handle,
int num,
IntPointer keys,
NDArrayHandle vals,
NDArrayHandle row_ids,
int priority) |
static int |
MXKVStorePullRowSparse(KVStoreHandle handle,
int num,
IntPointer keys,
PointerPointer vals,
PointerPointer row_ids,
int priority)
\brief pull a list of (key, value) pairs from the kvstore, where each key is an integer.
|
static int |
MXKVStorePullRowSparseEx(KVStoreHandle handle,
int num,
byte[] keys,
NDArrayHandle vals,
NDArrayHandle row_ids,
int priority) |
static int |
MXKVStorePullRowSparseEx(KVStoreHandle handle,
int num,
byte[] keys,
PointerPointer vals,
PointerPointer row_ids,
int priority) |
static int |
MXKVStorePullRowSparseEx(KVStoreHandle handle,
int num,
ByteBuffer keys,
NDArrayHandle vals,
NDArrayHandle row_ids,
int priority) |
static int |
MXKVStorePullRowSparseEx(KVStoreHandle handle,
int num,
ByteBuffer keys,
PointerPointer vals,
PointerPointer row_ids,
int priority) |
static int |
MXKVStorePullRowSparseEx(KVStoreHandle handle,
int num,
BytePointer keys,
NDArrayHandle vals,
NDArrayHandle row_ids,
int priority) |
static int |
MXKVStorePullRowSparseEx(KVStoreHandle handle,
int num,
BytePointer keys,
PointerPointer vals,
PointerPointer row_ids,
int priority) |
static int |
MXKVStorePullRowSparseEx(KVStoreHandle handle,
int num,
PointerPointer keys,
PointerPointer vals,
PointerPointer row_ids,
int priority)
\brief pull a list of (key, value) pairs from the kvstore, where each key is a string.
|
static int |
MXKVStorePullWithSparse(KVStoreHandle handle,
int num,
int[] keys,
NDArrayHandle vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparse(KVStoreHandle handle,
int num,
int[] keys,
PointerPointer vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparse(KVStoreHandle handle,
int num,
IntBuffer keys,
NDArrayHandle vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparse(KVStoreHandle handle,
int num,
IntBuffer keys,
PointerPointer vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparse(KVStoreHandle handle,
int num,
IntPointer keys,
NDArrayHandle vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparse(KVStoreHandle handle,
int num,
IntPointer keys,
PointerPointer vals,
int priority,
boolean ignore_sparse)
\brief pull a list of (key, value) pairs from the kvstore
|
static int |
MXKVStorePullWithSparseEx(KVStoreHandle handle,
int num,
byte[] keys,
NDArrayHandle vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparseEx(KVStoreHandle handle,
int num,
byte[] keys,
PointerPointer vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparseEx(KVStoreHandle handle,
int num,
ByteBuffer keys,
NDArrayHandle vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparseEx(KVStoreHandle handle,
int num,
ByteBuffer keys,
PointerPointer vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparseEx(KVStoreHandle handle,
int num,
BytePointer keys,
NDArrayHandle vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparseEx(KVStoreHandle handle,
int num,
BytePointer keys,
PointerPointer vals,
int priority,
boolean ignore_sparse) |
static int |
MXKVStorePullWithSparseEx(KVStoreHandle handle,
int num,
PointerPointer keys,
PointerPointer vals,
int priority,
boolean ignore_sparse)
\brief pull a list of (key, value) pairs from the kvstore, where each key is a string
|
static int |
MXKVStorePush(KVStoreHandle handle,
int num,
int[] keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePush(KVStoreHandle handle,
int num,
int[] keys,
PointerPointer vals,
int priority) |
static int |
MXKVStorePush(KVStoreHandle handle,
int num,
IntBuffer keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePush(KVStoreHandle handle,
int num,
IntBuffer keys,
PointerPointer vals,
int priority) |
static int |
MXKVStorePush(KVStoreHandle handle,
int num,
IntPointer keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePush(KVStoreHandle handle,
int num,
IntPointer keys,
PointerPointer vals,
int priority)
\brief Push a list of (key,value) pairs to kvstore
|
static int |
MXKVStorePushEx(KVStoreHandle handle,
int num,
byte[] keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePushEx(KVStoreHandle handle,
int num,
byte[] keys,
PointerPointer vals,
int priority) |
static int |
MXKVStorePushEx(KVStoreHandle handle,
int num,
ByteBuffer keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePushEx(KVStoreHandle handle,
int num,
ByteBuffer keys,
PointerPointer vals,
int priority) |
static int |
MXKVStorePushEx(KVStoreHandle handle,
int num,
BytePointer keys,
NDArrayHandle vals,
int priority) |
static int |
MXKVStorePushEx(KVStoreHandle handle,
int num,
BytePointer keys,
PointerPointer vals,
int priority) |
static int |
MXKVStorePushEx(KVStoreHandle handle,
int num,
PointerPointer keys,
PointerPointer vals,
int priority)
\brief Push a list of (key,value) pairs to kvstore, where each key is a string
|
static int |
MXKVStorePushPull(KVStoreHandle handle,
int vnum,
int[] vkeys,
int onum,
int[] okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStorePushPull(KVStoreHandle handle,
int vnum,
int[] vkeys,
int onum,
int[] okeys,
PointerPointer vals,
PointerPointer outs,
int priority) |
static int |
MXKVStorePushPull(KVStoreHandle handle,
int vnum,
IntBuffer vkeys,
int onum,
IntBuffer okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStorePushPull(KVStoreHandle handle,
int vnum,
IntBuffer vkeys,
int onum,
IntBuffer okeys,
PointerPointer vals,
PointerPointer outs,
int priority) |
static int |
MXKVStorePushPull(KVStoreHandle handle,
int vnum,
IntPointer vkeys,
int onum,
IntPointer okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStorePushPull(KVStoreHandle handle,
int vnum,
IntPointer vkeys,
int onum,
IntPointer okeys,
PointerPointer vals,
PointerPointer outs,
int priority)
\brief push and pull a list of (key, value) pairs from the kvstore
|
static int |
MXKVStorePushPullEx(KVStoreHandle handle,
int vnum,
byte[] vkeys,
int onum,
byte[] okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStorePushPullEx(KVStoreHandle handle,
int vnum,
byte[] vkeys,
int onum,
byte[] okeys,
PointerPointer vals,
PointerPointer outs,
int priority) |
static int |
MXKVStorePushPullEx(KVStoreHandle handle,
int vnum,
ByteBuffer vkeys,
int onum,
ByteBuffer okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStorePushPullEx(KVStoreHandle handle,
int vnum,
ByteBuffer vkeys,
int onum,
ByteBuffer okeys,
PointerPointer vals,
PointerPointer outs,
int priority) |
static int |
MXKVStorePushPullEx(KVStoreHandle handle,
int vnum,
BytePointer vkeys,
int onum,
BytePointer okeys,
NDArrayHandle vals,
NDArrayHandle outs,
int priority) |
static int |
MXKVStorePushPullEx(KVStoreHandle handle,
int vnum,
BytePointer vkeys,
int onum,
BytePointer okeys,
PointerPointer vals,
PointerPointer outs,
int priority) |
static int |
MXKVStorePushPullEx(KVStoreHandle handle,
int vnum,
PointerPointer vkeys,
int onum,
PointerPointer okeys,
PointerPointer vals,
PointerPointer outs,
int priority)
\brief push and pull a list of (key, value) pairs from the kvstore,
where each key is a string
|
static int |
MXKVStoreRunServer(KVStoreHandle handle,
MXKVStoreServerController controller,
Pointer controller_handle)
\brief Run as server (or scheduler)
|
static int |
MXKVStoreSendCommmandToServers(KVStoreHandle handle,
int cmd_id,
BytePointer cmd_body)
\brief Send a command to all server nodes
|
static int |
MXKVStoreSendCommmandToServers(KVStoreHandle handle,
int cmd_id,
String cmd_body) |
static int |
MXKVStoreSetBarrierBeforeExit(KVStoreHandle handle,
int barrier_before_exit)
\brief whether to do barrier when finalize
|
static int |
MXKVStoreSetGradientCompression(KVStoreHandle handle,
int num_params,
byte[] keys,
byte[] vals) |
static int |
MXKVStoreSetGradientCompression(KVStoreHandle handle,
int num_params,
ByteBuffer keys,
ByteBuffer vals) |
static int |
MXKVStoreSetGradientCompression(KVStoreHandle handle,
int num_params,
BytePointer keys,
BytePointer vals) |
static int |
MXKVStoreSetGradientCompression(KVStoreHandle handle,
int num_params,
PointerPointer keys,
PointerPointer vals)
\brief Set parameters to use low-bit compressed gradients
|
static int |
MXKVStoreSetUpdater(KVStoreHandle handle,
MXKVStoreUpdater updater,
Pointer updater_handle)
\brief register a push updater
|
static int |
MXKVStoreSetUpdaterEx(KVStoreHandle handle,
MXKVStoreUpdater updater,
MXKVStoreStrUpdater str_updater,
Pointer updater_handle)
\brief register a push updater with int keys and one with string keys
|
static int |
MXLibInfoFeatures(LibFeature libFeature,
SizeTPointer size) |
static int |
MXLibInfoFeatures(PointerPointer libFeature,
SizeTPointer size)
\brief Get list of features supported on the runtime
|
static int |
MXListAllOpNames(int[] out_size,
PointerPointer out_array) |
static int |
MXListAllOpNames(IntBuffer out_size,
PointerPointer out_array) |
static int |
MXListAllOpNames(IntPointer out_size,
PointerPointer out_array)
\brief list all the available operator names, include entries.
|
static int |
MXListDataIters(int[] out_size,
PointerPointer out_array) |
static int |
MXListDataIters(IntBuffer out_size,
PointerPointer out_array) |
static int |
MXListDataIters(IntPointer out_size,
PointerPointer out_array)
\brief List all the available iterator entries
|
static int |
MXListFunctions(int[] out_size,
PointerPointer out_array) |
static int |
MXListFunctions(IntBuffer out_size,
PointerPointer out_array) |
static int |
MXListFunctions(IntPointer out_size,
PointerPointer out_array)
\brief list all the available functions handles
most user can use it to list all the needed functions
|
static int |
MXLoadLib(BytePointer path,
int verbose)
\brief Load library dynamically
|
static int |
MXLoadLib(String path,
int verbose) |
static int |
MXNDArrayAt(NDArrayHandle handle,
int idx,
NDArrayHandle out) |
static int |
MXNDArrayAt(NDArrayHandle handle,
int idx,
PointerPointer out)
\brief Index the NDArray along axis 0.
|
static int |
MXNDArrayAt64(NDArrayHandle handle,
long idx,
NDArrayHandle out) |
static int |
MXNDArrayAt64(NDArrayHandle handle,
long idx,
PointerPointer out)
\brief Index the NDArray along axis 0.
|
static int |
MXNDArrayCallDLPackDeleter(DLManagedTensorHandle dlpack)
\brief Delete a dlpack tensor
|
static int |
MXNDArrayCreate(int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
NDArrayHandle out) |
static int |
MXNDArrayCreate(int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
PointerPointer out) |
static int |
MXNDArrayCreate(IntBuffer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
NDArrayHandle out) |
static int |
MXNDArrayCreate(IntBuffer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
PointerPointer out) |
static int |
MXNDArrayCreate(IntPointer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
NDArrayHandle out) |
static int |
MXNDArrayCreate(IntPointer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
PointerPointer out)
\brief create a NDArray with specified shape
|
static int |
MXNDArrayCreateEx(int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateEx(int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
PointerPointer out) |
static int |
MXNDArrayCreateEx(IntBuffer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateEx(IntBuffer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
PointerPointer out) |
static int |
MXNDArrayCreateEx(IntPointer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateEx(IntPointer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
PointerPointer out)
\brief create a NDArray with specified shape and data type
This api is available when MXNet is built with flag
USE_INT64_TENSOR_SIZE=0 (by default)
|
static int |
MXNDArrayCreateEx64(long[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateEx64(long[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
PointerPointer out) |
static int |
MXNDArrayCreateEx64(LongBuffer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateEx64(LongBuffer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
PointerPointer out) |
static int |
MXNDArrayCreateEx64(LongPointer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateEx64(LongPointer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
PointerPointer out)
\brief create a NDArray with specified shape and data type
This api is available when MXNet is built with flag
USE_INT64_TENSOR_SIZE=1 (not default) i.e.
|
static int |
MXNDArrayCreateFromSharedMem(int shared_pid,
int shared_id,
int[] shape,
int ndim,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateFromSharedMem(int shared_pid,
int shared_id,
int[] shape,
int ndim,
int dtype,
PointerPointer out) |
static int |
MXNDArrayCreateFromSharedMem(int shared_pid,
int shared_id,
IntBuffer shape,
int ndim,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateFromSharedMem(int shared_pid,
int shared_id,
IntBuffer shape,
int ndim,
int dtype,
PointerPointer out) |
static int |
MXNDArrayCreateFromSharedMem(int shared_pid,
int shared_id,
IntPointer shape,
int ndim,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateFromSharedMem(int shared_pid,
int shared_id,
IntPointer shape,
int ndim,
int dtype,
PointerPointer out)
\brief DEPRECATED.
|
static int |
MXNDArrayCreateFromSharedMemEx(int shared_pid,
int shared_id,
int[] shape,
int ndim,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateFromSharedMemEx(int shared_pid,
int shared_id,
int[] shape,
int ndim,
int dtype,
PointerPointer out) |
static int |
MXNDArrayCreateFromSharedMemEx(int shared_pid,
int shared_id,
IntBuffer shape,
int ndim,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateFromSharedMemEx(int shared_pid,
int shared_id,
IntBuffer shape,
int ndim,
int dtype,
PointerPointer out) |
static int |
MXNDArrayCreateFromSharedMemEx(int shared_pid,
int shared_id,
IntPointer shape,
int ndim,
int dtype,
NDArrayHandle out) |
static int |
MXNDArrayCreateFromSharedMemEx(int shared_pid,
int shared_id,
IntPointer shape,
int ndim,
int dtype,
PointerPointer out)
\brief Reconstruct NDArray from shared memory handle
|
static int |
MXNDArrayCreateNone(NDArrayHandle out) |
static int |
MXNDArrayCreateNone(PointerPointer out)
\brief create a NDArray handle that is not initialized
can be used to pass in as mutate variables
to hold the result of NDArray
|
static int |
MXNDArrayCreateSparseEx(int storage_type,
int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
int num_aux,
int[] aux_type,
int[] aux_ndims,
int[] aux_shape,
NDArrayHandle out) |
static int |
MXNDArrayCreateSparseEx(int storage_type,
int[] shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
int num_aux,
int[] aux_type,
int[] aux_ndims,
int[] aux_shape,
PointerPointer out) |
static int |
MXNDArrayCreateSparseEx(int storage_type,
IntBuffer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
int num_aux,
IntBuffer aux_type,
IntBuffer aux_ndims,
IntBuffer aux_shape,
NDArrayHandle out) |
static int |
MXNDArrayCreateSparseEx(int storage_type,
IntBuffer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
int num_aux,
IntBuffer aux_type,
IntBuffer aux_ndims,
IntBuffer aux_shape,
PointerPointer out) |
static int |
MXNDArrayCreateSparseEx(int storage_type,
IntPointer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
int num_aux,
IntPointer aux_type,
IntPointer aux_ndims,
IntPointer aux_shape,
NDArrayHandle out) |
static int |
MXNDArrayCreateSparseEx(int storage_type,
IntPointer shape,
int ndim,
int dev_type,
int dev_id,
int delay_alloc,
int dtype,
int num_aux,
IntPointer aux_type,
IntPointer aux_ndims,
IntPointer aux_shape,
PointerPointer out)
\brief create an empty sparse NDArray with specified shape and data type
This api is available when MXNet is built with flag
USE_INT64_TENSOR_SIZE=0 (by default)
|
static int |
MXNDArrayDetach(NDArrayHandle handle,
NDArrayHandle out) |
static int |
MXNDArrayDetach(NDArrayHandle handle,
PointerPointer out)
\brief detach and ndarray from computation graph by clearing entry_
|
static int |
MXNDArrayFree(NDArrayHandle handle)
\brief free the narray handle
|
static int |
MXNDArrayFromDLPack(DLManagedTensorHandle dlpack,
NDArrayHandle out_handle) |
static int |
MXNDArrayFromDLPack(DLManagedTensorHandle dlpack,
PointerPointer out_handle)
\brief DEPRECATED.
|
static int |
MXNDArrayFromDLPackEx(DLManagedTensorHandle dlpack,
boolean transient_handle,
NDArrayHandle out_handle) |
static int |
MXNDArrayFromDLPackEx(DLManagedTensorHandle dlpack,
boolean transient_handle,
PointerPointer out_handle)
\brief Create a NDArray backed by a dlpack tensor.
|
static int |
MXNDArrayGetAuxNDArray(NDArrayHandle handle,
int i,
NDArrayHandle out) |
static int |
MXNDArrayGetAuxNDArray(NDArrayHandle handle,
int i,
PointerPointer out)
\brief Get a deep copy of the ith aux data blob
This api is available when MXNet is built with flag
USE_INT64_TENSOR_SIZE=0 (by default)
in the form of an NDArray of default storage type.
|
static int |
MXNDArrayGetAuxType(NDArrayHandle handle,
int i,
int[] out_type) |
static int |
MXNDArrayGetAuxType(NDArrayHandle handle,
int i,
IntBuffer out_type) |
static int |
MXNDArrayGetAuxType(NDArrayHandle handle,
int i,
IntPointer out_type)
\brief get the type of the ith aux data in NDArray
This api is available when MXNet is built with flag
USE_INT64_TENSOR_SIZE=0 (by default)
|
static int |
MXNDArrayGetContext(NDArrayHandle handle,
int[] out_dev_type,
int[] out_dev_id) |
static int |
MXNDArrayGetContext(NDArrayHandle handle,
IntBuffer out_dev_type,
IntBuffer out_dev_id) |
static int |
MXNDArrayGetContext(NDArrayHandle handle,
IntPointer out_dev_type,
IntPointer out_dev_id)
\brief get the context of the NDArray
|
static int |
MXNDArrayGetData(NDArrayHandle handle,
Pointer out_pdata) |
static int |
MXNDArrayGetData(NDArrayHandle handle,
PointerPointer out_pdata)
\brief get the content of the data in NDArray
|
static int |
MXNDArrayGetDataNDArray(NDArrayHandle handle,
NDArrayHandle out) |
static int |
MXNDArrayGetDataNDArray(NDArrayHandle handle,
PointerPointer out)
\brief Get a deep copy of the data blob
in the form of an NDArray of default storage type.
|
static int |
MXNDArrayGetDType(NDArrayHandle handle,
int[] out_dtype) |
static int |
MXNDArrayGetDType(NDArrayHandle handle,
IntBuffer out_dtype) |
static int |
MXNDArrayGetDType(NDArrayHandle handle,
IntPointer out_dtype)
\brief get the type of the data in NDArray
|
static int |
MXNDArrayGetGrad(NDArrayHandle handle,
NDArrayHandle out) |
static int |
MXNDArrayGetGrad(NDArrayHandle handle,
PointerPointer out)
\brief return gradient buffer attached to this NDArray
|
static int |
MXNDArrayGetGradState(NDArrayHandle handle,
int[] out) |
static int |
MXNDArrayGetGradState(NDArrayHandle handle,
IntBuffer out) |
static int |
MXNDArrayGetGradState(NDArrayHandle handle,
IntPointer out)
\brief set the flag for gradient array state.
|
static int |
MXNDArrayGetShape(NDArrayHandle handle,
int[] out_dim,
int[] out_pdata) |
static int |
MXNDArrayGetShape(NDArrayHandle handle,
IntBuffer out_dim,
IntBuffer out_pdata) |
static int |
MXNDArrayGetShape(NDArrayHandle handle,
IntPointer out_dim,
IntPointer out_pdata) |
static int |
MXNDArrayGetShape(NDArrayHandle handle,
IntPointer out_dim,
PointerPointer out_pdata)
\brief DEPRECATED.
|
static int |
MXNDArrayGetShapeEx(NDArrayHandle handle,
int[] out_dim,
int[] out_pdata) |
static int |
MXNDArrayGetShapeEx(NDArrayHandle handle,
IntBuffer out_dim,
IntBuffer out_pdata) |
static int |
MXNDArrayGetShapeEx(NDArrayHandle handle,
IntPointer out_dim,
IntPointer out_pdata) |
static int |
MXNDArrayGetShapeEx(NDArrayHandle handle,
IntPointer out_dim,
PointerPointer out_pdata)
\brief get the shape of the array
This api is available when MXNet is built with flag
USE_INT64_TENSOR_SIZE=0 (by default)
|
static int |
MXNDArrayGetShapeEx64(NDArrayHandle handle,
int[] out_dim,
long[] out_pdata) |
static int |
MXNDArrayGetShapeEx64(NDArrayHandle handle,
IntBuffer out_dim,
LongBuffer out_pdata) |
static int |
MXNDArrayGetShapeEx64(NDArrayHandle handle,
IntPointer out_dim,
LongPointer out_pdata) |
static int |
MXNDArrayGetShapeEx64(NDArrayHandle handle,
IntPointer out_dim,
PointerPointer out_pdata)
\brief get the shape of the array
This api is available when MXNet is built with flag
USE_INT64_TENSOR_SIZE=1 (not default) i.e.
|
static int |
MXNDArrayGetSharedMemHandle(NDArrayHandle handle,
int[] shared_pid,
int[] shared_id) |
static int |
MXNDArrayGetSharedMemHandle(NDArrayHandle handle,
IntBuffer shared_pid,
IntBuffer shared_id) |
static int |
MXNDArrayGetSharedMemHandle(NDArrayHandle handle,
IntPointer shared_pid,
IntPointer shared_id)
\brief Get shared memory handle from NDArray
|
static int |
MXNDArrayGetStorageType(NDArrayHandle handle,
int[] out_storage_type) |
static int |
MXNDArrayGetStorageType(NDArrayHandle handle,
IntBuffer out_storage_type) |
static int |
MXNDArrayGetStorageType(NDArrayHandle handle,
IntPointer out_storage_type)
\brief get the storage type of the array
|
static int |
MXNDArrayLoad(BytePointer fname,
int[] out_size,
PointerPointer out_arr,
int[] out_name_size,
PointerPointer out_names) |
static int |
MXNDArrayLoad(BytePointer fname,
IntBuffer out_size,
PointerPointer out_arr,
IntBuffer out_name_size,
PointerPointer out_names) |
static int |
MXNDArrayLoad(BytePointer fname,
IntPointer out_size,
PointerPointer out_arr,
IntPointer out_name_size,
PointerPointer out_names)
\brief Load list of narray from the file.
|
static int |
MXNDArrayLoad(String fname,
int[] out_size,
PointerPointer out_arr,
int[] out_name_size,
PointerPointer out_names) |
static int |
MXNDArrayLoad(String fname,
IntBuffer out_size,
PointerPointer out_arr,
IntBuffer out_name_size,
PointerPointer out_names) |
static int |
MXNDArrayLoad(String fname,
IntPointer out_size,
PointerPointer out_arr,
IntPointer out_name_size,
PointerPointer out_names) |
static int |
MXNDArrayLoadFromBuffer(Pointer ndarray_buffer,
long size,
int[] out_size,
PointerPointer out_arr,
int[] out_name_size,
PointerPointer out_names) |
static int |
MXNDArrayLoadFromBuffer(Pointer ndarray_buffer,
long size,
IntBuffer out_size,
PointerPointer out_arr,
IntBuffer out_name_size,
PointerPointer out_names) |
static int |
MXNDArrayLoadFromBuffer(Pointer ndarray_buffer,
long size,
IntPointer out_size,
PointerPointer out_arr,
IntPointer out_name_size,
PointerPointer out_names)
\brief Load list / dictionary of narrays from file content loaded into memory.
|
static int |
MXNDArrayLoadFromRawBytes(Pointer buf,
long size,
NDArrayHandle out) |
static int |
MXNDArrayLoadFromRawBytes(Pointer buf,
long size,
PointerPointer out)
\brief create a NDArray handle that is loaded from raw bytes.
|
static int |
MXNDArrayReshape(NDArrayHandle handle,
int ndim,
int[] dims,
NDArrayHandle out) |
static int |
MXNDArrayReshape(NDArrayHandle handle,
int ndim,
int[] dims,
PointerPointer out) |
static int |
MXNDArrayReshape(NDArrayHandle handle,
int ndim,
IntBuffer dims,
NDArrayHandle out) |
static int |
MXNDArrayReshape(NDArrayHandle handle,
int ndim,
IntBuffer dims,
PointerPointer out) |
static int |
MXNDArrayReshape(NDArrayHandle handle,
int ndim,
IntPointer dims,
NDArrayHandle out) |
static int |
MXNDArrayReshape(NDArrayHandle handle,
int ndim,
IntPointer dims,
PointerPointer out)
\brief Reshape the NDArray.
|
static int |
MXNDArrayReshape64(NDArrayHandle handle,
int ndim,
long[] dims,
boolean reverse,
NDArrayHandle out) |
static int |
MXNDArrayReshape64(NDArrayHandle handle,
int ndim,
long[] dims,
boolean reverse,
PointerPointer out) |
static int |
MXNDArrayReshape64(NDArrayHandle handle,
int ndim,
LongBuffer dims,
boolean reverse,
NDArrayHandle out) |
static int |
MXNDArrayReshape64(NDArrayHandle handle,
int ndim,
LongBuffer dims,
boolean reverse,
PointerPointer out) |
static int |
MXNDArrayReshape64(NDArrayHandle handle,
int ndim,
LongPointer dims,
boolean reverse,
NDArrayHandle out) |
static int |
MXNDArrayReshape64(NDArrayHandle handle,
int ndim,
LongPointer dims,
boolean reverse,
PointerPointer out)
\brief Reshape the NDArray.
|
static int |
MXNDArraySave(BytePointer fname,
int num_args,
PointerPointer args,
byte[] keys) |
static int |
MXNDArraySave(BytePointer fname,
int num_args,
PointerPointer args,
ByteBuffer keys) |
static int |
MXNDArraySave(BytePointer fname,
int num_args,
PointerPointer args,
BytePointer keys) |
static int |
MXNDArraySave(BytePointer fname,
int num_args,
PointerPointer args,
PointerPointer keys)
\brief Save list of narray into the file.
|
static int |
MXNDArraySave(String fname,
int num_args,
NDArrayHandle args,
byte[] keys) |
static int |
MXNDArraySave(String fname,
int num_args,
NDArrayHandle args,
ByteBuffer keys) |
static int |
MXNDArraySave(String fname,
int num_args,
NDArrayHandle args,
BytePointer keys) |
static int |
MXNDArraySaveRawBytes(NDArrayHandle handle,
SizeTPointer out_size,
byte[] out_buf) |
static int |
MXNDArraySaveRawBytes(NDArrayHandle handle,
SizeTPointer out_size,
ByteBuffer out_buf) |
static int |
MXNDArraySaveRawBytes(NDArrayHandle handle,
SizeTPointer out_size,
BytePointer out_buf) |
static int |
MXNDArraySaveRawBytes(NDArrayHandle handle,
SizeTPointer out_size,
PointerPointer out_buf)
\brief save the NDArray into raw bytes.
|
static int |
MXNDArraySetGradState(NDArrayHandle handle,
int state)
\brief set the flag for gradient array state.
|
static int |
MXNDArraySlice(NDArrayHandle handle,
int slice_begin,
int slice_end,
NDArrayHandle out) |
static int |
MXNDArraySlice(NDArrayHandle handle,
int slice_begin,
int slice_end,
PointerPointer out)
\brief Slice the NDArray along axis 0.
|
static int |
MXNDArraySlice64(NDArrayHandle handle,
long slice_begin,
long slice_end,
NDArrayHandle out) |
static int |
MXNDArraySlice64(NDArrayHandle handle,
long slice_begin,
long slice_end,
PointerPointer out)
\brief Slice the NDArray along axis 0.
|
static int |
MXNDArraySyncCheckFormat(NDArrayHandle handle,
boolean full_check)
\brief check whether the NDArray format is valid
|
static int |
MXNDArraySyncCopyFromCPU(NDArrayHandle handle,
Pointer data,
long size)
\brief Perform a synchronize copy from a contiguous CPU memory region.
|
static int |
MXNDArraySyncCopyFromNDArray(NDArrayHandle handle_dst,
NDArrayHandle handle_src,
int i)
\brief Copy src.data() to dst.data() if i = -1, else dst.aux_data(i) if i >= 0
This function blocks.
|
static int |
MXNDArraySyncCopyToCPU(NDArrayHandle handle,
Pointer data,
long size)
\brief Perform a synchronize copyto a contiguous CPU memory region.
|
static int |
MXNDArrayToDLPack(NDArrayHandle handle,
DLManagedTensorHandle out_dlpack)
\brief Create a reference view of NDArray that
represents as DLManagedTensor
Notice: MXNet uses asynchronous execution.
|
static int |
MXNDArrayWaitAll()
\brief wait until all delayed operations in
the system is completed
|
static int |
MXNDArrayWaitToRead(NDArrayHandle handle)
\brief Wait until all the pending writes with respect NDArray are finished.
|
static int |
MXNDArrayWaitToWrite(NDArrayHandle handle)
\brief Wait until all the pending read/write with respect NDArray are finished.
|
static int |
MXNDListCreate(BytePointer nd_file_bytes,
int nd_file_size,
NDListHandle out,
int[] out_length) |
static int |
MXNDListCreate(BytePointer nd_file_bytes,
int nd_file_size,
NDListHandle out,
IntBuffer out_length) |
static int |
MXNDListCreate(BytePointer nd_file_bytes,
int nd_file_size,
NDListHandle out,
IntPointer out_length)
\brief Create a NDArray List by loading from ndarray file.
|
static int |
MXNDListCreate(String nd_file_bytes,
int nd_file_size,
NDListHandle out,
int[] out_length) |
static int |
MXNDListCreate(String nd_file_bytes,
int nd_file_size,
NDListHandle out,
IntBuffer out_length) |
static int |
MXNDListCreate(String nd_file_bytes,
int nd_file_size,
NDListHandle out,
IntPointer out_length) |
static int |
MXNDListFree(NDListHandle handle)
\brief Free a MXAPINDList
|
static int |
MXNDListGet(NDListHandle handle,
int index,
byte[] out_key,
float[] out_data,
int[] out_shape,
int[] out_ndim) |
static int |
MXNDListGet(NDListHandle handle,
int index,
ByteBuffer out_key,
FloatBuffer out_data,
IntBuffer out_shape,
IntBuffer out_ndim) |
static int |
MXNDListGet(NDListHandle handle,
int index,
BytePointer out_key,
FloatPointer out_data,
IntPointer out_shape,
IntPointer out_ndim) |
static int |
MXNDListGet(NDListHandle handle,
int index,
PointerPointer out_key,
PointerPointer out_data,
PointerPointer out_shape,
IntPointer out_ndim)
\brief Get an element from list
|
static int |
MXNotifyShutdown()
\brief Notify the engine about a shutdown,
This can help engine to print less messages into display.
|
static int |
MXOptimizeForBackend(SymbolHandle sym_handle,
BytePointer backend_name,
int dev_type,
PointerPointer ret_sym_handle,
int args_len,
PointerPointer in_args_handle,
int aux_len,
PointerPointer in_aux_handle,
int num_options,
byte[] keys,
byte[] vals,
int num_input_shapes,
byte[] input_shape_names,
long[] input_shape_data,
int[] input_shape_idx,
int num_input_dtypes,
byte[] input_dtype_names,
int[] input_dtypes,
int num_input_stypes,
byte[] input_stype_names,
int[] input_stypes,
boolean skip_infer,
int[] new_args_cnt,
PointerPointer new_args_handle,
PointerPointer new_arg_names_handle,
int[] new_aux_cnt,
PointerPointer new_aux_handle,
PointerPointer new_aux_names_handle) |
static int |
MXOptimizeForBackend(SymbolHandle sym_handle,
BytePointer backend_name,
int dev_type,
PointerPointer ret_sym_handle,
int args_len,
PointerPointer in_args_handle,
int aux_len,
PointerPointer in_aux_handle,
int num_options,
ByteBuffer keys,
ByteBuffer vals,
int num_input_shapes,
ByteBuffer input_shape_names,
LongBuffer input_shape_data,
IntBuffer input_shape_idx,
int num_input_dtypes,
ByteBuffer input_dtype_names,
IntBuffer input_dtypes,
int num_input_stypes,
ByteBuffer input_stype_names,
IntBuffer input_stypes,
boolean skip_infer,
IntBuffer new_args_cnt,
PointerPointer new_args_handle,
PointerPointer new_arg_names_handle,
IntBuffer new_aux_cnt,
PointerPointer new_aux_handle,
PointerPointer new_aux_names_handle) |
static int |
MXOptimizeForBackend(SymbolHandle sym_handle,
BytePointer backend_name,
int dev_type,
PointerPointer ret_sym_handle,
int args_len,
PointerPointer in_args_handle,
int aux_len,
PointerPointer in_aux_handle,
int num_options,
BytePointer keys,
BytePointer vals,
int num_input_shapes,
BytePointer input_shape_names,
LongPointer input_shape_data,
IntPointer input_shape_idx,
int num_input_dtypes,
BytePointer input_dtype_names,
IntPointer input_dtypes,
int num_input_stypes,
BytePointer input_stype_names,
IntPointer input_stypes,
boolean skip_infer,
IntPointer new_args_cnt,
PointerPointer new_args_handle,
PointerPointer new_arg_names_handle,
IntPointer new_aux_cnt,
PointerPointer new_aux_handle,
PointerPointer new_aux_names_handle) |
static int |
MXOptimizeForBackend(SymbolHandle sym_handle,
BytePointer backend_name,
int dev_type,
PointerPointer ret_sym_handle,
int args_len,
PointerPointer in_args_handle,
int aux_len,
PointerPointer in_aux_handle,
int num_options,
PointerPointer keys,
PointerPointer vals,
int num_input_shapes,
PointerPointer input_shape_names,
LongPointer input_shape_data,
IntPointer input_shape_idx,
int num_input_dtypes,
PointerPointer input_dtype_names,
IntPointer input_dtypes,
int num_input_stypes,
PointerPointer input_stype_names,
IntPointer input_stypes,
boolean skip_infer,
IntPointer new_args_cnt,
PointerPointer new_args_handle,
PointerPointer new_arg_names_handle,
IntPointer new_aux_cnt,
PointerPointer new_aux_handle,
PointerPointer new_aux_names_handle)
\brief Partitions symbol for given backend, potentially creating subgraphs
|
static int |
MXOptimizeForBackend(SymbolHandle sym_handle,
String backend_name,
int dev_type,
SymbolHandle ret_sym_handle,
int args_len,
NDArrayHandle in_args_handle,
int aux_len,
NDArrayHandle in_aux_handle,
int num_options,
byte[] keys,
byte[] vals,
int num_input_shapes,
byte[] input_shape_names,
long[] input_shape_data,
int[] input_shape_idx,
int num_input_dtypes,
byte[] input_dtype_names,
int[] input_dtypes,
int num_input_stypes,
byte[] input_stype_names,
int[] input_stypes,
boolean skip_infer,
int[] new_args_cnt,
PointerPointer new_args_handle,
PointerPointer new_arg_names_handle,
int[] new_aux_cnt,
PointerPointer new_aux_handle,
PointerPointer new_aux_names_handle) |
static int |
MXOptimizeForBackend(SymbolHandle sym_handle,
String backend_name,
int dev_type,
SymbolHandle ret_sym_handle,
int args_len,
NDArrayHandle in_args_handle,
int aux_len,
NDArrayHandle in_aux_handle,
int num_options,
ByteBuffer keys,
ByteBuffer vals,
int num_input_shapes,
ByteBuffer input_shape_names,
LongBuffer input_shape_data,
IntBuffer input_shape_idx,
int num_input_dtypes,
ByteBuffer input_dtype_names,
IntBuffer input_dtypes,
int num_input_stypes,
ByteBuffer input_stype_names,
IntBuffer input_stypes,
boolean skip_infer,
IntBuffer new_args_cnt,
PointerPointer new_args_handle,
PointerPointer new_arg_names_handle,
IntBuffer new_aux_cnt,
PointerPointer new_aux_handle,
PointerPointer new_aux_names_handle) |
static int |
MXOptimizeForBackend(SymbolHandle sym_handle,
String backend_name,
int dev_type,
SymbolHandle ret_sym_handle,
int args_len,
NDArrayHandle in_args_handle,
int aux_len,
NDArrayHandle in_aux_handle,
int num_options,
BytePointer keys,
BytePointer vals,
int num_input_shapes,
BytePointer input_shape_names,
LongPointer input_shape_data,
IntPointer input_shape_idx,
int num_input_dtypes,
BytePointer input_dtype_names,
IntPointer input_dtypes,
int num_input_stypes,
BytePointer input_stype_names,
IntPointer input_stypes,
boolean skip_infer,
IntPointer new_args_cnt,
PointerPointer new_args_handle,
PointerPointer new_arg_names_handle,
IntPointer new_aux_cnt,
PointerPointer new_aux_handle,
PointerPointer new_aux_names_handle) |
static int |
MXPredCreate(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
byte[] input_keys,
int[] input_shape_indptr,
int[] input_shape_data,
PointerPointer out) |
static int |
MXPredCreate(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
ByteBuffer input_keys,
IntBuffer input_shape_indptr,
IntBuffer input_shape_data,
PointerPointer out) |
static int |
MXPredCreate(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
BytePointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
PointerPointer out) |
static int |
MXPredCreate(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
PointerPointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
PointerPointer out)
\brief create a predictor
|
static int |
MXPredCreate(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
byte[] input_keys,
int[] input_shape_indptr,
int[] input_shape_data,
PredictorHandle out) |
static int |
MXPredCreate(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
ByteBuffer input_keys,
IntBuffer input_shape_indptr,
IntBuffer input_shape_data,
PredictorHandle out) |
static int |
MXPredCreate(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
BytePointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
PredictorHandle out) |
static int |
MXPredCreateEx(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
byte[] input_keys,
int[] input_shape_indptr,
int[] input_shape_data,
int num_provided_arg_dtypes,
byte[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
PointerPointer out) |
static int |
MXPredCreateEx(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
ByteBuffer input_keys,
IntBuffer input_shape_indptr,
IntBuffer input_shape_data,
int num_provided_arg_dtypes,
ByteBuffer provided_arg_dtype_names,
IntBuffer provided_arg_dtypes,
PointerPointer out) |
static int |
MXPredCreateEx(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
BytePointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
int num_provided_arg_dtypes,
BytePointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
PointerPointer out) |
static int |
MXPredCreateEx(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
PointerPointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
int num_provided_arg_dtypes,
PointerPointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
PointerPointer out)
\brief create a predictor
|
static int |
MXPredCreateEx(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
byte[] input_keys,
int[] input_shape_indptr,
int[] input_shape_data,
int num_provided_arg_dtypes,
byte[] provided_arg_dtype_names,
int[] provided_arg_dtypes,
PredictorHandle out) |
static int |
MXPredCreateEx(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
ByteBuffer input_keys,
IntBuffer input_shape_indptr,
IntBuffer input_shape_data,
int num_provided_arg_dtypes,
ByteBuffer provided_arg_dtype_names,
IntBuffer provided_arg_dtypes,
PredictorHandle out) |
static int |
MXPredCreateEx(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
BytePointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
int num_provided_arg_dtypes,
BytePointer provided_arg_dtype_names,
IntPointer provided_arg_dtypes,
PredictorHandle out) |
static int |
MXPredCreateMultiThread(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
byte[] input_keys,
int[] input_shape_indptr,
int[] input_shape_data,
int num_threads,
PointerPointer out) |
static int |
MXPredCreateMultiThread(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
ByteBuffer input_keys,
IntBuffer input_shape_indptr,
IntBuffer input_shape_data,
int num_threads,
PointerPointer out) |
static int |
MXPredCreateMultiThread(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
BytePointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
int num_threads,
PointerPointer out) |
static int |
MXPredCreateMultiThread(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
PointerPointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
int num_threads,
PointerPointer out)
\brief create predictors for multiple threads.
|
static int |
MXPredCreateMultiThread(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
byte[] input_keys,
int[] input_shape_indptr,
int[] input_shape_data,
int num_threads,
PredictorHandle out) |
static int |
MXPredCreateMultiThread(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
ByteBuffer input_keys,
IntBuffer input_shape_indptr,
IntBuffer input_shape_data,
int num_threads,
PredictorHandle out) |
static int |
MXPredCreateMultiThread(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
BytePointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
int num_threads,
PredictorHandle out) |
static int |
MXPredCreatePartialOut(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
byte[] input_keys,
int[] input_shape_indptr,
int[] input_shape_data,
int num_output_nodes,
byte[] output_keys,
PointerPointer out) |
static int |
MXPredCreatePartialOut(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
ByteBuffer input_keys,
IntBuffer input_shape_indptr,
IntBuffer input_shape_data,
int num_output_nodes,
ByteBuffer output_keys,
PointerPointer out) |
static int |
MXPredCreatePartialOut(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
BytePointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
int num_output_nodes,
BytePointer output_keys,
PointerPointer out) |
static int |
MXPredCreatePartialOut(BytePointer symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
PointerPointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
int num_output_nodes,
PointerPointer output_keys,
PointerPointer out)
\brief create a predictor wich customized outputs
|
static int |
MXPredCreatePartialOut(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
byte[] input_keys,
int[] input_shape_indptr,
int[] input_shape_data,
int num_output_nodes,
byte[] output_keys,
PredictorHandle out) |
static int |
MXPredCreatePartialOut(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
ByteBuffer input_keys,
IntBuffer input_shape_indptr,
IntBuffer input_shape_data,
int num_output_nodes,
ByteBuffer output_keys,
PredictorHandle out) |
static int |
MXPredCreatePartialOut(String symbol_json_str,
Pointer param_bytes,
int param_size,
int dev_type,
int dev_id,
int num_input_nodes,
BytePointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
int num_output_nodes,
BytePointer output_keys,
PredictorHandle out) |
static int |
MXPredForward(PredictorHandle handle)
\brief Run a forward pass to get the output.
|
static int |
MXPredFree(PredictorHandle handle)
\brief Free a predictor handle.
|
static int |
MXPredGetOutput(PredictorHandle handle,
int index,
float[] data,
int size) |
static int |
MXPredGetOutput(PredictorHandle handle,
int index,
FloatBuffer data,
int size) |
static int |
MXPredGetOutput(PredictorHandle handle,
int index,
FloatPointer data,
int size)
\brief Get the output value of prediction.
|
static int |
MXPredGetOutputShape(PredictorHandle handle,
int index,
int[] shape_data,
int[] shape_ndim) |
static int |
MXPredGetOutputShape(PredictorHandle handle,
int index,
IntBuffer shape_data,
IntBuffer shape_ndim) |
static int |
MXPredGetOutputShape(PredictorHandle handle,
int index,
IntPointer shape_data,
IntPointer shape_ndim) |
static int |
MXPredGetOutputShape(PredictorHandle handle,
int index,
PointerPointer shape_data,
IntPointer shape_ndim)
\brief Get the shape of output node.
|
static int |
MXPredGetOutputType(PredictorHandle handle,
int out_index,
int[] out_dtype) |
static int |
MXPredGetOutputType(PredictorHandle handle,
int out_index,
IntBuffer out_dtype) |
static int |
MXPredGetOutputType(PredictorHandle handle,
int out_index,
IntPointer out_dtype)
\brief Get the dtype of output node.
|
static int |
MXPredPartialForward(PredictorHandle handle,
int step,
int[] step_left) |
static int |
MXPredPartialForward(PredictorHandle handle,
int step,
IntBuffer step_left) |
static int |
MXPredPartialForward(PredictorHandle handle,
int step,
IntPointer step_left)
\brief Run a interactive forward pass to get the output.
|
static int |
MXPredReshape(int num_input_nodes,
byte[] input_keys,
int[] input_shape_indptr,
int[] input_shape_data,
PredictorHandle handle,
PointerPointer out) |
static int |
MXPredReshape(int num_input_nodes,
byte[] input_keys,
int[] input_shape_indptr,
int[] input_shape_data,
PredictorHandle handle,
PredictorHandle out) |
static int |
MXPredReshape(int num_input_nodes,
ByteBuffer input_keys,
IntBuffer input_shape_indptr,
IntBuffer input_shape_data,
PredictorHandle handle,
PointerPointer out) |
static int |
MXPredReshape(int num_input_nodes,
ByteBuffer input_keys,
IntBuffer input_shape_indptr,
IntBuffer input_shape_data,
PredictorHandle handle,
PredictorHandle out) |
static int |
MXPredReshape(int num_input_nodes,
BytePointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
PredictorHandle handle,
PointerPointer out) |
static int |
MXPredReshape(int num_input_nodes,
BytePointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
PredictorHandle handle,
PredictorHandle out) |
static int |
MXPredReshape(int num_input_nodes,
PointerPointer input_keys,
IntPointer input_shape_indptr,
IntPointer input_shape_data,
PredictorHandle handle,
PointerPointer out)
\brief Change the input shape of an existing predictor.
|
static int |
MXPredSetInput(PredictorHandle handle,
BytePointer key,
float[] data,
int size) |
static int |
MXPredSetInput(PredictorHandle handle,
BytePointer key,
FloatBuffer data,
int size) |
static int |
MXPredSetInput(PredictorHandle handle,
BytePointer key,
FloatPointer data,
int size)
\brief Set the input data of predictor.
|
static int |
MXPredSetInput(PredictorHandle handle,
String key,
float[] data,
int size) |
static int |
MXPredSetInput(PredictorHandle handle,
String key,
FloatBuffer data,
int size) |
static int |
MXPredSetInput(PredictorHandle handle,
String key,
FloatPointer data,
int size) |
static int |
MXPredSetMonitorCallback(PredictorHandle handle,
PredMonitorCallback callback,
Pointer callback_handle,
boolean monitor_all)
\brief set a call back to notify the completion of operation and allow for
additional monitoring
|
static int |
MXProcessProfilePause(int paused,
int profile_process,
KVStoreHandle kvStoreHandle)
\brief Pause profiler tuning collection
|
static int |
MXProfileAdjustCounter(ProfileHandle counter_handle,
long value)
\brief Adjust a counter by the given amount, given its handle
|
static int |
MXProfileCreateCounter(ProfileHandle domain,
BytePointer counter_name,
ProfileHandle out)
\brief Create profile counter
|
static int |
MXProfileCreateCounter(ProfileHandle domain,
String counter_name,
ProfileHandle out) |
static int |
MXProfileCreateDomain(BytePointer domain,
ProfileHandle out)
\brief Create profiling domain
|
static int |
MXProfileCreateDomain(String domain,
ProfileHandle out) |
static int |
MXProfileCreateEvent(BytePointer event_name,
ProfileHandle out)
\brief Create profile event
|
static int |
MXProfileCreateEvent(String event_name,
ProfileHandle out) |
static int |
MXProfileCreateFrame(ProfileHandle domain,
BytePointer frame_name,
ProfileHandle out)
\brief Create profile frame
|
static int |
MXProfileCreateFrame(ProfileHandle domain,
String frame_name,
ProfileHandle out) |
static int |
MXProfileCreateTask(ProfileHandle domain,
BytePointer task_name,
ProfileHandle out)
\brief Create profile task
|
static int |
MXProfileCreateTask(ProfileHandle domain,
String task_name,
ProfileHandle out) |
static int |
MXProfileDestroyHandle(ProfileHandle frame_handle)
\brief Destroy a frame
|
static int |
MXProfileDurationStart(ProfileHandle duration_handle)
\brief Start timing the duration of a profile duration object such as an event, task or frame
|
static int |
MXProfileDurationStop(ProfileHandle duration_handle)
\brief Stop timing the duration of a profile duration object such as an event, task or frame
|
static int |
MXProfilePause(int paused)
\brief Pause profiler tuning collection for worker/current process
|
static int |
MXProfileSetCounter(ProfileHandle counter_handle,
long value)
\brief Set a counter, given its handle
|
static int |
MXProfileSetMarker(ProfileHandle domain,
BytePointer instant_marker_name,
BytePointer scope)
\brief Mark a single instant in time
|
static int |
MXProfileSetMarker(ProfileHandle domain,
String instant_marker_name,
String scope) |
static int |
MXQuantizeSymbol(SymbolHandle sym_handle,
PointerPointer ret_sym_handle,
int[] dev_type,
int num_excluded_sym_names,
byte[] excluded_sym_names,
int num_excluded_op_names,
byte[] excluded_op_names,
int num_offline,
byte[] offline_params,
BytePointer quantized_dtype,
boolean calib_quantize,
BytePointer quantize_mode,
BytePointer quantize_granularity,
int[] out_num_calib_names,
PointerPointer out_calib_names) |
static int |
MXQuantizeSymbol(SymbolHandle sym_handle,
PointerPointer ret_sym_handle,
IntBuffer dev_type,
int num_excluded_sym_names,
ByteBuffer excluded_sym_names,
int num_excluded_op_names,
ByteBuffer excluded_op_names,
int num_offline,
ByteBuffer offline_params,
BytePointer quantized_dtype,
boolean calib_quantize,
BytePointer quantize_mode,
BytePointer quantize_granularity,
IntBuffer out_num_calib_names,
PointerPointer out_calib_names) |
static int |
MXQuantizeSymbol(SymbolHandle sym_handle,
PointerPointer ret_sym_handle,
IntPointer dev_type,
int num_excluded_sym_names,
BytePointer excluded_sym_names,
int num_excluded_op_names,
BytePointer excluded_op_names,
int num_offline,
BytePointer offline_params,
BytePointer quantized_dtype,
boolean calib_quantize,
BytePointer quantize_mode,
BytePointer quantize_granularity,
IntPointer out_num_calib_names,
PointerPointer out_calib_names) |
static int |
MXQuantizeSymbol(SymbolHandle sym_handle,
PointerPointer ret_sym_handle,
IntPointer dev_type,
int num_excluded_sym_names,
PointerPointer excluded_sym_names,
int num_excluded_op_names,
PointerPointer excluded_op_names,
int num_offline,
PointerPointer offline_params,
BytePointer quantized_dtype,
boolean calib_quantize,
BytePointer quantize_mode,
BytePointer quantize_granularity,
IntPointer out_num_calib_names,
PointerPointer out_calib_names)
\brief Convert a symbol into a quantized symbol where FP32 operators are replaced with INT8
|
static int |
MXQuantizeSymbol(SymbolHandle sym_handle,
SymbolHandle ret_sym_handle,
int[] dev_type,
int num_excluded_sym_names,
byte[] excluded_sym_names,
int num_excluded_op_names,
byte[] excluded_op_names,
int num_offline,
byte[] offline_params,
String quantized_dtype,
boolean calib_quantize,
String quantize_mode,
String quantize_granularity,
int[] out_num_calib_names,
PointerPointer out_calib_names) |
static int |
MXQuantizeSymbol(SymbolHandle sym_handle,
SymbolHandle ret_sym_handle,
IntBuffer dev_type,
int num_excluded_sym_names,
ByteBuffer excluded_sym_names,
int num_excluded_op_names,
ByteBuffer excluded_op_names,
int num_offline,
ByteBuffer offline_params,
String quantized_dtype,
boolean calib_quantize,
String quantize_mode,
String quantize_granularity,
IntBuffer out_num_calib_names,
PointerPointer out_calib_names) |
static int |
MXQuantizeSymbol(SymbolHandle sym_handle,
SymbolHandle ret_sym_handle,
IntPointer dev_type,
int num_excluded_sym_names,
BytePointer excluded_sym_names,
int num_excluded_op_names,
BytePointer excluded_op_names,
int num_offline,
BytePointer offline_params,
String quantized_dtype,
boolean calib_quantize,
String quantize_mode,
String quantize_granularity,
IntPointer out_num_calib_names,
PointerPointer out_calib_names) |
static int |
MXRandomSeed(int seed)
\brief Seed all global random number generators in mxnet.
|
static int |
MXRandomSeedContext(int seed,
int dev_type,
int dev_id)
\brief Seed the global random number generator of the given device.
|
static int |
MXRecordIOReaderCreate(BytePointer uri,
PointerPointer out)
\brief Create a RecordIO reader object
|
static int |
MXRecordIOReaderCreate(String uri,
RecordIOHandle out) |
static int |
MXRecordIOReaderFree(RecordIOHandle handle)
\brief Delete a RecordIO reader object
|
static int |
MXRecordIOReaderReadRecord(RecordIOHandle handle,
byte[] buf,
SizeTPointer size) |
static int |
MXRecordIOReaderReadRecord(RecordIOHandle handle,
ByteBuffer buf,
SizeTPointer size) |
static int |
MXRecordIOReaderReadRecord(RecordIOHandle handle,
BytePointer buf,
SizeTPointer size) |
static int |
MXRecordIOReaderReadRecord(RecordIOHandle handle,
PointerPointer buf,
SizeTPointer size)
\brief Write a record to a RecordIO object
|
static int |
MXRecordIOReaderSeek(RecordIOHandle handle,
long pos)
\brief Set the current reader pointer position
|
static int |
MXRecordIOReaderTell(RecordIOHandle handle,
SizeTPointer pos)
\brief Get the current writer pointer position
|
static int |
MXRecordIOWriterCreate(BytePointer uri,
PointerPointer out)
\brief Create a RecordIO writer object
|
static int |
MXRecordIOWriterCreate(String uri,
RecordIOHandle out) |
static int |
MXRecordIOWriterFree(RecordIOHandle handle)
\brief Delete a RecordIO writer object
|
static int |
MXRecordIOWriterTell(RecordIOHandle handle,
SizeTPointer pos)
\brief Get the current writer pointer position
|
static int |
MXRecordIOWriterWriteRecord(RecordIOHandle handle,
BytePointer buf,
long size)
\brief Write a record to a RecordIO object
|
static int |
MXRecordIOWriterWriteRecord(RecordIOHandle handle,
String buf,
long size) |
static int |
MXReducePrecisionSymbol(SymbolHandle sym_handle,
PointerPointer 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,
byte[] target_dtype_op_names,
byte[] fp32_op_names,
byte[] widest_dtype_op_names,
byte[] conditional_fp32_op_names,
byte[] excluded_symbols,
byte[] conditional_param_names,
byte[] conditional_param_vals,
byte[] model_param_names,
byte[] arg_names) |
static int |
MXReducePrecisionSymbol(SymbolHandle sym_handle,
PointerPointer ret_sym_handle,
int num_args,
IntBuffer arg_type_data,
int num_ind_ptr,
IntBuffer ind_ptr,
IntBuffer 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,
ByteBuffer target_dtype_op_names,
ByteBuffer fp32_op_names,
ByteBuffer widest_dtype_op_names,
ByteBuffer conditional_fp32_op_names,
ByteBuffer excluded_symbols,
ByteBuffer conditional_param_names,
ByteBuffer conditional_param_vals,
ByteBuffer model_param_names,
ByteBuffer arg_names) |
static int |
MXReducePrecisionSymbol(SymbolHandle sym_handle,
PointerPointer ret_sym_handle,
int num_args,
IntPointer arg_type_data,
int num_ind_ptr,
IntPointer ind_ptr,
IntPointer 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,
BytePointer target_dtype_op_names,
BytePointer fp32_op_names,
BytePointer widest_dtype_op_names,
BytePointer conditional_fp32_op_names,
BytePointer excluded_symbols,
BytePointer conditional_param_names,
BytePointer conditional_param_vals,
BytePointer model_param_names,
BytePointer arg_names) |
static int |
MXReducePrecisionSymbol(SymbolHandle sym_handle,
PointerPointer ret_sym_handle,
int num_args,
IntPointer arg_type_data,
int num_ind_ptr,
IntPointer ind_ptr,
IntPointer 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,
PointerPointer target_dtype_op_names,
PointerPointer fp32_op_names,
PointerPointer widest_dtype_op_names,
PointerPointer conditional_fp32_op_names,
PointerPointer excluded_symbols,
PointerPointer conditional_param_names,
PointerPointer conditional_param_vals,
PointerPointer model_param_names,
PointerPointer arg_names)
\brief Convert a symbol into a mixed precision symbol with cast operators for target dtype casting
|
static int |
MXReducePrecisionSymbol(SymbolHandle sym_handle,
SymbolHandle 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,
byte[] target_dtype_op_names,
byte[] fp32_op_names,
byte[] widest_dtype_op_names,
byte[] conditional_fp32_op_names,
byte[] excluded_symbols,
byte[] conditional_param_names,
byte[] conditional_param_vals,
byte[] model_param_names,
byte[] arg_names) |
static int |
MXReducePrecisionSymbol(SymbolHandle sym_handle,
SymbolHandle ret_sym_handle,
int num_args,
IntBuffer arg_type_data,
int num_ind_ptr,
IntBuffer ind_ptr,
IntBuffer 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,
ByteBuffer target_dtype_op_names,
ByteBuffer fp32_op_names,
ByteBuffer widest_dtype_op_names,
ByteBuffer conditional_fp32_op_names,
ByteBuffer excluded_symbols,
ByteBuffer conditional_param_names,
ByteBuffer conditional_param_vals,
ByteBuffer model_param_names,
ByteBuffer arg_names) |
static int |
MXReducePrecisionSymbol(SymbolHandle sym_handle,
SymbolHandle ret_sym_handle,
int num_args,
IntPointer arg_type_data,
int num_ind_ptr,
IntPointer ind_ptr,
IntPointer 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,
BytePointer target_dtype_op_names,
BytePointer fp32_op_names,
BytePointer widest_dtype_op_names,
BytePointer conditional_fp32_op_names,
BytePointer excluded_symbols,
BytePointer conditional_param_names,
BytePointer conditional_param_vals,
BytePointer model_param_names,
BytePointer arg_names) |
static int |
MXRtcCreate(byte[] name,
int num_input,
int num_output,
byte[] input_names,
byte[] output_names,
NDArrayHandle inputs,
NDArrayHandle outputs,
byte[] kernel,
RtcHandle out) |
static int |
MXRtcCreate(byte[] name,
int num_input,
int num_output,
byte[] input_names,
byte[] output_names,
PointerPointer inputs,
PointerPointer outputs,
byte[] kernel,
PointerPointer out) |
static int |
MXRtcCreate(ByteBuffer name,
int num_input,
int num_output,
ByteBuffer input_names,
ByteBuffer output_names,
NDArrayHandle inputs,
NDArrayHandle outputs,
ByteBuffer kernel,
RtcHandle out) |
static int |
MXRtcCreate(ByteBuffer name,
int num_input,
int num_output,
ByteBuffer input_names,
ByteBuffer output_names,
PointerPointer inputs,
PointerPointer outputs,
ByteBuffer kernel,
PointerPointer out) |
static int |
MXRtcCreate(BytePointer name,
int num_input,
int num_output,
BytePointer input_names,
BytePointer output_names,
NDArrayHandle inputs,
NDArrayHandle outputs,
BytePointer kernel,
RtcHandle out) |
static int |
MXRtcCreate(BytePointer name,
int num_input,
int num_output,
BytePointer input_names,
BytePointer output_names,
PointerPointer inputs,
PointerPointer outputs,
BytePointer kernel,
PointerPointer out) |
static int |
MXRtcCreate(BytePointer name,
int num_input,
int num_output,
PointerPointer input_names,
PointerPointer output_names,
PointerPointer inputs,
PointerPointer outputs,
BytePointer kernel,
PointerPointer out)
\brief Create a MXRtc object
|
static int |
MXRtcCudaKernelCall(CudaKernelHandle handle,
int dev_id,
Pointer 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) |
static int |
MXRtcCudaKernelCall(CudaKernelHandle handle,
int dev_id,
PointerPointer 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) |
static int |
MXRtcCudaKernelCreate(CudaModuleHandle handle,
BytePointer name,
int num_args,
int[] is_ndarray,
int[] is_const,
int[] arg_types,
CudaKernelHandle out) |
static int |
MXRtcCudaKernelCreate(CudaModuleHandle handle,
BytePointer name,
int num_args,
IntBuffer is_ndarray,
IntBuffer is_const,
IntBuffer arg_types,
CudaKernelHandle out) |
static int |
MXRtcCudaKernelCreate(CudaModuleHandle handle,
BytePointer name,
int num_args,
IntPointer is_ndarray,
IntPointer is_const,
IntPointer arg_types,
CudaKernelHandle out) |
static int |
MXRtcCudaKernelCreate(CudaModuleHandle handle,
String name,
int num_args,
int[] is_ndarray,
int[] is_const,
int[] arg_types,
CudaKernelHandle out) |
static int |
MXRtcCudaKernelCreate(CudaModuleHandle handle,
String name,
int num_args,
IntBuffer is_ndarray,
IntBuffer is_const,
IntBuffer arg_types,
CudaKernelHandle out) |
static int |
MXRtcCudaKernelCreate(CudaModuleHandle handle,
String name,
int num_args,
IntPointer is_ndarray,
IntPointer is_const,
IntPointer arg_types,
CudaKernelHandle out) |
static int |
MXRtcCudaKernelFree(CudaKernelHandle handle) |
static int |
MXRtcCudaModuleCreate(BytePointer source,
int num_options,
byte[] options,
int num_exports,
byte[] exports,
CudaModuleHandle out) |
static int |
MXRtcCudaModuleCreate(BytePointer source,
int num_options,
ByteBuffer options,
int num_exports,
ByteBuffer exports,
CudaModuleHandle out) |
static int |
MXRtcCudaModuleCreate(BytePointer source,
int num_options,
BytePointer options,
int num_exports,
BytePointer exports,
CudaModuleHandle out) |
static int |
MXRtcCudaModuleCreate(BytePointer source,
int num_options,
PointerPointer options,
int num_exports,
PointerPointer exports,
CudaModuleHandle out) |
static int |
MXRtcCudaModuleCreate(String source,
int num_options,
byte[] options,
int num_exports,
byte[] exports,
CudaModuleHandle out) |
static int |
MXRtcCudaModuleCreate(String source,
int num_options,
ByteBuffer options,
int num_exports,
ByteBuffer exports,
CudaModuleHandle out) |
static int |
MXRtcCudaModuleCreate(String source,
int num_options,
BytePointer options,
int num_exports,
BytePointer exports,
CudaModuleHandle out) |
static int |
MXRtcCudaModuleFree(CudaModuleHandle handle) |
static int |
MXRtcFree(RtcHandle handle)
\brief Delete a MXRtc object
|
static int |
MXRtcPush(RtcHandle handle,
int num_input,
int num_output,
NDArrayHandle inputs,
NDArrayHandle outputs,
int gridDimX,
int gridDimY,
int gridDimZ,
int blockDimX,
int blockDimY,
int blockDimZ) |
static int |
MXRtcPush(RtcHandle handle,
int num_input,
int num_output,
PointerPointer inputs,
PointerPointer outputs,
int gridDimX,
int gridDimY,
int gridDimZ,
int blockDimX,
int blockDimY,
int blockDimZ)
\brief Run cuda kernel
|
static int |
MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle,
int num_layers,
byte[] layer_names,
float[] low_quantiles,
float[] high_quantiles,
PointerPointer ret_sym_handle) |
static int |
MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle,
int num_layers,
byte[] layer_names,
float[] low_quantiles,
float[] high_quantiles,
SymbolHandle ret_sym_handle) |
static int |
MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle,
int num_layers,
ByteBuffer layer_names,
FloatBuffer low_quantiles,
FloatBuffer high_quantiles,
PointerPointer ret_sym_handle) |
static int |
MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle,
int num_layers,
ByteBuffer layer_names,
FloatBuffer low_quantiles,
FloatBuffer high_quantiles,
SymbolHandle ret_sym_handle) |
static int |
MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle,
int num_layers,
BytePointer layer_names,
FloatPointer low_quantiles,
FloatPointer high_quantiles,
PointerPointer ret_sym_handle) |
static int |
MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle,
int num_layers,
BytePointer layer_names,
FloatPointer low_quantiles,
FloatPointer high_quantiles,
SymbolHandle ret_sym_handle) |
static int |
MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle,
int num_layers,
PointerPointer layer_names,
FloatPointer low_quantiles,
FloatPointer high_quantiles,
PointerPointer ret_sym_handle)
\brief Set calibration table to node attributes in the sym
|
static int |
MXSetIsNumpyShape(int is_np_shape,
int[] prev) |
static int |
MXSetIsNumpyShape(int is_np_shape,
IntBuffer prev) |
static int |
MXSetIsNumpyShape(int is_np_shape,
IntPointer prev)
\brief set numpy compatibility switch
|
static int |
MXSetNumOMPThreads(int thread_num)
\brief Set the number of OMP threads to use
|
static int |
MXSetProcessProfilerConfig(int num_params,
byte[] keys,
byte[] vals,
KVStoreHandle kvstoreHandle) |
static int |
MXSetProcessProfilerConfig(int num_params,
ByteBuffer keys,
ByteBuffer vals,
KVStoreHandle kvstoreHandle) |
static int |
MXSetProcessProfilerConfig(int num_params,
BytePointer keys,
BytePointer vals,
KVStoreHandle kvstoreHandle) |
static int |
MXSetProcessProfilerConfig(int num_params,
PointerPointer keys,
PointerPointer vals,
KVStoreHandle kvstoreHandle)
\brief Set up configuration of profiler for the process passed as profile_process in keys
|
static int |
MXSetProcessProfilerState(int state,
int profile_process,
KVStoreHandle kvStoreHandle)
\brief Set up state of profiler for either worker or server process
|
static int |
MXSetProfilerConfig(int num_params,
byte[] keys,
byte[] vals) |
static int |
MXSetProfilerConfig(int num_params,
ByteBuffer keys,
ByteBuffer vals) |
static int |
MXSetProfilerConfig(int num_params,
BytePointer keys,
BytePointer vals) |
static int |
MXSetProfilerConfig(int num_params,
PointerPointer keys,
PointerPointer vals)
\brief Set up configuration of profiler for worker/current process
|
static int |
MXSetProfilerState(int state)
\brief Set up state of profiler for current process
|
static int |
MXShallowCopyNDArray(NDArrayHandle src,
NDArrayHandle out) |
static int |
MXShallowCopyNDArray(NDArrayHandle src,
PointerPointer out)
\brief Create an NDArray from source sharing the same data chunk.
|
static int |
MXShallowCopySymbol(SymbolHandle src,
PointerPointer out)
\brief Create an Symbol from source sharing the same graph structure.
|
static int |
MXShallowCopySymbol(SymbolHandle src,
SymbolHandle out) |
static int |
MXStorageEmptyCache(int dev_type,
int dev_id)
\brief Release all unreferenced memory from the devices storage managers memory pool
|
static int |
MXSymbolCompose(SymbolHandle sym,
BytePointer name,
int num_args,
byte[] keys,
PointerPointer args) |
static int |
MXSymbolCompose(SymbolHandle sym,
BytePointer name,
int num_args,
ByteBuffer keys,
PointerPointer args) |
static int |
MXSymbolCompose(SymbolHandle sym,
BytePointer name,
int num_args,
BytePointer keys,
PointerPointer args) |
static int |
MXSymbolCompose(SymbolHandle sym,
BytePointer name,
int num_args,
PointerPointer keys,
PointerPointer args)
\brief Compose the symbol on other symbols.
|
static int |
MXSymbolCompose(SymbolHandle sym,
String name,
int num_args,
byte[] keys,
SymbolHandle args) |
static int |
MXSymbolCompose(SymbolHandle sym,
String name,
int num_args,
ByteBuffer keys,
SymbolHandle args) |
static int |
MXSymbolCompose(SymbolHandle sym,
String name,
int num_args,
BytePointer keys,
SymbolHandle args) |
static int |
MXSymbolCopy(SymbolHandle symbol,
PointerPointer out)
\brief Copy the symbol to another handle
|
static int |
MXSymbolCopy(SymbolHandle symbol,
SymbolHandle out) |
static int |
MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator,
int num_param,
byte[] keys,
byte[] vals,
PointerPointer out) |
static int |
MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator,
int num_param,
byte[] keys,
byte[] vals,
SymbolHandle out) |
static int |
MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator,
int num_param,
ByteBuffer keys,
ByteBuffer vals,
PointerPointer out) |
static int |
MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator,
int num_param,
ByteBuffer keys,
ByteBuffer vals,
SymbolHandle out) |
static int |
MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator,
int num_param,
BytePointer keys,
BytePointer vals,
PointerPointer out) |
static int |
MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator,
int num_param,
BytePointer keys,
BytePointer vals,
SymbolHandle out) |
static int |
MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator,
int num_param,
PointerPointer keys,
PointerPointer vals,
PointerPointer out)
\brief Create an AtomicSymbol.
|
static int |
MXSymbolCreateFromFile(BytePointer fname,
PointerPointer out)
\brief Load a symbol from a json file.
|
static int |
MXSymbolCreateFromFile(String fname,
SymbolHandle out) |
static int |
MXSymbolCreateFromJSON(BytePointer json,
PointerPointer out)
\brief Load a symbol from a json string.
|
static int |
MXSymbolCreateFromJSON(String json,
SymbolHandle out) |
static int |
MXSymbolCreateGroup(int num_symbols,
PointerPointer symbols,
PointerPointer out)
\brief Create a Symbol by grouping list of symbols together
|
static int |
MXSymbolCreateGroup(int num_symbols,
SymbolHandle symbols,
SymbolHandle out) |
static int |
MXSymbolCreateVariable(BytePointer name,
PointerPointer out)
\brief Create a Variable Symbol.
|
static int |
MXSymbolCreateVariable(String name,
SymbolHandle out) |
static int |
MXSymbolCutSubgraph(SymbolHandle sym,
PointerPointer inputs,
int[] input_size) |
static int |
MXSymbolCutSubgraph(SymbolHandle sym,
PointerPointer inputs,
IntBuffer input_size) |
static int |
MXSymbolCutSubgraph(SymbolHandle sym,
PointerPointer inputs,
IntPointer input_size)
\brief Cut a subgraph whose nodes are marked with a subgraph attribute.
|
static int |
MXSymbolFree(SymbolHandle symbol)
\brief Free the symbol handle.
|
static int |
MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator,
byte[] name,
byte[] description,
int[] num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
byte[] key_var_num_args) |
static int |
MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator,
byte[] name,
byte[] description,
int[] num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
byte[] key_var_num_args,
byte[] return_type) |
static int |
MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator,
ByteBuffer name,
ByteBuffer description,
IntBuffer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
ByteBuffer key_var_num_args) |
static int |
MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator,
ByteBuffer name,
ByteBuffer description,
IntBuffer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
ByteBuffer key_var_num_args,
ByteBuffer return_type) |
static int |
MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator,
BytePointer name,
BytePointer description,
IntPointer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
BytePointer key_var_num_args) |
static int |
MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator,
BytePointer name,
BytePointer description,
IntPointer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
BytePointer key_var_num_args,
BytePointer return_type) |
static int |
MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator,
PointerPointer name,
PointerPointer description,
IntPointer num_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
PointerPointer key_var_num_args,
PointerPointer return_type)
\brief Get the detailed information about atomic symbol.
|
static int |
MXSymbolGetAtomicSymbolName(AtomicSymbolCreator creator,
byte[] name) |
static int |
MXSymbolGetAtomicSymbolName(AtomicSymbolCreator creator,
ByteBuffer name) |
static int |
MXSymbolGetAtomicSymbolName(AtomicSymbolCreator creator,
BytePointer name) |
static int |
MXSymbolGetAtomicSymbolName(AtomicSymbolCreator creator,
PointerPointer name)
\brief Get the name of an atomic symbol.
|
static int |
MXSymbolGetAttr(SymbolHandle symbol,
BytePointer key,
byte[] out,
int[] success) |
static int |
MXSymbolGetAttr(SymbolHandle symbol,
BytePointer key,
ByteBuffer out,
IntBuffer success) |
static int |
MXSymbolGetAttr(SymbolHandle symbol,
BytePointer key,
BytePointer out,
IntPointer success) |
static int |
MXSymbolGetAttr(SymbolHandle symbol,
BytePointer key,
PointerPointer out,
IntPointer success)
\brief Get string attribute from symbol
|
static int |
MXSymbolGetAttr(SymbolHandle symbol,
String key,
byte[] out,
int[] success) |
static int |
MXSymbolGetAttr(SymbolHandle symbol,
String key,
ByteBuffer out,
IntBuffer success) |
static int |
MXSymbolGetAttr(SymbolHandle symbol,
String key,
BytePointer out,
IntPointer success) |
static int |
MXSymbolGetChildren(SymbolHandle symbol,
PointerPointer out)
\brief Get a symbol that contains only direct children.
|
static int |
MXSymbolGetChildren(SymbolHandle symbol,
SymbolHandle out) |
static int |
MXSymbolGetInputSymbols(SymbolHandle sym,
PointerPointer inputs,
int[] input_size) |
static int |
MXSymbolGetInputSymbols(SymbolHandle sym,
PointerPointer inputs,
IntBuffer input_size) |
static int |
MXSymbolGetInputSymbols(SymbolHandle sym,
PointerPointer inputs,
IntPointer input_size)
\brief Get the input symbols of the graph.
|
static int |
MXSymbolGetInternals(SymbolHandle symbol,
PointerPointer out)
\brief Get a symbol that contains all the internals.
|
static int |
MXSymbolGetInternals(SymbolHandle symbol,
SymbolHandle out) |
static int |
MXSymbolGetName(SymbolHandle symbol,
byte[] out,
int[] success) |
static int |
MXSymbolGetName(SymbolHandle symbol,
ByteBuffer out,
IntBuffer success) |
static int |
MXSymbolGetName(SymbolHandle symbol,
BytePointer out,
IntPointer success) |
static int |
MXSymbolGetName(SymbolHandle symbol,
PointerPointer out,
IntPointer success)
\brief Get string name from symbol
|
static int |
MXSymbolGetNumOutputs(SymbolHandle symbol,
int[] output_count) |
static int |
MXSymbolGetNumOutputs(SymbolHandle symbol,
IntBuffer output_count) |
static int |
MXSymbolGetNumOutputs(SymbolHandle symbol,
IntPointer output_count)
\brief Get number of outputs of the symbol.
|
static int |
MXSymbolGetOutput(SymbolHandle symbol,
int index,
PointerPointer out)
\brief Get index-th outputs of the symbol.
|
static int |
MXSymbolGetOutput(SymbolHandle symbol,
int index,
SymbolHandle out) |
static int |
MXSymbolGrad(SymbolHandle sym,
int num_wrt,
byte[] wrt,
PointerPointer out) |
static int |
MXSymbolGrad(SymbolHandle sym,
int num_wrt,
byte[] wrt,
SymbolHandle out) |
static int |
MXSymbolGrad(SymbolHandle sym,
int num_wrt,
ByteBuffer wrt,
PointerPointer out) |
static int |
MXSymbolGrad(SymbolHandle sym,
int num_wrt,
ByteBuffer wrt,
SymbolHandle out) |
static int |
MXSymbolGrad(SymbolHandle sym,
int num_wrt,
BytePointer wrt,
PointerPointer out) |
static int |
MXSymbolGrad(SymbolHandle sym,
int num_wrt,
BytePointer wrt,
SymbolHandle out) |
static int |
MXSymbolGrad(SymbolHandle sym,
int num_wrt,
PointerPointer wrt,
PointerPointer out)
\brief Get the gradient graph of the symbol
|
static int |
MXSymbolInferShape(SymbolHandle sym,
int num_args,
byte[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
int[] in_shape_size,
int[] in_shape_ndim,
PointerPointer in_shape_data,
int[] out_shape_size,
int[] out_shape_ndim,
PointerPointer out_shape_data,
int[] aux_shape_size,
int[] aux_shape_ndim,
PointerPointer aux_shape_data,
int[] complete) |
static int |
MXSymbolInferShape(SymbolHandle sym,
int num_args,
ByteBuffer keys,
IntBuffer arg_ind_ptr,
IntBuffer arg_shape_data,
IntBuffer in_shape_size,
IntBuffer in_shape_ndim,
PointerPointer in_shape_data,
IntBuffer out_shape_size,
IntBuffer out_shape_ndim,
PointerPointer out_shape_data,
IntBuffer aux_shape_size,
IntBuffer aux_shape_ndim,
PointerPointer aux_shape_data,
IntBuffer complete) |
static int |
MXSymbolInferShape(SymbolHandle sym,
int num_args,
BytePointer keys,
IntPointer arg_ind_ptr,
IntPointer arg_shape_data,
IntPointer in_shape_size,
IntPointer in_shape_ndim,
PointerPointer in_shape_data,
IntPointer out_shape_size,
IntPointer out_shape_ndim,
PointerPointer out_shape_data,
IntPointer aux_shape_size,
IntPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete) |
static int |
MXSymbolInferShape(SymbolHandle sym,
int num_args,
PointerPointer keys,
IntPointer arg_ind_ptr,
IntPointer arg_shape_data,
IntPointer in_shape_size,
PointerPointer in_shape_ndim,
PointerPointer in_shape_data,
IntPointer out_shape_size,
PointerPointer out_shape_ndim,
PointerPointer out_shape_data,
IntPointer aux_shape_size,
PointerPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete)
\brief DEPRECATED.
|
static int |
MXSymbolInferShapeEx(SymbolHandle sym,
int num_args,
byte[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
int[] in_shape_size,
int[] in_shape_ndim,
PointerPointer in_shape_data,
int[] out_shape_size,
int[] out_shape_ndim,
PointerPointer out_shape_data,
int[] aux_shape_size,
int[] aux_shape_ndim,
PointerPointer aux_shape_data,
int[] complete) |
static int |
MXSymbolInferShapeEx(SymbolHandle sym,
int num_args,
ByteBuffer keys,
IntBuffer arg_ind_ptr,
IntBuffer arg_shape_data,
IntBuffer in_shape_size,
IntBuffer in_shape_ndim,
PointerPointer in_shape_data,
IntBuffer out_shape_size,
IntBuffer out_shape_ndim,
PointerPointer out_shape_data,
IntBuffer aux_shape_size,
IntBuffer aux_shape_ndim,
PointerPointer aux_shape_data,
IntBuffer complete) |
static int |
MXSymbolInferShapeEx(SymbolHandle sym,
int num_args,
BytePointer keys,
IntPointer arg_ind_ptr,
IntPointer arg_shape_data,
IntPointer in_shape_size,
IntPointer in_shape_ndim,
PointerPointer in_shape_data,
IntPointer out_shape_size,
IntPointer out_shape_ndim,
PointerPointer out_shape_data,
IntPointer aux_shape_size,
IntPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete) |
static int |
MXSymbolInferShapeEx(SymbolHandle sym,
int num_args,
PointerPointer keys,
IntPointer arg_ind_ptr,
IntPointer arg_shape_data,
IntPointer in_shape_size,
PointerPointer in_shape_ndim,
PointerPointer in_shape_data,
IntPointer out_shape_size,
PointerPointer out_shape_ndim,
PointerPointer out_shape_data,
IntPointer aux_shape_size,
PointerPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete)
\brief infer shape of unknown input shapes given the known one.
|
static int |
MXSymbolInferShapeEx64(SymbolHandle sym,
int num_args,
byte[] keys,
long[] arg_ind_ptr,
long[] arg_shape_data,
SizeTPointer in_shape_size,
int[] in_shape_ndim,
PointerPointer in_shape_data,
SizeTPointer out_shape_size,
int[] out_shape_ndim,
PointerPointer out_shape_data,
SizeTPointer aux_shape_size,
int[] aux_shape_ndim,
PointerPointer aux_shape_data,
int[] complete) |
static int |
MXSymbolInferShapeEx64(SymbolHandle sym,
int num_args,
ByteBuffer keys,
LongBuffer arg_ind_ptr,
LongBuffer arg_shape_data,
SizeTPointer in_shape_size,
IntBuffer in_shape_ndim,
PointerPointer in_shape_data,
SizeTPointer out_shape_size,
IntBuffer out_shape_ndim,
PointerPointer out_shape_data,
SizeTPointer aux_shape_size,
IntBuffer aux_shape_ndim,
PointerPointer aux_shape_data,
IntBuffer complete) |
static int |
MXSymbolInferShapeEx64(SymbolHandle sym,
int num_args,
BytePointer keys,
LongPointer arg_ind_ptr,
LongPointer arg_shape_data,
SizeTPointer in_shape_size,
IntPointer in_shape_ndim,
PointerPointer in_shape_data,
SizeTPointer out_shape_size,
IntPointer out_shape_ndim,
PointerPointer out_shape_data,
SizeTPointer aux_shape_size,
IntPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete) |
static int |
MXSymbolInferShapeEx64(SymbolHandle sym,
int num_args,
PointerPointer keys,
LongPointer arg_ind_ptr,
LongPointer arg_shape_data,
SizeTPointer in_shape_size,
PointerPointer in_shape_ndim,
PointerPointer in_shape_data,
SizeTPointer out_shape_size,
PointerPointer out_shape_ndim,
PointerPointer out_shape_data,
SizeTPointer aux_shape_size,
PointerPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete)
\brief infer shape of unknown input shapes given the known one.
|
static int |
MXSymbolInferShapePartial(SymbolHandle sym,
int num_args,
byte[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
int[] in_shape_size,
int[] in_shape_ndim,
PointerPointer in_shape_data,
int[] out_shape_size,
int[] out_shape_ndim,
PointerPointer out_shape_data,
int[] aux_shape_size,
int[] aux_shape_ndim,
PointerPointer aux_shape_data,
int[] complete) |
static int |
MXSymbolInferShapePartial(SymbolHandle sym,
int num_args,
ByteBuffer keys,
IntBuffer arg_ind_ptr,
IntBuffer arg_shape_data,
IntBuffer in_shape_size,
IntBuffer in_shape_ndim,
PointerPointer in_shape_data,
IntBuffer out_shape_size,
IntBuffer out_shape_ndim,
PointerPointer out_shape_data,
IntBuffer aux_shape_size,
IntBuffer aux_shape_ndim,
PointerPointer aux_shape_data,
IntBuffer complete) |
static int |
MXSymbolInferShapePartial(SymbolHandle sym,
int num_args,
BytePointer keys,
IntPointer arg_ind_ptr,
IntPointer arg_shape_data,
IntPointer in_shape_size,
IntPointer in_shape_ndim,
PointerPointer in_shape_data,
IntPointer out_shape_size,
IntPointer out_shape_ndim,
PointerPointer out_shape_data,
IntPointer aux_shape_size,
IntPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete) |
static int |
MXSymbolInferShapePartial(SymbolHandle sym,
int num_args,
PointerPointer keys,
IntPointer arg_ind_ptr,
IntPointer arg_shape_data,
IntPointer in_shape_size,
PointerPointer in_shape_ndim,
PointerPointer in_shape_data,
IntPointer out_shape_size,
PointerPointer out_shape_ndim,
PointerPointer out_shape_data,
IntPointer aux_shape_size,
PointerPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete)
\brief DEPRECATED.
|
static int |
MXSymbolInferShapePartialEx(SymbolHandle sym,
int num_args,
byte[] keys,
int[] arg_ind_ptr,
int[] arg_shape_data,
int[] in_shape_size,
int[] in_shape_ndim,
PointerPointer in_shape_data,
int[] out_shape_size,
int[] out_shape_ndim,
PointerPointer out_shape_data,
int[] aux_shape_size,
int[] aux_shape_ndim,
PointerPointer aux_shape_data,
int[] complete) |
static int |
MXSymbolInferShapePartialEx(SymbolHandle sym,
int num_args,
ByteBuffer keys,
IntBuffer arg_ind_ptr,
IntBuffer arg_shape_data,
IntBuffer in_shape_size,
IntBuffer in_shape_ndim,
PointerPointer in_shape_data,
IntBuffer out_shape_size,
IntBuffer out_shape_ndim,
PointerPointer out_shape_data,
IntBuffer aux_shape_size,
IntBuffer aux_shape_ndim,
PointerPointer aux_shape_data,
IntBuffer complete) |
static int |
MXSymbolInferShapePartialEx(SymbolHandle sym,
int num_args,
BytePointer keys,
IntPointer arg_ind_ptr,
IntPointer arg_shape_data,
IntPointer in_shape_size,
IntPointer in_shape_ndim,
PointerPointer in_shape_data,
IntPointer out_shape_size,
IntPointer out_shape_ndim,
PointerPointer out_shape_data,
IntPointer aux_shape_size,
IntPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete) |
static int |
MXSymbolInferShapePartialEx(SymbolHandle sym,
int num_args,
PointerPointer keys,
IntPointer arg_ind_ptr,
IntPointer arg_shape_data,
IntPointer in_shape_size,
PointerPointer in_shape_ndim,
PointerPointer in_shape_data,
IntPointer out_shape_size,
PointerPointer out_shape_ndim,
PointerPointer out_shape_data,
IntPointer aux_shape_size,
PointerPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete)
\brief partially infer shape of unknown input shapes given the known one.
|
static int |
MXSymbolInferShapePartialEx64(SymbolHandle sym,
int num_args,
byte[] keys,
long[] arg_ind_ptr,
long[] arg_shape_data,
SizeTPointer in_shape_size,
int[] in_shape_ndim,
PointerPointer in_shape_data,
SizeTPointer out_shape_size,
int[] out_shape_ndim,
PointerPointer out_shape_data,
SizeTPointer aux_shape_size,
int[] aux_shape_ndim,
PointerPointer aux_shape_data,
int[] complete) |
static int |
MXSymbolInferShapePartialEx64(SymbolHandle sym,
int num_args,
ByteBuffer keys,
LongBuffer arg_ind_ptr,
LongBuffer arg_shape_data,
SizeTPointer in_shape_size,
IntBuffer in_shape_ndim,
PointerPointer in_shape_data,
SizeTPointer out_shape_size,
IntBuffer out_shape_ndim,
PointerPointer out_shape_data,
SizeTPointer aux_shape_size,
IntBuffer aux_shape_ndim,
PointerPointer aux_shape_data,
IntBuffer complete) |
static int |
MXSymbolInferShapePartialEx64(SymbolHandle sym,
int num_args,
BytePointer keys,
LongPointer arg_ind_ptr,
LongPointer arg_shape_data,
SizeTPointer in_shape_size,
IntPointer in_shape_ndim,
PointerPointer in_shape_data,
SizeTPointer out_shape_size,
IntPointer out_shape_ndim,
PointerPointer out_shape_data,
SizeTPointer aux_shape_size,
IntPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete) |
static int |
MXSymbolInferShapePartialEx64(SymbolHandle sym,
int num_args,
PointerPointer keys,
LongPointer arg_ind_ptr,
LongPointer arg_shape_data,
SizeTPointer in_shape_size,
PointerPointer in_shape_ndim,
PointerPointer in_shape_data,
SizeTPointer out_shape_size,
PointerPointer out_shape_ndim,
PointerPointer out_shape_data,
SizeTPointer aux_shape_size,
PointerPointer aux_shape_ndim,
PointerPointer aux_shape_data,
IntPointer complete)
\brief partially infer shape of unknown input shapes given the known one.
|
static int |
MXSymbolInferType(SymbolHandle sym,
int num_args,
byte[] keys,
int[] arg_type_data,
int[] in_type_size,
int[] in_type_data,
int[] out_type_size,
int[] out_type_data,
int[] aux_type_size,
int[] aux_type_data,
int[] complete) |
static int |
MXSymbolInferType(SymbolHandle sym,
int num_args,
ByteBuffer keys,
IntBuffer arg_type_data,
IntBuffer in_type_size,
IntBuffer in_type_data,
IntBuffer out_type_size,
IntBuffer out_type_data,
IntBuffer aux_type_size,
IntBuffer aux_type_data,
IntBuffer complete) |
static int |
MXSymbolInferType(SymbolHandle sym,
int num_args,
BytePointer keys,
IntPointer arg_type_data,
IntPointer in_type_size,
IntPointer in_type_data,
IntPointer out_type_size,
IntPointer out_type_data,
IntPointer aux_type_size,
IntPointer aux_type_data,
IntPointer complete) |
static int |
MXSymbolInferType(SymbolHandle sym,
int num_args,
PointerPointer keys,
IntPointer arg_type_data,
IntPointer in_type_size,
PointerPointer in_type_data,
IntPointer out_type_size,
PointerPointer out_type_data,
IntPointer aux_type_size,
PointerPointer aux_type_data,
IntPointer complete)
\brief infer type of unknown input types given the known one.
|
static int |
MXSymbolInferTypePartial(SymbolHandle sym,
int num_args,
byte[] keys,
int[] arg_type_data,
int[] in_type_size,
int[] in_type_data,
int[] out_type_size,
int[] out_type_data,
int[] aux_type_size,
int[] aux_type_data,
int[] complete) |
static int |
MXSymbolInferTypePartial(SymbolHandle sym,
int num_args,
ByteBuffer keys,
IntBuffer arg_type_data,
IntBuffer in_type_size,
IntBuffer in_type_data,
IntBuffer out_type_size,
IntBuffer out_type_data,
IntBuffer aux_type_size,
IntBuffer aux_type_data,
IntBuffer complete) |
static int |
MXSymbolInferTypePartial(SymbolHandle sym,
int num_args,
BytePointer keys,
IntPointer arg_type_data,
IntPointer in_type_size,
IntPointer in_type_data,
IntPointer out_type_size,
IntPointer out_type_data,
IntPointer aux_type_size,
IntPointer aux_type_data,
IntPointer complete) |
static int |
MXSymbolInferTypePartial(SymbolHandle sym,
int num_args,
PointerPointer keys,
IntPointer arg_type_data,
IntPointer in_type_size,
PointerPointer in_type_data,
IntPointer out_type_size,
PointerPointer out_type_data,
IntPointer aux_type_size,
PointerPointer aux_type_data,
IntPointer complete)
\brief partially infer type of unknown input types given the known one.
|
static int |
MXSymbolListArguments(SymbolHandle symbol,
int[] out_size,
PointerPointer out_str_array) |
static int |
MXSymbolListArguments(SymbolHandle symbol,
IntBuffer out_size,
PointerPointer out_str_array) |
static int |
MXSymbolListArguments(SymbolHandle symbol,
IntPointer out_size,
PointerPointer out_str_array)
\brief List arguments in the symbol.
|
static int |
MXSymbolListAtomicSymbolCreators(int[] out_size,
PointerPointer out_array) |
static int |
MXSymbolListAtomicSymbolCreators(IntBuffer out_size,
PointerPointer out_array) |
static int |
MXSymbolListAtomicSymbolCreators(IntPointer out_size,
PointerPointer out_array)
\brief list all the available AtomicSymbolEntry
|
static int |
MXSymbolListAttr(SymbolHandle symbol,
int[] out_size,
PointerPointer out) |
static int |
MXSymbolListAttr(SymbolHandle symbol,
IntBuffer out_size,
PointerPointer out) |
static int |
MXSymbolListAttr(SymbolHandle symbol,
IntPointer out_size,
PointerPointer out)
\brief Get all attributes from symbol, including all descendents.
|
static int |
MXSymbolListAttrShallow(SymbolHandle symbol,
int[] out_size,
PointerPointer out) |
static int |
MXSymbolListAttrShallow(SymbolHandle symbol,
IntBuffer out_size,
PointerPointer out) |
static int |
MXSymbolListAttrShallow(SymbolHandle symbol,
IntPointer out_size,
PointerPointer out)
\brief Get all attributes from symbol, excluding descendents.
|
static int |
MXSymbolListAuxiliaryStates(SymbolHandle symbol,
int[] out_size,
PointerPointer out_str_array) |
static int |
MXSymbolListAuxiliaryStates(SymbolHandle symbol,
IntBuffer out_size,
PointerPointer out_str_array) |
static int |
MXSymbolListAuxiliaryStates(SymbolHandle symbol,
IntPointer out_size,
PointerPointer out_str_array)
\brief List auxiliary states in the symbol.
|
static int |
MXSymbolListOutputs(SymbolHandle symbol,
int[] out_size,
PointerPointer out_str_array) |
static int |
MXSymbolListOutputs(SymbolHandle symbol,
IntBuffer out_size,
PointerPointer out_str_array) |
static int |
MXSymbolListOutputs(SymbolHandle symbol,
IntPointer out_size,
PointerPointer out_str_array)
\brief List returns in the symbol.
|
static int |
MXSymbolPrint(SymbolHandle symbol,
byte[] out_str) |
static int |
MXSymbolPrint(SymbolHandle symbol,
ByteBuffer out_str) |
static int |
MXSymbolPrint(SymbolHandle symbol,
BytePointer out_str) |
static int |
MXSymbolPrint(SymbolHandle symbol,
PointerPointer out_str)
\brief Print the content of symbol, used for debug.
|
static int |
MXSymbolRemoveAmpCast(SymbolHandle sym_handle,
PointerPointer ret_sym_handle)
\brief Remove the operators amp_cast and amp_multicast
|
static int |
MXSymbolRemoveAmpCast(SymbolHandle sym_handle,
SymbolHandle ret_sym_handle) |
static int |
MXSymbolSaveToFile(SymbolHandle symbol,
BytePointer fname)
\brief Save a symbol into a json file.
|
static int |
MXSymbolSaveToFile(SymbolHandle symbol,
String fname) |
static int |
MXSymbolSaveToJSON(SymbolHandle symbol,
byte[] out_json) |
static int |
MXSymbolSaveToJSON(SymbolHandle symbol,
ByteBuffer out_json) |
static int |
MXSymbolSaveToJSON(SymbolHandle symbol,
BytePointer out_json) |
static int |
MXSymbolSaveToJSON(SymbolHandle symbol,
PointerPointer out_json)
\brief Save a symbol into a json string
|
static int |
MXSymbolSetAttr(SymbolHandle symbol,
BytePointer key,
BytePointer value)
\brief Set string attribute from symbol.
|
static int |
MXSymbolSetAttr(SymbolHandle symbol,
String key,
String value) |
static int |
NNAddControlDeps(SymbolHandle handle,
SymbolHandle src_dep)
\brief Add src_dep to the handle as control dep.
|
static void |
NNAPISetLastError(BytePointer msg)
\brief Set the last error message needed by C API
|
static void |
NNAPISetLastError(String msg) |
static BytePointer |
NNGetLastError()
\brief return str message of the last error
all function in this file will return 0 when success
and -1 when an error occurred,
NNGetLastError can be called to retrieve the error
this function is threadsafe and can be called by different thread
|
static int |
NNGetOpHandle(BytePointer op_name,
OpHandle op_out)
\brief Get operator handle given name.
|
static int |
NNGetOpHandle(String op_name,
OpHandle op_out) |
static int |
NNGetOpInfo(OpHandle op,
byte[] real_name,
byte[] description,
int[] num_doc_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
byte[] return_type) |
static int |
NNGetOpInfo(OpHandle op,
ByteBuffer real_name,
ByteBuffer description,
IntBuffer num_doc_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
ByteBuffer return_type) |
static int |
NNGetOpInfo(OpHandle op,
BytePointer real_name,
BytePointer description,
IntPointer num_doc_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
BytePointer return_type) |
static int |
NNGetOpInfo(OpHandle op,
PointerPointer real_name,
PointerPointer description,
IntPointer num_doc_args,
PointerPointer arg_names,
PointerPointer arg_type_infos,
PointerPointer arg_descriptions,
PointerPointer return_type)
\brief Get the detailed information about atomic symbol.
|
static int |
NNGraphApplyPasses(GraphHandle src,
int num_pass,
byte[] pass_names,
GraphHandle dst) |
static int |
NNGraphApplyPasses(GraphHandle src,
int num_pass,
ByteBuffer pass_names,
GraphHandle dst) |
static int |
NNGraphApplyPasses(GraphHandle src,
int num_pass,
BytePointer pass_names,
GraphHandle dst) |
static int |
NNGraphApplyPasses(GraphHandle src,
int num_pass,
PointerPointer pass_names,
GraphHandle dst)
\brief Apply passes on the src graph.
|
static int |
NNGraphCreate(SymbolHandle symbol,
GraphHandle graph)
\brief create a graph handle from symbol
|
static int |
NNGraphFree(GraphHandle handle)
\brief free the graph handle
|
static int |
NNGraphGetJSONAttr(GraphHandle handle,
BytePointer key,
byte[] json_out,
int[] success) |
static int |
NNGraphGetJSONAttr(GraphHandle handle,
BytePointer key,
ByteBuffer json_out,
IntBuffer success) |
static int |
NNGraphGetJSONAttr(GraphHandle handle,
BytePointer key,
BytePointer json_out,
IntPointer success) |
static int |
NNGraphGetJSONAttr(GraphHandle handle,
BytePointer key,
PointerPointer json_out,
IntPointer success)
\brief Get a serialized attrirbute from graph.
|
static int |
NNGraphGetJSONAttr(GraphHandle handle,
String key,
byte[] json_out,
int[] success) |
static int |
NNGraphGetJSONAttr(GraphHandle handle,
String key,
ByteBuffer json_out,
IntBuffer success) |
static int |
NNGraphGetJSONAttr(GraphHandle handle,
String key,
BytePointer json_out,
IntPointer success) |
static int |
NNGraphGetSymbol(GraphHandle graph,
PointerPointer symbol)
\brief Get a new symbol from the graph.
|
static int |
NNGraphGetSymbol(GraphHandle graph,
SymbolHandle symbol) |
static int |
NNGraphSetJSONAttr(GraphHandle handle,
BytePointer key,
BytePointer json_value)
\brief Get Set a attribute in json format.
|
static int |
NNGraphSetJSONAttr(GraphHandle handle,
String key,
String json_value) |
static int |
NNGraphSetNodeEntryListAttr_(GraphHandle handle,
BytePointer key,
SymbolHandle list)
\brief Set a attribute whose type is std::vector
|
static int |
NNGraphSetNodeEntryListAttr_(GraphHandle handle,
String key,
SymbolHandle list) |
static int |
NNListAllOpNames(int[] out_size,
PointerPointer out_array) |
static int |
NNListAllOpNames(IntBuffer out_size,
PointerPointer out_array) |
static int |
NNListAllOpNames(IntPointer out_size,
PointerPointer out_array)
\brief list all the available operator names, include entries.
|
static int |
NNListUniqueOps(int[] out_size,
PointerPointer out_array) |
static int |
NNListUniqueOps(IntBuffer out_size,
PointerPointer out_array) |
static int |
NNListUniqueOps(IntPointer out_size,
PointerPointer out_array)
\brief list all the available operators.
|
static int |
NNSymbolCompose(SymbolHandle sym,
BytePointer name,
int num_args,
byte[] keys,
PointerPointer args) |
static int |
NNSymbolCompose(SymbolHandle sym,
BytePointer name,
int num_args,
ByteBuffer keys,
PointerPointer args) |
static int |
NNSymbolCompose(SymbolHandle sym,
BytePointer name,
int num_args,
BytePointer keys,
PointerPointer args) |
static int |
NNSymbolCompose(SymbolHandle sym,
BytePointer name,
int num_args,
PointerPointer keys,
PointerPointer args)
\brief Compose the symbol on other symbols.
|
static int |
NNSymbolCompose(SymbolHandle sym,
String name,
int num_args,
byte[] keys,
SymbolHandle args) |
static int |
NNSymbolCompose(SymbolHandle sym,
String name,
int num_args,
ByteBuffer keys,
SymbolHandle args) |
static int |
NNSymbolCompose(SymbolHandle sym,
String name,
int num_args,
BytePointer keys,
SymbolHandle args) |
static int |
NNSymbolCopy(SymbolHandle symbol,
PointerPointer out)
\brief Copy the symbol to another handle
|
static int |
NNSymbolCopy(SymbolHandle symbol,
SymbolHandle out) |
static int |
NNSymbolCreateAtomicSymbol(OpHandle op,
int num_param,
byte[] keys,
byte[] vals,
PointerPointer out) |
static int |
NNSymbolCreateAtomicSymbol(OpHandle op,
int num_param,
byte[] keys,
byte[] vals,
SymbolHandle out) |
static int |
NNSymbolCreateAtomicSymbol(OpHandle op,
int num_param,
ByteBuffer keys,
ByteBuffer vals,
PointerPointer out) |
static int |
NNSymbolCreateAtomicSymbol(OpHandle op,
int num_param,
ByteBuffer keys,
ByteBuffer vals,
SymbolHandle out) |
static int |
NNSymbolCreateAtomicSymbol(OpHandle op,
int num_param,
BytePointer keys,
BytePointer vals,
PointerPointer out) |
static int |
NNSymbolCreateAtomicSymbol(OpHandle op,
int num_param,
BytePointer keys,
BytePointer vals,
SymbolHandle out) |
static int |
NNSymbolCreateAtomicSymbol(OpHandle op,
int num_param,
PointerPointer keys,
PointerPointer vals,
PointerPointer out)
\brief Create an AtomicSymbol functor.
|
static int |
NNSymbolCreateGroup(int num_symbols,
PointerPointer symbols,
PointerPointer out)
\brief Create a Symbol by grouping list of symbols together
|
static int |
NNSymbolCreateGroup(int num_symbols,
SymbolHandle symbols,
SymbolHandle out) |
static int |
NNSymbolCreateVariable(BytePointer name,
PointerPointer out)
\brief Create a Variable Symbol.
|
static int |
NNSymbolCreateVariable(String name,
SymbolHandle out) |
static int |
NNSymbolFree(SymbolHandle symbol)
\brief Free the symbol handle.
|
static int |
NNSymbolGetAttr(SymbolHandle symbol,
BytePointer key,
byte[] out,
int[] success) |
static int |
NNSymbolGetAttr(SymbolHandle symbol,
BytePointer key,
ByteBuffer out,
IntBuffer success) |
static int |
NNSymbolGetAttr(SymbolHandle symbol,
BytePointer key,
BytePointer out,
IntPointer success) |
static int |
NNSymbolGetAttr(SymbolHandle symbol,
BytePointer key,
PointerPointer out,
IntPointer success)
\brief Get string attribute from symbol
|
static int |
NNSymbolGetAttr(SymbolHandle symbol,
String key,
byte[] out,
int[] success) |
static int |
NNSymbolGetAttr(SymbolHandle symbol,
String key,
ByteBuffer out,
IntBuffer success) |
static int |
NNSymbolGetAttr(SymbolHandle symbol,
String key,
BytePointer out,
IntPointer success) |
static int |
NNSymbolGetChildren(SymbolHandle symbol,
PointerPointer out)
\brief Get a symbol that contains only direct children.
|
static int |
NNSymbolGetChildren(SymbolHandle symbol,
SymbolHandle out) |
static int |
NNSymbolGetInternals(SymbolHandle symbol,
PointerPointer out)
\brief Get a symbol that contains all the internals.
|
static int |
NNSymbolGetInternals(SymbolHandle symbol,
SymbolHandle out) |
static int |
NNSymbolGetNumOutputs(SymbolHandle symbol,
int[] output_count) |
static int |
NNSymbolGetNumOutputs(SymbolHandle symbol,
IntBuffer output_count) |
static int |
NNSymbolGetNumOutputs(SymbolHandle symbol,
IntPointer output_count)
\brief Supply number of outputs of the symbol.
|
static int |
NNSymbolGetOutput(SymbolHandle symbol,
int index,
PointerPointer out)
\brief Get index-th outputs of the symbol.
|
static int |
NNSymbolGetOutput(SymbolHandle symbol,
int index,
SymbolHandle out) |
static int |
NNSymbolListAttrs(SymbolHandle symbol,
int recursive_option,
int[] out_size,
PointerPointer out) |
static int |
NNSymbolListAttrs(SymbolHandle symbol,
int recursive_option,
IntBuffer out_size,
PointerPointer out) |
static int |
NNSymbolListAttrs(SymbolHandle symbol,
int recursive_option,
IntPointer out_size,
PointerPointer out)
\brief Get all attributes from symbol, including all descendents.
|
static int |
NNSymbolListInputNames(SymbolHandle symbol,
int option,
int[] out_size,
PointerPointer out_str_array) |
static int |
NNSymbolListInputNames(SymbolHandle symbol,
int option,
IntBuffer out_size,
PointerPointer out_str_array) |
static int |
NNSymbolListInputNames(SymbolHandle symbol,
int option,
IntPointer out_size,
PointerPointer out_str_array)
\brief List input names in the symbol.
|
static int |
NNSymbolListInputVariables(SymbolHandle symbol,
int option,
int[] out_size,
PointerPointer out_sym_array) |
static int |
NNSymbolListInputVariables(SymbolHandle symbol,
int option,
IntBuffer out_size,
PointerPointer out_sym_array) |
static int |
NNSymbolListInputVariables(SymbolHandle symbol,
int option,
IntPointer out_size,
PointerPointer out_sym_array)
\brief List inputs variables in the symbol.
|
static int |
NNSymbolListOutputNames(SymbolHandle symbol,
int[] out_size,
PointerPointer out_str_array) |
static int |
NNSymbolListOutputNames(SymbolHandle symbol,
IntBuffer out_size,
PointerPointer out_str_array) |
static int |
NNSymbolListOutputNames(SymbolHandle symbol,
IntPointer out_size,
PointerPointer out_str_array)
\brief List returns names in the symbol.
|
static int |
NNSymbolPrint(SymbolHandle symbol,
byte[] out_str) |
static int |
NNSymbolPrint(SymbolHandle symbol,
ByteBuffer out_str) |
static int |
NNSymbolPrint(SymbolHandle symbol,
BytePointer out_str) |
static int |
NNSymbolPrint(SymbolHandle symbol,
PointerPointer out_str)
\brief Print the content of symbol, used for debug.
|
static int |
NNSymbolSetAttrs(SymbolHandle symbol,
int num_param,
byte[] keys,
byte[] values) |
static int |
NNSymbolSetAttrs(SymbolHandle symbol,
int num_param,
ByteBuffer keys,
ByteBuffer values) |
static int |
NNSymbolSetAttrs(SymbolHandle symbol,
int num_param,
BytePointer keys,
BytePointer values) |
static int |
NNSymbolSetAttrs(SymbolHandle symbol,
int num_param,
PointerPointer keys,
PointerPointer values)
\brief Set string attribute from symbol.
|
public static final int kCustomOpDelete
public static final int kCustomOpForward
public static final int kCustomOpBackward
public static final int kCustomOpPropDelete
public static final int kCustomOpPropListArguments
public static final int kCustomOpPropListOutputs
public static final int kCustomOpPropListAuxiliaryStates
public static final int kCustomOpPropInferShape
public static final int kCustomOpPropDeclareBackwardDependency
public static final int kCustomOpPropCreateOperator
public static final int kCustomOpPropInferType
public static final int kCustomOpPropInferStorageType
public static final int kCustomOpPropBackwardInferStorageType
public static final int kCustomFunctionBackward
public static final int kCustomFunctionDelete
@Cast(value="const char*") public static BytePointer MXGetLastError()
public static int MXLoadLib(@Cast(value="const char*") BytePointer path, @Cast(value="unsigned") int verbose)
path
- to the library .so fileverbose
- 0 for quiet, 1 for verbosepublic static int MXLibInfoFeatures(@Cast(value="const LibFeature**") PointerPointer libFeature, @Cast(value="size_t*") SizeTPointer size)
libFeature
- pointer to array of LibFeaturesize
- of the arraypublic static int MXLibInfoFeatures(@Const @ByPtrPtr LibFeature libFeature, @Cast(value="size_t*") SizeTPointer size)
public static int MXRandomSeed(int seed)
seed
- the random number seed.public static int MXRandomSeedContext(int seed, int dev_type, int dev_id)
seed
- the random number seed.public static int MXNotifyShutdown()
public static int MXSetProcessProfilerConfig(int num_params, @Cast(value="const char*const*") PointerPointer keys, @Cast(value="const char*const*") PointerPointer vals, KVStoreHandle kvstoreHandle)
num_params
- Number of parameterskeys
- array of parameter keysvals
- array of parameter valueskvstoreHandle
- handle to kvstorepublic static int MXSetProcessProfilerConfig(int num_params, @Cast(value="const char*const*") @ByPtrPtr BytePointer keys, @Cast(value="const char*const*") @ByPtrPtr BytePointer vals, KVStoreHandle kvstoreHandle)
public static int MXSetProcessProfilerConfig(int num_params, @Cast(value="const char*const*") @ByPtrPtr ByteBuffer keys, @Cast(value="const char*const*") @ByPtrPtr ByteBuffer vals, KVStoreHandle kvstoreHandle)
public static int MXSetProcessProfilerConfig(int num_params, @Cast(value="const char*const*") @ByPtrPtr byte[] keys, @Cast(value="const char*const*") @ByPtrPtr byte[] vals, KVStoreHandle kvstoreHandle)
public static int MXSetProfilerConfig(int num_params, @Cast(value="const char*const*") PointerPointer keys, @Cast(value="const char*const*") PointerPointer vals)
num_params
- Number of parameterskeys
- array of parameter keysvals
- array of parameter valuespublic static int MXSetProfilerConfig(int num_params, @Cast(value="const char*const*") @ByPtrPtr BytePointer keys, @Cast(value="const char*const*") @ByPtrPtr BytePointer vals)
public static int MXSetProfilerConfig(int num_params, @Cast(value="const char*const*") @ByPtrPtr ByteBuffer keys, @Cast(value="const char*const*") @ByPtrPtr ByteBuffer vals)
public static int MXSetProfilerConfig(int num_params, @Cast(value="const char*const*") @ByPtrPtr byte[] keys, @Cast(value="const char*const*") @ByPtrPtr byte[] vals)
public static int MXSetProcessProfilerState(int state, int profile_process, KVStoreHandle kvStoreHandle)
state
- indicate the working state of profiler,
profiler not running when state == 0,
profiler running when state == 1profile_process
- an int,
when 0 command is for worker/current process,
when 1 command is for server processkvstoreHandle
- handle to kvstore, needed for server process profilingpublic static int MXSetProfilerState(int state)
state
- indicate the working state of profiler,
profiler not running when state == 0,
profiler running when state == 1public static int MXDumpProcessProfile(int finished, int profile_process, KVStoreHandle kvStoreHandle)
finished
- true if stat output should stop after this pointprofile_process
- an int,
when 0 command is for worker/current process,
when 1 command is for server processkvstoreHandle
- handle to kvstorepublic static int MXDumpProfile(int finished)
finished
- true if stat output should stop after this pointpublic static int MXAggregateProfileStatsPrint(@Cast(value="const char**") PointerPointer out_str, int reset)
out_str
- Will receive a pointer to the output stringreset
- Clear the aggregate stats after printingpublic static int MXAggregateProfileStatsPrint(@Cast(value="const char**") @ByPtrPtr BytePointer out_str, int reset)
public static int MXAggregateProfileStatsPrint(@Cast(value="const char**") @ByPtrPtr ByteBuffer out_str, int reset)
public static int MXAggregateProfileStatsPrint(@Cast(value="const char**") @ByPtrPtr byte[] out_str, int reset)
public static int MXAggregateProfileStatsPrintEx(@Cast(value="const char**") PointerPointer out_str, int reset, int format, int sort_by, int ascending)
out_str
- will receive a pointer to the output stringreset
- clear the aggregate stats after printingformat
- whether to return in tabular or json formatsort_by
- sort by total, avg, min, max, or countascending
- whether to sort ascendinglypublic static int MXAggregateProfileStatsPrintEx(@Cast(value="const char**") @ByPtrPtr BytePointer out_str, int reset, int format, int sort_by, int ascending)
public static int MXAggregateProfileStatsPrintEx(@Cast(value="const char**") @ByPtrPtr ByteBuffer out_str, int reset, int format, int sort_by, int ascending)
public static int MXAggregateProfileStatsPrintEx(@Cast(value="const char**") @ByPtrPtr byte[] out_str, int reset, int format, int sort_by, int ascending)
public static int MXProcessProfilePause(int paused, int profile_process, KVStoreHandle kvStoreHandle)
paused
- If nonzero, profiling pauses. Otherwise, profiling resumes/continuesprofile_process
- integer which denotes whether to process worker or server processkvstoreHandle
- handle to kvstorepublic static int MXProfilePause(int paused)
paused
- If nonzero, profiling pauses. Otherwise, profiling resumes/continuespublic static int MXProfileCreateDomain(@Cast(value="const char*") BytePointer domain, @ByPtrPtr ProfileHandle out)
domain
- String representing the domain name to createout
- Return domain objectpublic static int MXProfileCreateDomain(String domain, @ByPtrPtr ProfileHandle out)
public static int MXProfileCreateTask(ProfileHandle domain, @Cast(value="const char*") BytePointer task_name, @ByPtrPtr ProfileHandle out)
name
- Name of the taskdomain
- Domain of the taskout
- Output handlepublic static int MXProfileCreateTask(ProfileHandle domain, String task_name, @ByPtrPtr ProfileHandle out)
public static int MXProfileCreateFrame(ProfileHandle domain, @Cast(value="const char*") BytePointer frame_name, @ByPtrPtr ProfileHandle out)
name
- Name of the framedomain
- Domain of the frameout
- Output handlepublic static int MXProfileCreateFrame(ProfileHandle domain, String frame_name, @ByPtrPtr ProfileHandle out)
public static int MXProfileCreateEvent(@Cast(value="const char*") BytePointer event_name, @ByPtrPtr ProfileHandle out)
name
- Name of the eventout
- Output handlepublic static int MXProfileCreateEvent(String event_name, @ByPtrPtr ProfileHandle out)
public static int MXProfileCreateCounter(ProfileHandle domain, @Cast(value="const char*") BytePointer counter_name, @ByPtrPtr ProfileHandle out)
name
- Name of the counterdomain
- Domain of the counterout
- Output handlepublic static int MXProfileCreateCounter(ProfileHandle domain, String counter_name, @ByPtrPtr ProfileHandle out)
public static int MXProfileDestroyHandle(ProfileHandle frame_handle)
frame_handle
- Handle to frame to destroypublic static int MXProfileDurationStart(ProfileHandle duration_handle)
duration_handle
- handle to the duration objectpublic static int MXProfileDurationStop(ProfileHandle duration_handle)
duration_handle
- handle to the duration objectpublic static int MXProfileSetCounter(ProfileHandle counter_handle, @Cast(value="uint64_t") long value)
counter_handle
- Handle to counter to setvalue
- Value to set the counter to (64-bit unsigned integer)public static int MXProfileAdjustCounter(ProfileHandle counter_handle, @Cast(value="int64_t") long value)
counter_handle
- Handle to counter to adjustvalue
- Value to adjust the counter by (64-bit signed integer)public static int MXProfileSetMarker(ProfileHandle domain, @Cast(value="const char*") BytePointer instant_marker_name, @Cast(value="const char*") BytePointer scope)
domain
- Domain of the markerinstant_marker_name
- Name of the markerscope
- Scope of marker ('global', 'process', 'thread', 'task', 'marker')public static int MXProfileSetMarker(ProfileHandle domain, String instant_marker_name, String scope)
public static int MXSetNumOMPThreads(int thread_num)
thread_num
- Number of OMP threads desiredpublic static int MXEngineSetBulkSize(int bulk_size, IntPointer prev_bulk_size)
bulk_size
- new bulk_sizeprev_bulk_size
- previous bulk_sizepublic static int MXEngineSetBulkSize(int bulk_size, IntBuffer prev_bulk_size)
public static int MXEngineSetBulkSize(int bulk_size, int[] prev_bulk_size)
public static int MXGetGPUCount(IntPointer out)
pointer
- to int that will hold the number of GPUs available.public static int MXGetGPUCount(IntBuffer out)
public static int MXGetGPUCount(int[] out)
public static int MXGetGPUMemoryInformation(int dev, IntPointer free_mem, IntPointer total_mem)
dev
- the GPU number to queryfree_mem
- pointer to the integer holding free GPU memorytotal_mem
- pointer to the integer holding total GPU memorypublic static int MXGetGPUMemoryInformation(int dev, IntBuffer free_mem, IntBuffer total_mem)
public static int MXGetGPUMemoryInformation(int dev, int[] free_mem, int[] total_mem)
public static int MXGetGPUMemoryInformation64(int dev, @Cast(value="uint64_t*") LongPointer free_mem, @Cast(value="uint64_t*") LongPointer total_mem)
dev
- the GPU number to queryfree_mem
- pointer to the uint64_t holding free GPU memorytotal_mem
- pointer to the uint64_t holding total GPU memorypublic static int MXGetGPUMemoryInformation64(int dev, @Cast(value="uint64_t*") LongBuffer free_mem, @Cast(value="uint64_t*") LongBuffer total_mem)
public static int MXGetGPUMemoryInformation64(int dev, @Cast(value="uint64_t*") long[] free_mem, @Cast(value="uint64_t*") long[] total_mem)
public static int MXGetVersion(IntPointer out)
pointer
- to the integer holding the version numberpublic static int MXGetVersion(IntBuffer out)
public static int MXGetVersion(int[] out)
public static int MXNDArrayCreateNone(PointerPointer out)
out
- the returning handlepublic static int MXNDArrayCreateNone(@Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreate(@Cast(value="const uint32_t*") IntPointer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, PointerPointer out)
shape
- the pointer to the shapendim
- the dimension of the shapedev_type
- device type, specify device we want to takedev_id
- the device id of the specific devicedelay_alloc
- whether to delay allocation until
the narray is first mutatedout
- the returning handlepublic static int MXNDArrayCreate(@Cast(value="const uint32_t*") IntBuffer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreate(@Cast(value="const uint32_t*") int[] shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, PointerPointer out)
public static int MXNDArrayCreate(@Cast(value="const uint32_t*") IntPointer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreate(@Cast(value="const uint32_t*") IntBuffer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, PointerPointer out)
public static int MXNDArrayCreate(@Cast(value="const uint32_t*") int[] shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateEx(@Cast(value="const uint32_t*") IntPointer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, PointerPointer out)
shape
- the pointer to the shapendim
- the dimension of the shapedev_type
- device type, specify device we want to takedev_id
- the device id of the specific devicedelay_alloc
- whether to delay allocation until
the narray is first mutateddtype
- data type of created arrayout
- the returning handlepublic static int MXNDArrayCreateEx(@Cast(value="const uint32_t*") IntBuffer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateEx(@Cast(value="const uint32_t*") int[] shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, PointerPointer out)
public static int MXNDArrayCreateEx(@Cast(value="const uint32_t*") IntPointer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateEx(@Cast(value="const uint32_t*") IntBuffer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, PointerPointer out)
public static int MXNDArrayCreateEx(@Cast(value="const uint32_t*") int[] shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateEx64(@Cast(value="const int64_t*") LongPointer shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, PointerPointer out)
shape
- the pointer to int64_t shapendim
- the dimension of the shapedev_type
- device type, specify device we want to takedev_id
- the device id of the specific devicedelay_alloc
- whether to delay allocation until
the narray is first mutateddtype
- data type of created arrayout
- the returning handlepublic static int MXNDArrayCreateEx64(@Cast(value="const int64_t*") LongBuffer shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateEx64(@Cast(value="const int64_t*") long[] shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, PointerPointer out)
public static int MXNDArrayCreateEx64(@Cast(value="const int64_t*") LongPointer shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateEx64(@Cast(value="const int64_t*") LongBuffer shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, PointerPointer out)
public static int MXNDArrayCreateEx64(@Cast(value="const int64_t*") long[] shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateSparseEx(int storage_type, @Cast(value="const uint32_t*") IntPointer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="uint32_t") int num_aux, IntPointer aux_type, @Cast(value="uint32_t*") IntPointer aux_ndims, @Cast(value="const uint32_t*") IntPointer aux_shape, PointerPointer out)
storage_type
- the storage type of the ndarrayshape
- the pointer to the shapendim
- the dimension of the shapedev_type
- device type, specify device we want to takedev_id
- the device id of the specific devicedelay_alloc
- whether to delay allocation until
the narray is first mutateddtype
- data type of created arraynum_aux
- the number of aux data to support this ndarrayaux_type
- data type of the aux data for the created arrayaux_ndims
- the dimension of the shapes of aux dataaux_shape
- the shapes of aux dataout
- the returning handlepublic static int MXNDArrayCreateSparseEx(int storage_type, @Cast(value="const uint32_t*") IntBuffer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="uint32_t") int num_aux, IntBuffer aux_type, @Cast(value="uint32_t*") IntBuffer aux_ndims, @Cast(value="const uint32_t*") IntBuffer aux_shape, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateSparseEx(int storage_type, @Cast(value="const uint32_t*") int[] shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="uint32_t") int num_aux, int[] aux_type, @Cast(value="uint32_t*") int[] aux_ndims, @Cast(value="const uint32_t*") int[] aux_shape, PointerPointer out)
public static int MXNDArrayCreateSparseEx(int storage_type, @Cast(value="const uint32_t*") IntPointer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="uint32_t") int num_aux, IntPointer aux_type, @Cast(value="uint32_t*") IntPointer aux_ndims, @Cast(value="const uint32_t*") IntPointer aux_shape, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateSparseEx(int storage_type, @Cast(value="const uint32_t*") IntBuffer shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="uint32_t") int num_aux, IntBuffer aux_type, @Cast(value="uint32_t*") IntBuffer aux_ndims, @Cast(value="const uint32_t*") IntBuffer aux_shape, PointerPointer out)
public static int MXNDArrayCreateSparseEx(int storage_type, @Cast(value="const uint32_t*") int[] shape, @Cast(value="uint32_t") int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, @Cast(value="uint32_t") int num_aux, int[] aux_type, @Cast(value="uint32_t*") int[] aux_ndims, @Cast(value="const uint32_t*") int[] aux_shape, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayLoadFromRawBytes(@Const Pointer buf, @Cast(value="size_t") long size, PointerPointer out)
buf
- the head of the raw bytessize
- size of the raw bytesout
- the returning handlepublic static int MXNDArrayLoadFromRawBytes(@Const Pointer buf, @Cast(value="size_t") long size, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArraySaveRawBytes(NDArrayHandle handle, @Cast(value="size_t*") SizeTPointer out_size, @Cast(value="const char**") PointerPointer out_buf)
handle
- the NDArray handleout_size
- size of the raw bytesout_buf
- the head of returning memory bytes.public static int MXNDArraySaveRawBytes(NDArrayHandle handle, @Cast(value="size_t*") SizeTPointer out_size, @Cast(value="const char**") @ByPtrPtr BytePointer out_buf)
public static int MXNDArraySaveRawBytes(NDArrayHandle handle, @Cast(value="size_t*") SizeTPointer out_size, @Cast(value="const char**") @ByPtrPtr ByteBuffer out_buf)
public static int MXNDArraySaveRawBytes(NDArrayHandle handle, @Cast(value="size_t*") SizeTPointer out_size, @Cast(value="const char**") @ByPtrPtr byte[] out_buf)
public static int MXNDArraySave(@Cast(value="const char*") BytePointer fname, @Cast(value="uint32_t") int num_args, PointerPointer args, @Cast(value="const char**") PointerPointer keys)
fname
- name of the file.num_args
- number of arguments to save.args
- the array of NDArrayHandles to be saved.keys
- the name of the NDArray, optional, can be NULLpublic static int MXNDArraySave(@Cast(value="const char*") BytePointer fname, @Cast(value="uint32_t") int num_args, PointerPointer args, @Cast(value="const char**") @ByPtrPtr BytePointer keys)
public static int MXNDArraySave(String fname, @Cast(value="uint32_t") int num_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys)
public static int MXNDArraySave(@Cast(value="const char*") BytePointer fname, @Cast(value="uint32_t") int num_args, PointerPointer args, @Cast(value="const char**") @ByPtrPtr byte[] keys)
public static int MXNDArraySave(String fname, @Cast(value="uint32_t") int num_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle args, @Cast(value="const char**") @ByPtrPtr BytePointer keys)
public static int MXNDArraySave(@Cast(value="const char*") BytePointer fname, @Cast(value="uint32_t") int num_args, PointerPointer args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys)
public static int MXNDArraySave(String fname, @Cast(value="uint32_t") int num_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle args, @Cast(value="const char**") @ByPtrPtr byte[] keys)
public static int MXNDArrayLoad(@Cast(value="const char*") BytePointer fname, @Cast(value="uint32_t*") IntPointer out_size, @ByPtrPtr PointerPointer out_arr, @Cast(value="uint32_t*") IntPointer out_name_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_names)
fname
- name of the file.out_size
- number of narray loaded.out_arr
- head of the returning narray handles.out_name_size
- size of output name arrray.out_names
- the names of returning NDArrays, can be NULLpublic static int MXNDArrayLoad(String fname, @Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer out_arr, @Cast(value="uint32_t*") IntBuffer out_name_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_names)
public static int MXNDArrayLoad(@Cast(value="const char*") BytePointer fname, @Cast(value="uint32_t*") int[] out_size, @ByPtrPtr PointerPointer out_arr, @Cast(value="uint32_t*") int[] out_name_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_names)
public static int MXNDArrayLoad(String fname, @Cast(value="uint32_t*") IntPointer out_size, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer out_arr, @Cast(value="uint32_t*") IntPointer out_name_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_names)
public static int MXNDArrayLoad(@Cast(value="const char*") BytePointer fname, @Cast(value="uint32_t*") IntBuffer out_size, @ByPtrPtr PointerPointer out_arr, @Cast(value="uint32_t*") IntBuffer out_name_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_names)
public static int MXNDArrayLoad(String fname, @Cast(value="uint32_t*") int[] out_size, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer out_arr, @Cast(value="uint32_t*") int[] out_name_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_names)
public static int MXNDArrayLoadFromBuffer(@Const Pointer ndarray_buffer, @Cast(value="size_t") long size, @Cast(value="uint32_t*") IntPointer out_size, @ByPtrPtr PointerPointer out_arr, @Cast(value="uint32_t*") IntPointer out_name_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_names)
ndarray_buffer
- pointer to the start of the ndarray file contentsize
- size of the fileout_size
- number of narray loaded.out_arr
- head of the returning narray handles.out_name_size
- size of output name arrray.out_names
- the names of returning NDArrays, can be NULLpublic static int MXNDArrayLoadFromBuffer(@Const Pointer ndarray_buffer, @Cast(value="size_t") long size, @Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer out_arr, @Cast(value="uint32_t*") IntBuffer out_name_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_names)
public static int MXNDArrayLoadFromBuffer(@Const Pointer ndarray_buffer, @Cast(value="size_t") long size, @Cast(value="uint32_t*") int[] out_size, @ByPtrPtr PointerPointer out_arr, @Cast(value="uint32_t*") int[] out_name_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_names)
public static int MXNDArraySyncCopyFromCPU(NDArrayHandle handle, @Const Pointer data, @Cast(value="size_t") long size)
handle
- the NDArray handledata
- the data source to copy from.size
- the memory size we want to copy from.public static int MXNDArraySyncCopyToCPU(NDArrayHandle handle, Pointer data, @Cast(value="size_t") long size)
handle
- the NDArray handledata
- the data source to copy into.size
- the memory size we want to copy into.public static int MXNDArraySyncCopyFromNDArray(NDArrayHandle handle_dst, NDArrayHandle handle_src, int i)
handle_dst
- handle of a dst ndarray whose data/aux_data has been allocatedhandle_src
- handle of a src ndarray which has default storage typei
- dst data blob indicatorpublic static int MXNDArraySyncCheckFormat(NDArrayHandle handle, @Cast(value="const bool") boolean full_check)
full_check
- if True
, rigorous check, O(N) operations
Otherwise basic check, O(1) operationspublic static int MXNDArrayWaitToRead(NDArrayHandle handle)
handle
- the NDArray handlepublic static int MXNDArrayWaitToWrite(NDArrayHandle handle)
handle
- the NDArray handlepublic static int MXNDArrayWaitAll()
public static int MXNDArrayFree(NDArrayHandle handle)
handle
- the handle to be freedpublic static int MXNDArraySlice(NDArrayHandle handle, @Cast(value="uint32_t") int slice_begin, @Cast(value="uint32_t") int slice_end, PointerPointer out)
handle
- the handle to the NDArrayslice_begin
- The beginning index of sliceslice_end
- The ending index of sliceout
- The NDArrayHandle of sliced NDArraypublic static int MXNDArraySlice(NDArrayHandle handle, @Cast(value="uint32_t") int slice_begin, @Cast(value="uint32_t") int slice_end, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArraySlice64(NDArrayHandle handle, @Cast(value="int64_t") long slice_begin, @Cast(value="int64_t") long slice_end, PointerPointer out)
handle
- the handle to the NDArrayslice_begin
- The beginning index of sliceslice_end
- The ending index of sliceout
- The NDArrayHandle of sliced NDArraypublic static int MXNDArraySlice64(NDArrayHandle handle, @Cast(value="int64_t") long slice_begin, @Cast(value="int64_t") long slice_end, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayAt(NDArrayHandle handle, @Cast(value="uint32_t") int idx, PointerPointer out)
handle
- the handle to the NDArrayidx
- the indexout
- The NDArrayHandle of output NDArraypublic static int MXNDArrayAt(NDArrayHandle handle, @Cast(value="uint32_t") int idx, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayAt64(NDArrayHandle handle, @Cast(value="int64_t") long idx, PointerPointer out)
handle
- the handle to the NDArrayidx
- the indexout
- The NDArrayHandle of output NDArraypublic static int MXNDArrayAt64(NDArrayHandle handle, @Cast(value="int64_t") long idx, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayGetStorageType(NDArrayHandle handle, IntPointer out_storage_type)
public static int MXNDArrayGetStorageType(NDArrayHandle handle, IntBuffer out_storage_type)
public static int MXNDArrayGetStorageType(NDArrayHandle handle, int[] out_storage_type)
public static int MXNDArrayReshape(NDArrayHandle handle, int ndim, IntPointer dims, PointerPointer out)
handle
- the handle to the narrayndim
- number of dimensions of new shapedims
- new shapeout
- the NDArrayHandle of reshaped NDArraypublic static int MXNDArrayReshape(NDArrayHandle handle, int ndim, IntBuffer dims, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayReshape(NDArrayHandle handle, int ndim, int[] dims, PointerPointer out)
public static int MXNDArrayReshape(NDArrayHandle handle, int ndim, IntPointer dims, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayReshape(NDArrayHandle handle, int ndim, IntBuffer dims, PointerPointer out)
public static int MXNDArrayReshape(NDArrayHandle handle, int ndim, int[] dims, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayReshape64(NDArrayHandle handle, int ndim, @Cast(value="dim_t*") LongPointer dims, @Cast(value="bool") boolean reverse, PointerPointer out)
handle
- the handle to the narrayndim
- number of dimensions of new shapedims
- new shapeout
- the NDArrayHandle of reshaped NDArraypublic static int MXNDArrayReshape64(NDArrayHandle handle, int ndim, @Cast(value="dim_t*") LongBuffer dims, @Cast(value="bool") boolean reverse, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayReshape64(NDArrayHandle handle, int ndim, @Cast(value="dim_t*") long[] dims, @Cast(value="bool") boolean reverse, PointerPointer out)
public static int MXNDArrayReshape64(NDArrayHandle handle, int ndim, @Cast(value="dim_t*") LongPointer dims, @Cast(value="bool") boolean reverse, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayReshape64(NDArrayHandle handle, int ndim, @Cast(value="dim_t*") LongBuffer dims, @Cast(value="bool") boolean reverse, PointerPointer out)
public static int MXNDArrayReshape64(NDArrayHandle handle, int ndim, @Cast(value="dim_t*") long[] dims, @Cast(value="bool") boolean reverse, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayGetShape(NDArrayHandle handle, @Cast(value="uint32_t*") IntPointer out_dim, @Cast(value="const uint32_t**") PointerPointer out_pdata)
handle
- the handle to the narrayout_dim
- the output dimensionout_pdata
- pointer holder to get data pointer of the shapepublic static int MXNDArrayGetShape(NDArrayHandle handle, @Cast(value="uint32_t*") IntPointer out_dim, @Cast(value="const uint32_t**") @ByPtrPtr IntPointer out_pdata)
public static int MXNDArrayGetShape(NDArrayHandle handle, @Cast(value="uint32_t*") IntBuffer out_dim, @Cast(value="const uint32_t**") @ByPtrPtr IntBuffer out_pdata)
public static int MXNDArrayGetShape(NDArrayHandle handle, @Cast(value="uint32_t*") int[] out_dim, @Cast(value="const uint32_t**") @ByPtrPtr int[] out_pdata)
public static int MXNDArrayGetShapeEx(NDArrayHandle handle, IntPointer out_dim, @Cast(value="const int**") PointerPointer out_pdata)
handle
- the handle to the narrayout_dim
- the output dimensionout_pdata
- pointer holder to get data pointer of the shapepublic static int MXNDArrayGetShapeEx(NDArrayHandle handle, IntPointer out_dim, @Const @ByPtrPtr IntPointer out_pdata)
public static int MXNDArrayGetShapeEx(NDArrayHandle handle, IntBuffer out_dim, @Const @ByPtrPtr IntBuffer out_pdata)
public static int MXNDArrayGetShapeEx(NDArrayHandle handle, int[] out_dim, @Const @ByPtrPtr int[] out_pdata)
public static int MXNDArrayGetShapeEx64(NDArrayHandle handle, IntPointer out_dim, @Cast(value="const int64_t**") PointerPointer out_pdata)
handle
- the handle to the narrayout_dim
- the output dimensionout_pdata
- pointer holder to get data pointer of the shapepublic static int MXNDArrayGetShapeEx64(NDArrayHandle handle, IntPointer out_dim, @Cast(value="const int64_t**") @ByPtrPtr LongPointer out_pdata)
public static int MXNDArrayGetShapeEx64(NDArrayHandle handle, IntBuffer out_dim, @Cast(value="const int64_t**") @ByPtrPtr LongBuffer out_pdata)
public static int MXNDArrayGetShapeEx64(NDArrayHandle handle, int[] out_dim, @Cast(value="const int64_t**") @ByPtrPtr long[] out_pdata)
public static int MXNDArrayGetData(NDArrayHandle handle, @Cast(value="void**") PointerPointer out_pdata)
handle
- the handle to the ndarrayout_pdata
- pointer holder to get pointer of datapublic static int MXNDArrayGetData(NDArrayHandle handle, @Cast(value="void**") @ByPtrPtr Pointer out_pdata)
public static int MXNDArrayToDLPack(NDArrayHandle handle, @ByPtrPtr DLManagedTensorHandle out_dlpack)
handle
- the handle to the ndarrayout_dlpack
- pointer holder to get pointer of DLManagedTensorpublic static int MXNDArrayFromDLPack(DLManagedTensorHandle dlpack, PointerPointer out_handle)
This allows us to create a NDArray using the memory allocated by an external deep learning framework that is DLPack compatible. The memory is retained until the NDArray went out of scope.
dlpack
- the pointer of the input DLManagedTensortransient_handle
- whether the handle will be destructed before calling the deleterout_handle
- pointer holder to get pointer of NDArraypublic static int MXNDArrayFromDLPack(DLManagedTensorHandle dlpack, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out_handle)
public static int MXNDArrayFromDLPackEx(DLManagedTensorHandle dlpack, @Cast(value="const bool") boolean transient_handle, PointerPointer out_handle)
dlpack
- the pointer of the input DLManagedTensortransient_handle
- whether the handle will be destructed before calling the deleterout_handle
- pointer holder to get pointer of NDArraypublic static int MXNDArrayFromDLPackEx(DLManagedTensorHandle dlpack, @Cast(value="const bool") boolean transient_handle, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out_handle)
public static int MXNDArrayCallDLPackDeleter(DLManagedTensorHandle dlpack)
dlpack
- the pointer of the input DLManagedTensorpublic static int MXNDArrayGetDType(NDArrayHandle handle, IntPointer out_dtype)
handle
- the handle to the narrayout_dtype
- pointer holder to get type of datapublic static int MXNDArrayGetDType(NDArrayHandle handle, IntBuffer out_dtype)
public static int MXNDArrayGetDType(NDArrayHandle handle, int[] out_dtype)
public static int MXNDArrayGetAuxType(NDArrayHandle handle, @Cast(value="uint32_t") int i, IntPointer out_type)
handle
- the handle to the narrayi
- the index of the aux dataout_type
- pointer holder to get type of aux datapublic static int MXNDArrayGetAuxType(NDArrayHandle handle, @Cast(value="uint32_t") int i, IntBuffer out_type)
public static int MXNDArrayGetAuxType(NDArrayHandle handle, @Cast(value="uint32_t") int i, int[] out_type)
public static int MXNDArrayGetAuxNDArray(NDArrayHandle handle, @Cast(value="uint32_t") int i, PointerPointer out)
public static int MXNDArrayGetAuxNDArray(NDArrayHandle handle, @Cast(value="uint32_t") int i, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayGetDataNDArray(NDArrayHandle handle, PointerPointer out)
public static int MXNDArrayGetDataNDArray(NDArrayHandle handle, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayGetContext(NDArrayHandle handle, IntPointer out_dev_type, IntPointer out_dev_id)
handle
- the handle to the narrayout_dev_type
- the output device typeout_dev_id
- the output device idpublic static int MXNDArrayGetContext(NDArrayHandle handle, IntBuffer out_dev_type, IntBuffer out_dev_id)
public static int MXNDArrayGetContext(NDArrayHandle handle, int[] out_dev_type, int[] out_dev_id)
public static int MXNDArrayGetGrad(NDArrayHandle handle, PointerPointer out)
handle
- NDArray handlepublic static int MXNDArrayGetGrad(NDArrayHandle handle, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayDetach(NDArrayHandle handle, PointerPointer out)
handle
- NDArray handlepublic static int MXNDArrayDetach(NDArrayHandle handle, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArraySetGradState(NDArrayHandle handle, int state)
handle
- NDArray handlestate
- the new state.public static int MXNDArrayGetGradState(NDArrayHandle handle, IntPointer out)
handle
- NDArray handlestate
- the new state.public static int MXNDArrayGetGradState(NDArrayHandle handle, IntBuffer out)
public static int MXNDArrayGetGradState(NDArrayHandle handle, int[] out)
public static int MXListFunctions(@Cast(value="uint32_t*") IntPointer out_size, @Const @ByPtrPtr PointerPointer out_array)
out_size
- the size of returned arrayout_array
- the output function arraypublic static int MXListFunctions(@Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="FunctionHandle**") @Const @ByPtrPtr PointerPointer out_array)
public static int MXListFunctions(@Cast(value="uint32_t*") int[] out_size, @Const @ByPtrPtr PointerPointer out_array)
public static int MXGetFunction(@Cast(value="const char*") BytePointer name, @Const PointerPointer out)
name
- the name of the functionout
- the corresponding function handlepublic static int MXGetFunction(String name, @Const @Cast(value="FunctionHandle*") @ByPtrPtr FunctionHandle out)
public static int MXFuncGetInfo(@Const FunctionHandle fun, @Cast(value="const char**") PointerPointer name, @Cast(value="const char**") PointerPointer description, @Cast(value="uint32_t*") IntPointer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") PointerPointer return_type)
fun
- The function handle.name
- The returned name of the function.description
- The returned description of the function.num_args
- Number of arguments.arg_names
- Name of the arguments.arg_type_infos
- Type information about the arguments.arg_descriptions
- Description information about the arguments.return_type
- Return type of the function.public static int MXFuncGetInfo(@Const FunctionHandle fun, @Cast(value="const char**") @ByPtrPtr BytePointer name, @Cast(value="const char**") @ByPtrPtr BytePointer description, @Cast(value="uint32_t*") IntPointer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions)
public static int MXFuncGetInfo(@Const FunctionHandle fun, @Cast(value="const char**") @ByPtrPtr BytePointer name, @Cast(value="const char**") @ByPtrPtr BytePointer description, @Cast(value="uint32_t*") IntPointer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr BytePointer return_type)
public static int MXFuncGetInfo(@Const FunctionHandle fun, @Cast(value="const char**") @ByPtrPtr ByteBuffer name, @Cast(value="const char**") @ByPtrPtr ByteBuffer description, @Cast(value="uint32_t*") IntBuffer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr ByteBuffer return_type)
public static int MXFuncGetInfo(@Const FunctionHandle fun, @Cast(value="const char**") @ByPtrPtr ByteBuffer name, @Cast(value="const char**") @ByPtrPtr ByteBuffer description, @Cast(value="uint32_t*") IntBuffer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions)
public static int MXFuncGetInfo(@Const FunctionHandle fun, @Cast(value="const char**") @ByPtrPtr byte[] name, @Cast(value="const char**") @ByPtrPtr byte[] description, @Cast(value="uint32_t*") int[] num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr byte[] return_type)
public static int MXFuncGetInfo(@Const FunctionHandle fun, @Cast(value="const char**") @ByPtrPtr byte[] name, @Cast(value="const char**") @ByPtrPtr byte[] description, @Cast(value="uint32_t*") int[] num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions)
public static int MXFuncDescribe(@Const FunctionHandle fun, @Cast(value="uint32_t*") IntPointer num_use_vars, @Cast(value="uint32_t*") IntPointer num_scalars, @Cast(value="uint32_t*") IntPointer num_mutate_vars, IntPointer type_mask)
fun
- input function handlenum_use_vars
- how many NDArrays to be passed in as used_varsnum_scalars
- scalar variable is needednum_mutate_vars
- how many NDArrays to be passed in as mutate_varstype_mask
- the type mask of this functionMXFuncInvoke
public static int MXFuncDescribe(@Const FunctionHandle fun, @Cast(value="uint32_t*") IntBuffer num_use_vars, @Cast(value="uint32_t*") IntBuffer num_scalars, @Cast(value="uint32_t*") IntBuffer num_mutate_vars, IntBuffer type_mask)
public static int MXFuncDescribe(@Const FunctionHandle fun, @Cast(value="uint32_t*") int[] num_use_vars, @Cast(value="uint32_t*") int[] num_scalars, @Cast(value="uint32_t*") int[] num_mutate_vars, int[] type_mask)
public static int MXFuncInvoke(@Const FunctionHandle fun, PointerPointer use_vars, FloatPointer scalar_args, PointerPointer mutate_vars)
fun
- the functionuse_vars
- the normal arguments passed to functionscalar_args
- the scalar qargumentsmutate_vars
- the mutate argumentsMXFuncDescribeArgs
public static int MXFuncInvoke(@Const FunctionHandle fun, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle use_vars, FloatBuffer scalar_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle mutate_vars)
public static int MXFuncInvoke(@Const FunctionHandle fun, PointerPointer use_vars, float[] scalar_args, PointerPointer mutate_vars)
public static int MXFuncInvoke(@Const FunctionHandle fun, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle use_vars, FloatPointer scalar_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle mutate_vars)
public static int MXFuncInvoke(@Const FunctionHandle fun, PointerPointer use_vars, FloatBuffer scalar_args, PointerPointer mutate_vars)
public static int MXFuncInvoke(@Const FunctionHandle fun, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle use_vars, float[] scalar_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle mutate_vars)
public static int MXFuncInvokeEx(@Const FunctionHandle fun, PointerPointer use_vars, FloatPointer scalar_args, PointerPointer mutate_vars, int num_params, @Cast(value="char**") PointerPointer param_keys, @Cast(value="char**") PointerPointer param_vals)
fun
- the functionuse_vars
- the normal arguments passed to functionscalar_args
- the scalar qargumentsmutate_vars
- the mutate argumentsnum_params
- number of keyword parametersparam_keys
- keys for keyword parametersparam_vals
- values for keyword parametersMXFuncDescribeArgs
public static int MXFuncInvokeEx(@Const FunctionHandle fun, PointerPointer use_vars, FloatPointer scalar_args, PointerPointer mutate_vars, int num_params, @Cast(value="char**") @ByPtrPtr BytePointer param_keys, @Cast(value="char**") @ByPtrPtr BytePointer param_vals)
public static int MXFuncInvokeEx(@Const FunctionHandle fun, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle use_vars, FloatBuffer scalar_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle mutate_vars, int num_params, @Cast(value="char**") @ByPtrPtr ByteBuffer param_keys, @Cast(value="char**") @ByPtrPtr ByteBuffer param_vals)
public static int MXFuncInvokeEx(@Const FunctionHandle fun, PointerPointer use_vars, float[] scalar_args, PointerPointer mutate_vars, int num_params, @Cast(value="char**") @ByPtrPtr byte[] param_keys, @Cast(value="char**") @ByPtrPtr byte[] param_vals)
public static int MXFuncInvokeEx(@Const FunctionHandle fun, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle use_vars, FloatPointer scalar_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle mutate_vars, int num_params, @Cast(value="char**") @ByPtrPtr BytePointer param_keys, @Cast(value="char**") @ByPtrPtr BytePointer param_vals)
public static int MXFuncInvokeEx(@Const FunctionHandle fun, PointerPointer use_vars, FloatBuffer scalar_args, PointerPointer mutate_vars, int num_params, @Cast(value="char**") @ByPtrPtr ByteBuffer param_keys, @Cast(value="char**") @ByPtrPtr ByteBuffer param_vals)
public static int MXFuncInvokeEx(@Const FunctionHandle fun, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle use_vars, float[] scalar_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle mutate_vars, int num_params, @Cast(value="char**") @ByPtrPtr byte[] param_keys, @Cast(value="char**") @ByPtrPtr byte[] param_vals)
public static int MXImperativeInvoke(AtomicSymbolCreator creator, int num_inputs, PointerPointer inputs, IntPointer num_outputs, @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") PointerPointer param_keys, @Cast(value="const char**") PointerPointer param_vals)
creator
- the opnum_inputs
- number of input NDArraysinputs
- input NDArraysnum_outputs
- number of output NDArraysoutputs
- output NDArraysnum_params
- number of keyword parametersparam_keys
- keys for keyword parametersparam_vals
- values for keyword parameterspublic static int MXImperativeInvoke(AtomicSymbolCreator creator, int num_inputs, PointerPointer inputs, IntPointer num_outputs, @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr BytePointer param_keys, @Cast(value="const char**") @ByPtrPtr BytePointer param_vals)
public static int MXImperativeInvoke(AtomicSymbolCreator creator, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, IntBuffer num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr ByteBuffer param_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer param_vals)
public static int MXImperativeInvoke(AtomicSymbolCreator creator, int num_inputs, PointerPointer inputs, int[] num_outputs, @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr byte[] param_keys, @Cast(value="const char**") @ByPtrPtr byte[] param_vals)
public static int MXImperativeInvoke(AtomicSymbolCreator creator, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, IntPointer num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr BytePointer param_keys, @Cast(value="const char**") @ByPtrPtr BytePointer param_vals)
public static int MXImperativeInvoke(AtomicSymbolCreator creator, int num_inputs, PointerPointer inputs, IntBuffer num_outputs, @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr ByteBuffer param_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer param_vals)
public static int MXImperativeInvoke(AtomicSymbolCreator creator, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, int[] num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr byte[] param_keys, @Cast(value="const char**") @ByPtrPtr byte[] param_vals)
public static int MXImperativeInvokeEx(AtomicSymbolCreator creator, int num_inputs, PointerPointer inputs, IntPointer num_outputs, @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") PointerPointer param_keys, @Cast(value="const char**") PointerPointer param_vals, @Cast(value="const int**") PointerPointer out_stypes)
creator
- the opnum_inputs
- number of input NDArraysinputs
- input NDArraysnum_outputs
- number of output NDArraysoutputs
- output NDArraysnum_params
- number of keyword parametersparam_keys
- keys for keyword parametersparam_vals
- values for keyword parametersout_stypes
- output ndarrays' stypespublic static int MXImperativeInvokeEx(AtomicSymbolCreator creator, int num_inputs, PointerPointer inputs, IntPointer num_outputs, @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr BytePointer param_keys, @Cast(value="const char**") @ByPtrPtr BytePointer param_vals, @Const @ByPtrPtr IntPointer out_stypes)
public static int MXImperativeInvokeEx(AtomicSymbolCreator creator, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, IntBuffer num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr ByteBuffer param_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer param_vals, @Const @ByPtrPtr IntBuffer out_stypes)
public static int MXImperativeInvokeEx(AtomicSymbolCreator creator, int num_inputs, PointerPointer inputs, int[] num_outputs, @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr byte[] param_keys, @Cast(value="const char**") @ByPtrPtr byte[] param_vals, @Const @ByPtrPtr int[] out_stypes)
public static int MXImperativeInvokeEx(AtomicSymbolCreator creator, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, IntPointer num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr BytePointer param_keys, @Cast(value="const char**") @ByPtrPtr BytePointer param_vals, @Const @ByPtrPtr IntPointer out_stypes)
public static int MXImperativeInvokeEx(AtomicSymbolCreator creator, int num_inputs, PointerPointer inputs, IntBuffer num_outputs, @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr ByteBuffer param_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer param_vals, @Const @ByPtrPtr IntBuffer out_stypes)
public static int MXImperativeInvokeEx(AtomicSymbolCreator creator, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, int[] num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs, int num_params, @Cast(value="const char**") @ByPtrPtr byte[] param_keys, @Cast(value="const char**") @ByPtrPtr byte[] param_vals, @Const @ByPtrPtr int[] out_stypes)
public static int MXAutogradSetIsRecording(int is_recording, IntPointer prev)
is_recording
- 1 when recording, 0 when not recording.prev
- returns the previous status before this set.public static int MXAutogradSetIsRecording(int is_recording, IntBuffer prev)
public static int MXAutogradSetIsRecording(int is_recording, int[] prev)
public static int MXAutogradSetIsTraining(int is_training, IntPointer prev)
is_training
- 1 when training, 0 when testingprev
- returns the previous status before this set.public static int MXAutogradSetIsTraining(int is_training, IntBuffer prev)
public static int MXAutogradSetIsTraining(int is_training, int[] prev)
public static int MXAutogradIsRecording(@Cast(value="bool*") BoolPointer curr)
curr
- returns the current status.public static int MXAutogradIsTraining(@Cast(value="bool*") BoolPointer curr)
curr
- returns the current status.public static int MXIsNumpyShape(IntPointer curr)
curr
- returns the current statuspublic static int MXIsNumpyShape(IntBuffer curr)
public static int MXIsNumpyShape(int[] curr)
public static int MXSetIsNumpyShape(int is_np_shape, IntPointer prev)
is_np_shape
- 1 when numpy shape semantics is thread local on,
2 when numpy shape semantics is global on and 0 when offprev
- returns the previous status before this setpublic static int MXSetIsNumpyShape(int is_np_shape, IntBuffer prev)
public static int MXSetIsNumpyShape(int is_np_shape, int[] prev)
public static int MXAutogradMarkVariables(@Cast(value="uint32_t") int num_var, PointerPointer var_handles, @Cast(value="uint32_t*") IntPointer reqs_array, PointerPointer grad_handles)
num_var
- number of variable NDArraysvar_handles
- variable NDArrayspublic static int MXAutogradMarkVariables(@Cast(value="uint32_t") int num_var, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle var_handles, @Cast(value="uint32_t*") IntBuffer reqs_array, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle grad_handles)
public static int MXAutogradMarkVariables(@Cast(value="uint32_t") int num_var, PointerPointer var_handles, @Cast(value="uint32_t*") int[] reqs_array, PointerPointer grad_handles)
public static int MXAutogradMarkVariables(@Cast(value="uint32_t") int num_var, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle var_handles, @Cast(value="uint32_t*") IntPointer reqs_array, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle grad_handles)
public static int MXAutogradMarkVariables(@Cast(value="uint32_t") int num_var, PointerPointer var_handles, @Cast(value="uint32_t*") IntBuffer reqs_array, PointerPointer grad_handles)
public static int MXAutogradMarkVariables(@Cast(value="uint32_t") int num_var, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle var_handles, @Cast(value="uint32_t*") int[] reqs_array, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle grad_handles)
public static int MXAutogradComputeGradient(@Cast(value="uint32_t") int num_output, PointerPointer output_handles)
num_output
- number of output NDArrayoutput_handles
- output NDArrayspublic static int MXAutogradComputeGradient(@Cast(value="uint32_t") int num_output, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle output_handles)
public static int MXAutogradBackward(@Cast(value="uint32_t") int num_output, PointerPointer output_handles, PointerPointer ograd_handles, int retain_graph)
num_output
- number of output NDArrayoutput_handles
- output NDArraysograd_handles
- head gradient for NDArraysretain_graph
- whether to keep the graph after backwardpublic static int MXAutogradBackward(@Cast(value="uint32_t") int num_output, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle output_handles, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle ograd_handles, int retain_graph)
public static int MXAutogradBackwardEx(@Cast(value="uint32_t") int num_output, PointerPointer output_handles, PointerPointer ograd_handles, @Cast(value="uint32_t") int num_variables, PointerPointer var_handles, int retain_graph, int create_graph, int is_train, @ByPtrPtr PointerPointer grad_handles, @Cast(value="int**") PointerPointer grad_stypes)
num_output
- number of output NDArrayoutput_handles
- output NDArraysograd_handles
- head gradient for NDArraysnum_variables
- number of variablesretain_graph
- whether to keep the graph after backwardis_train
- whether to do backward for training or inferencepublic static int MXAutogradBackwardEx(@Cast(value="uint32_t") int num_output, PointerPointer output_handles, PointerPointer ograd_handles, @Cast(value="uint32_t") int num_variables, PointerPointer var_handles, int retain_graph, int create_graph, int is_train, @ByPtrPtr PointerPointer grad_handles, @ByPtrPtr IntPointer grad_stypes)
public static int MXAutogradBackwardEx(@Cast(value="uint32_t") int num_output, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle output_handles, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle ograd_handles, @Cast(value="uint32_t") int num_variables, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle var_handles, int retain_graph, int create_graph, int is_train, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer grad_handles, @ByPtrPtr IntBuffer grad_stypes)
public static int MXAutogradBackwardEx(@Cast(value="uint32_t") int num_output, PointerPointer output_handles, PointerPointer ograd_handles, @Cast(value="uint32_t") int num_variables, PointerPointer var_handles, int retain_graph, int create_graph, int is_train, @ByPtrPtr PointerPointer grad_handles, @ByPtrPtr int[] grad_stypes)
public static int MXAutogradBackwardEx(@Cast(value="uint32_t") int num_output, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle output_handles, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle ograd_handles, @Cast(value="uint32_t") int num_variables, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle var_handles, int retain_graph, int create_graph, int is_train, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer grad_handles, @ByPtrPtr IntPointer grad_stypes)
public static int MXAutogradBackwardEx(@Cast(value="uint32_t") int num_output, PointerPointer output_handles, PointerPointer ograd_handles, @Cast(value="uint32_t") int num_variables, PointerPointer var_handles, int retain_graph, int create_graph, int is_train, @ByPtrPtr PointerPointer grad_handles, @ByPtrPtr IntBuffer grad_stypes)
public static int MXAutogradBackwardEx(@Cast(value="uint32_t") int num_output, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle output_handles, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle ograd_handles, @Cast(value="uint32_t") int num_variables, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle var_handles, int retain_graph, int create_graph, int is_train, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer grad_handles, @ByPtrPtr int[] grad_stypes)
public static int MXAutogradGetSymbol(NDArrayHandle handle, PointerPointer out)
public static int MXAutogradGetSymbol(NDArrayHandle handle, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXCreateCachedOp(SymbolHandle handle, @ByPtrPtr CachedOpHandle out)
public static int MXCreateCachedOpEx(SymbolHandle handle, int num_flags, @Cast(value="const char**") PointerPointer keys, @Cast(value="const char**") PointerPointer vals, @ByPtrPtr CachedOpHandle out)
public static int MXCreateCachedOpEx(SymbolHandle handle, int num_flags, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, @ByPtrPtr CachedOpHandle out)
public static int MXCreateCachedOpEx(SymbolHandle handle, int num_flags, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, @ByPtrPtr CachedOpHandle out)
public static int MXCreateCachedOpEx(SymbolHandle handle, int num_flags, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, @ByPtrPtr CachedOpHandle out)
public static int MXCreateCachedOpEX(SymbolHandle handle, int num_flags, @Cast(value="const char**") PointerPointer keys, @Cast(value="const char**") PointerPointer vals, @ByPtrPtr CachedOpHandle out, @Cast(value="bool") boolean thread_safe)
public static int MXCreateCachedOpEX(SymbolHandle handle, int num_flags, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, @ByPtrPtr CachedOpHandle out)
public static int MXCreateCachedOpEX(SymbolHandle handle, int num_flags, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, @ByPtrPtr CachedOpHandle out, @Cast(value="bool") boolean thread_safe)
public static int MXCreateCachedOpEX(SymbolHandle handle, int num_flags, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, @ByPtrPtr CachedOpHandle out, @Cast(value="bool") boolean thread_safe)
public static int MXCreateCachedOpEX(SymbolHandle handle, int num_flags, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, @ByPtrPtr CachedOpHandle out)
public static int MXCreateCachedOpEX(SymbolHandle handle, int num_flags, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, @ByPtrPtr CachedOpHandle out, @Cast(value="bool") boolean thread_safe)
public static int MXCreateCachedOpEX(SymbolHandle handle, int num_flags, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, @ByPtrPtr CachedOpHandle out)
public static int MXFreeCachedOp(CachedOpHandle handle)
public static int MXInvokeCachedOp(CachedOpHandle handle, int num_inputs, PointerPointer inputs, IntPointer num_outputs, @ByPtrPtr PointerPointer outputs)
public static int MXInvokeCachedOp(CachedOpHandle handle, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, IntBuffer num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs)
public static int MXInvokeCachedOp(CachedOpHandle handle, int num_inputs, PointerPointer inputs, int[] num_outputs, @ByPtrPtr PointerPointer outputs)
public static int MXInvokeCachedOp(CachedOpHandle handle, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, IntPointer num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs)
public static int MXInvokeCachedOp(CachedOpHandle handle, int num_inputs, PointerPointer inputs, IntBuffer num_outputs, @ByPtrPtr PointerPointer outputs)
public static int MXInvokeCachedOp(CachedOpHandle handle, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, int[] num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs)
public static int MXInvokeCachedOpEx(CachedOpHandle handle, int num_inputs, PointerPointer inputs, IntPointer num_outputs, @ByPtrPtr PointerPointer outputs, @Cast(value="const int**") PointerPointer out_stypes)
handle
- the handle to the cached opnum_inputs
- number of input NDArraysinputs
- input NDArraysnum_outputs
- number of output NDArraysoutputs
- output NDArraysout_stypes
- output ndarrays' stypespublic static int MXInvokeCachedOpEx(CachedOpHandle handle, int num_inputs, PointerPointer inputs, IntPointer num_outputs, @ByPtrPtr PointerPointer outputs, @Const @ByPtrPtr IntPointer out_stypes)
public static int MXInvokeCachedOpEx(CachedOpHandle handle, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, IntBuffer num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs, @Const @ByPtrPtr IntBuffer out_stypes)
public static int MXInvokeCachedOpEx(CachedOpHandle handle, int num_inputs, PointerPointer inputs, int[] num_outputs, @ByPtrPtr PointerPointer outputs, @Const @ByPtrPtr int[] out_stypes)
public static int MXInvokeCachedOpEx(CachedOpHandle handle, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, IntPointer num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs, @Const @ByPtrPtr IntPointer out_stypes)
public static int MXInvokeCachedOpEx(CachedOpHandle handle, int num_inputs, PointerPointer inputs, IntBuffer num_outputs, @ByPtrPtr PointerPointer outputs, @Const @ByPtrPtr IntBuffer out_stypes)
public static int MXInvokeCachedOpEx(CachedOpHandle handle, int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, int[] num_outputs, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer outputs, @Const @ByPtrPtr int[] out_stypes)
public static int MXCachedOpRegisterOpHook(NDArrayHandle handle, CachedOpMonitorCallback callback, @Cast(value="bool") boolean monitor_all)
public static int MXListAllOpNames(@Cast(value="uint32_t*") IntPointer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_array)
out_size
- the size of returned arrayout_array
- the output operator name array.public static int MXListAllOpNames(@Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_array)
public static int MXListAllOpNames(@Cast(value="uint32_t*") int[] out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_array)
public static int MXSymbolListAtomicSymbolCreators(@Cast(value="uint32_t*") IntPointer out_size, @ByPtrPtr PointerPointer out_array)
out_size
- the size of returned arrayout_array
- the output AtomicSymbolCreator arraypublic static int MXSymbolListAtomicSymbolCreators(@Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="AtomicSymbolCreator**") @ByPtrPtr PointerPointer out_array)
public static int MXSymbolListAtomicSymbolCreators(@Cast(value="uint32_t*") int[] out_size, @ByPtrPtr PointerPointer out_array)
public static int MXSymbolGetAtomicSymbolName(AtomicSymbolCreator creator, @Cast(value="const char**") PointerPointer name)
creator
- the AtomicSymbolCreator.name
- The returned name of the creator.public static int MXSymbolGetAtomicSymbolName(AtomicSymbolCreator creator, @Cast(value="const char**") @ByPtrPtr BytePointer name)
public static int MXSymbolGetAtomicSymbolName(AtomicSymbolCreator creator, @Cast(value="const char**") @ByPtrPtr ByteBuffer name)
public static int MXSymbolGetAtomicSymbolName(AtomicSymbolCreator creator, @Cast(value="const char**") @ByPtrPtr byte[] name)
public static int MXSymbolGetInputSymbols(SymbolHandle sym, @ByPtrPtr PointerPointer inputs, IntPointer input_size)
sym
- The graph.inputs
- The input symbols of the graph.input_size
- the number of input symbols returned.public static int MXSymbolGetInputSymbols(SymbolHandle sym, @Cast(value="SymbolHandle**") @ByPtrPtr PointerPointer inputs, IntBuffer input_size)
public static int MXSymbolGetInputSymbols(SymbolHandle sym, @ByPtrPtr PointerPointer inputs, int[] input_size)
public static int MXSymbolCutSubgraph(SymbolHandle sym, @ByPtrPtr PointerPointer inputs, IntPointer input_size)
sym
- The graph.inputs
- The nodes that connect to the subgraph.input_size
- The number of such nodes.public static int MXSymbolCutSubgraph(SymbolHandle sym, @Cast(value="SymbolHandle**") @ByPtrPtr PointerPointer inputs, IntBuffer input_size)
public static int MXSymbolCutSubgraph(SymbolHandle sym, @ByPtrPtr PointerPointer inputs, int[] input_size)
public static int MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator, @Cast(value="const char**") PointerPointer name, @Cast(value="const char**") PointerPointer description, @Cast(value="uint32_t*") IntPointer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") PointerPointer key_var_num_args, @Cast(value="const char**") PointerPointer return_type)
creator
- the AtomicSymbolCreator.name
- The returned name of the creator.description
- The returned description of the symbol.num_args
- Number of arguments.arg_names
- Name of the arguments.arg_type_infos
- Type informations about the arguments.arg_descriptions
- Description information about the arguments.key_var_num_args
- The keyword argument for specifying variable number of arguments.
When this parameter has non-zero length, the function allows variable number
of positional arguments, and will need the caller to pass it in in
MXSymbolCreateAtomicSymbol,
With key = key_var_num_args, and value = number of positional arguments.return_type
- Return type of the function, can be Symbol or Symbol[]public static int MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator, @Cast(value="const char**") @ByPtrPtr BytePointer name, @Cast(value="const char**") @ByPtrPtr BytePointer description, @Cast(value="uint32_t*") IntPointer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr BytePointer key_var_num_args)
public static int MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator, @Cast(value="const char**") @ByPtrPtr BytePointer name, @Cast(value="const char**") @ByPtrPtr BytePointer description, @Cast(value="uint32_t*") IntPointer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr BytePointer key_var_num_args, @Cast(value="const char**") @ByPtrPtr BytePointer return_type)
public static int MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator, @Cast(value="const char**") @ByPtrPtr ByteBuffer name, @Cast(value="const char**") @ByPtrPtr ByteBuffer description, @Cast(value="uint32_t*") IntBuffer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr ByteBuffer key_var_num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer return_type)
public static int MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator, @Cast(value="const char**") @ByPtrPtr ByteBuffer name, @Cast(value="const char**") @ByPtrPtr ByteBuffer description, @Cast(value="uint32_t*") IntBuffer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr ByteBuffer key_var_num_args)
public static int MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator, @Cast(value="const char**") @ByPtrPtr byte[] name, @Cast(value="const char**") @ByPtrPtr byte[] description, @Cast(value="uint32_t*") int[] num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr byte[] key_var_num_args, @Cast(value="const char**") @ByPtrPtr byte[] return_type)
public static int MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator, @Cast(value="const char**") @ByPtrPtr byte[] name, @Cast(value="const char**") @ByPtrPtr byte[] description, @Cast(value="uint32_t*") int[] num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr byte[] key_var_num_args)
public static int MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") PointerPointer keys, @Cast(value="const char**") PointerPointer vals, PointerPointer out)
creator
- the AtomicSymbolCreatornum_param
- the number of parameterskeys
- the keys to the paramsvals
- the vals of the paramsout
- pointer to the created symbol handlepublic static int MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, PointerPointer out)
public static int MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, PointerPointer out)
public static int MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, PointerPointer out)
public static int MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolCreateVariable(@Cast(value="const char*") BytePointer name, PointerPointer out)
name
- name of the variableout
- pointer to the created symbol handlepublic static int MXSymbolCreateVariable(String name, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolCreateGroup(@Cast(value="uint32_t") int num_symbols, PointerPointer symbols, PointerPointer out)
num_symbols
- number of symbols to be groupedsymbols
- array of symbol handlesout
- pointer to the created symbol handlepublic static int MXSymbolCreateGroup(@Cast(value="uint32_t") int num_symbols, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle symbols, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolCreateFromFile(@Cast(value="const char*") BytePointer fname, PointerPointer out)
fname
- the file name.out
- the output symbol.public static int MXSymbolCreateFromFile(String fname, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolCreateFromJSON(@Cast(value="const char*") BytePointer json, PointerPointer out)
json
- the json string.out
- the output symbol.public static int MXSymbolCreateFromJSON(String json, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolRemoveAmpCast(SymbolHandle sym_handle, PointerPointer ret_sym_handle)
sym_handle
- the input symbol.ret_sym_handle
- the output symbol.public static int MXSymbolRemoveAmpCast(SymbolHandle sym_handle, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle)
public static int MXSymbolSaveToFile(SymbolHandle symbol, @Cast(value="const char*") BytePointer fname)
symbol
- the input symbol.fname
- the file name.public static int MXSymbolSaveToFile(SymbolHandle symbol, String fname)
public static int MXSymbolSaveToJSON(SymbolHandle symbol, @Cast(value="const char**") PointerPointer out_json)
symbol
- the input symbol.out_json
- output json string.public static int MXSymbolSaveToJSON(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr BytePointer out_json)
public static int MXSymbolSaveToJSON(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr ByteBuffer out_json)
public static int MXSymbolSaveToJSON(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr byte[] out_json)
public static int MXSymbolFree(SymbolHandle symbol)
symbol
- the symbolpublic static int MXSymbolCopy(SymbolHandle symbol, PointerPointer out)
symbol
- the source symbolout
- used to hold the result of copypublic static int MXSymbolCopy(SymbolHandle symbol, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolPrint(SymbolHandle symbol, @Cast(value="const char**") PointerPointer out_str)
symbol
- the symbolout_str
- pointer to hold the output string of the printing.public static int MXSymbolPrint(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr BytePointer out_str)
public static int MXSymbolPrint(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr ByteBuffer out_str)
public static int MXSymbolPrint(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr byte[] out_str)
public static int MXSymbolGetName(SymbolHandle symbol, @Cast(value="const char**") PointerPointer out, IntPointer success)
symbol
- the source symbolout
- The result name.success
- Whether the result is contained in out.public static int MXSymbolGetName(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr BytePointer out, IntPointer success)
public static int MXSymbolGetName(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr ByteBuffer out, IntBuffer success)
public static int MXSymbolGetName(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr byte[] out, int[] success)
public static int MXSymbolGetAttr(SymbolHandle symbol, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") PointerPointer out, IntPointer success)
symbol
- the source symbolkey
- The key of the symbol.out
- The result attribute, can be NULL if the attribute do not exist.success
- Whether the result is contained in out.public static int MXSymbolGetAttr(SymbolHandle symbol, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") @ByPtrPtr BytePointer out, IntPointer success)
public static int MXSymbolGetAttr(SymbolHandle symbol, String key, @Cast(value="const char**") @ByPtrPtr ByteBuffer out, IntBuffer success)
public static int MXSymbolGetAttr(SymbolHandle symbol, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") @ByPtrPtr byte[] out, int[] success)
public static int MXSymbolGetAttr(SymbolHandle symbol, String key, @Cast(value="const char**") @ByPtrPtr BytePointer out, IntPointer success)
public static int MXSymbolGetAttr(SymbolHandle symbol, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") @ByPtrPtr ByteBuffer out, IntBuffer success)
public static int MXSymbolGetAttr(SymbolHandle symbol, String key, @Cast(value="const char**") @ByPtrPtr byte[] out, int[] success)
public static int MXSymbolSetAttr(SymbolHandle symbol, @Cast(value="const char*") BytePointer key, @Cast(value="const char*") BytePointer value)
symbol
- the source symbolkey
- The key of the symbol.value
- The value to be saved.public static int MXSymbolSetAttr(SymbolHandle symbol, String key, String value)
public static int MXSymbolListAttr(SymbolHandle symbol, @Cast(value="uint32_t*") IntPointer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out)
symbol
- the source symbolout_size
- The number of output attributesout
- 2*out_size strings representing key value pairs.public static int MXSymbolListAttr(SymbolHandle symbol, @Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out)
public static int MXSymbolListAttr(SymbolHandle symbol, @Cast(value="uint32_t*") int[] out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out)
public static int MXSymbolListAttrShallow(SymbolHandle symbol, @Cast(value="uint32_t*") IntPointer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out)
symbol
- the source symbolout_size
- The number of output attributesout
- 2*out_size strings representing key value pairs.public static int MXSymbolListAttrShallow(SymbolHandle symbol, @Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out)
public static int MXSymbolListAttrShallow(SymbolHandle symbol, @Cast(value="uint32_t*") int[] out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out)
public static int MXSymbolListArguments(SymbolHandle symbol, @Cast(value="uint32_t*") IntPointer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
symbol
- the symbolout_size
- output sizeout_str_array
- pointer to hold the output string arraypublic static int MXSymbolListArguments(SymbolHandle symbol, @Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
public static int MXSymbolListArguments(SymbolHandle symbol, @Cast(value="uint32_t*") int[] out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
public static int MXSymbolListOutputs(SymbolHandle symbol, @Cast(value="uint32_t*") IntPointer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
symbol
- the symbolout_size
- output sizeout_str_array
- pointer to hold the output string arraypublic static int MXSymbolListOutputs(SymbolHandle symbol, @Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
public static int MXSymbolListOutputs(SymbolHandle symbol, @Cast(value="uint32_t*") int[] out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
public static int MXSymbolGetNumOutputs(SymbolHandle symbol, @Cast(value="uint32_t*") IntPointer output_count)
symbol
- The symbolout_size
- number of outputspublic static int MXSymbolGetNumOutputs(SymbolHandle symbol, @Cast(value="uint32_t*") IntBuffer output_count)
public static int MXSymbolGetNumOutputs(SymbolHandle symbol, @Cast(value="uint32_t*") int[] output_count)
public static int MXSymbolGetInternals(SymbolHandle symbol, PointerPointer out)
symbol
- The symbolout
- The output symbol whose outputs are all the internals.public static int MXSymbolGetInternals(SymbolHandle symbol, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolGetChildren(SymbolHandle symbol, PointerPointer out)
symbol
- The symbolout
- The output symbol whose outputs are the direct children.public static int MXSymbolGetChildren(SymbolHandle symbol, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolGetOutput(SymbolHandle symbol, @Cast(value="uint32_t") int index, PointerPointer out)
symbol
- The symbolindex
- the Index of the output.out
- The output symbol whose outputs are the index-th symbol.public static int MXSymbolGetOutput(SymbolHandle symbol, @Cast(value="uint32_t") int index, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolListAuxiliaryStates(SymbolHandle symbol, @Cast(value="uint32_t*") IntPointer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
symbol
- the symbolout_size
- output sizeout_str_array
- pointer to hold the output string arraypublic static int MXSymbolListAuxiliaryStates(SymbolHandle symbol, @Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
public static int MXSymbolListAuxiliaryStates(SymbolHandle symbol, @Cast(value="uint32_t*") int[] out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
public static int MXSymbolCompose(SymbolHandle sym, @Cast(value="const char*") BytePointer name, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") PointerPointer keys, PointerPointer args)
sym
- the symbol to applyname
- the name of symbolnum_args
- number of argumentskeys
- the key of keyword args (optional)args
- arguments to sympublic static int MXSymbolCompose(SymbolHandle sym, @Cast(value="const char*") BytePointer name, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, PointerPointer args)
public static int MXSymbolCompose(SymbolHandle sym, String name, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle args)
public static int MXSymbolCompose(SymbolHandle sym, @Cast(value="const char*") BytePointer name, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, PointerPointer args)
public static int MXSymbolCompose(SymbolHandle sym, String name, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle args)
public static int MXSymbolCompose(SymbolHandle sym, @Cast(value="const char*") BytePointer name, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, PointerPointer args)
public static int MXSymbolCompose(SymbolHandle sym, String name, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle args)
public static int MXSymbolGrad(SymbolHandle sym, @Cast(value="uint32_t") int num_wrt, @Cast(value="const char**") PointerPointer wrt, PointerPointer out)
sym
- the symbol to get gradientnum_wrt
- number of arguments to get gradientwrt
- the name of the arguments to get gradientout
- the returned symbol that has gradientpublic static int MXSymbolGrad(SymbolHandle sym, @Cast(value="uint32_t") int num_wrt, @Cast(value="const char**") @ByPtrPtr BytePointer wrt, PointerPointer out)
public static int MXSymbolGrad(SymbolHandle sym, @Cast(value="uint32_t") int num_wrt, @Cast(value="const char**") @ByPtrPtr ByteBuffer wrt, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolGrad(SymbolHandle sym, @Cast(value="uint32_t") int num_wrt, @Cast(value="const char**") @ByPtrPtr byte[] wrt, PointerPointer out)
public static int MXSymbolGrad(SymbolHandle sym, @Cast(value="uint32_t") int num_wrt, @Cast(value="const char**") @ByPtrPtr BytePointer wrt, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolGrad(SymbolHandle sym, @Cast(value="uint32_t") int num_wrt, @Cast(value="const char**") @ByPtrPtr ByteBuffer wrt, PointerPointer out)
public static int MXSymbolGrad(SymbolHandle sym, @Cast(value="uint32_t") int num_wrt, @Cast(value="const char**") @ByPtrPtr byte[] wrt, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXSymbolInferShape(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") PointerPointer keys, @Cast(value="const uint32_t*") IntPointer arg_ind_ptr, @Cast(value="const uint32_t*") IntPointer arg_shape_data, @Cast(value="uint32_t*") IntPointer in_shape_size, @Cast(value="const uint32_t**") PointerPointer in_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntPointer out_shape_size, @Cast(value="const uint32_t**") PointerPointer out_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntPointer aux_shape_size, @Cast(value="const uint32_t**") PointerPointer aux_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
sym
- symbol handlenum_args
- numbe of input arguments.keys
- the key of keyword args (optional)arg_ind_ptr
- the head pointer of the rows in CSRarg_shape_data
- the content of the CSRin_shape_size
- sizeof the returning array of in_shapesin_shape_ndim
- returning array of shape dimensions of each input shape.in_shape_data
- returning array of pointers to head of the input shape.out_shape_size
- sizeof the returning array of out_shapesout_shape_ndim
- returning array of shape dimensions of each output shape.out_shape_data
- returning array of pointers to head of the output shape.aux_shape_size
- sizeof the returning array of aux_shapesaux_shape_ndim
- returning array of shape dimensions of each auxiliary shape.aux_shape_data
- returning array of pointers to head of the auxiliary shape.complete
- whether infer shape completes or more information is needed.public static int MXSymbolInferShape(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const uint32_t*") IntPointer arg_ind_ptr, @Cast(value="const uint32_t*") IntPointer arg_shape_data, @Cast(value="uint32_t*") IntPointer in_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntPointer in_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntPointer out_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntPointer out_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntPointer aux_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntPointer aux_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
public static int MXSymbolInferShape(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const uint32_t*") IntBuffer arg_ind_ptr, @Cast(value="const uint32_t*") IntBuffer arg_shape_data, @Cast(value="uint32_t*") IntBuffer in_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntBuffer in_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntBuffer out_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntBuffer out_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntBuffer aux_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntBuffer aux_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer aux_shape_data, IntBuffer complete)
public static int MXSymbolInferShape(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const uint32_t*") int[] arg_ind_ptr, @Cast(value="const uint32_t*") int[] arg_shape_data, @Cast(value="uint32_t*") int[] in_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr int[] in_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") int[] out_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr int[] out_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") int[] aux_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr int[] aux_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer aux_shape_data, int[] complete)
public static int MXSymbolInferShapeEx(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") PointerPointer keys, @Cast(value="const uint32_t*") IntPointer arg_ind_ptr, @Const IntPointer arg_shape_data, @Cast(value="uint32_t*") IntPointer in_shape_size, @Cast(value="const int**") PointerPointer in_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntPointer out_shape_size, @Cast(value="const int**") PointerPointer out_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntPointer aux_shape_size, @Cast(value="const int**") PointerPointer aux_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
sym
- symbol handlenum_args
- number of input arguments.keys
- the key of keyword args (optional)arg_ind_ptr
- the head pointer of the rows in CSRarg_shape_data
- the content of the CSRin_shape_size
- sizeof the returning array of in_shapesin_shape_ndim
- returning array of shape dimensions of eachs input shape.in_shape_data
- returning array of pointers to head of the input shape.out_shape_size
- sizeof the returning array of out_shapesout_shape_ndim
- returning array of shape dimensions of each output shape.out_shape_data
- returning array of pointers to head of the output shape.aux_shape_size
- sizeof the returning array of aux_shapesaux_shape_ndim
- returning array of shape dimensions of each auxiliary shape.aux_shape_data
- returning array of pointers to head of the auxiliary shape.complete
- whether infer shape completes or more information is needed.public static int MXSymbolInferShapeEx(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const uint32_t*") IntPointer arg_ind_ptr, @Const IntPointer arg_shape_data, @Cast(value="uint32_t*") IntPointer in_shape_size, @Const @ByPtrPtr IntPointer in_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntPointer out_shape_size, @Const @ByPtrPtr IntPointer out_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntPointer aux_shape_size, @Const @ByPtrPtr IntPointer aux_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
public static int MXSymbolInferShapeEx(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const uint32_t*") IntBuffer arg_ind_ptr, @Const IntBuffer arg_shape_data, @Cast(value="uint32_t*") IntBuffer in_shape_size, @Const @ByPtrPtr IntBuffer in_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntBuffer out_shape_size, @Const @ByPtrPtr IntBuffer out_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntBuffer aux_shape_size, @Const @ByPtrPtr IntBuffer aux_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer aux_shape_data, IntBuffer complete)
public static int MXSymbolInferShapeEx(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const uint32_t*") int[] arg_ind_ptr, @Const int[] arg_shape_data, @Cast(value="uint32_t*") int[] in_shape_size, @Const @ByPtrPtr int[] in_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") int[] out_shape_size, @Const @ByPtrPtr int[] out_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") int[] aux_shape_size, @Const @ByPtrPtr int[] aux_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer aux_shape_data, int[] complete)
public static int MXSymbolInferShapeEx64(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") PointerPointer keys, @Cast(value="const int64_t*") LongPointer arg_ind_ptr, @Cast(value="const int64_t*") LongPointer arg_shape_data, @Cast(value="size_t*") SizeTPointer in_shape_size, @Cast(value="const int**") PointerPointer in_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="size_t*") SizeTPointer out_shape_size, @Cast(value="const int**") PointerPointer out_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="size_t*") SizeTPointer aux_shape_size, @Cast(value="const int**") PointerPointer aux_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
sym
- symbol handlenum_args
- number of input arguments.keys
- the key of keyword args (optional)arg_ind_ptr
- the head pointer of the rows in CSRarg_shape_data
- the content of the CSRin_shape_size
- sizeof the returning array of in_shapesin_shape_ndim
- returning array of shape dimensions of each input shape.in_shape_data
- returning array of pointers to head of the input shape.out_shape_size
- sizeof the returning array of out_shapesout_shape_ndim
- returning array of shape dimensions of each output shape.out_shape_data
- returning array of pointers to head of the output shape.aux_shape_size
- sizeof the returning array of aux_shapesaux_shape_ndim
- returning array of shape dimensions of each auxiliary shape.aux_shape_data
- returning array of pointers to head of the auxiliary shape.complete
- whether infer shape completes or more information is needed.public static int MXSymbolInferShapeEx64(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const int64_t*") LongPointer arg_ind_ptr, @Cast(value="const int64_t*") LongPointer arg_shape_data, @Cast(value="size_t*") SizeTPointer in_shape_size, @Const @ByPtrPtr IntPointer in_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="size_t*") SizeTPointer out_shape_size, @Const @ByPtrPtr IntPointer out_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="size_t*") SizeTPointer aux_shape_size, @Const @ByPtrPtr IntPointer aux_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
public static int MXSymbolInferShapeEx64(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const int64_t*") LongBuffer arg_ind_ptr, @Cast(value="const int64_t*") LongBuffer arg_shape_data, @Cast(value="size_t*") SizeTPointer in_shape_size, @Const @ByPtrPtr IntBuffer in_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="size_t*") SizeTPointer out_shape_size, @Const @ByPtrPtr IntBuffer out_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="size_t*") SizeTPointer aux_shape_size, @Const @ByPtrPtr IntBuffer aux_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer aux_shape_data, IntBuffer complete)
public static int MXSymbolInferShapeEx64(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const int64_t*") long[] arg_ind_ptr, @Cast(value="const int64_t*") long[] arg_shape_data, @Cast(value="size_t*") SizeTPointer in_shape_size, @Const @ByPtrPtr int[] in_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="size_t*") SizeTPointer out_shape_size, @Const @ByPtrPtr int[] out_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="size_t*") SizeTPointer aux_shape_size, @Const @ByPtrPtr int[] aux_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer aux_shape_data, int[] complete)
public static int MXSymbolInferShapePartial(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") PointerPointer keys, @Cast(value="const uint32_t*") IntPointer arg_ind_ptr, @Cast(value="const uint32_t*") IntPointer arg_shape_data, @Cast(value="uint32_t*") IntPointer in_shape_size, @Cast(value="const uint32_t**") PointerPointer in_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntPointer out_shape_size, @Cast(value="const uint32_t**") PointerPointer out_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntPointer aux_shape_size, @Cast(value="const uint32_t**") PointerPointer aux_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
sym
- symbol handlenum_args
- numbe of input arguments.keys
- the key of keyword args (optional)arg_ind_ptr
- the head pointer of the rows in CSRarg_shape_data
- the content of the CSRin_shape_size
- sizeof the returning array of in_shapesin_shape_ndim
- returning array of shape dimensions of each input shape.in_shape_data
- returning array of pointers to head of the input shape.out_shape_size
- sizeof the returning array of out_shapesout_shape_ndim
- returning array of shape dimensions of each output shape.out_shape_data
- returning array of pointers to head of the output shape.aux_shape_size
- sizeof the returning array of aux_shapesaux_shape_ndim
- returning array of shape dimensions of each auxiliary shape.aux_shape_data
- returning array of pointers to head of the auxiliary shape.complete
- whether infer shape completes or more information is needed.public static int MXSymbolInferShapePartial(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const uint32_t*") IntPointer arg_ind_ptr, @Cast(value="const uint32_t*") IntPointer arg_shape_data, @Cast(value="uint32_t*") IntPointer in_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntPointer in_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntPointer out_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntPointer out_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntPointer aux_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntPointer aux_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
public static int MXSymbolInferShapePartial(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const uint32_t*") IntBuffer arg_ind_ptr, @Cast(value="const uint32_t*") IntBuffer arg_shape_data, @Cast(value="uint32_t*") IntBuffer in_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntBuffer in_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntBuffer out_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntBuffer out_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntBuffer aux_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr IntBuffer aux_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer aux_shape_data, IntBuffer complete)
public static int MXSymbolInferShapePartial(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const uint32_t*") int[] arg_ind_ptr, @Cast(value="const uint32_t*") int[] arg_shape_data, @Cast(value="uint32_t*") int[] in_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr int[] in_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") int[] out_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr int[] out_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") int[] aux_shape_size, @Cast(value="const uint32_t**") @ByPtrPtr int[] aux_shape_ndim, @Cast(value="const uint32_t***") @ByPtrPtr PointerPointer aux_shape_data, int[] complete)
public static int MXSymbolInferShapePartialEx(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") PointerPointer keys, @Cast(value="const uint32_t*") IntPointer arg_ind_ptr, @Const IntPointer arg_shape_data, @Cast(value="uint32_t*") IntPointer in_shape_size, @Cast(value="const int**") PointerPointer in_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntPointer out_shape_size, @Cast(value="const int**") PointerPointer out_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntPointer aux_shape_size, @Cast(value="const int**") PointerPointer aux_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
sym
- symbol handlenum_args
- number of input arguments.keys
- the key of keyword args (optional)arg_ind_ptr
- the head pointer of the rows in CSRarg_shape_data
- the content of the CSRin_shape_size
- sizeof the returning array of in_shapesin_shape_ndim
- returning array of shape dimensions of each input shape.in_shape_data
- returning array of pointers to head of the input shape.out_shape_size
- sizeof the returning array of out_shapesout_shape_ndim
- returning array of shape dimensions of each output shape.out_shape_data
- returning array of pointers to head of the output shape.aux_shape_size
- sizeof the returning array of aux_shapesaux_shape_ndim
- returning array of shape dimensions of each auxiliary shape.aux_shape_data
- returning array of pointers to head of the auxiliary shape.complete
- whether infer shape completes or more information is needed.public static int MXSymbolInferShapePartialEx(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const uint32_t*") IntPointer arg_ind_ptr, @Const IntPointer arg_shape_data, @Cast(value="uint32_t*") IntPointer in_shape_size, @Const @ByPtrPtr IntPointer in_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntPointer out_shape_size, @Const @ByPtrPtr IntPointer out_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntPointer aux_shape_size, @Const @ByPtrPtr IntPointer aux_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
public static int MXSymbolInferShapePartialEx(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const uint32_t*") IntBuffer arg_ind_ptr, @Const IntBuffer arg_shape_data, @Cast(value="uint32_t*") IntBuffer in_shape_size, @Const @ByPtrPtr IntBuffer in_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") IntBuffer out_shape_size, @Const @ByPtrPtr IntBuffer out_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") IntBuffer aux_shape_size, @Const @ByPtrPtr IntBuffer aux_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer aux_shape_data, IntBuffer complete)
public static int MXSymbolInferShapePartialEx(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const uint32_t*") int[] arg_ind_ptr, @Const int[] arg_shape_data, @Cast(value="uint32_t*") int[] in_shape_size, @Const @ByPtrPtr int[] in_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="uint32_t*") int[] out_shape_size, @Const @ByPtrPtr int[] out_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="uint32_t*") int[] aux_shape_size, @Const @ByPtrPtr int[] aux_shape_ndim, @Cast(value="const int***") @ByPtrPtr PointerPointer aux_shape_data, int[] complete)
public static int MXSymbolInferShapePartialEx64(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") PointerPointer keys, @Cast(value="const int64_t*") LongPointer arg_ind_ptr, @Cast(value="const int64_t*") LongPointer arg_shape_data, @Cast(value="size_t*") SizeTPointer in_shape_size, @Cast(value="const int**") PointerPointer in_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="size_t*") SizeTPointer out_shape_size, @Cast(value="const int**") PointerPointer out_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="size_t*") SizeTPointer aux_shape_size, @Cast(value="const int**") PointerPointer aux_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
sym
- symbol handlenum_args
- number of input arguments.keys
- the key of keyword args (optional)arg_ind_ptr
- the head pointer of the rows in CSRarg_shape_data
- the content of the CSRin_shape_size
- sizeof the returning array of in_shapesin_shape_ndim
- returning array of shape dimensions of each input shape.in_shape_data
- returning array of pointers to head of the input shape.out_shape_size
- sizeof the returning array of out_shapesout_shape_ndim
- returning array of shape dimensions of each output shape.out_shape_data
- returning array of pointers to head of the output shape.aux_shape_size
- sizeof the returning array of aux_shapesaux_shape_ndim
- returning array of shape dimensions of each auxiliary shape.aux_shape_data
- returning array of pointers to head of the auxiliary shape.complete
- whether infer shape completes or more information is needed.public static int MXSymbolInferShapePartialEx64(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const int64_t*") LongPointer arg_ind_ptr, @Cast(value="const int64_t*") LongPointer arg_shape_data, @Cast(value="size_t*") SizeTPointer in_shape_size, @Const @ByPtrPtr IntPointer in_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="size_t*") SizeTPointer out_shape_size, @Const @ByPtrPtr IntPointer out_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="size_t*") SizeTPointer aux_shape_size, @Const @ByPtrPtr IntPointer aux_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer aux_shape_data, IntPointer complete)
public static int MXSymbolInferShapePartialEx64(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const int64_t*") LongBuffer arg_ind_ptr, @Cast(value="const int64_t*") LongBuffer arg_shape_data, @Cast(value="size_t*") SizeTPointer in_shape_size, @Const @ByPtrPtr IntBuffer in_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="size_t*") SizeTPointer out_shape_size, @Const @ByPtrPtr IntBuffer out_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="size_t*") SizeTPointer aux_shape_size, @Const @ByPtrPtr IntBuffer aux_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer aux_shape_data, IntBuffer complete)
public static int MXSymbolInferShapePartialEx64(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const int64_t*") long[] arg_ind_ptr, @Cast(value="const int64_t*") long[] arg_shape_data, @Cast(value="size_t*") SizeTPointer in_shape_size, @Const @ByPtrPtr int[] in_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer in_shape_data, @Cast(value="size_t*") SizeTPointer out_shape_size, @Const @ByPtrPtr int[] out_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer out_shape_data, @Cast(value="size_t*") SizeTPointer aux_shape_size, @Const @ByPtrPtr int[] aux_shape_ndim, @Cast(value="const int64_t***") @ByPtrPtr PointerPointer aux_shape_data, int[] complete)
public static int MXSymbolInferType(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") PointerPointer keys, @Const IntPointer arg_type_data, @Cast(value="uint32_t*") IntPointer in_type_size, @Cast(value="const int**") PointerPointer in_type_data, @Cast(value="uint32_t*") IntPointer out_type_size, @Cast(value="const int**") PointerPointer out_type_data, @Cast(value="uint32_t*") IntPointer aux_type_size, @Cast(value="const int**") PointerPointer aux_type_data, IntPointer complete)
sym
- symbol handlenum_args
- numbe of input arguments.keys
- the key of keyword args (optional)arg_type_data
- the content of the CSRin_type_size
- sizeof the returning array of in_typesin_type_data
- returning array of pointers to head of the input type.out_type_size
- sizeof the returning array of out_typesout_type_data
- returning array of pointers to head of the output type.aux_type_size
- sizeof the returning array of aux_typesaux_type_data
- returning array of pointers to head of the auxiliary type.complete
- whether infer type completes or more information is needed.public static int MXSymbolInferType(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Const IntPointer arg_type_data, @Cast(value="uint32_t*") IntPointer in_type_size, @Const @ByPtrPtr IntPointer in_type_data, @Cast(value="uint32_t*") IntPointer out_type_size, @Const @ByPtrPtr IntPointer out_type_data, @Cast(value="uint32_t*") IntPointer aux_type_size, @Const @ByPtrPtr IntPointer aux_type_data, IntPointer complete)
public static int MXSymbolInferType(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Const IntBuffer arg_type_data, @Cast(value="uint32_t*") IntBuffer in_type_size, @Const @ByPtrPtr IntBuffer in_type_data, @Cast(value="uint32_t*") IntBuffer out_type_size, @Const @ByPtrPtr IntBuffer out_type_data, @Cast(value="uint32_t*") IntBuffer aux_type_size, @Const @ByPtrPtr IntBuffer aux_type_data, IntBuffer complete)
public static int MXSymbolInferType(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Const int[] arg_type_data, @Cast(value="uint32_t*") int[] in_type_size, @Const @ByPtrPtr int[] in_type_data, @Cast(value="uint32_t*") int[] out_type_size, @Const @ByPtrPtr int[] out_type_data, @Cast(value="uint32_t*") int[] aux_type_size, @Const @ByPtrPtr int[] aux_type_data, int[] complete)
public static int MXSymbolInferTypePartial(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") PointerPointer keys, @Const IntPointer arg_type_data, @Cast(value="uint32_t*") IntPointer in_type_size, @Cast(value="const int**") PointerPointer in_type_data, @Cast(value="uint32_t*") IntPointer out_type_size, @Cast(value="const int**") PointerPointer out_type_data, @Cast(value="uint32_t*") IntPointer aux_type_size, @Cast(value="const int**") PointerPointer aux_type_data, IntPointer complete)
sym
- symbol handlenum_args
- numbe of input arguments.keys
- the key of keyword args (optional)arg_type_data
- the content of the CSRin_type_size
- sizeof the returning array of in_typesin_type_data
- returning array of pointers to head of the input type.out_type_size
- sizeof the returning array of out_typesout_type_data
- returning array of pointers to head of the output type.aux_type_size
- sizeof the returning array of aux_typesaux_type_data
- returning array of pointers to head of the auxiliary type.complete
- whether infer type completes or more information is needed.public static int MXSymbolInferTypePartial(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Const IntPointer arg_type_data, @Cast(value="uint32_t*") IntPointer in_type_size, @Const @ByPtrPtr IntPointer in_type_data, @Cast(value="uint32_t*") IntPointer out_type_size, @Const @ByPtrPtr IntPointer out_type_data, @Cast(value="uint32_t*") IntPointer aux_type_size, @Const @ByPtrPtr IntPointer aux_type_data, IntPointer complete)
public static int MXSymbolInferTypePartial(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Const IntBuffer arg_type_data, @Cast(value="uint32_t*") IntBuffer in_type_size, @Const @ByPtrPtr IntBuffer in_type_data, @Cast(value="uint32_t*") IntBuffer out_type_size, @Const @ByPtrPtr IntBuffer out_type_data, @Cast(value="uint32_t*") IntBuffer aux_type_size, @Const @ByPtrPtr IntBuffer aux_type_data, IntBuffer complete)
public static int MXSymbolInferTypePartial(SymbolHandle sym, @Cast(value="uint32_t") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Const int[] arg_type_data, @Cast(value="uint32_t*") int[] in_type_size, @Const @ByPtrPtr int[] in_type_data, @Cast(value="uint32_t*") int[] out_type_size, @Const @ByPtrPtr int[] out_type_data, @Cast(value="uint32_t*") int[] aux_type_size, @Const @ByPtrPtr int[] aux_type_data, int[] complete)
public static int MXQuantizeSymbol(SymbolHandle sym_handle, PointerPointer ret_sym_handle, @Const IntPointer dev_type, @Cast(value="const uint32_t") int num_excluded_sym_names, @Cast(value="const char**") PointerPointer excluded_sym_names, @Cast(value="const uint32_t") int num_excluded_op_names, @Cast(value="const char**") PointerPointer excluded_op_names, @Cast(value="const uint32_t") int num_offline, @Cast(value="const char**") PointerPointer offline_params, @Cast(value="const char*") BytePointer quantized_dtype, @Cast(value="const bool") boolean calib_quantize, @Cast(value="const char*") BytePointer quantize_mode, @Cast(value="const char*") BytePointer quantize_granularity, @Cast(value="uint32_t*") IntPointer out_num_calib_names, @Cast(value="const char***") @ByPtrPtr PointerPointer out_calib_names)
sym_handle
- symbol to be convertedret_sym_handle
- quantized symbol resultdev_type
- device typenum_excluded_sym_names
- number of layers excluded from being quantized in the input symbolexcluded_sym_names
- node names to be excluded from being quantizednum_excluded_op_names
- number of operators excluded from being quantized in the input symbolexcluded_op_names
- operator names to be excluded from being quantizednum_offline
- number of parameters that are quantized offlineoffline_params
- array of c strings representing the names of params quantized offlinequantized_dtype
- the quantized destination type for input datacalib_quantize
- **Deprecated**. quantize op will always be calibrated if couldquantize_mode
- quantize mode to be used in quantize passquantize_granularity
- quantize granularity, tensor-wise or channel-wiseout_num_calib_names
- return the number of nodes to be calibratedout_calib_names
- return the node names to be calibratedpublic static int MXQuantizeSymbol(SymbolHandle sym_handle, PointerPointer ret_sym_handle, @Const IntPointer dev_type, @Cast(value="const uint32_t") int num_excluded_sym_names, @Cast(value="const char**") @ByPtrPtr BytePointer excluded_sym_names, @Cast(value="const uint32_t") int num_excluded_op_names, @Cast(value="const char**") @ByPtrPtr BytePointer excluded_op_names, @Cast(value="const uint32_t") int num_offline, @Cast(value="const char**") @ByPtrPtr BytePointer offline_params, @Cast(value="const char*") BytePointer quantized_dtype, @Cast(value="const bool") boolean calib_quantize, @Cast(value="const char*") BytePointer quantize_mode, @Cast(value="const char*") BytePointer quantize_granularity, @Cast(value="uint32_t*") IntPointer out_num_calib_names, @Cast(value="const char***") @ByPtrPtr PointerPointer out_calib_names)
public static int MXQuantizeSymbol(SymbolHandle sym_handle, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle, @Const IntBuffer dev_type, @Cast(value="const uint32_t") int num_excluded_sym_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer excluded_sym_names, @Cast(value="const uint32_t") int num_excluded_op_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer excluded_op_names, @Cast(value="const uint32_t") int num_offline, @Cast(value="const char**") @ByPtrPtr ByteBuffer offline_params, String quantized_dtype, @Cast(value="const bool") boolean calib_quantize, String quantize_mode, String quantize_granularity, @Cast(value="uint32_t*") IntBuffer out_num_calib_names, @Cast(value="const char***") @ByPtrPtr PointerPointer out_calib_names)
public static int MXQuantizeSymbol(SymbolHandle sym_handle, PointerPointer ret_sym_handle, @Const int[] dev_type, @Cast(value="const uint32_t") int num_excluded_sym_names, @Cast(value="const char**") @ByPtrPtr byte[] excluded_sym_names, @Cast(value="const uint32_t") int num_excluded_op_names, @Cast(value="const char**") @ByPtrPtr byte[] excluded_op_names, @Cast(value="const uint32_t") int num_offline, @Cast(value="const char**") @ByPtrPtr byte[] offline_params, @Cast(value="const char*") BytePointer quantized_dtype, @Cast(value="const bool") boolean calib_quantize, @Cast(value="const char*") BytePointer quantize_mode, @Cast(value="const char*") BytePointer quantize_granularity, @Cast(value="uint32_t*") int[] out_num_calib_names, @Cast(value="const char***") @ByPtrPtr PointerPointer out_calib_names)
public static int MXQuantizeSymbol(SymbolHandle sym_handle, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle, @Const IntPointer dev_type, @Cast(value="const uint32_t") int num_excluded_sym_names, @Cast(value="const char**") @ByPtrPtr BytePointer excluded_sym_names, @Cast(value="const uint32_t") int num_excluded_op_names, @Cast(value="const char**") @ByPtrPtr BytePointer excluded_op_names, @Cast(value="const uint32_t") int num_offline, @Cast(value="const char**") @ByPtrPtr BytePointer offline_params, String quantized_dtype, @Cast(value="const bool") boolean calib_quantize, String quantize_mode, String quantize_granularity, @Cast(value="uint32_t*") IntPointer out_num_calib_names, @Cast(value="const char***") @ByPtrPtr PointerPointer out_calib_names)
public static int MXQuantizeSymbol(SymbolHandle sym_handle, PointerPointer ret_sym_handle, @Const IntBuffer dev_type, @Cast(value="const uint32_t") int num_excluded_sym_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer excluded_sym_names, @Cast(value="const uint32_t") int num_excluded_op_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer excluded_op_names, @Cast(value="const uint32_t") int num_offline, @Cast(value="const char**") @ByPtrPtr ByteBuffer offline_params, @Cast(value="const char*") BytePointer quantized_dtype, @Cast(value="const bool") boolean calib_quantize, @Cast(value="const char*") BytePointer quantize_mode, @Cast(value="const char*") BytePointer quantize_granularity, @Cast(value="uint32_t*") IntBuffer out_num_calib_names, @Cast(value="const char***") @ByPtrPtr PointerPointer out_calib_names)
public static int MXQuantizeSymbol(SymbolHandle sym_handle, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle, @Const int[] dev_type, @Cast(value="const uint32_t") int num_excluded_sym_names, @Cast(value="const char**") @ByPtrPtr byte[] excluded_sym_names, @Cast(value="const uint32_t") int num_excluded_op_names, @Cast(value="const char**") @ByPtrPtr byte[] excluded_op_names, @Cast(value="const uint32_t") int num_offline, @Cast(value="const char**") @ByPtrPtr byte[] offline_params, String quantized_dtype, @Cast(value="const bool") boolean calib_quantize, String quantize_mode, String quantize_granularity, @Cast(value="uint32_t*") int[] out_num_calib_names, @Cast(value="const char***") @ByPtrPtr PointerPointer out_calib_names)
public static int MXReducePrecisionSymbol(SymbolHandle sym_handle, PointerPointer ret_sym_handle, @Cast(value="uint32_t") int num_args, @Const IntPointer arg_type_data, @Cast(value="uint32_t") int num_ind_ptr, @Const IntPointer ind_ptr, @Const IntPointer target_dtype, int cast_optional_params, @Cast(value="const uint32_t") int num_target_dtype_op_names, @Cast(value="const uint32_t") int num_fp32_op_names, @Cast(value="const uint32_t") int num_widest_dtype_op_names, @Cast(value="const uint32_t") int num_conditional_fp32_op_names, @Cast(value="const uint32_t") int num_excluded_symbols, @Cast(value="const uint32_t") int num_model_params, @Cast(value="const char**") PointerPointer target_dtype_op_names, @Cast(value="const char**") PointerPointer fp32_op_names, @Cast(value="const char**") PointerPointer widest_dtype_op_names, @Cast(value="const char**") PointerPointer conditional_fp32_op_names, @Cast(value="const char**") PointerPointer excluded_symbols, @Cast(value="const char**") PointerPointer conditional_param_names, @Cast(value="const char**") PointerPointer conditional_param_vals, @Cast(value="const char**") PointerPointer model_param_names, @Cast(value="const char**") PointerPointer arg_names)
sym_handle
- symbol to be convertedret_sym_handle
- mixed precision symbol resultnum_args
- number of arguments for known dtypesarg_type_data
- arg types of the argumentstarget_dtype
- target_dtype for mixed precision symbolcast_optional_params
- whether to cast optional params to target_dtypenum_target_dtype_op_names
- number of ops to be casted to target_dtypenum_fp32_op_names
- number of ops to be casted to FP32num_widest_dtype_op_names
- number of ops to be casted to widest dtypenum_conditional_fp32_op_names
- number of ops to be casted to FP32 based on a conditionnum_excluded_symbols
- number of symbols to be excluded from castingnum_model_params
- number of model parametersnum_widest_dtype_op_names
- number of ops to be casted to the widest dtypenum_conditional_fp32_op_names
- number of ops to be cast to fp32 based on precisiontarget_dtype_op_names
- op names to be casted to target_dtypefp32_op_names
- op names to be casted to fp32widest_dtype_op_names
- names to be casted to widest dtypeconditional_fp32_op_names
- names to be casted to FP32 conditionallyexcluded_symbols
- symbol names to be excluded from castingparam_names
- param names for conditional FP32 castingparam_values
- param values for conditional FP32 castingarg_names
- argument names for which type information is providedmodel_param_names
- names for model parameterspublic static int MXReducePrecisionSymbol(SymbolHandle sym_handle, PointerPointer ret_sym_handle, @Cast(value="uint32_t") int num_args, @Const IntPointer arg_type_data, @Cast(value="uint32_t") int num_ind_ptr, @Const IntPointer ind_ptr, @Const IntPointer target_dtype, int cast_optional_params, @Cast(value="const uint32_t") int num_target_dtype_op_names, @Cast(value="const uint32_t") int num_fp32_op_names, @Cast(value="const uint32_t") int num_widest_dtype_op_names, @Cast(value="const uint32_t") int num_conditional_fp32_op_names, @Cast(value="const uint32_t") int num_excluded_symbols, @Cast(value="const uint32_t") int num_model_params, @Cast(value="const char**") @ByPtrPtr BytePointer target_dtype_op_names, @Cast(value="const char**") @ByPtrPtr BytePointer fp32_op_names, @Cast(value="const char**") @ByPtrPtr BytePointer widest_dtype_op_names, @Cast(value="const char**") @ByPtrPtr BytePointer conditional_fp32_op_names, @Cast(value="const char**") @ByPtrPtr BytePointer excluded_symbols, @Cast(value="const char**") @ByPtrPtr BytePointer conditional_param_names, @Cast(value="const char**") @ByPtrPtr BytePointer conditional_param_vals, @Cast(value="const char**") @ByPtrPtr BytePointer model_param_names, @Cast(value="const char**") @ByPtrPtr BytePointer arg_names)
public static int MXReducePrecisionSymbol(SymbolHandle sym_handle, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle, @Cast(value="uint32_t") int num_args, @Const IntBuffer arg_type_data, @Cast(value="uint32_t") int num_ind_ptr, @Const IntBuffer ind_ptr, @Const IntBuffer target_dtype, int cast_optional_params, @Cast(value="const uint32_t") int num_target_dtype_op_names, @Cast(value="const uint32_t") int num_fp32_op_names, @Cast(value="const uint32_t") int num_widest_dtype_op_names, @Cast(value="const uint32_t") int num_conditional_fp32_op_names, @Cast(value="const uint32_t") int num_excluded_symbols, @Cast(value="const uint32_t") int num_model_params, @Cast(value="const char**") @ByPtrPtr ByteBuffer target_dtype_op_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer fp32_op_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer widest_dtype_op_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer conditional_fp32_op_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer excluded_symbols, @Cast(value="const char**") @ByPtrPtr ByteBuffer conditional_param_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer conditional_param_vals, @Cast(value="const char**") @ByPtrPtr ByteBuffer model_param_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer arg_names)
public static int MXReducePrecisionSymbol(SymbolHandle sym_handle, PointerPointer ret_sym_handle, @Cast(value="uint32_t") int num_args, @Const int[] arg_type_data, @Cast(value="uint32_t") int num_ind_ptr, @Const int[] ind_ptr, @Const int[] target_dtype, int cast_optional_params, @Cast(value="const uint32_t") int num_target_dtype_op_names, @Cast(value="const uint32_t") int num_fp32_op_names, @Cast(value="const uint32_t") int num_widest_dtype_op_names, @Cast(value="const uint32_t") int num_conditional_fp32_op_names, @Cast(value="const uint32_t") int num_excluded_symbols, @Cast(value="const uint32_t") int num_model_params, @Cast(value="const char**") @ByPtrPtr byte[] target_dtype_op_names, @Cast(value="const char**") @ByPtrPtr byte[] fp32_op_names, @Cast(value="const char**") @ByPtrPtr byte[] widest_dtype_op_names, @Cast(value="const char**") @ByPtrPtr byte[] conditional_fp32_op_names, @Cast(value="const char**") @ByPtrPtr byte[] excluded_symbols, @Cast(value="const char**") @ByPtrPtr byte[] conditional_param_names, @Cast(value="const char**") @ByPtrPtr byte[] conditional_param_vals, @Cast(value="const char**") @ByPtrPtr byte[] model_param_names, @Cast(value="const char**") @ByPtrPtr byte[] arg_names)
public static int MXReducePrecisionSymbol(SymbolHandle sym_handle, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle, @Cast(value="uint32_t") int num_args, @Const IntPointer arg_type_data, @Cast(value="uint32_t") int num_ind_ptr, @Const IntPointer ind_ptr, @Const IntPointer target_dtype, int cast_optional_params, @Cast(value="const uint32_t") int num_target_dtype_op_names, @Cast(value="const uint32_t") int num_fp32_op_names, @Cast(value="const uint32_t") int num_widest_dtype_op_names, @Cast(value="const uint32_t") int num_conditional_fp32_op_names, @Cast(value="const uint32_t") int num_excluded_symbols, @Cast(value="const uint32_t") int num_model_params, @Cast(value="const char**") @ByPtrPtr BytePointer target_dtype_op_names, @Cast(value="const char**") @ByPtrPtr BytePointer fp32_op_names, @Cast(value="const char**") @ByPtrPtr BytePointer widest_dtype_op_names, @Cast(value="const char**") @ByPtrPtr BytePointer conditional_fp32_op_names, @Cast(value="const char**") @ByPtrPtr BytePointer excluded_symbols, @Cast(value="const char**") @ByPtrPtr BytePointer conditional_param_names, @Cast(value="const char**") @ByPtrPtr BytePointer conditional_param_vals, @Cast(value="const char**") @ByPtrPtr BytePointer model_param_names, @Cast(value="const char**") @ByPtrPtr BytePointer arg_names)
public static int MXReducePrecisionSymbol(SymbolHandle sym_handle, PointerPointer ret_sym_handle, @Cast(value="uint32_t") int num_args, @Const IntBuffer arg_type_data, @Cast(value="uint32_t") int num_ind_ptr, @Const IntBuffer ind_ptr, @Const IntBuffer target_dtype, int cast_optional_params, @Cast(value="const uint32_t") int num_target_dtype_op_names, @Cast(value="const uint32_t") int num_fp32_op_names, @Cast(value="const uint32_t") int num_widest_dtype_op_names, @Cast(value="const uint32_t") int num_conditional_fp32_op_names, @Cast(value="const uint32_t") int num_excluded_symbols, @Cast(value="const uint32_t") int num_model_params, @Cast(value="const char**") @ByPtrPtr ByteBuffer target_dtype_op_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer fp32_op_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer widest_dtype_op_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer conditional_fp32_op_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer excluded_symbols, @Cast(value="const char**") @ByPtrPtr ByteBuffer conditional_param_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer conditional_param_vals, @Cast(value="const char**") @ByPtrPtr ByteBuffer model_param_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer arg_names)
public static int MXReducePrecisionSymbol(SymbolHandle sym_handle, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle, @Cast(value="uint32_t") int num_args, @Const int[] arg_type_data, @Cast(value="uint32_t") int num_ind_ptr, @Const int[] ind_ptr, @Const int[] target_dtype, int cast_optional_params, @Cast(value="const uint32_t") int num_target_dtype_op_names, @Cast(value="const uint32_t") int num_fp32_op_names, @Cast(value="const uint32_t") int num_widest_dtype_op_names, @Cast(value="const uint32_t") int num_conditional_fp32_op_names, @Cast(value="const uint32_t") int num_excluded_symbols, @Cast(value="const uint32_t") int num_model_params, @Cast(value="const char**") @ByPtrPtr byte[] target_dtype_op_names, @Cast(value="const char**") @ByPtrPtr byte[] fp32_op_names, @Cast(value="const char**") @ByPtrPtr byte[] widest_dtype_op_names, @Cast(value="const char**") @ByPtrPtr byte[] conditional_fp32_op_names, @Cast(value="const char**") @ByPtrPtr byte[] excluded_symbols, @Cast(value="const char**") @ByPtrPtr byte[] conditional_param_names, @Cast(value="const char**") @ByPtrPtr byte[] conditional_param_vals, @Cast(value="const char**") @ByPtrPtr byte[] model_param_names, @Cast(value="const char**") @ByPtrPtr byte[] arg_names)
public static int MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle, @Cast(value="const uint32_t") int num_layers, @Cast(value="const char**") PointerPointer layer_names, @Const FloatPointer low_quantiles, @Const FloatPointer high_quantiles, PointerPointer ret_sym_handle)
sym_handle
- symbol whose node attributes are to be set by calibration tablenum_layers
- number of layers in the calibration tablelayer
- names stored as keys in the calibration tablelow_quantiles
- low quantiles of layers stored in the calibration tablehigh_quantiles
- high quantiles of layers stored in the calibration tableret_sym_handle
- returned symbolpublic static int MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle, @Cast(value="const uint32_t") int num_layers, @Cast(value="const char**") @ByPtrPtr BytePointer layer_names, @Const FloatPointer low_quantiles, @Const FloatPointer high_quantiles, PointerPointer ret_sym_handle)
public static int MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle, @Cast(value="const uint32_t") int num_layers, @Cast(value="const char**") @ByPtrPtr ByteBuffer layer_names, @Const FloatBuffer low_quantiles, @Const FloatBuffer high_quantiles, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle)
public static int MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle, @Cast(value="const uint32_t") int num_layers, @Cast(value="const char**") @ByPtrPtr byte[] layer_names, @Const float[] low_quantiles, @Const float[] high_quantiles, PointerPointer ret_sym_handle)
public static int MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle, @Cast(value="const uint32_t") int num_layers, @Cast(value="const char**") @ByPtrPtr BytePointer layer_names, @Const FloatPointer low_quantiles, @Const FloatPointer high_quantiles, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle)
public static int MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle, @Cast(value="const uint32_t") int num_layers, @Cast(value="const char**") @ByPtrPtr ByteBuffer layer_names, @Const FloatBuffer low_quantiles, @Const FloatBuffer high_quantiles, PointerPointer ret_sym_handle)
public static int MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle, @Cast(value="const uint32_t") int num_layers, @Cast(value="const char**") @ByPtrPtr byte[] layer_names, @Const float[] low_quantiles, @Const float[] high_quantiles, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle)
public static int MXGenBackendSubgraph(SymbolHandle sym_handle, @Cast(value="const char*") BytePointer backend, PointerPointer ret_sym_handle)
sym_handle
- symbol to be convertedbackend
- backend names for subgraph passret_sym_handle
- returned symbolpublic static int MXGenBackendSubgraph(SymbolHandle sym_handle, String backend, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle)
public static int MXGenAtomicSymbolFromSymbol(SymbolHandle sym_handle, PointerPointer ret_sym_handle)
sym_handle
- source symbolret_sym_handle
- returned atomic symbolpublic static int MXGenAtomicSymbolFromSymbol(SymbolHandle sym_handle, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle)
public static int MXOptimizeForBackend(SymbolHandle sym_handle, @Cast(value="const char*") BytePointer backend_name, int dev_type, PointerPointer ret_sym_handle, @Cast(value="const mx_uint") int args_len, PointerPointer in_args_handle, @Cast(value="const mx_uint") int aux_len, PointerPointer in_aux_handle, @Cast(value="const mx_uint") int num_options, @Cast(value="const char**") PointerPointer keys, @Cast(value="const char**") PointerPointer vals, @Cast(value="const uint32_t") int num_input_shapes, @Cast(value="const char**") PointerPointer input_shape_names, @Cast(value="const int64_t*") LongPointer input_shape_data, @Cast(value="const uint32_t*") IntPointer input_shape_idx, @Cast(value="const uint32_t") int num_input_dtypes, @Cast(value="const char**") PointerPointer input_dtype_names, @Const IntPointer input_dtypes, @Cast(value="const uint32_t") int num_input_stypes, @Cast(value="const char**") PointerPointer input_stype_names, @Const IntPointer input_stypes, @Cast(value="bool") boolean skip_infer, IntPointer new_args_cnt, @ByPtrPtr PointerPointer new_args_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_arg_names_handle, IntPointer new_aux_cnt, @ByPtrPtr PointerPointer new_aux_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_aux_names_handle)
sym_handle
- symbol to be partitioneddev_type
- context device typebackend_name
- backend nameret_sym_handle
- partitioned symbol returnedlen
- number of argsin_args_handle
- args arraynum_options
- number of key value pairskeys
- keys for optionsvals
- values corresponding to keysnum_input_shapes
- number of input shapesinput_shape_names
- names of the input shapesinput_shape_data
- pointer to the contiguous data shapesinput_shape_idx
- array of per shape starting idx, the shape length for the i-th input shape
is calculate as input_shape_idx[i+1] - input_shape_idx[i]num_input_dtypes
- number of input data typesinput_dtype_names
- array of names of the input data typesinput_dtypes
- array of values of the input data typesnum_input_stypesnumber
- of input storage typesinput_stype_names
- array of names of the input storage typesinput_stypes
- array of values of input storage typesskip_infer
- if the optimization should skip the attribute inferences
(to use if the backend does not require shape inference)new_args_cnt
- pointer a number to store the number of new argsnew_args_handle
- pointer on array to store the new args handlesnew_arg_names_handle
- pointer on array to store the new args namesnew_aux_cnt
- pointer a number to store the number of new auxnew_aux_handle
- pointer on array to store the new aux handlesnew_aux_names_handle
- pointer on array to store the new aux namespublic static int MXOptimizeForBackend(SymbolHandle sym_handle, @Cast(value="const char*") BytePointer backend_name, int dev_type, PointerPointer ret_sym_handle, @Cast(value="const mx_uint") int args_len, PointerPointer in_args_handle, @Cast(value="const mx_uint") int aux_len, PointerPointer in_aux_handle, @Cast(value="const mx_uint") int num_options, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, @Cast(value="const uint32_t") int num_input_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer input_shape_names, @Cast(value="const int64_t*") LongPointer input_shape_data, @Cast(value="const uint32_t*") IntPointer input_shape_idx, @Cast(value="const uint32_t") int num_input_dtypes, @Cast(value="const char**") @ByPtrPtr BytePointer input_dtype_names, @Const IntPointer input_dtypes, @Cast(value="const uint32_t") int num_input_stypes, @Cast(value="const char**") @ByPtrPtr BytePointer input_stype_names, @Const IntPointer input_stypes, @Cast(value="bool") boolean skip_infer, IntPointer new_args_cnt, @ByPtrPtr PointerPointer new_args_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_arg_names_handle, IntPointer new_aux_cnt, @ByPtrPtr PointerPointer new_aux_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_aux_names_handle)
public static int MXOptimizeForBackend(SymbolHandle sym_handle, String backend_name, int dev_type, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle, @Cast(value="const mx_uint") int args_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args_handle, @Cast(value="const mx_uint") int aux_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_aux_handle, @Cast(value="const mx_uint") int num_options, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, @Cast(value="const uint32_t") int num_input_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_shape_names, @Cast(value="const int64_t*") LongBuffer input_shape_data, @Cast(value="const uint32_t*") IntBuffer input_shape_idx, @Cast(value="const uint32_t") int num_input_dtypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_dtype_names, @Const IntBuffer input_dtypes, @Cast(value="const uint32_t") int num_input_stypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_stype_names, @Const IntBuffer input_stypes, @Cast(value="bool") boolean skip_infer, IntBuffer new_args_cnt, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer new_args_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_arg_names_handle, IntBuffer new_aux_cnt, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer new_aux_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_aux_names_handle)
public static int MXOptimizeForBackend(SymbolHandle sym_handle, @Cast(value="const char*") BytePointer backend_name, int dev_type, PointerPointer ret_sym_handle, @Cast(value="const mx_uint") int args_len, PointerPointer in_args_handle, @Cast(value="const mx_uint") int aux_len, PointerPointer in_aux_handle, @Cast(value="const mx_uint") int num_options, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, @Cast(value="const uint32_t") int num_input_shapes, @Cast(value="const char**") @ByPtrPtr byte[] input_shape_names, @Cast(value="const int64_t*") long[] input_shape_data, @Cast(value="const uint32_t*") int[] input_shape_idx, @Cast(value="const uint32_t") int num_input_dtypes, @Cast(value="const char**") @ByPtrPtr byte[] input_dtype_names, @Const int[] input_dtypes, @Cast(value="const uint32_t") int num_input_stypes, @Cast(value="const char**") @ByPtrPtr byte[] input_stype_names, @Const int[] input_stypes, @Cast(value="bool") boolean skip_infer, int[] new_args_cnt, @ByPtrPtr PointerPointer new_args_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_arg_names_handle, int[] new_aux_cnt, @ByPtrPtr PointerPointer new_aux_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_aux_names_handle)
public static int MXOptimizeForBackend(SymbolHandle sym_handle, String backend_name, int dev_type, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle, @Cast(value="const mx_uint") int args_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args_handle, @Cast(value="const mx_uint") int aux_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_aux_handle, @Cast(value="const mx_uint") int num_options, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, @Cast(value="const uint32_t") int num_input_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer input_shape_names, @Cast(value="const int64_t*") LongPointer input_shape_data, @Cast(value="const uint32_t*") IntPointer input_shape_idx, @Cast(value="const uint32_t") int num_input_dtypes, @Cast(value="const char**") @ByPtrPtr BytePointer input_dtype_names, @Const IntPointer input_dtypes, @Cast(value="const uint32_t") int num_input_stypes, @Cast(value="const char**") @ByPtrPtr BytePointer input_stype_names, @Const IntPointer input_stypes, @Cast(value="bool") boolean skip_infer, IntPointer new_args_cnt, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer new_args_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_arg_names_handle, IntPointer new_aux_cnt, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer new_aux_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_aux_names_handle)
public static int MXOptimizeForBackend(SymbolHandle sym_handle, @Cast(value="const char*") BytePointer backend_name, int dev_type, PointerPointer ret_sym_handle, @Cast(value="const mx_uint") int args_len, PointerPointer in_args_handle, @Cast(value="const mx_uint") int aux_len, PointerPointer in_aux_handle, @Cast(value="const mx_uint") int num_options, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, @Cast(value="const uint32_t") int num_input_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_shape_names, @Cast(value="const int64_t*") LongBuffer input_shape_data, @Cast(value="const uint32_t*") IntBuffer input_shape_idx, @Cast(value="const uint32_t") int num_input_dtypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_dtype_names, @Const IntBuffer input_dtypes, @Cast(value="const uint32_t") int num_input_stypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_stype_names, @Const IntBuffer input_stypes, @Cast(value="bool") boolean skip_infer, IntBuffer new_args_cnt, @ByPtrPtr PointerPointer new_args_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_arg_names_handle, IntBuffer new_aux_cnt, @ByPtrPtr PointerPointer new_aux_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_aux_names_handle)
public static int MXOptimizeForBackend(SymbolHandle sym_handle, String backend_name, int dev_type, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle ret_sym_handle, @Cast(value="const mx_uint") int args_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args_handle, @Cast(value="const mx_uint") int aux_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_aux_handle, @Cast(value="const mx_uint") int num_options, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, @Cast(value="const uint32_t") int num_input_shapes, @Cast(value="const char**") @ByPtrPtr byte[] input_shape_names, @Cast(value="const int64_t*") long[] input_shape_data, @Cast(value="const uint32_t*") int[] input_shape_idx, @Cast(value="const uint32_t") int num_input_dtypes, @Cast(value="const char**") @ByPtrPtr byte[] input_dtype_names, @Const int[] input_dtypes, @Cast(value="const uint32_t") int num_input_stypes, @Cast(value="const char**") @ByPtrPtr byte[] input_stype_names, @Const int[] input_stypes, @Cast(value="bool") boolean skip_infer, int[] new_args_cnt, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer new_args_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_arg_names_handle, int[] new_aux_cnt, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer new_aux_handle, @Cast(value="char***") @ByPtrPtr PointerPointer new_aux_names_handle)
public static int MXExecutorFree(ExecutorHandle handle)
handle
- the executor.public static int MXExecutorPrint(ExecutorHandle handle, @Cast(value="const char**") PointerPointer out_str)
handle
- the executor.out_str
- pointer to hold the output string of the printing.public static int MXExecutorPrint(ExecutorHandle handle, @Cast(value="const char**") @ByPtrPtr BytePointer out_str)
public static int MXExecutorPrint(ExecutorHandle handle, @Cast(value="const char**") @ByPtrPtr ByteBuffer out_str)
public static int MXExecutorPrint(ExecutorHandle handle, @Cast(value="const char**") @ByPtrPtr byte[] out_str)
public static int MXExecutorForward(ExecutorHandle handle, int is_train)
handle
- executor handleis_train
- int value to indicate whether the forward pass is for evaluationpublic static int MXExecutorBackward(ExecutorHandle handle, @Cast(value="uint32_t") int len, PointerPointer head_grads)
handle
- execute handlelen
- lenthhead_grads
- NDArray handle for heads' gradientpublic static int MXExecutorBackward(ExecutorHandle handle, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle head_grads)
public static int MXExecutorBackwardEx(ExecutorHandle handle, @Cast(value="uint32_t") int len, PointerPointer head_grads, int is_train)
handle
- execute handlelen
- lenthhead_grads
- NDArray handle for heads' gradientis_train
- int value to indicate whether the backward pass is for evaluationpublic static int MXExecutorBackwardEx(ExecutorHandle handle, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle head_grads, int is_train)
public static int MXExecutorOutputs(ExecutorHandle handle, @Cast(value="uint32_t*") IntPointer out_size, @ByPtrPtr PointerPointer out)
handle
- executor handleout_size
- output narray vector sizeout
- out put narray handlespublic static int MXExecutorOutputs(ExecutorHandle handle, @Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer out)
public static int MXExecutorOutputs(ExecutorHandle handle, @Cast(value="uint32_t*") int[] out_size, @ByPtrPtr PointerPointer out)
public static int MXExecutorBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") IntPointer grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, PointerPointer out)
symbol_handle
- symbol handledev_type
- device typedev_id
- device idlen
- lengthin_args
- in args arrayarg_grad_store
- arg grads handle arraygrad_req_type
- grad req arrayaux_states_len
- length of auxiliary statesaux_states
- auxiliary states arrayout
- output executor handlepublic static int MXExecutorBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle arg_grad_store, @Cast(value="uint32_t*") IntBuffer grad_req_type, @Cast(value="uint32_t") int aux_states_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle aux_states, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") int[] grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, PointerPointer out)
public static int MXExecutorBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle arg_grad_store, @Cast(value="uint32_t*") IntPointer grad_req_type, @Cast(value="uint32_t") int aux_states_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle aux_states, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") IntBuffer grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, PointerPointer out)
public static int MXExecutorBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle arg_grad_store, @Cast(value="uint32_t*") int[] grad_req_type, @Cast(value="uint32_t") int aux_states_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle aux_states, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorBindX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") PointerPointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") IntPointer grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, PointerPointer out)
symbol_handle
- symbol handledev_type
- device type of default contextdev_id
- device id of default contextnum_map_keys
- size of group2ctx mapmap_keys
- keys of group2ctx mapmap_dev_types
- device type of group2ctx mapmap_dev_ids
- device id of group2ctx maplen
- lengthin_args
- in args arrayarg_grad_store
- arg grads handle arraygrad_req_type
- grad req arrayaux_states_len
- length of auxiliary statesaux_states
- auxiliary states arrayout
- output executor handlepublic static int MXExecutorBindX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr BytePointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") IntPointer grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, PointerPointer out)
public static int MXExecutorBindX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer map_keys, @Const IntBuffer map_dev_types, @Const IntBuffer map_dev_ids, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle arg_grad_store, @Cast(value="uint32_t*") IntBuffer grad_req_type, @Cast(value="uint32_t") int aux_states_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle aux_states, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorBindX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr byte[] map_keys, @Const int[] map_dev_types, @Const int[] map_dev_ids, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") int[] grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, PointerPointer out)
public static int MXExecutorBindX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr BytePointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle arg_grad_store, @Cast(value="uint32_t*") IntPointer grad_req_type, @Cast(value="uint32_t") int aux_states_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle aux_states, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorBindX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer map_keys, @Const IntBuffer map_dev_types, @Const IntBuffer map_dev_ids, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") IntBuffer grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, PointerPointer out)
public static int MXExecutorBindX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr byte[] map_keys, @Const int[] map_dev_types, @Const int[] map_dev_ids, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle arg_grad_store, @Cast(value="uint32_t*") int[] grad_req_type, @Cast(value="uint32_t") int aux_states_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle aux_states, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorBindEX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") PointerPointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") IntPointer grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
symbol_handle
- symbol handledev_type
- device type of default contextdev_id
- device id of default contextnum_map_keys
- size of group2ctx mapmap_keys
- keys of group2ctx mapmap_dev_types
- device type of group2ctx mapmap_dev_ids
- device id of group2ctx maplen
- lengthin_args
- in args arrayarg_grad_store
- arg grads handle arraygrad_req_type
- grad req arrayaux_states_len
- length of auxiliary statesaux_states
- auxiliary states arrayshared_exec
- input executor handle for memory sharingout
- output executor handlepublic static int MXExecutorBindEX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr BytePointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") IntPointer grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
public static int MXExecutorBindEX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer map_keys, @Const IntBuffer map_dev_types, @Const IntBuffer map_dev_ids, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle arg_grad_store, @Cast(value="uint32_t*") IntBuffer grad_req_type, @Cast(value="uint32_t") int aux_states_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle aux_states, ExecutorHandle shared_exec, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorBindEX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr byte[] map_keys, @Const int[] map_dev_types, @Const int[] map_dev_ids, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") int[] grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
public static int MXExecutorBindEX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr BytePointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle arg_grad_store, @Cast(value="uint32_t*") IntPointer grad_req_type, @Cast(value="uint32_t") int aux_states_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle aux_states, ExecutorHandle shared_exec, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorBindEX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer map_keys, @Const IntBuffer map_dev_types, @Const IntBuffer map_dev_ids, @Cast(value="uint32_t") int len, PointerPointer in_args, PointerPointer arg_grad_store, @Cast(value="uint32_t*") IntBuffer grad_req_type, @Cast(value="uint32_t") int aux_states_len, PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
public static int MXExecutorBindEX(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr byte[] map_keys, @Const int[] map_dev_types, @Const int[] map_dev_ids, @Cast(value="uint32_t") int len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle in_args, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle arg_grad_store, @Cast(value="uint32_t*") int[] grad_req_type, @Cast(value="uint32_t") int aux_states_len, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle aux_states, ExecutorHandle shared_exec, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorSimpleBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") PointerPointer g2c_keys, @Const IntPointer g2c_dev_types, @Const IntPointer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") PointerPointer provided_grad_req_names, @Cast(value="const char**") PointerPointer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") PointerPointer provided_arg_shape_names, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") PointerPointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") PointerPointer provided_arg_stype_names, @Const IntPointer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") PointerPointer shared_arg_name_list, IntPointer shared_buffer_len, @Cast(value="const char**") PointerPointer shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntPointer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr BytePointer g2c_keys, @Const IntPointer g2c_dev_types, @Const IntPointer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_shape_names, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_stype_names, @Const IntPointer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr BytePointer shared_arg_name_list, IntPointer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr BytePointer shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntPointer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer g2c_keys, @Const IntBuffer g2c_dev_types, @Const IntBuffer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_shape_names, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_data, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_dtype_names, @Const IntBuffer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_stype_names, @Const IntBuffer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_arg_name_list, IntBuffer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_buffer_name_list, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntBuffer num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntBuffer num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorSimpleBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr byte[] g2c_keys, @Const int[] g2c_dev_types, @Const int[] g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_shape_names, @Cast(value="const uint32_t*") int[] provided_arg_shape_data, @Cast(value="const uint32_t*") int[] provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_dtype_names, @Const int[] provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_stype_names, @Const int[] provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr byte[] shared_arg_name_list, int[] shared_buffer_len, @Cast(value="const char**") @ByPtrPtr byte[] shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") int[] num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") int[] num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr BytePointer g2c_keys, @Const IntPointer g2c_dev_types, @Const IntPointer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_shape_names, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_stype_names, @Const IntPointer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr BytePointer shared_arg_name_list, IntPointer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr BytePointer shared_buffer_name_list, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntPointer num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorSimpleBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer g2c_keys, @Const IntBuffer g2c_dev_types, @Const IntBuffer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_shape_names, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_data, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_dtype_names, @Const IntBuffer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_stype_names, @Const IntBuffer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_arg_name_list, IntBuffer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntBuffer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntBuffer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBind(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr byte[] g2c_keys, @Const int[] g2c_dev_types, @Const int[] g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_shape_names, @Cast(value="const uint32_t*") int[] provided_arg_shape_data, @Cast(value="const uint32_t*") int[] provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_dtype_names, @Const int[] provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_stype_names, @Const int[] provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr byte[] shared_arg_name_list, int[] shared_buffer_len, @Cast(value="const char**") @ByPtrPtr byte[] shared_buffer_name_list, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") int[] num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") int[] num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorSimpleBindEx(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") PointerPointer g2c_keys, @Const IntPointer g2c_dev_types, @Const IntPointer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") PointerPointer provided_grad_req_names, @Cast(value="const char**") PointerPointer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") PointerPointer provided_arg_shape_names, @Const IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") PointerPointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") PointerPointer provided_arg_stype_names, @Const IntPointer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") PointerPointer shared_arg_name_list, IntPointer shared_buffer_len, @Cast(value="const char**") PointerPointer shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntPointer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBindEx(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr BytePointer g2c_keys, @Const IntPointer g2c_dev_types, @Const IntPointer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_shape_names, @Const IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_stype_names, @Const IntPointer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr BytePointer shared_arg_name_list, IntPointer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr BytePointer shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntPointer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBindEx(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer g2c_keys, @Const IntBuffer g2c_dev_types, @Const IntBuffer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_shape_names, @Const IntBuffer provided_arg_shape_data, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_dtype_names, @Const IntBuffer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_stype_names, @Const IntBuffer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_arg_name_list, IntBuffer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_buffer_name_list, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntBuffer num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntBuffer num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorSimpleBindEx(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr byte[] g2c_keys, @Const int[] g2c_dev_types, @Const int[] g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_shape_names, @Const int[] provided_arg_shape_data, @Cast(value="const uint32_t*") int[] provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_dtype_names, @Const int[] provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_stype_names, @Const int[] provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr byte[] shared_arg_name_list, int[] shared_buffer_len, @Cast(value="const char**") @ByPtrPtr byte[] shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") int[] num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") int[] num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBindEx(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr BytePointer g2c_keys, @Const IntPointer g2c_dev_types, @Const IntPointer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_shape_names, @Const IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_stype_names, @Const IntPointer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr BytePointer shared_arg_name_list, IntPointer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr BytePointer shared_buffer_name_list, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntPointer num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorSimpleBindEx(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer g2c_keys, @Const IntBuffer g2c_dev_types, @Const IntBuffer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_shape_names, @Const IntBuffer provided_arg_shape_data, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_dtype_names, @Const IntBuffer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_stype_names, @Const IntBuffer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_arg_name_list, IntBuffer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntBuffer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntBuffer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBindEx(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr byte[] g2c_keys, @Const int[] g2c_dev_types, @Const int[] g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_shape_names, @Const int[] provided_arg_shape_data, @Cast(value="const uint32_t*") int[] provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_dtype_names, @Const int[] provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_stype_names, @Const int[] provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr byte[] shared_arg_name_list, int[] shared_buffer_len, @Cast(value="const char**") @ByPtrPtr byte[] shared_buffer_name_list, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") int[] num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") int[] num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorSimpleBindEx64(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") PointerPointer g2c_keys, @Const IntPointer g2c_dev_types, @Const IntPointer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") PointerPointer provided_grad_req_names, @Cast(value="const char**") PointerPointer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") PointerPointer provided_arg_shape_names, @Cast(value="const int64_t*") LongPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") PointerPointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") PointerPointer provided_arg_stype_names, @Const IntPointer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") PointerPointer shared_arg_name_list, IntPointer shared_buffer_len, @Cast(value="const char**") PointerPointer shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntPointer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBindEx64(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr BytePointer g2c_keys, @Const IntPointer g2c_dev_types, @Const IntPointer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_shape_names, @Cast(value="const int64_t*") LongPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_stype_names, @Const IntPointer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr BytePointer shared_arg_name_list, IntPointer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr BytePointer shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntPointer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBindEx64(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer g2c_keys, @Const IntBuffer g2c_dev_types, @Const IntBuffer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_shape_names, @Cast(value="const int64_t*") LongBuffer provided_arg_shape_data, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_dtype_names, @Const IntBuffer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_stype_names, @Const IntBuffer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_arg_name_list, IntBuffer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_buffer_name_list, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntBuffer num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntBuffer num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorSimpleBindEx64(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr byte[] g2c_keys, @Const int[] g2c_dev_types, @Const int[] g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_shape_names, @Cast(value="const int64_t*") long[] provided_arg_shape_data, @Cast(value="const uint32_t*") int[] provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_dtype_names, @Const int[] provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_stype_names, @Const int[] provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr byte[] shared_arg_name_list, int[] shared_buffer_len, @Cast(value="const char**") @ByPtrPtr byte[] shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") int[] num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") int[] num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBindEx64(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr BytePointer g2c_keys, @Const IntPointer g2c_dev_types, @Const IntPointer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr BytePointer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_shape_names, @Cast(value="const int64_t*") LongPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_stype_names, @Const IntPointer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr BytePointer shared_arg_name_list, IntPointer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr BytePointer shared_buffer_name_list, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntPointer num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorSimpleBindEx64(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer g2c_keys, @Const IntBuffer g2c_dev_types, @Const IntBuffer g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_shape_names, @Cast(value="const int64_t*") LongBuffer provided_arg_shape_data, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_dtype_names, @Const IntBuffer provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_stype_names, @Const IntBuffer provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_arg_name_list, IntBuffer shared_buffer_len, @Cast(value="const char**") @ByPtrPtr ByteBuffer shared_buffer_name_list, PointerPointer shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") IntBuffer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntBuffer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, PointerPointer out)
public static int MXExecutorSimpleBindEx64(SymbolHandle symbol_handle, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_g2c_keys, @Cast(value="const char**") @ByPtrPtr byte[] g2c_keys, @Const int[] g2c_dev_types, @Const int[] g2c_dev_ids, @Cast(value="const uint32_t") int provided_grad_req_list_len, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_names, @Cast(value="const char**") @ByPtrPtr byte[] provided_grad_req_types, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_shape_names, @Cast(value="const int64_t*") long[] provided_arg_shape_data, @Cast(value="const uint32_t*") int[] provided_arg_shape_idx, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_dtype_names, @Const int[] provided_arg_dtypes, @Cast(value="const uint32_t") int num_provided_arg_stypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_stype_names, @Const int[] provided_arg_stypes, @Cast(value="const uint32_t") int num_shared_arg_names, @Cast(value="const char**") @ByPtrPtr byte[] shared_arg_name_list, int[] shared_buffer_len, @Cast(value="const char**") @ByPtrPtr byte[] shared_buffer_name_list, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle shared_buffer_handle_list, @Cast(value="const char***") @ByPtrPtr PointerPointer updated_shared_buffer_name_list, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer updated_shared_buffer_handle_list, @Cast(value="uint32_t*") int[] num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") int[] num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec_handle, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorReshape(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") PointerPointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") PointerPointer provided_arg_shape_names, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="uint32_t*") IntPointer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
partial_shaping
- Whether to allow changing the shape of unspecified arguments.allow_up_sizing
- Whether to allow allocating new ndarrays that's larger than the original.dev_type
- device type of default contextdev_id
- device id of default contextnum_map_keys
- size of group2ctx mapmap_keys
- keys of group2ctx mapmap_dev_types
- device type of group2ctx mapmap_dev_ids
- device id of group2ctx mapnum_in_args
- length of in_argsin_args
- in args arrayarg_grads
- arg grads handle arraynum_aux_states
- length of auxiliary statesaux_states
- auxiliary states arrayshared_exec
- input executor handle for memory sharingout
- output executor handlepublic static int MXExecutorReshape(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr BytePointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_shape_names, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="uint32_t*") IntPointer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
public static int MXExecutorReshape(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer map_keys, @Const IntBuffer map_dev_types, @Const IntBuffer map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_shape_names, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_data, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_idx, @Cast(value="uint32_t*") IntBuffer num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntBuffer num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorReshape(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr byte[] map_keys, @Const int[] map_dev_types, @Const int[] map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_shape_names, @Cast(value="const uint32_t*") int[] provided_arg_shape_data, @Cast(value="const uint32_t*") int[] provided_arg_shape_idx, @Cast(value="uint32_t*") int[] num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") int[] num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
public static int MXExecutorReshape(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr BytePointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_shape_names, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="uint32_t*") IntPointer num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorReshape(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer map_keys, @Const IntBuffer map_dev_types, @Const IntBuffer map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_shape_names, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_data, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_idx, @Cast(value="uint32_t*") IntBuffer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntBuffer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
public static int MXExecutorReshape(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr byte[] map_keys, @Const int[] map_dev_types, @Const int[] map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_shape_names, @Cast(value="const uint32_t*") int[] provided_arg_shape_data, @Cast(value="const uint32_t*") int[] provided_arg_shape_idx, @Cast(value="uint32_t*") int[] num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") int[] num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorReshapeEx(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") PointerPointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") PointerPointer provided_arg_shape_names, @Const IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="uint32_t*") IntPointer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
partial_shaping
- Whether to allow changing the shape of unspecified arguments.allow_up_sizing
- Whether to allow allocating new ndarrays that's larger than the original.dev_type
- device type of default contextdev_id
- device id of default contextnum_map_keys
- size of group2ctx mapmap_keys
- keys of group2ctx mapmap_dev_types
- device type of group2ctx mapmap_dev_ids
- device id of group2ctx mapnum_in_args
- length of in_argsin_args
- in args arrayarg_grads
- arg grads handle arraynum_aux_states
- length of auxiliary statesaux_states
- auxiliary states arrayshared_exec
- input executor handle for memory sharingout
- output executor handlepublic static int MXExecutorReshapeEx(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr BytePointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_shape_names, @Const IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="uint32_t*") IntPointer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
public static int MXExecutorReshapeEx(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer map_keys, @Const IntBuffer map_dev_types, @Const IntBuffer map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_shape_names, @Const IntBuffer provided_arg_shape_data, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_idx, @Cast(value="uint32_t*") IntBuffer num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntBuffer num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorReshapeEx(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr byte[] map_keys, @Const int[] map_dev_types, @Const int[] map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_shape_names, @Const int[] provided_arg_shape_data, @Cast(value="const uint32_t*") int[] provided_arg_shape_idx, @Cast(value="uint32_t*") int[] num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") int[] num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
public static int MXExecutorReshapeEx(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr BytePointer map_keys, @Const IntPointer map_dev_types, @Const IntPointer map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_shape_names, @Const IntPointer provided_arg_shape_data, @Cast(value="const uint32_t*") IntPointer provided_arg_shape_idx, @Cast(value="uint32_t*") IntPointer num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntPointer num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorReshapeEx(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer map_keys, @Const IntBuffer map_dev_types, @Const IntBuffer map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_shape_names, @Const IntBuffer provided_arg_shape_data, @Cast(value="const uint32_t*") IntBuffer provided_arg_shape_idx, @Cast(value="uint32_t*") IntBuffer num_in_args, @ByPtrPtr PointerPointer in_args, @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") IntBuffer num_aux_states, @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, PointerPointer out)
public static int MXExecutorReshapeEx(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, @Cast(value="uint32_t") int num_map_keys, @Cast(value="const char**") @ByPtrPtr byte[] map_keys, @Const int[] map_dev_types, @Const int[] map_dev_ids, @Cast(value="const uint32_t") int num_provided_arg_shapes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_shape_names, @Const int[] provided_arg_shape_data, @Cast(value="const uint32_t*") int[] provided_arg_shape_idx, @Cast(value="uint32_t*") int[] num_in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer in_args, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer arg_grads, @Cast(value="uint32_t*") int[] num_aux_states, @Cast(value="NDArrayHandle**") @ByPtrPtr PointerPointer aux_states, ExecutorHandle shared_exec, @Cast(value="ExecutorHandle*") @ByPtrPtr ExecutorHandle out)
public static int MXExecutorGetOptimizedSymbol(ExecutorHandle handle, PointerPointer out)
public static int MXExecutorGetOptimizedSymbol(ExecutorHandle handle, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXExecutorSetMonitorCallback(ExecutorHandle handle, ExecutorMonitorCallback callback, Pointer callback_handle)
public static int MXExecutorSetMonitorCallbackEX(ExecutorHandle handle, ExecutorMonitorCallback callback, Pointer callback_handle, @Cast(value="bool") boolean monitor_all)
monitor_all
- If true, monitor both input and output, otherwise monitor output only.public static int MXListDataIters(@Cast(value="uint32_t*") IntPointer out_size, @ByPtrPtr PointerPointer out_array)
out_size
- the size of returned iteratorsout_array
- the output iteratos entriespublic static int MXListDataIters(@Cast(value="uint32_t*") IntBuffer out_size, @Cast(value="DataIterCreator**") @ByPtrPtr PointerPointer out_array)
public static int MXListDataIters(@Cast(value="uint32_t*") int[] out_size, @ByPtrPtr PointerPointer out_array)
public static int MXDataIterCreateIter(DataIterCreator handle, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") PointerPointer keys, @Cast(value="const char**") PointerPointer vals, PointerPointer out)
handle
- of the iterator creatornum_param
- number of parameterkeys
- parameter keysvals
- parameter valuesout
- resulting iteratorpublic static int MXDataIterCreateIter(DataIterCreator handle, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, PointerPointer out)
public static int MXDataIterCreateIter(DataIterCreator handle, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, @Cast(value="DataIterHandle*") @ByPtrPtr DataIterHandle out)
public static int MXDataIterCreateIter(DataIterCreator handle, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, PointerPointer out)
public static int MXDataIterCreateIter(DataIterCreator handle, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, @Cast(value="DataIterHandle*") @ByPtrPtr DataIterHandle out)
public static int MXDataIterCreateIter(DataIterCreator handle, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, PointerPointer out)
public static int MXDataIterCreateIter(DataIterCreator handle, @Cast(value="uint32_t") int num_param, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, @Cast(value="DataIterHandle*") @ByPtrPtr DataIterHandle out)
public static int MXDataIterGetIterInfo(DataIterCreator creator, @Cast(value="const char**") PointerPointer name, @Cast(value="const char**") PointerPointer description, @Cast(value="uint32_t*") IntPointer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions)
creator
- the DataIterCreator.name
- The returned name of the creator.description
- The returned description of the symbol.num_args
- Number of arguments.arg_names
- Name of the arguments.arg_type_infos
- Type informations about the arguments.arg_descriptions
- Description information about the arguments.public static int MXDataIterGetIterInfo(DataIterCreator creator, @Cast(value="const char**") @ByPtrPtr BytePointer name, @Cast(value="const char**") @ByPtrPtr BytePointer description, @Cast(value="uint32_t*") IntPointer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions)
public static int MXDataIterGetIterInfo(DataIterCreator creator, @Cast(value="const char**") @ByPtrPtr ByteBuffer name, @Cast(value="const char**") @ByPtrPtr ByteBuffer description, @Cast(value="uint32_t*") IntBuffer num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions)
public static int MXDataIterGetIterInfo(DataIterCreator creator, @Cast(value="const char**") @ByPtrPtr byte[] name, @Cast(value="const char**") @ByPtrPtr byte[] description, @Cast(value="uint32_t*") int[] num_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions)
public static int MXDataIterFree(DataIterHandle handle)
handle
- the handle pointer to the data iteratorpublic static int MXDataIterNext(DataIterHandle handle, IntPointer out)
handle
- the handle to iteratorout
- return value of nextpublic static int MXDataIterNext(DataIterHandle handle, IntBuffer out)
public static int MXDataIterNext(DataIterHandle handle, int[] out)
public static int MXDataIterBeforeFirst(DataIterHandle handle)
handle
- the handle to iteratorpublic static int MXDataIterGetData(DataIterHandle handle, PointerPointer out)
handle
- the handle pointer to the data iteratorout
- handle to underlying data NDArraypublic static int MXDataIterGetData(DataIterHandle handle, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXDataIterGetIndex(DataIterHandle handle, @Cast(value="uint64_t**") PointerPointer out_index, @Cast(value="uint64_t*") LongPointer out_size)
handle
- the handle pointer to the data iteratorout_index
- output index of the array.out_size
- output size of the array.public static int MXDataIterGetIndex(DataIterHandle handle, @Cast(value="uint64_t**") @ByPtrPtr LongPointer out_index, @Cast(value="uint64_t*") LongPointer out_size)
public static int MXDataIterGetIndex(DataIterHandle handle, @Cast(value="uint64_t**") @ByPtrPtr LongBuffer out_index, @Cast(value="uint64_t*") LongBuffer out_size)
public static int MXDataIterGetIndex(DataIterHandle handle, @Cast(value="uint64_t**") @ByPtrPtr long[] out_index, @Cast(value="uint64_t*") long[] out_size)
public static int MXDataIterGetPadNum(DataIterHandle handle, IntPointer pad)
handle
- the handle pointer to the data iteratorpad
- pad number ptrpublic static int MXDataIterGetPadNum(DataIterHandle handle, IntBuffer pad)
public static int MXDataIterGetPadNum(DataIterHandle handle, int[] pad)
public static int MXDataIterGetLabel(DataIterHandle handle, PointerPointer out)
handle
- the handle pointer to the data iteratorout
- the handle to underlying label NDArraypublic static int MXDataIterGetLabel(DataIterHandle handle, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXInitPSEnv(@Cast(value="uint32_t") int num_vars, @Cast(value="const char**") PointerPointer keys, @Cast(value="const char**") PointerPointer vals)
num_vars
- number of variables to initializekeys
- environment keysvals
- environment valuespublic static int MXInitPSEnv(@Cast(value="uint32_t") int num_vars, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals)
public static int MXInitPSEnv(@Cast(value="uint32_t") int num_vars, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals)
public static int MXInitPSEnv(@Cast(value="uint32_t") int num_vars, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals)
public static int MXKVStoreCreate(@Cast(value="const char*") BytePointer type, PointerPointer out)
type
- the type of KVStoreout
- The output type of KVStorepublic static int MXKVStoreCreate(String type, @Cast(value="KVStoreHandle*") @ByPtrPtr KVStoreHandle out)
public static int MXKVStoreSetGradientCompression(KVStoreHandle handle, @Cast(value="uint32_t") int num_params, @Cast(value="const char**") PointerPointer keys, @Cast(value="const char**") PointerPointer vals)
handle
- handle to the kvstorekeys
- keys for compression parametersvals
- values for compression parameterspublic static int MXKVStoreSetGradientCompression(KVStoreHandle handle, @Cast(value="uint32_t") int num_params, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals)
public static int MXKVStoreSetGradientCompression(KVStoreHandle handle, @Cast(value="uint32_t") int num_params, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals)
public static int MXKVStoreSetGradientCompression(KVStoreHandle handle, @Cast(value="uint32_t") int num_params, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals)
public static int MXKVStoreFree(KVStoreHandle handle)
handle
- handle to the kvstorepublic static int MXKVStoreInit(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntPointer keys, PointerPointer vals)
handle
- handle to the kvstorenum
- the number of key-value pairskeys
- the list of keysvals
- the list of valuespublic static int MXKVStoreInit(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntBuffer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals)
public static int MXKVStoreInit(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const int[] keys, PointerPointer vals)
public static int MXKVStoreInit(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntPointer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals)
public static int MXKVStoreInit(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntBuffer keys, PointerPointer vals)
public static int MXKVStoreInit(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const int[] keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals)
public static int MXKVStoreInitEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") PointerPointer keys, PointerPointer vals)
handle
- handle to the kvstorenum
- the number of key-value pairskeys
- the list of keysvals
- the list of valuespublic static int MXKVStoreInitEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr BytePointer keys, PointerPointer vals)
public static int MXKVStoreInitEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals)
public static int MXKVStoreInitEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr byte[] keys, PointerPointer vals)
public static int MXKVStoreInitEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals)
public static int MXKVStoreInitEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, PointerPointer vals)
public static int MXKVStoreInitEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals)
public static int MXKVStorePush(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntPointer keys, PointerPointer vals, int priority)
handle
- handle to the kvstorenum
- the number of key-value pairskeys
- the list of keysvals
- the list of valuespriority
- the priority of the actionpublic static int MXKVStorePush(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntBuffer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePush(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const int[] keys, PointerPointer vals, int priority)
public static int MXKVStorePush(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntPointer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePush(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntBuffer keys, PointerPointer vals, int priority)
public static int MXKVStorePush(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const int[] keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePushEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") PointerPointer keys, PointerPointer vals, int priority)
handle
- handle to the kvstorenum
- the number of key-value pairskeys
- the list of keysvals
- the list of valuespriority
- the priority of the actionpublic static int MXKVStorePushEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr BytePointer keys, PointerPointer vals, int priority)
public static int MXKVStorePushEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePushEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr byte[] keys, PointerPointer vals, int priority)
public static int MXKVStorePushEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePushEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, PointerPointer vals, int priority)
public static int MXKVStorePushEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePullWithSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntPointer keys, PointerPointer vals, int priority, @Cast(value="bool") boolean ignore_sparse)
handle
- handle to the kvstorenum
- the number of key-value pairskeys
- the list of keysvals
- the list of valuespriority
- the priority of the actionignore_sparse
- whether to ignore sparse arrays in the requestpublic static int MXKVStorePullWithSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntBuffer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePullWithSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const int[] keys, PointerPointer vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePullWithSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntPointer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePullWithSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntBuffer keys, PointerPointer vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePullWithSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const int[] keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePullWithSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") PointerPointer keys, PointerPointer vals, int priority, @Cast(value="bool") boolean ignore_sparse)
handle
- handle to the kvstorenum
- the number of key-value pairskeys
- the list of keysvals
- the list of valuespriority
- the priority of the actionignore_sparse
- whether to ignore sparse arrays in the requestpublic static int MXKVStorePullWithSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr BytePointer keys, PointerPointer vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePullWithSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePullWithSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr byte[] keys, PointerPointer vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePullWithSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePullWithSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, PointerPointer vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePullWithSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority, @Cast(value="bool") boolean ignore_sparse)
public static int MXKVStorePull(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntPointer keys, PointerPointer vals, int priority)
handle
- handle to the kvstorenum
- the number of key-value pairskeys
- the list of keysvals
- the list of valuespriority
- the priority of the actionpublic static int MXKVStorePull(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntBuffer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePull(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const int[] keys, PointerPointer vals, int priority)
public static int MXKVStorePull(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntPointer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePull(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntBuffer keys, PointerPointer vals, int priority)
public static int MXKVStorePull(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const int[] keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePullEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") PointerPointer keys, PointerPointer vals, int priority)
handle
- handle to the kvstorenum
- the number of key-value pairskeys
- the list of keysvals
- the list of valuespriority
- the priority of the actionpublic static int MXKVStorePullEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr BytePointer keys, PointerPointer vals, int priority)
public static int MXKVStorePullEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePullEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr byte[] keys, PointerPointer vals, int priority)
public static int MXKVStorePullEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePullEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, PointerPointer vals, int priority)
public static int MXKVStorePullEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, int priority)
public static int MXKVStorePullRowSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntPointer keys, PointerPointer vals, @Cast(value="NDArrayHandle*") PointerPointer row_ids, int priority)
handle
- handle to the kvstorenum
- the number of key-value pairskeys
- the list of keysvals
- the list of valuesrow_ids
- the list of row_id NDArrayspriority
- the priority of the actionpublic static int MXKVStorePullRowSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntBuffer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle row_ids, int priority)
public static int MXKVStorePullRowSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const int[] keys, PointerPointer vals, @Cast(value="NDArrayHandle*") PointerPointer row_ids, int priority)
public static int MXKVStorePullRowSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntPointer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle row_ids, int priority)
public static int MXKVStorePullRowSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const IntBuffer keys, PointerPointer vals, @Cast(value="NDArrayHandle*") PointerPointer row_ids, int priority)
public static int MXKVStorePullRowSparse(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Const int[] keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle row_ids, int priority)
public static int MXKVStorePullRowSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") PointerPointer keys, PointerPointer vals, @Cast(value="NDArrayHandle*") PointerPointer row_ids, int priority)
handle
- handle to the kvstorenum
- the number of key-value pairskeys
- the list of keysvals
- the list of valuesrow_ids
- the list of row_id NDArrayspriority
- the priority of the actionpublic static int MXKVStorePullRowSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr BytePointer keys, PointerPointer vals, @Cast(value="NDArrayHandle*") PointerPointer row_ids, int priority)
public static int MXKVStorePullRowSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle row_ids, int priority)
public static int MXKVStorePullRowSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr byte[] keys, PointerPointer vals, @Cast(value="NDArrayHandle*") PointerPointer row_ids, int priority)
public static int MXKVStorePullRowSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle row_ids, int priority)
public static int MXKVStorePullRowSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, PointerPointer vals, @Cast(value="NDArrayHandle*") PointerPointer row_ids, int priority)
public static int MXKVStorePullRowSparseEx(KVStoreHandle handle, @Cast(value="uint32_t") int num, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle row_ids, int priority)
public static int MXKVStoreBroadcast(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const IntPointer vkeys, @Cast(value="mx_uint") int onum, @Const IntPointer okeys, PointerPointer vals, PointerPointer outs, int priority)
handle
- handle to the kvstorevnum
- the number of key-value pairs corresponding to vkeysvkeys
- the list of keys for the values to be pushedonum
- the number of key-value pairs corresponding to okeysokeys
- the list of keys for the values to be pulledvals
- the list of valuesouts
- the list of outputspriority
- the priority of the actionpublic static int MXKVStoreBroadcast(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const IntBuffer vkeys, @Cast(value="mx_uint") int onum, @Const IntBuffer okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStoreBroadcast(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const int[] vkeys, @Cast(value="mx_uint") int onum, @Const int[] okeys, PointerPointer vals, PointerPointer outs, int priority)
public static int MXKVStoreBroadcast(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const IntPointer vkeys, @Cast(value="mx_uint") int onum, @Const IntPointer okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStoreBroadcast(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const IntBuffer vkeys, @Cast(value="mx_uint") int onum, @Const IntBuffer okeys, PointerPointer vals, PointerPointer outs, int priority)
public static int MXKVStoreBroadcast(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const int[] vkeys, @Cast(value="mx_uint") int onum, @Const int[] okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStoreBroadcastEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") PointerPointer vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") PointerPointer okeys, PointerPointer vals, PointerPointer outs, int priority)
handle
- handle to the kvstorevnum
- the number of key-value pairs corresponding to vkeysvkeys
- the list of keys for the values to be pushedonum
- the number of key-value pairs corresponding to okeysokeys
- the list of keys for the values to be pulledvals
- the list of valuesouts
- the list of outputspriority
- the priority of the actionpublic static int MXKVStoreBroadcastEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr BytePointer vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr BytePointer okeys, PointerPointer vals, PointerPointer outs, int priority)
public static int MXKVStoreBroadcastEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr ByteBuffer vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr ByteBuffer okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStoreBroadcastEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr byte[] vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr byte[] okeys, PointerPointer vals, PointerPointer outs, int priority)
public static int MXKVStoreBroadcastEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr BytePointer vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr BytePointer okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStoreBroadcastEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr ByteBuffer vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr ByteBuffer okeys, PointerPointer vals, PointerPointer outs, int priority)
public static int MXKVStoreBroadcastEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr byte[] vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr byte[] okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStorePushPull(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const IntPointer vkeys, @Cast(value="mx_uint") int onum, @Const IntPointer okeys, PointerPointer vals, PointerPointer outs, int priority)
handle
- handle to the kvstorevnum
- the number of key-value pairs corresponding to vkeysvkeys
- the list of keys for the values to be pushedonum
- the number of key-value pairs corresponding to okeysokeys
- the list of keys for the values to be pulledvals
- the list of valuesouts
- the list of outputspriority
- the priority of the actionpublic static int MXKVStorePushPull(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const IntBuffer vkeys, @Cast(value="mx_uint") int onum, @Const IntBuffer okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStorePushPull(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const int[] vkeys, @Cast(value="mx_uint") int onum, @Const int[] okeys, PointerPointer vals, PointerPointer outs, int priority)
public static int MXKVStorePushPull(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const IntPointer vkeys, @Cast(value="mx_uint") int onum, @Const IntPointer okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStorePushPull(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const IntBuffer vkeys, @Cast(value="mx_uint") int onum, @Const IntBuffer okeys, PointerPointer vals, PointerPointer outs, int priority)
public static int MXKVStorePushPull(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Const int[] vkeys, @Cast(value="mx_uint") int onum, @Const int[] okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStorePushPullEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") PointerPointer vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") PointerPointer okeys, PointerPointer vals, PointerPointer outs, int priority)
handle
- handle to the kvstorevnum
- the number of key-value pairs corresponding to vkeysvkeys
- the list of keys for the values to be pushedonum
- the number of key-value pairs corresponding to okeysokeys
- the list of keys for the values to be pulledvals
- the list of valuesouts
- the list of outputspriority
- the priority of the actionpublic static int MXKVStorePushPullEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr BytePointer vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr BytePointer okeys, PointerPointer vals, PointerPointer outs, int priority)
public static int MXKVStorePushPullEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr ByteBuffer vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr ByteBuffer okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStorePushPullEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr byte[] vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr byte[] okeys, PointerPointer vals, PointerPointer outs, int priority)
public static int MXKVStorePushPullEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr BytePointer vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr BytePointer okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStorePushPullEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr ByteBuffer vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr ByteBuffer okeys, PointerPointer vals, PointerPointer outs, int priority)
public static int MXKVStorePushPullEx(KVStoreHandle handle, @Cast(value="mx_uint") int vnum, @Cast(value="const char**") @ByPtrPtr byte[] vkeys, @Cast(value="mx_uint") int onum, @Cast(value="const char**") @ByPtrPtr byte[] okeys, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle vals, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outs, int priority)
public static int MXKVStoreSetUpdater(KVStoreHandle handle, MXKVStoreUpdater updater, Pointer updater_handle)
handle
- handle to the KVStoreupdater
- udpater functionupdater_handle
- The additional handle used to invoke the updaterpublic static int MXKVStoreSetUpdaterEx(KVStoreHandle handle, MXKVStoreUpdater updater, MXKVStoreStrUpdater str_updater, Pointer updater_handle)
handle
- handle to the KVStoreupdater
- updater function with int keysstr_updater
- updater function with string keysupdater_handle
- The additional handle used to invoke the updaterpublic static int MXKVStoreGetType(KVStoreHandle handle, @Cast(value="const char**") PointerPointer type)
handle
- handle to the KVStoretype
- a string typepublic static int MXKVStoreGetType(KVStoreHandle handle, @Cast(value="const char**") @ByPtrPtr BytePointer type)
public static int MXKVStoreGetType(KVStoreHandle handle, @Cast(value="const char**") @ByPtrPtr ByteBuffer type)
public static int MXKVStoreGetType(KVStoreHandle handle, @Cast(value="const char**") @ByPtrPtr byte[] type)
public static int MXKVStoreGetRank(KVStoreHandle handle, IntPointer ret)
handle
- handle to the KVStoreret
- the node rankpublic static int MXKVStoreGetRank(KVStoreHandle handle, IntBuffer ret)
public static int MXKVStoreGetRank(KVStoreHandle handle, int[] ret)
public static int MXKVStoreGetGroupSize(KVStoreHandle handle, IntPointer ret)
IsWorkerNode() == true
,
- number of servers if if IsServerNode() == true
,
- 1 if IsSchedulerNode() == true
,handle
- handle to the KVStoreret
- the group sizepublic static int MXKVStoreGetGroupSize(KVStoreHandle handle, IntBuffer ret)
public static int MXKVStoreGetGroupSize(KVStoreHandle handle, int[] ret)
public static int MXKVStoreIsWorkerNode(IntPointer ret)
ret
- 1 for yes, 0 for nopublic static int MXKVStoreIsWorkerNode(IntBuffer ret)
public static int MXKVStoreIsWorkerNode(int[] ret)
public static int MXKVStoreIsServerNode(IntPointer ret)
ret
- 1 for yes, 0 for nopublic static int MXKVStoreIsServerNode(IntBuffer ret)
public static int MXKVStoreIsServerNode(int[] ret)
public static int MXKVStoreIsSchedulerNode(IntPointer ret)
ret
- 1 for yes, 0 for nopublic static int MXKVStoreIsSchedulerNode(IntBuffer ret)
public static int MXKVStoreIsSchedulerNode(int[] ret)
public static int MXKVStoreBarrier(KVStoreHandle handle)
handle
- handle to the KVStorepublic static int MXKVStoreSetBarrierBeforeExit(KVStoreHandle handle, int barrier_before_exit)
handle
- handle to the KVStorebarrier_before_exit
- whether to do barrier when kvstore finalizepublic static int MXKVStoreRunServer(KVStoreHandle handle, MXKVStoreServerController controller, Pointer controller_handle)
handle
- handle to the KVStorecontroller
- the user-defined server controllercontroller_handle
- helper handle for implementing controllerpublic static int MXKVStoreSendCommmandToServers(KVStoreHandle handle, int cmd_id, @Cast(value="const char*") BytePointer cmd_body)
handle
- handle to the KVStorecmd_id
- the head of the commandcmd_body
- the body of the commandpublic static int MXKVStoreSendCommmandToServers(KVStoreHandle handle, int cmd_id, String cmd_body)
public static int MXKVStoreGetNumDeadNode(KVStoreHandle handle, int node_id, IntPointer number, int timeout_sec)
handle
- handle to the KVStorenode_id
- Can be a node group or a single node.
kScheduler = 1, kServerGroup = 2, kWorkerGroup = 4number
- Ouptut number of dead nodestimeout_sec
- A node fails to send heartbeart in {timeout_sec} seconds
will be presumed as 'dead'public static int MXKVStoreGetNumDeadNode(KVStoreHandle handle, int node_id, IntPointer number)
public static int MXKVStoreGetNumDeadNode(KVStoreHandle handle, int node_id, IntBuffer number, int timeout_sec)
public static int MXKVStoreGetNumDeadNode(KVStoreHandle handle, int node_id, IntBuffer number)
public static int MXKVStoreGetNumDeadNode(KVStoreHandle handle, int node_id, int[] number, int timeout_sec)
public static int MXKVStoreGetNumDeadNode(KVStoreHandle handle, int node_id, int[] number)
public static int MXRecordIOWriterCreate(@Cast(value="const char*") BytePointer uri, PointerPointer out)
uri
- path to fileout
- handle pointer to the created objectpublic static int MXRecordIOWriterCreate(String uri, @Cast(value="RecordIOHandle*") @ByPtrPtr RecordIOHandle out)
public static int MXRecordIOWriterFree(RecordIOHandle handle)
handle
- handle to RecordIO objectpublic static int MXRecordIOWriterWriteRecord(RecordIOHandle handle, @Cast(value="const char*") BytePointer buf, @Cast(value="size_t") long size)
handle
- handle to RecordIO objectbuf
- buffer to writesize
- size of bufferpublic static int MXRecordIOWriterWriteRecord(RecordIOHandle handle, String buf, @Cast(value="size_t") long size)
public static int MXRecordIOWriterTell(RecordIOHandle handle, @Cast(value="size_t*") SizeTPointer pos)
handle
- handle to RecordIO objectpos
- handle to output positionpublic static int MXRecordIOReaderCreate(@Cast(value="const char*") BytePointer uri, PointerPointer out)
uri
- path to fileout
- handle pointer to the created objectpublic static int MXRecordIOReaderCreate(String uri, @Cast(value="RecordIOHandle*") @ByPtrPtr RecordIOHandle out)
public static int MXRecordIOReaderFree(RecordIOHandle handle)
handle
- handle to RecordIO objectpublic static int MXRecordIOReaderReadRecord(RecordIOHandle handle, @Cast(value="const char**") PointerPointer buf, @Cast(value="size_t*") SizeTPointer size)
handle
- handle to RecordIO objectbuf
- pointer to return buffersize
- point to size of bufferpublic static int MXRecordIOReaderReadRecord(RecordIOHandle handle, @Cast(value="const char**") @ByPtrPtr BytePointer buf, @Cast(value="size_t*") SizeTPointer size)
public static int MXRecordIOReaderReadRecord(RecordIOHandle handle, @Cast(value="const char**") @ByPtrPtr ByteBuffer buf, @Cast(value="size_t*") SizeTPointer size)
public static int MXRecordIOReaderReadRecord(RecordIOHandle handle, @Cast(value="const char**") @ByPtrPtr byte[] buf, @Cast(value="size_t*") SizeTPointer size)
public static int MXRecordIOReaderSeek(RecordIOHandle handle, @Cast(value="size_t") long pos)
handle
- handle to RecordIO objectpos
- target positionpublic static int MXRecordIOReaderTell(RecordIOHandle handle, @Cast(value="size_t*") SizeTPointer pos)
handle
- handle to RecordIO objectpos
- handle to output positionpublic static int MXRtcCreate(@Cast(value="char*") BytePointer name, @Cast(value="uint32_t") int num_input, @Cast(value="uint32_t") int num_output, @Cast(value="char**") PointerPointer input_names, @Cast(value="char**") PointerPointer output_names, PointerPointer inputs, PointerPointer outputs, @Cast(value="char*") BytePointer kernel, PointerPointer out)
public static int MXRtcCreate(@Cast(value="char*") BytePointer name, @Cast(value="uint32_t") int num_input, @Cast(value="uint32_t") int num_output, @Cast(value="char**") @ByPtrPtr BytePointer input_names, @Cast(value="char**") @ByPtrPtr BytePointer output_names, PointerPointer inputs, PointerPointer outputs, @Cast(value="char*") BytePointer kernel, PointerPointer out)
public static int MXRtcCreate(@Cast(value="char*") ByteBuffer name, @Cast(value="uint32_t") int num_input, @Cast(value="uint32_t") int num_output, @Cast(value="char**") @ByPtrPtr ByteBuffer input_names, @Cast(value="char**") @ByPtrPtr ByteBuffer output_names, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outputs, @Cast(value="char*") ByteBuffer kernel, @Cast(value="RtcHandle*") @ByPtrPtr RtcHandle out)
public static int MXRtcCreate(@Cast(value="char*") byte[] name, @Cast(value="uint32_t") int num_input, @Cast(value="uint32_t") int num_output, @Cast(value="char**") @ByPtrPtr byte[] input_names, @Cast(value="char**") @ByPtrPtr byte[] output_names, PointerPointer inputs, PointerPointer outputs, @Cast(value="char*") byte[] kernel, PointerPointer out)
public static int MXRtcCreate(@Cast(value="char*") BytePointer name, @Cast(value="uint32_t") int num_input, @Cast(value="uint32_t") int num_output, @Cast(value="char**") @ByPtrPtr BytePointer input_names, @Cast(value="char**") @ByPtrPtr BytePointer output_names, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outputs, @Cast(value="char*") BytePointer kernel, @Cast(value="RtcHandle*") @ByPtrPtr RtcHandle out)
public static int MXRtcCreate(@Cast(value="char*") ByteBuffer name, @Cast(value="uint32_t") int num_input, @Cast(value="uint32_t") int num_output, @Cast(value="char**") @ByPtrPtr ByteBuffer input_names, @Cast(value="char**") @ByPtrPtr ByteBuffer output_names, PointerPointer inputs, PointerPointer outputs, @Cast(value="char*") ByteBuffer kernel, PointerPointer out)
public static int MXRtcCreate(@Cast(value="char*") byte[] name, @Cast(value="uint32_t") int num_input, @Cast(value="uint32_t") int num_output, @Cast(value="char**") @ByPtrPtr byte[] input_names, @Cast(value="char**") @ByPtrPtr byte[] output_names, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outputs, @Cast(value="char*") byte[] kernel, @Cast(value="RtcHandle*") @ByPtrPtr RtcHandle out)
public static int MXRtcPush(RtcHandle handle, @Cast(value="uint32_t") int num_input, @Cast(value="uint32_t") int num_output, PointerPointer inputs, PointerPointer outputs, @Cast(value="uint32_t") int gridDimX, @Cast(value="uint32_t") int gridDimY, @Cast(value="uint32_t") int gridDimZ, @Cast(value="uint32_t") int blockDimX, @Cast(value="uint32_t") int blockDimY, @Cast(value="uint32_t") int blockDimZ)
public static int MXRtcPush(RtcHandle handle, @Cast(value="uint32_t") int num_input, @Cast(value="uint32_t") int num_output, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outputs, @Cast(value="uint32_t") int gridDimX, @Cast(value="uint32_t") int gridDimY, @Cast(value="uint32_t") int gridDimZ, @Cast(value="uint32_t") int blockDimX, @Cast(value="uint32_t") int blockDimY, @Cast(value="uint32_t") int blockDimZ)
public static int MXRtcFree(RtcHandle handle)
public static int MXCustomOpRegister(@Cast(value="const char*") BytePointer op_type, CustomOpPropCreator creator)
public static int MXCustomOpRegister(String op_type, CustomOpPropCreator creator)
public static int MXCustomFunctionRecord(int num_inputs, PointerPointer inputs, int num_outputs, PointerPointer outputs, MXCallbackList callbacks)
public static int MXCustomFunctionRecord(int num_inputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle inputs, int num_outputs, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle outputs, MXCallbackList callbacks)
public static int MXRtcCudaModuleCreate(@Cast(value="const char*") BytePointer source, int num_options, @Cast(value="const char**") PointerPointer options, int num_exports, @Cast(value="const char**") PointerPointer exports, @ByPtrPtr CudaModuleHandle out)
public static int MXRtcCudaModuleCreate(@Cast(value="const char*") BytePointer source, int num_options, @Cast(value="const char**") @ByPtrPtr BytePointer options, int num_exports, @Cast(value="const char**") @ByPtrPtr BytePointer exports, @ByPtrPtr CudaModuleHandle out)
public static int MXRtcCudaModuleCreate(String source, int num_options, @Cast(value="const char**") @ByPtrPtr ByteBuffer options, int num_exports, @Cast(value="const char**") @ByPtrPtr ByteBuffer exports, @ByPtrPtr CudaModuleHandle out)
public static int MXRtcCudaModuleCreate(@Cast(value="const char*") BytePointer source, int num_options, @Cast(value="const char**") @ByPtrPtr byte[] options, int num_exports, @Cast(value="const char**") @ByPtrPtr byte[] exports, @ByPtrPtr CudaModuleHandle out)
public static int MXRtcCudaModuleCreate(String source, int num_options, @Cast(value="const char**") @ByPtrPtr BytePointer options, int num_exports, @Cast(value="const char**") @ByPtrPtr BytePointer exports, @ByPtrPtr CudaModuleHandle out)
public static int MXRtcCudaModuleCreate(@Cast(value="const char*") BytePointer source, int num_options, @Cast(value="const char**") @ByPtrPtr ByteBuffer options, int num_exports, @Cast(value="const char**") @ByPtrPtr ByteBuffer exports, @ByPtrPtr CudaModuleHandle out)
public static int MXRtcCudaModuleCreate(String source, int num_options, @Cast(value="const char**") @ByPtrPtr byte[] options, int num_exports, @Cast(value="const char**") @ByPtrPtr byte[] exports, @ByPtrPtr CudaModuleHandle out)
public static int MXRtcCudaModuleFree(CudaModuleHandle handle)
public static int MXRtcCudaKernelCreate(CudaModuleHandle handle, @Cast(value="const char*") BytePointer name, int num_args, IntPointer is_ndarray, IntPointer is_const, IntPointer arg_types, @ByPtrPtr CudaKernelHandle out)
public static int MXRtcCudaKernelCreate(CudaModuleHandle handle, String name, int num_args, IntBuffer is_ndarray, IntBuffer is_const, IntBuffer arg_types, @ByPtrPtr CudaKernelHandle out)
public static int MXRtcCudaKernelCreate(CudaModuleHandle handle, @Cast(value="const char*") BytePointer name, int num_args, int[] is_ndarray, int[] is_const, int[] arg_types, @ByPtrPtr CudaKernelHandle out)
public static int MXRtcCudaKernelCreate(CudaModuleHandle handle, String name, int num_args, IntPointer is_ndarray, IntPointer is_const, IntPointer arg_types, @ByPtrPtr CudaKernelHandle out)
public static int MXRtcCudaKernelCreate(CudaModuleHandle handle, @Cast(value="const char*") BytePointer name, int num_args, IntBuffer is_ndarray, IntBuffer is_const, IntBuffer arg_types, @ByPtrPtr CudaKernelHandle out)
public static int MXRtcCudaKernelCreate(CudaModuleHandle handle, String name, int num_args, int[] is_ndarray, int[] is_const, int[] arg_types, @ByPtrPtr CudaKernelHandle out)
public static int MXRtcCudaKernelFree(CudaKernelHandle handle)
public static int MXRtcCudaKernelCall(CudaKernelHandle handle, int dev_id, @Cast(value="void**") PointerPointer args, @Cast(value="uint32_t") int grid_dim_x, @Cast(value="uint32_t") int grid_dim_y, @Cast(value="uint32_t") int grid_dim_z, @Cast(value="uint32_t") int block_dim_x, @Cast(value="uint32_t") int block_dim_y, @Cast(value="uint32_t") int block_dim_z, @Cast(value="uint32_t") int shared_mem)
public static int MXRtcCudaKernelCall(CudaKernelHandle handle, int dev_id, @Cast(value="void**") @ByPtrPtr Pointer args, @Cast(value="uint32_t") int grid_dim_x, @Cast(value="uint32_t") int grid_dim_y, @Cast(value="uint32_t") int grid_dim_z, @Cast(value="uint32_t") int block_dim_x, @Cast(value="uint32_t") int block_dim_y, @Cast(value="uint32_t") int block_dim_z, @Cast(value="uint32_t") int shared_mem)
public static int MXNDArrayGetSharedMemHandle(NDArrayHandle handle, IntPointer shared_pid, IntPointer shared_id)
handle
- NDArray handle.shared_pid
- output PIDshared_id
- output shared memory id.public static int MXNDArrayGetSharedMemHandle(NDArrayHandle handle, IntBuffer shared_pid, IntBuffer shared_id)
public static int MXNDArrayGetSharedMemHandle(NDArrayHandle handle, int[] shared_pid, int[] shared_id)
public static int MXNDArrayCreateFromSharedMem(int shared_pid, int shared_id, @Cast(value="const uint32_t*") IntPointer shape, @Cast(value="uint32_t") int ndim, int dtype, PointerPointer out)
shared_pid
- shared PIDshared_id
- shared memory idshape
- pointer to NDArray dimensionsndim
- number of NDArray dimensionsdtype
- data type of NDArrayout
- constructed NDArraypublic static int MXNDArrayCreateFromSharedMem(int shared_pid, int shared_id, @Cast(value="const uint32_t*") IntBuffer shape, @Cast(value="uint32_t") int ndim, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateFromSharedMem(int shared_pid, int shared_id, @Cast(value="const uint32_t*") int[] shape, @Cast(value="uint32_t") int ndim, int dtype, PointerPointer out)
public static int MXNDArrayCreateFromSharedMem(int shared_pid, int shared_id, @Cast(value="const uint32_t*") IntPointer shape, @Cast(value="uint32_t") int ndim, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateFromSharedMem(int shared_pid, int shared_id, @Cast(value="const uint32_t*") IntBuffer shape, @Cast(value="uint32_t") int ndim, int dtype, PointerPointer out)
public static int MXNDArrayCreateFromSharedMem(int shared_pid, int shared_id, @Cast(value="const uint32_t*") int[] shape, @Cast(value="uint32_t") int ndim, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXStorageEmptyCache(int dev_type, int dev_id)
dev_type
- device type, specify device we want to takedev_id
- the device id of the specific devicepublic static int MXNDArrayCreateFromSharedMemEx(int shared_pid, int shared_id, @Const IntPointer shape, int ndim, int dtype, PointerPointer out)
shared_pid
- shared PIDshared_id
- shared memory idshape
- pointer to NDArray dimensionsndim
- number of NDArray dimensionsdtype
- data type of NDArrayout
- constructed NDArraypublic static int MXNDArrayCreateFromSharedMemEx(int shared_pid, int shared_id, @Const IntBuffer shape, int ndim, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateFromSharedMemEx(int shared_pid, int shared_id, @Const int[] shape, int ndim, int dtype, PointerPointer out)
public static int MXNDArrayCreateFromSharedMemEx(int shared_pid, int shared_id, @Const IntPointer shape, int ndim, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXNDArrayCreateFromSharedMemEx(int shared_pid, int shared_id, @Const IntBuffer shape, int ndim, int dtype, PointerPointer out)
public static int MXNDArrayCreateFromSharedMemEx(int shared_pid, int shared_id, @Const int[] shape, int ndim, int dtype, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXEnginePushAsync(EngineAsyncFunc async_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, EngineVarHandle const_vars_handle, int num_const_vars, EngineVarHandle mutable_vars_handle, int num_mutable_vars, EngineFnPropertyHandle prop_handle, int priority, @Cast(value="const char*") BytePointer opr_name, @Cast(value="bool") boolean _wait)
async_func
- Execution function whici takes a parameter on_complete
that must be called when the execution ompletes.func_param
- The parameter set on calling async_func, can be NULL.deleter
- The callback to free func_param, can be NULL.ctx_handle
- Execution context.const_vars_handle
- The variables that current operation will use
but not mutate.num_const_vars
- The number of const_vars_handle.mutable_vars_handle
- The variables that current operation will mutate.num_mutable_vars
- The number of mutable_vars_handle.prop_handle
- Property of the function.priority
- Priority of the action, as hint to the engine.opr_name
- The operation name.wait
- Whether this is a WaitForVar operation.public static int MXEnginePushAsync(EngineAsyncFunc async_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, EngineVarHandle const_vars_handle, int num_const_vars, EngineVarHandle mutable_vars_handle, int num_mutable_vars)
public static int MXEnginePushAsync(EngineAsyncFunc async_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, EngineVarHandle const_vars_handle, int num_const_vars, EngineVarHandle mutable_vars_handle, int num_mutable_vars, EngineFnPropertyHandle prop_handle, int priority, String opr_name, @Cast(value="bool") boolean _wait)
public static int MXEnginePushSync(EngineSyncFunc sync_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, EngineVarHandle const_vars_handle, int num_const_vars, EngineVarHandle mutable_vars_handle, int num_mutable_vars, EngineFnPropertyHandle prop_handle, int priority, @Cast(value="const char*") BytePointer opr_name)
sync_func
- Execution function that executes the operation.func_param
- The parameter set on calling sync_func, can be NULL.deleter
- The callback to free func_param, can be NULL.ctx_handle
- Execution context.const_vars_handle
- The variables that current operation will use
but not mutate.num_const_vars
- The number of const_vars_handle.mutable_vars_handle
- The variables that current operation will mutate.num_mutable_vars
- The number of mutable_vars_handle.prop_handle
- Property of the function.priority
- Priority of the action, as hint to the engine.opr_name
- The operation name.public static int MXEnginePushSync(EngineSyncFunc sync_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, EngineVarHandle const_vars_handle, int num_const_vars, EngineVarHandle mutable_vars_handle, int num_mutable_vars)
public static int MXEnginePushSync(EngineSyncFunc sync_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, EngineVarHandle const_vars_handle, int num_const_vars, EngineVarHandle mutable_vars_handle, int num_mutable_vars, EngineFnPropertyHandle prop_handle, int priority, String opr_name)
public static int MXShallowCopyNDArray(NDArrayHandle src, PointerPointer out)
src
- source NDArrayout
- new NDArray sharing the same data chunck with srcpublic static int MXShallowCopyNDArray(NDArrayHandle src, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle out)
public static int MXShallowCopySymbol(SymbolHandle src, PointerPointer out)
src
- source Symbolout
- new Symbol sharing the same graph structure with srcpublic static int MXShallowCopySymbol(SymbolHandle src, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int MXEnginePushAsyncND(EngineAsyncFunc async_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, PointerPointer const_nds_handle, int num_const_nds, PointerPointer mutable_nds_handle, int num_mutable_nds, EngineFnPropertyHandle prop_handle, int priority, @Cast(value="const char*") BytePointer opr_name, @Cast(value="bool") boolean _wait)
async_func
- Execution function whici takes a parameter on_complete
that must be called when the execution ompletes.func_param
- The parameter set on calling async_func, can be NULL.deleter
- The callback to free func_param, can be NULL.ctx_handle
- Execution context.const_nds_handle
- The NDArrays that current operation will use
but not mutate.num_const_nds
- The number of const_nds_handle.mutable_nds_handle
- The NDArrays that current operation will mutate.num_mutable_nds
- The number of mutable_nds_handle.prop_handle
- Property of the function.priority
- Priority of the action, as hint to the engine.opr_name
- The operation name.wait
- Whether this is a WaitForVar operation.public static int MXEnginePushAsyncND(EngineAsyncFunc async_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, PointerPointer const_nds_handle, int num_const_nds, PointerPointer mutable_nds_handle, int num_mutable_nds)
public static int MXEnginePushAsyncND(EngineAsyncFunc async_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle const_nds_handle, int num_const_nds, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle mutable_nds_handle, int num_mutable_nds, EngineFnPropertyHandle prop_handle, int priority, String opr_name, @Cast(value="bool") boolean _wait)
public static int MXEnginePushAsyncND(EngineAsyncFunc async_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle const_nds_handle, int num_const_nds, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle mutable_nds_handle, int num_mutable_nds)
public static int MXEnginePushSyncND(EngineSyncFunc sync_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, PointerPointer const_nds_handle, int num_const_nds, PointerPointer mutable_nds_handle, int num_mutable_nds, EngineFnPropertyHandle prop_handle, int priority, @Cast(value="const char*") BytePointer opr_name)
sync_func
- Execution function that executes the operation.func_param
- The parameter set on calling sync_func, can be NULL.deleter
- The callback to free func_param, can be NULL.ctx_handle
- Execution context.const_nds_handle
- The NDArrays that current operation will use
but not mutate.num_const_nds
- The number of const_nds_handle.mutable_nds_handle
- The NDArrays that current operation will mutate.num_mutable_nds
- The number of mutable_nds_handle.prop_handle
- Property of the function.priority
- Priority of the action, as hint to the engine.opr_name
- The operation name.public static int MXEnginePushSyncND(EngineSyncFunc sync_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, PointerPointer const_nds_handle, int num_const_nds, PointerPointer mutable_nds_handle, int num_mutable_nds)
public static int MXEnginePushSyncND(EngineSyncFunc sync_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle const_nds_handle, int num_const_nds, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle mutable_nds_handle, int num_mutable_nds, EngineFnPropertyHandle prop_handle, int priority, String opr_name)
public static int MXEnginePushSyncND(EngineSyncFunc sync_func, Pointer func_param, EngineFuncParamDeleter deleter, ContextHandle ctx_handle, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle const_nds_handle, int num_const_nds, @Cast(value="NDArrayHandle*") @ByPtrPtr NDArrayHandle mutable_nds_handle, int num_mutable_nds)
public static int MXPredCreate(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") PointerPointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, PointerPointer out)
symbol_json_str
- The JSON string of the symbol.param_bytes
- The in-memory raw bytes of parameter ndarray file.param_size
- The size of parameter ndarray file.dev_type
- The device type, 1: cpu, 2:gpudev_id
- The device id of the predictor.num_input_nodes
- Number of input nodes to the net,
For feedforward net, this is 1.input_keys
- The name of input argument.
For feedforward net, this is {"data"}input_shape_indptr
- Index pointer of shapes of each input node.
The length of this array = num_input_nodes + 1.
For feedforward net that takes 4 dimensional input, this is {0, 4}.input_shape_data
- A flattened data of shapes of each input node.
For feedforward net that takes 4 dimensional input, this is the shape data.out
- The created predictor handle.public static int MXPredCreate(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, PointerPointer out)
public static int MXPredCreate(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_keys, @Cast(value="const uint32_t*") IntBuffer input_shape_indptr, @Cast(value="const uint32_t*") IntBuffer input_shape_data, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreate(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr byte[] input_keys, @Cast(value="const uint32_t*") int[] input_shape_indptr, @Cast(value="const uint32_t*") int[] input_shape_data, PointerPointer out)
public static int MXPredCreate(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreate(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_keys, @Cast(value="const uint32_t*") IntBuffer input_shape_indptr, @Cast(value="const uint32_t*") IntBuffer input_shape_data, PointerPointer out)
public static int MXPredCreate(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr byte[] input_keys, @Cast(value="const uint32_t*") int[] input_shape_indptr, @Cast(value="const uint32_t*") int[] input_shape_data, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreateEx(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_input_nodes, @Cast(value="const char**") PointerPointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") PointerPointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, PointerPointer out)
symbol_json_str
- The JSON string of the symbol.param_bytes
- The in-memory raw bytes of parameter ndarray file.param_size
- The size of parameter ndarray file.dev_type
- The device type, 1: cpu, 2: gpudev_id
- The device id of the predictor.num_input_nodes
- Number of input nodes to the net.
For feedforward net, this is 1.input_keys
- The name of the input argument.
For feedforward net, this is {"data"}input_shape_indptr
- Index pointer of shapes of each input node.
The length of this array = num_input_nodes + 1.
For feedforward net that takes 4 dimensional input, this is {0, 4}.input_shape_data
- A flattened data of shapes of each input node.
For feedforward net that takes 4 dimensional input, this is the shape data.num_provided_arg_dtypes
- The length of provided_arg_dtypes.provided_arg_dtype_names
- The provided_arg_dtype_names the names of args for which dtypes are provided.provided_arg_dtypes
- The provided_arg_dtypes the dtype providedout
- The created predictor handle.public static int MXPredCreateEx(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, PointerPointer out)
public static int MXPredCreateEx(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_keys, @Cast(value="const uint32_t*") IntBuffer input_shape_indptr, @Cast(value="const uint32_t*") IntBuffer input_shape_data, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_dtype_names, @Const IntBuffer provided_arg_dtypes, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreateEx(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr byte[] input_keys, @Cast(value="const uint32_t*") int[] input_shape_indptr, @Cast(value="const uint32_t*") int[] input_shape_data, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_dtype_names, @Const int[] provided_arg_dtypes, PointerPointer out)
public static int MXPredCreateEx(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr BytePointer provided_arg_dtype_names, @Const IntPointer provided_arg_dtypes, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreateEx(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_keys, @Cast(value="const uint32_t*") IntBuffer input_shape_indptr, @Cast(value="const uint32_t*") IntBuffer input_shape_data, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr ByteBuffer provided_arg_dtype_names, @Const IntBuffer provided_arg_dtypes, PointerPointer out)
public static int MXPredCreateEx(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="const uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr byte[] input_keys, @Cast(value="const uint32_t*") int[] input_shape_indptr, @Cast(value="const uint32_t*") int[] input_shape_data, @Cast(value="const uint32_t") int num_provided_arg_dtypes, @Cast(value="const char**") @ByPtrPtr byte[] provided_arg_dtype_names, @Const int[] provided_arg_dtypes, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreatePartialOut(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") PointerPointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, @Cast(value="uint32_t") int num_output_nodes, @Cast(value="const char**") PointerPointer output_keys, PointerPointer out)
symbol_json_str
- The JSON string of the symbol.param_bytes
- The in-memory raw bytes of parameter ndarray file.param_size
- The size of parameter ndarray file.dev_type
- The device type, 1: cpu, 2:gpudev_id
- The device id of the predictor.num_input_nodes
- Number of input nodes to the net,
For feedforward net, this is 1.input_keys
- The name of input argument.
For feedforward net, this is {"data"}input_shape_indptr
- Index pointer of shapes of each input node.
The length of this array = num_input_nodes + 1.
For feedforward net that takes 4 dimensional input, this is {0, 4}.input_shape_data
- A flattened data of shapes of each input node.
For feedforward net that takes 4 dimensional input, this is the shape data.num_output_nodes
- Number of output nodes to the net,output_keys
- The name of output argument.
For example {"global_pool"}out
- The created predictor handle.public static int MXPredCreatePartialOut(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, @Cast(value="uint32_t") int num_output_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer output_keys, PointerPointer out)
public static int MXPredCreatePartialOut(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_keys, @Cast(value="const uint32_t*") IntBuffer input_shape_indptr, @Cast(value="const uint32_t*") IntBuffer input_shape_data, @Cast(value="uint32_t") int num_output_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer output_keys, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreatePartialOut(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr byte[] input_keys, @Cast(value="const uint32_t*") int[] input_shape_indptr, @Cast(value="const uint32_t*") int[] input_shape_data, @Cast(value="uint32_t") int num_output_nodes, @Cast(value="const char**") @ByPtrPtr byte[] output_keys, PointerPointer out)
public static int MXPredCreatePartialOut(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, @Cast(value="uint32_t") int num_output_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer output_keys, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreatePartialOut(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_keys, @Cast(value="const uint32_t*") IntBuffer input_shape_indptr, @Cast(value="const uint32_t*") IntBuffer input_shape_data, @Cast(value="uint32_t") int num_output_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer output_keys, PointerPointer out)
public static int MXPredCreatePartialOut(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr byte[] input_keys, @Cast(value="const uint32_t*") int[] input_shape_indptr, @Cast(value="const uint32_t*") int[] input_shape_data, @Cast(value="uint32_t") int num_output_nodes, @Cast(value="const char**") @ByPtrPtr byte[] output_keys, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreateMultiThread(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") PointerPointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, int num_threads, PointerPointer out)
symbol_json_str
- The JSON string of the symbol.param_bytes
- The in-memory raw bytes of parameter ndarray file.param_size
- The size of parameter ndarray file.dev_type
- The device type, 1: cpu, 2:gpudev_id
- The device id of the predictor.num_input_nodes
- Number of input nodes to the net,
For feedforward net, this is 1.input_keys
- The name of input argument.
For feedforward net, this is {"data"}input_shape_indptr
- Index pointer of shapes of each input node.
The length of this array = num_input_nodes + 1.
For feedforward net that takes 4 dimensional input, this is {0, 4}.input_shape_data
- A flattened data of shapes of each input node.
For feedforward net that takes 4 dimensional input, this is the shape data.num_threads
- The number of threads that we'll run the predictors.out
- An array of created predictor handles. The array has to be large
enough to keep num_threads
predictors.public static int MXPredCreateMultiThread(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, int num_threads, PointerPointer out)
public static int MXPredCreateMultiThread(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_keys, @Cast(value="const uint32_t*") IntBuffer input_shape_indptr, @Cast(value="const uint32_t*") IntBuffer input_shape_data, int num_threads, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreateMultiThread(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr byte[] input_keys, @Cast(value="const uint32_t*") int[] input_shape_indptr, @Cast(value="const uint32_t*") int[] input_shape_data, int num_threads, PointerPointer out)
public static int MXPredCreateMultiThread(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, int num_threads, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredCreateMultiThread(@Cast(value="const char*") BytePointer symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_keys, @Cast(value="const uint32_t*") IntBuffer input_shape_indptr, @Cast(value="const uint32_t*") IntBuffer input_shape_data, int num_threads, PointerPointer out)
public static int MXPredCreateMultiThread(String symbol_json_str, @Const Pointer param_bytes, int param_size, int dev_type, int dev_id, @Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr byte[] input_keys, @Cast(value="const uint32_t*") int[] input_shape_indptr, @Cast(value="const uint32_t*") int[] input_shape_data, int num_threads, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredReshape(@Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") PointerPointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, PredictorHandle handle, PointerPointer out)
num_input_nodes
- Number of input nodes to the net,
For feedforward net, this is 1.input_keys
- The name of input argument.
For feedforward net, this is {"data"}input_shape_indptr
- Index pointer of shapes of each input node.
The length of this array = num_input_nodes + 1.
For feedforward net that takes 4 dimensional input, this is {0, 4}.input_shape_data
- A flattened data of shapes of each input node.
For feedforward net that takes 4 dimensional input, this is the shape data.handle
- The original predictor handle.out
- The reshaped predictor handle.public static int MXPredReshape(@Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, PredictorHandle handle, PointerPointer out)
public static int MXPredReshape(@Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_keys, @Cast(value="const uint32_t*") IntBuffer input_shape_indptr, @Cast(value="const uint32_t*") IntBuffer input_shape_data, PredictorHandle handle, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredReshape(@Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr byte[] input_keys, @Cast(value="const uint32_t*") int[] input_shape_indptr, @Cast(value="const uint32_t*") int[] input_shape_data, PredictorHandle handle, PointerPointer out)
public static int MXPredReshape(@Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr BytePointer input_keys, @Cast(value="const uint32_t*") IntPointer input_shape_indptr, @Cast(value="const uint32_t*") IntPointer input_shape_data, PredictorHandle handle, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredReshape(@Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr ByteBuffer input_keys, @Cast(value="const uint32_t*") IntBuffer input_shape_indptr, @Cast(value="const uint32_t*") IntBuffer input_shape_data, PredictorHandle handle, PointerPointer out)
public static int MXPredReshape(@Cast(value="uint32_t") int num_input_nodes, @Cast(value="const char**") @ByPtrPtr byte[] input_keys, @Cast(value="const uint32_t*") int[] input_shape_indptr, @Cast(value="const uint32_t*") int[] input_shape_data, PredictorHandle handle, @Cast(value="PredictorHandle*") @ByPtrPtr PredictorHandle out)
public static int MXPredGetOutputShape(PredictorHandle handle, @Cast(value="uint32_t") int index, @Cast(value="uint32_t**") PointerPointer shape_data, @Cast(value="uint32_t*") IntPointer shape_ndim)
handle
- The handle of the predictor.index
- The index of output node, set to 0 if there is only one output.shape_data
- Used to hold pointer to the shape datashape_ndim
- Used to hold shape dimension.public static int MXPredGetOutputShape(PredictorHandle handle, @Cast(value="uint32_t") int index, @Cast(value="uint32_t**") @ByPtrPtr IntPointer shape_data, @Cast(value="uint32_t*") IntPointer shape_ndim)
public static int MXPredGetOutputShape(PredictorHandle handle, @Cast(value="uint32_t") int index, @Cast(value="uint32_t**") @ByPtrPtr IntBuffer shape_data, @Cast(value="uint32_t*") IntBuffer shape_ndim)
public static int MXPredGetOutputShape(PredictorHandle handle, @Cast(value="uint32_t") int index, @Cast(value="uint32_t**") @ByPtrPtr int[] shape_data, @Cast(value="uint32_t*") int[] shape_ndim)
public static int MXPredGetOutputType(PredictorHandle handle, @Cast(value="uint32_t") int out_index, IntPointer out_dtype)
handle
- The handle of the predictor.out_index
- The index of the output node, set to 0 if there is only one output.out_dtype
- The dtype of the output nodepublic static int MXPredGetOutputType(PredictorHandle handle, @Cast(value="uint32_t") int out_index, IntBuffer out_dtype)
public static int MXPredGetOutputType(PredictorHandle handle, @Cast(value="uint32_t") int out_index, int[] out_dtype)
public static int MXPredSetInput(PredictorHandle handle, @Cast(value="const char*") BytePointer key, @Const FloatPointer data, @Cast(value="uint32_t") int size)
handle
- The predictor handle.key
- The name of input node to set.
For feedforward net, this is "data".data
- The pointer to the data to be set, with the shape specified in MXPredCreate.size
- The size of data array, used for safety check.public static int MXPredSetInput(PredictorHandle handle, String key, @Const FloatBuffer data, @Cast(value="uint32_t") int size)
public static int MXPredSetInput(PredictorHandle handle, @Cast(value="const char*") BytePointer key, @Const float[] data, @Cast(value="uint32_t") int size)
public static int MXPredSetInput(PredictorHandle handle, String key, @Const FloatPointer data, @Cast(value="uint32_t") int size)
public static int MXPredSetInput(PredictorHandle handle, @Cast(value="const char*") BytePointer key, @Const FloatBuffer data, @Cast(value="uint32_t") int size)
public static int MXPredSetInput(PredictorHandle handle, String key, @Const float[] data, @Cast(value="uint32_t") int size)
public static int MXPredForward(PredictorHandle handle)
handle
- The handle of the predictor.public static int MXPredPartialForward(PredictorHandle handle, int step, IntPointer step_left)
int step_left = 1;
for (int step = 0; step_left != 0; ++step) {
MXPredPartialForward(handle, step, &step_left);
printf("Current progress [%d/%d]\n", step, step + step_left + 1);
}
handle
- The handle of the predictor.step
- The current step to run forward on.step_left
- The number of steps leftpublic static int MXPredPartialForward(PredictorHandle handle, int step, IntBuffer step_left)
public static int MXPredPartialForward(PredictorHandle handle, int step, int[] step_left)
public static int MXPredGetOutput(PredictorHandle handle, @Cast(value="uint32_t") int index, FloatPointer data, @Cast(value="uint32_t") int size)
handle
- The handle of the predictor.index
- The index of output node, set to 0 if there is only one output.data
- User allocated data to hold the output.size
- The size of data array, used for safe checking.public static int MXPredGetOutput(PredictorHandle handle, @Cast(value="uint32_t") int index, FloatBuffer data, @Cast(value="uint32_t") int size)
public static int MXPredGetOutput(PredictorHandle handle, @Cast(value="uint32_t") int index, float[] data, @Cast(value="uint32_t") int size)
public static int MXPredFree(PredictorHandle handle)
handle
- The handle of the predictor.public static int MXNDListCreate(@Cast(value="const char*") BytePointer nd_file_bytes, int nd_file_size, @ByPtrPtr NDListHandle out, @Cast(value="uint32_t*") IntPointer out_length)
nd_file_bytes
- The byte contents of nd file to be loaded.nd_file_size
- The size of the nd file to be loaded.out
- The out put NDListHandleout_length
- Length of the list.public static int MXNDListCreate(String nd_file_bytes, int nd_file_size, @ByPtrPtr NDListHandle out, @Cast(value="uint32_t*") IntBuffer out_length)
public static int MXNDListCreate(@Cast(value="const char*") BytePointer nd_file_bytes, int nd_file_size, @ByPtrPtr NDListHandle out, @Cast(value="uint32_t*") int[] out_length)
public static int MXNDListCreate(String nd_file_bytes, int nd_file_size, @ByPtrPtr NDListHandle out, @Cast(value="uint32_t*") IntPointer out_length)
public static int MXNDListCreate(@Cast(value="const char*") BytePointer nd_file_bytes, int nd_file_size, @ByPtrPtr NDListHandle out, @Cast(value="uint32_t*") IntBuffer out_length)
public static int MXNDListCreate(String nd_file_bytes, int nd_file_size, @ByPtrPtr NDListHandle out, @Cast(value="uint32_t*") int[] out_length)
public static int MXNDListGet(NDListHandle handle, @Cast(value="uint32_t") int index, @Cast(value="const char**") PointerPointer out_key, @Cast(value="const float**") PointerPointer out_data, @Cast(value="const uint32_t**") PointerPointer out_shape, @Cast(value="uint32_t*") IntPointer out_ndim)
handle
- The handle to the NDArrayindex
- The index in the listout_key
- The output key of the itemout_data
- The data region of the itemout_shape
- The shape of the item.out_ndim
- The number of dimension in the shape.public static int MXNDListGet(NDListHandle handle, @Cast(value="uint32_t") int index, @Cast(value="const char**") @ByPtrPtr BytePointer out_key, @Const @ByPtrPtr FloatPointer out_data, @Cast(value="const uint32_t**") @ByPtrPtr IntPointer out_shape, @Cast(value="uint32_t*") IntPointer out_ndim)
public static int MXNDListGet(NDListHandle handle, @Cast(value="uint32_t") int index, @Cast(value="const char**") @ByPtrPtr ByteBuffer out_key, @Const @ByPtrPtr FloatBuffer out_data, @Cast(value="const uint32_t**") @ByPtrPtr IntBuffer out_shape, @Cast(value="uint32_t*") IntBuffer out_ndim)
public static int MXNDListGet(NDListHandle handle, @Cast(value="uint32_t") int index, @Cast(value="const char**") @ByPtrPtr byte[] out_key, @Const @ByPtrPtr float[] out_data, @Cast(value="const uint32_t**") @ByPtrPtr int[] out_shape, @Cast(value="uint32_t*") int[] out_ndim)
public static int MXPredSetMonitorCallback(PredictorHandle handle, PredMonitorCallback callback, Pointer callback_handle, @Cast(value="bool") boolean monitor_all)
public static int MXNDListFree(NDListHandle handle)
handle
- The handle of the MXAPINDList.public static void NNAPISetLastError(@Cast(value="const char*") BytePointer msg)
msg
- The error message to set.public static void NNAPISetLastError(String msg)
@Cast(value="const char*") public static BytePointer NNGetLastError()
public static int NNListAllOpNames(@Cast(value="nn_uint*") IntPointer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_array)
out_size
- the size of returned arrayout_array
- the output operator name array.public static int NNListAllOpNames(@Cast(value="nn_uint*") IntBuffer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_array)
public static int NNListAllOpNames(@Cast(value="nn_uint*") int[] out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_array)
public static int NNGetOpHandle(@Cast(value="const char*") BytePointer op_name, @ByPtrPtr OpHandle op_out)
op_name
- The name of the operator.op_out
- The returnning op handle.public static int NNListUniqueOps(@Cast(value="nn_uint*") IntPointer out_size, @Cast(value="OpHandle**") @ByPtrPtr PointerPointer out_array)
out_size
- the size of returned arrayout_array
- the output AtomicSymbolCreator arraypublic static int NNListUniqueOps(@Cast(value="nn_uint*") IntBuffer out_size, @Cast(value="OpHandle**") @ByPtrPtr PointerPointer out_array)
public static int NNListUniqueOps(@Cast(value="nn_uint*") int[] out_size, @Cast(value="OpHandle**") @ByPtrPtr PointerPointer out_array)
public static int NNGetOpInfo(OpHandle op, @Cast(value="const char**") PointerPointer real_name, @Cast(value="const char**") PointerPointer description, @Cast(value="nn_uint*") IntPointer num_doc_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") PointerPointer return_type)
op
- The operator handle.real_name
- The returned name of the creator.
This name is not the alias name of the atomic symbol.description
- The returned description of the symbol.num_doc_args
- Number of arguments that contain documents.arg_names
- Name of the arguments of doc argsarg_type_infos
- Type informations about the arguments.arg_descriptions
- Description information about the arguments.return_type
- Return type of the function, if any.public static int NNGetOpInfo(OpHandle op, @Cast(value="const char**") @ByPtrPtr BytePointer real_name, @Cast(value="const char**") @ByPtrPtr BytePointer description, @Cast(value="nn_uint*") IntPointer num_doc_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr BytePointer return_type)
public static int NNGetOpInfo(OpHandle op, @Cast(value="const char**") @ByPtrPtr ByteBuffer real_name, @Cast(value="const char**") @ByPtrPtr ByteBuffer description, @Cast(value="nn_uint*") IntBuffer num_doc_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr ByteBuffer return_type)
public static int NNGetOpInfo(OpHandle op, @Cast(value="const char**") @ByPtrPtr byte[] real_name, @Cast(value="const char**") @ByPtrPtr byte[] description, @Cast(value="nn_uint*") int[] num_doc_args, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_names, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_type_infos, @Cast(value="const char***") @ByPtrPtr PointerPointer arg_descriptions, @Cast(value="const char**") @ByPtrPtr byte[] return_type)
public static int NNSymbolCreateAtomicSymbol(OpHandle op, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") PointerPointer keys, @Cast(value="const char**") PointerPointer vals, PointerPointer out)
op
- The operator handlenum_param
- the number of parameterskeys
- the keys to the paramsvals
- the vals of the paramsout
- pointer to the created symbol handlepublic static int NNSymbolCreateAtomicSymbol(OpHandle op, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, PointerPointer out)
public static int NNSymbolCreateAtomicSymbol(OpHandle op, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int NNSymbolCreateAtomicSymbol(OpHandle op, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, PointerPointer out)
public static int NNSymbolCreateAtomicSymbol(OpHandle op, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer vals, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int NNSymbolCreateAtomicSymbol(OpHandle op, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer vals, PointerPointer out)
public static int NNSymbolCreateAtomicSymbol(OpHandle op, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] vals, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int NNSymbolCreateVariable(@Cast(value="const char*") BytePointer name, PointerPointer out)
name
- name of the variableout
- pointer to the created symbol handlepublic static int NNSymbolCreateVariable(String name, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int NNSymbolCreateGroup(@Cast(value="nn_uint") int num_symbols, PointerPointer symbols, PointerPointer out)
num_symbols
- number of symbols to be groupedsymbols
- array of symbol handlesout
- pointer to the created symbol handlepublic static int NNSymbolCreateGroup(@Cast(value="nn_uint") int num_symbols, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle symbols, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int NNAddControlDeps(SymbolHandle handle, SymbolHandle src_dep)
handle
- The symbol to add dependency edges on.src_dep
- the source handles.public static int NNSymbolFree(SymbolHandle symbol)
symbol
- the symbolpublic static int NNSymbolCopy(SymbolHandle symbol, PointerPointer out)
symbol
- the source symbolout
- used to hold the result of copypublic static int NNSymbolCopy(SymbolHandle symbol, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int NNSymbolPrint(SymbolHandle symbol, @Cast(value="const char**") PointerPointer out_str)
symbol
- the symbolout_str
- pointer to hold the output string of the printing.public static int NNSymbolPrint(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr BytePointer out_str)
public static int NNSymbolPrint(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr ByteBuffer out_str)
public static int NNSymbolPrint(SymbolHandle symbol, @Cast(value="const char**") @ByPtrPtr byte[] out_str)
public static int NNSymbolGetAttr(SymbolHandle symbol, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") PointerPointer out, IntPointer success)
symbol
- the source symbolkey
- The key of the symbol.out
- The result attribute, can be NULL if the attribute do not exist.success
- Whether the result is contained in out.public static int NNSymbolGetAttr(SymbolHandle symbol, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") @ByPtrPtr BytePointer out, IntPointer success)
public static int NNSymbolGetAttr(SymbolHandle symbol, String key, @Cast(value="const char**") @ByPtrPtr ByteBuffer out, IntBuffer success)
public static int NNSymbolGetAttr(SymbolHandle symbol, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") @ByPtrPtr byte[] out, int[] success)
public static int NNSymbolGetAttr(SymbolHandle symbol, String key, @Cast(value="const char**") @ByPtrPtr BytePointer out, IntPointer success)
public static int NNSymbolGetAttr(SymbolHandle symbol, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") @ByPtrPtr ByteBuffer out, IntBuffer success)
public static int NNSymbolGetAttr(SymbolHandle symbol, String key, @Cast(value="const char**") @ByPtrPtr byte[] out, int[] success)
public static int NNSymbolSetAttrs(SymbolHandle symbol, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") PointerPointer keys, @Cast(value="const char**") PointerPointer values)
symbol
- the source symbolnum_param
- Number of parameters to set.keys
- The keys of the attributevalues
- The value to be setpublic static int NNSymbolSetAttrs(SymbolHandle symbol, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="const char**") @ByPtrPtr BytePointer values)
public static int NNSymbolSetAttrs(SymbolHandle symbol, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="const char**") @ByPtrPtr ByteBuffer values)
public static int NNSymbolSetAttrs(SymbolHandle symbol, @Cast(value="nn_uint") int num_param, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="const char**") @ByPtrPtr byte[] values)
public static int NNSymbolListAttrs(SymbolHandle symbol, int recursive_option, @Cast(value="nn_uint*") IntPointer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out)
symbol
- the source symbolrecursive_option
- 0 for recursive, 1 for shallow.out_size
- The number of output attributesout
- 2*out_size strings representing key value pairs.public static int NNSymbolListAttrs(SymbolHandle symbol, int recursive_option, @Cast(value="nn_uint*") IntBuffer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out)
public static int NNSymbolListAttrs(SymbolHandle symbol, int recursive_option, @Cast(value="nn_uint*") int[] out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out)
public static int NNSymbolListInputVariables(SymbolHandle symbol, int option, @Cast(value="nn_uint*") IntPointer out_size, @ByPtrPtr PointerPointer out_sym_array)
symbol
- the symboloption
- The option to list the inputs
option=0 means list all arguments.
option=1 means list arguments that are readed only by the graph.
option=2 means list arguments that are mutated by the graph.out_size
- output sizeout_sym_array
- the output array.public static int NNSymbolListInputVariables(SymbolHandle symbol, int option, @Cast(value="nn_uint*") IntBuffer out_size, @Cast(value="SymbolHandle**") @ByPtrPtr PointerPointer out_sym_array)
public static int NNSymbolListInputVariables(SymbolHandle symbol, int option, @Cast(value="nn_uint*") int[] out_size, @ByPtrPtr PointerPointer out_sym_array)
public static int NNSymbolListInputNames(SymbolHandle symbol, int option, @Cast(value="nn_uint*") IntPointer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
symbol
- the symboloption
- The option to list the inputs
option=0 means list all arguments.
option=1 means list arguments that are readed only by the graph.
option=2 means list arguments that are mutated by the graph.out_size
- output sizeout_str_array
- pointer to hold the output string arraypublic static int NNSymbolListInputNames(SymbolHandle symbol, int option, @Cast(value="nn_uint*") IntBuffer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
public static int NNSymbolListInputNames(SymbolHandle symbol, int option, @Cast(value="nn_uint*") int[] out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
public static int NNSymbolListOutputNames(SymbolHandle symbol, @Cast(value="nn_uint*") IntPointer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
symbol
- the symbolout_size
- output sizeout_str_array
- pointer to hold the output string arraypublic static int NNSymbolListOutputNames(SymbolHandle symbol, @Cast(value="nn_uint*") IntBuffer out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
public static int NNSymbolListOutputNames(SymbolHandle symbol, @Cast(value="nn_uint*") int[] out_size, @Cast(value="const char***") @ByPtrPtr PointerPointer out_str_array)
public static int NNSymbolGetNumOutputs(SymbolHandle symbol, @Cast(value="nn_uint*") IntPointer output_count)
symbol
- the symboloutput_count
- number of outputspublic static int NNSymbolGetNumOutputs(SymbolHandle symbol, @Cast(value="nn_uint*") IntBuffer output_count)
public static int NNSymbolGetNumOutputs(SymbolHandle symbol, @Cast(value="nn_uint*") int[] output_count)
public static int NNSymbolGetInternals(SymbolHandle symbol, PointerPointer out)
symbol
- The symbolout
- The output symbol whose outputs are all the internals.public static int NNSymbolGetInternals(SymbolHandle symbol, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int NNSymbolGetChildren(SymbolHandle symbol, PointerPointer out)
symbol
- The symbolout
- The output symbol whose outputs are the direct children.public static int NNSymbolGetChildren(SymbolHandle symbol, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int NNSymbolGetOutput(SymbolHandle symbol, @Cast(value="nn_uint") int index, PointerPointer out)
symbol
- The symbolindex
- the Index of the output.out
- The output symbol whose outputs are the index-th symbol.public static int NNSymbolGetOutput(SymbolHandle symbol, @Cast(value="nn_uint") int index, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle out)
public static int NNSymbolCompose(SymbolHandle sym, @Cast(value="const char*") BytePointer name, @Cast(value="nn_uint") int num_args, @Cast(value="const char**") PointerPointer keys, PointerPointer args)
sym
- the symbol to applyname
- the name of symbolnum_args
- number of argumentskeys
- the key of keyword args (optional)args
- arguments to sympublic static int NNSymbolCompose(SymbolHandle sym, @Cast(value="const char*") BytePointer name, @Cast(value="nn_uint") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, PointerPointer args)
public static int NNSymbolCompose(SymbolHandle sym, String name, @Cast(value="nn_uint") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle args)
public static int NNSymbolCompose(SymbolHandle sym, @Cast(value="const char*") BytePointer name, @Cast(value="nn_uint") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, PointerPointer args)
public static int NNSymbolCompose(SymbolHandle sym, String name, @Cast(value="nn_uint") int num_args, @Cast(value="const char**") @ByPtrPtr BytePointer keys, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle args)
public static int NNSymbolCompose(SymbolHandle sym, @Cast(value="const char*") BytePointer name, @Cast(value="nn_uint") int num_args, @Cast(value="const char**") @ByPtrPtr ByteBuffer keys, PointerPointer args)
public static int NNSymbolCompose(SymbolHandle sym, String name, @Cast(value="nn_uint") int num_args, @Cast(value="const char**") @ByPtrPtr byte[] keys, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle args)
public static int NNGraphCreate(SymbolHandle symbol, @ByPtrPtr GraphHandle graph)
symbol
- The symbol representing the graph.graph
- The graph handle created.public static int NNGraphFree(GraphHandle handle)
handle
- The handle to be freed.public static int NNGraphGetSymbol(GraphHandle graph, PointerPointer symbol)
graph
- The graph handle.symbol
- The corresponding symbolpublic static int NNGraphGetSymbol(GraphHandle graph, @Cast(value="SymbolHandle*") @ByPtrPtr SymbolHandle symbol)
public static int NNGraphSetJSONAttr(GraphHandle handle, @Cast(value="const char*") BytePointer key, @Cast(value="const char*") BytePointer json_value)
handle
- The graph handle.key
- The key to the attribute.json_value
- The value need to be in format [type_name, value],
Where type_name is a registered type string in C++ side via DMLC_JSON_ENABLE_ANY.public static int NNGraphSetJSONAttr(GraphHandle handle, String key, String json_value)
public static int NNGraphGetJSONAttr(GraphHandle handle, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") PointerPointer json_out, IntPointer success)
handle
- The graph handle.key
- The key to the attribute.json_out
- The result attribute, can be NULL if the attribute do not exist.
The json_out is an array of [type_name, value].
Where the type_name is a registered type string in C++ side via DMLC_JSON_ENABLE_ANY.success
- Whether the result is contained in out.public static int NNGraphGetJSONAttr(GraphHandle handle, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") @ByPtrPtr BytePointer json_out, IntPointer success)
public static int NNGraphGetJSONAttr(GraphHandle handle, String key, @Cast(value="const char**") @ByPtrPtr ByteBuffer json_out, IntBuffer success)
public static int NNGraphGetJSONAttr(GraphHandle handle, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") @ByPtrPtr byte[] json_out, int[] success)
public static int NNGraphGetJSONAttr(GraphHandle handle, String key, @Cast(value="const char**") @ByPtrPtr BytePointer json_out, IntPointer success)
public static int NNGraphGetJSONAttr(GraphHandle handle, @Cast(value="const char*") BytePointer key, @Cast(value="const char**") @ByPtrPtr ByteBuffer json_out, IntBuffer success)
public static int NNGraphGetJSONAttr(GraphHandle handle, String key, @Cast(value="const char**") @ByPtrPtr byte[] json_out, int[] success)
public static int NNGraphSetNodeEntryListAttr_(GraphHandle handle, @Cast(value="const char*") BytePointer key, SymbolHandle list)
handle
- The graph handle.key
- The key to the attribute.list
- The symbol whose outputs represents the list of NodeEntry to be passed.public static int NNGraphSetNodeEntryListAttr_(GraphHandle handle, String key, SymbolHandle list)
public static int NNGraphApplyPasses(GraphHandle src, @Cast(value="nn_uint") int num_pass, @Cast(value="const char**") PointerPointer pass_names, @ByPtrPtr GraphHandle dst)
src
- The source graph handle.num_pass
- The number of pass to be applied.pass_names
- The names of the pass.dst
- The result graph.public static int NNGraphApplyPasses(GraphHandle src, @Cast(value="nn_uint") int num_pass, @Cast(value="const char**") @ByPtrPtr BytePointer pass_names, @ByPtrPtr GraphHandle dst)
public static int NNGraphApplyPasses(GraphHandle src, @Cast(value="nn_uint") int num_pass, @Cast(value="const char**") @ByPtrPtr ByteBuffer pass_names, @ByPtrPtr GraphHandle dst)
public static int NNGraphApplyPasses(GraphHandle src, @Cast(value="nn_uint") int num_pass, @Cast(value="const char**") @ByPtrPtr byte[] pass_names, @ByPtrPtr GraphHandle dst)
Copyright © 2022. All rights reserved.