24 #ifndef MXNET_C_API_H_ 25 #define MXNET_C_API_H_ 34 #define DEFAULT(x) = x 48 #define MXNET_DLL __declspec(dllexport) 50 #define MXNET_DLL __declspec(dllimport) 120 void (*
forward)(int,
float**,
int*,
unsigned**,
int*,
void*);
121 void (*
backward)(int,
float**,
int*,
unsigned**,
int*,
void*);
139 bool (*declare_backward_dependency)(
const int*,
const int*,
const int*,
154 int (**callbacks)(void);
184 const int* ,
const int ,
200 unsigned** ,
const int* ,
204 const char** ,
const char** ,
214 const int* ,
const int ,
281 const char*
const* vals,
282 KVStoreHandle kvstoreHandle);
305 KVStoreHandle kvStoreHandle);
357 int sort_by,
int ascending);
393 const char *task_name,
404 const char *frame_name,
423 const char *counter_name,
471 const char *instant_marker_name,
527 MXNET_DLL int MXLoadTVMOp(
const char *libpath);
529 struct OtherOptionEntity {
533 struct OtherOptionSpace {
534 OtherOptionEntity* entities;
540 char** entity_map_key;
541 OtherOptionEntity* entity_map_val;
543 char** space_map_key;
544 OtherOptionSpace* space_map_val;
547 typedef struct ConfigSpaces {
550 ConfigSpace* spaces_val;
553 MXNET_DLL int MXLoadTVMConfig(ConfigSpaces config);
554 #endif // MXNET_USE_TVM_OP 650 const uint32_t *shape,
659 const uint32_t *aux_shape,
682 const int64_t *shape,
691 const int64_t *aux_shape,
713 const char **out_buf);
737 NDArrayHandle** out_arr,
738 uint32_t *out_name_size,
739 const char*** out_names);
758 NDArrayHandle** out_arr,
759 uint32_t *out_name_size,
760 const char*** out_names);
799 const NDArrayHandle handle_src,
850 uint32_t slice_begin,
899 int *out_storage_type);
937 const uint32_t **out_pdata);
950 const int **out_pdata);
963 const int64_t **out_pdata);
983 DLManagedTensorHandle *out_dlpack);
1001 NDArrayHandle *out_handle);
1018 const bool transient_handle,
1019 NDArrayHandle *out_handle);
1072 NDArrayHandle *out);
1083 NDArrayHandle *out);
1091 NDArrayHandle *out);
1139 FunctionHandle **out_array);
1148 FunctionHandle *out);
1163 const char **description,
1165 const char ***arg_names,
1166 const char ***arg_type_infos,
1167 const char ***arg_descriptions,
1168 const char **return_type
DEFAULT(NULL));
1180 uint32_t *num_use_vars,
1181 uint32_t *num_scalars,
1182 uint32_t *num_mutate_vars,
1195 NDArrayHandle *use_vars,
1197 NDArrayHandle *mutate_vars);
1212 NDArrayHandle *use_vars,
1214 NDArrayHandle *mutate_vars,
1232 NDArrayHandle *inputs,
1234 NDArrayHandle **outputs,
1236 const char **param_keys,
1237 const char **param_vals);
1253 NDArrayHandle *inputs,
1255 NDArrayHandle **outputs,
1257 const char **param_keys,
1258 const char **param_vals,
1259 const int **out_stypes);
1307 NDArrayHandle *var_handles,
1308 uint32_t *reqs_array,
1309 NDArrayHandle *grad_handles);
1317 NDArrayHandle* output_handles);
1327 NDArrayHandle* output_handles,
1328 NDArrayHandle* ograd_handles,
1342 NDArrayHandle *output_handles,
1343 NDArrayHandle *ograd_handles,
1344 uint32_t num_variables,
1345 NDArrayHandle *var_handles,
1349 NDArrayHandle **grad_handles,
1368 CachedOpHandle *out);
1378 CachedOpHandle *out,
1379 bool thread_safe
DEFAULT(
false));
1391 NDArrayHandle *inputs,
1393 NDArrayHandle **outputs);
1407 NDArrayHandle *inputs,
1409 NDArrayHandle **outputs,
1410 const int** out_stypes);
1429 const char ***out_array);
1438 AtomicSymbolCreator **out_array);
1488 const char **description,
1490 const char ***arg_names,
1491 const char ***arg_type_infos,
1492 const char ***arg_descriptions,
1493 const char **key_var_num_args,
1494 const char **return_type
DEFAULT(NULL));
1524 SymbolHandle *symbols,
1651 const char ***out_str_array);
1662 const char ***out_str_array);
1671 uint32_t *output_count);
1709 const char ***out_str_array);
1729 SymbolHandle* args);
1769 const uint32_t *arg_ind_ptr,
1770 const uint32_t *arg_shape_data,
1771 uint32_t *in_shape_size,
1772 const uint32_t **in_shape_ndim,
1773 const uint32_t ***in_shape_data,
1774 uint32_t *out_shape_size,
1775 const uint32_t **out_shape_ndim,
1776 const uint32_t ***out_shape_data,
1777 uint32_t *aux_shape_size,
1778 const uint32_t **aux_shape_ndim,
1779 const uint32_t ***aux_shape_data,
1808 const uint32_t *arg_ind_ptr,
1809 const int *arg_shape_data,
1810 uint32_t *in_shape_size,
1811 const int **in_shape_ndim,
1812 const int ***in_shape_data,
1813 uint32_t *out_shape_size,
1814 const int **out_shape_ndim,
1815 const int ***out_shape_data,
1816 uint32_t *aux_shape_size,
1817 const int **aux_shape_ndim,
1818 const int ***aux_shape_data,
1847 const int64_t *arg_ind_ptr,
1848 const int64_t *arg_shape_data,
1849 size_t *in_shape_size,
1850 const int **in_shape_ndim,
1851 const int64_t ***in_shape_data,
1852 size_t *out_shape_size,
1853 const int **out_shape_ndim,
1854 const int64_t ***out_shape_data,
1855 size_t *aux_shape_size,
1856 const int **aux_shape_ndim,
1857 const int64_t ***aux_shape_data,
1888 const uint32_t *arg_ind_ptr,
1889 const uint32_t *arg_shape_data,
1890 uint32_t *in_shape_size,
1891 const uint32_t **in_shape_ndim,
1892 const uint32_t ***in_shape_data,
1893 uint32_t *out_shape_size,
1894 const uint32_t **out_shape_ndim,
1895 const uint32_t ***out_shape_data,
1896 uint32_t *aux_shape_size,
1897 const uint32_t **aux_shape_ndim,
1898 const uint32_t ***aux_shape_data,
1930 const uint32_t *arg_ind_ptr,
1931 const int *arg_shape_data,
1932 uint32_t *in_shape_size,
1933 const int **in_shape_ndim,
1934 const int ***in_shape_data,
1935 uint32_t *out_shape_size,
1936 const int **out_shape_ndim,
1937 const int ***out_shape_data,
1938 uint32_t *aux_shape_size,
1939 const int **aux_shape_ndim,
1940 const int ***aux_shape_data,
1972 const int64_t *arg_ind_ptr,
1973 const int64_t *arg_shape_data,
1974 size_t *in_shape_size,
1975 const int **in_shape_ndim,
1976 const int64_t ***in_shape_data,
1977 size_t *out_shape_size,
1978 const int **out_shape_ndim,
1979 const int64_t ***out_shape_data,
1980 size_t *aux_shape_size,
1981 const int **aux_shape_ndim,
1982 const int64_t ***aux_shape_data,
2006 const int *arg_type_data,
2007 uint32_t *in_type_size,
2008 const int **in_type_data,
2009 uint32_t *out_type_size,
2010 const int **out_type_data,
2011 uint32_t *aux_type_size,
2012 const int **aux_type_data,
2038 const int *arg_type_data,
2039 uint32_t *in_type_size,
2040 const int **in_type_data,
2041 uint32_t *out_type_size,
2042 const int **out_type_data,
2043 uint32_t *aux_type_size,
2044 const int **aux_type_data,
2066 SymbolHandle *ret_sym_handle,
2067 const int* dev_type,
2068 const uint32_t num_excluded_sym_names,
2069 const char **excluded_sym_names,
2070 const uint32_t num_excluded_op_names,
2071 const char **excluded_op_names,
2072 const uint32_t num_offline,
const char **offline_params,
2073 const char *quantized_dtype,
const bool calib_quantize,
2074 const char *quantize_mode,
const char *quantize_granularity,
2075 uint32_t* out_num_calib_names,
const char ***out_calib_names);
2104 SymbolHandle *ret_sym_handle,
2106 const int* arg_type_data,
2107 uint32_t num_ind_ptr,
2109 const int* target_dtype,
2110 const int cast_optional_params,
2111 const uint32_t num_target_dtype_op_names,
2112 const uint32_t num_fp32_op_names,
2113 const uint32_t num_widest_dtype_op_names,
2114 const uint32_t num_conditional_fp32_op_names,
2115 const uint32_t num_excluded_symbols,
2116 const uint32_t num_model_params,
2117 const char **target_dtype_op_names,
2118 const char **fp32_op_names,
2119 const char **widest_dtype_op_names,
2120 const char **conditional_fp32_op_names,
2121 const char **excluded_symbols,
2122 const char **conditional_param_names,
2123 const char **conditional_param_vals,
2124 const char **model_param_names,
2125 const char **arg_names);
2136 const uint32_t num_layers,
2137 const char** layer_names,
2138 const float* low_quantiles,
2139 const float* high_quantiles,
2140 SymbolHandle* ret_sym_handle);
2149 SymbolHandle *ret_sym_handle);
2189 const char* backend_name,
2191 SymbolHandle* ret_sym_handle,
2192 const mx_uint args_len,
2193 NDArrayHandle* in_args_handle,
2194 const mx_uint aux_len,
2195 NDArrayHandle* in_aux_handle,
2196 const mx_uint num_options,
2199 const uint32_t num_input_shapes,
2200 const char** input_shape_names,
2201 const int64_t* input_shape_data,
2202 const uint32_t* input_shape_idx,
2203 const uint32_t num_input_dtypes,
2204 const char** input_dtype_names,
2205 const int* input_dtypes,
2206 const uint32_t num_input_stypes,
2207 const char** input_stype_names,
2208 const int* input_stypes,
2211 NDArrayHandle** new_args_handle,
2212 char*** new_arg_names_handle,
2214 NDArrayHandle** new_aux_handle,
2215 char*** new_aux_names_handle);
2253 NDArrayHandle *head_grads);
2266 NDArrayHandle *head_grads,
2278 NDArrayHandle **out);
2299 NDArrayHandle *in_args,
2300 NDArrayHandle *arg_grad_store,
2301 uint32_t *grad_req_type,
2302 uint32_t aux_states_len,
2303 NDArrayHandle *aux_states,
2304 ExecutorHandle *out);
2329 uint32_t num_map_keys,
2330 const char** map_keys,
2331 const int* map_dev_types,
2332 const int* map_dev_ids,
2334 NDArrayHandle *in_args,
2335 NDArrayHandle *arg_grad_store,
2336 uint32_t *grad_req_type,
2337 uint32_t aux_states_len,
2338 NDArrayHandle *aux_states,
2339 ExecutorHandle *out);
2365 uint32_t num_map_keys,
2366 const char** map_keys,
2367 const int* map_dev_types,
2368 const int* map_dev_ids,
2370 NDArrayHandle *in_args,
2371 NDArrayHandle *arg_grad_store,
2372 uint32_t *grad_req_type,
2373 uint32_t aux_states_len,
2374 NDArrayHandle *aux_states,
2375 ExecutorHandle shared_exec,
2376 ExecutorHandle *out);
2382 const uint32_t num_g2c_keys,
2383 const char** g2c_keys,
2384 const int* g2c_dev_types,
2385 const int* g2c_dev_ids,
2386 const uint32_t provided_grad_req_list_len,
2387 const char** provided_grad_req_names,
2388 const char** provided_grad_req_types,
2389 const uint32_t num_provided_arg_shapes,
2390 const char** provided_arg_shape_names,
2391 const uint32_t* provided_arg_shape_data,
2392 const uint32_t* provided_arg_shape_idx,
2393 const uint32_t num_provided_arg_dtypes,
2394 const char** provided_arg_dtype_names,
2395 const int* provided_arg_dtypes,
2396 const uint32_t num_provided_arg_stypes,
2397 const char** provided_arg_stype_names,
2398 const int* provided_arg_stypes,
2399 const uint32_t num_shared_arg_names,
2400 const char** shared_arg_name_list,
2401 int* shared_buffer_len,
2402 const char** shared_buffer_name_list,
2403 NDArrayHandle* shared_buffer_handle_list,
2404 const char*** updated_shared_buffer_name_list,
2405 NDArrayHandle** updated_shared_buffer_handle_list,
2406 uint32_t* num_in_args,
2407 NDArrayHandle** in_args,
2408 NDArrayHandle** arg_grads,
2409 uint32_t* num_aux_states,
2410 NDArrayHandle** aux_states,
2411 ExecutorHandle shared_exec_handle,
2412 ExecutorHandle* out);
2418 const uint32_t num_g2c_keys,
2419 const char** g2c_keys,
2420 const int* g2c_dev_types,
2421 const int* g2c_dev_ids,
2422 const uint32_t provided_grad_req_list_len,
2423 const char** provided_grad_req_names,
2424 const char** provided_grad_req_types,
2425 const uint32_t num_provided_arg_shapes,
2426 const char** provided_arg_shape_names,
2427 const int* provided_arg_shape_data,
2428 const uint32_t* provided_arg_shape_idx,
2429 const uint32_t num_provided_arg_dtypes,
2430 const char** provided_arg_dtype_names,
2431 const int* provided_arg_dtypes,
2432 const uint32_t num_provided_arg_stypes,
2433 const char** provided_arg_stype_names,
2434 const int* provided_arg_stypes,
2435 const uint32_t num_shared_arg_names,
2436 const char** shared_arg_name_list,
2437 int* shared_buffer_len,
2438 const char** shared_buffer_name_list,
2439 NDArrayHandle* shared_buffer_handle_list,
2440 const char*** updated_shared_buffer_name_list,
2441 NDArrayHandle** updated_shared_buffer_handle_list,
2442 uint32_t* num_in_args,
2443 NDArrayHandle** in_args,
2444 NDArrayHandle** arg_grads,
2445 uint32_t* num_aux_states,
2446 NDArrayHandle** aux_states,
2447 ExecutorHandle shared_exec_handle,
2448 ExecutorHandle* out);
2454 const uint32_t num_g2c_keys,
2455 const char** g2c_keys,
2456 const int* g2c_dev_types,
2457 const int* g2c_dev_ids,
2458 const uint32_t provided_grad_req_list_len,
2459 const char** provided_grad_req_names,
2460 const char** provided_grad_req_types,
2461 const uint32_t num_provided_arg_shapes,
2462 const char** provided_arg_shape_names,
2463 const int64_t* provided_arg_shape_data,
2464 const uint32_t* provided_arg_shape_idx,
2465 const uint32_t num_provided_arg_dtypes,
2466 const char** provided_arg_dtype_names,
2467 const int* provided_arg_dtypes,
2468 const uint32_t num_provided_arg_stypes,
2469 const char** provided_arg_stype_names,
2470 const int* provided_arg_stypes,
2471 const uint32_t num_shared_arg_names,
2472 const char** shared_arg_name_list,
2473 int* shared_buffer_len,
2474 const char** shared_buffer_name_list,
2475 NDArrayHandle* shared_buffer_handle_list,
2476 const char*** updated_shared_buffer_name_list,
2477 NDArrayHandle** updated_shared_buffer_handle_list,
2478 uint32_t* num_in_args,
2479 NDArrayHandle** in_args,
2480 NDArrayHandle** arg_grads,
2481 uint32_t* num_aux_states,
2482 NDArrayHandle** aux_states,
2483 ExecutorHandle shared_exec_handle,
2484 ExecutorHandle* out);
2510 int allow_up_sizing,
2513 uint32_t num_map_keys,
2514 const char** map_keys,
2515 const int* map_dev_types,
2516 const int* map_dev_ids,
2517 const uint32_t num_provided_arg_shapes,
2518 const char** provided_arg_shape_names,
2519 const uint32_t* provided_arg_shape_data,
2520 const uint32_t* provided_arg_shape_idx,
2521 uint32_t* num_in_args,
2522 NDArrayHandle** in_args,
2523 NDArrayHandle** arg_grads,
2524 uint32_t* num_aux_states,
2525 NDArrayHandle** aux_states,
2526 ExecutorHandle shared_exec,
2527 ExecutorHandle *out);
2550 int allow_up_sizing,
2553 uint32_t num_map_keys,
2554 const char** map_keys,
2555 const int* map_dev_types,
2556 const int* map_dev_ids,
2557 const uint32_t num_provided_arg_shapes,
2558 const char** provided_arg_shape_names,
2559 const int* provided_arg_shape_data,
2560 const uint32_t* provided_arg_shape_idx,
2561 uint32_t* num_in_args,
2562 NDArrayHandle** in_args,
2563 NDArrayHandle** arg_grads,
2564 uint32_t* num_aux_states,
2565 NDArrayHandle** aux_states,
2566 ExecutorHandle shared_exec,
2567 ExecutorHandle *out);
2579 void* callback_handle);
2587 void *callback_handle,
bool monitor_all);
2598 DataIterCreator **out_array);
2613 DataIterHandle *out);
2627 const char **description,
2629 const char ***arg_names,
2630 const char ***arg_type_infos,
2631 const char ***arg_descriptions);
2660 NDArrayHandle *out);
2669 uint64_t **out_index,
2670 uint64_t *out_size);
2687 NDArrayHandle *out);
2709 KVStoreHandle *out);
2719 uint32_t num_params,
2740 NDArrayHandle* vals);
2753 NDArrayHandle* vals);
2767 NDArrayHandle* vals,
2781 NDArrayHandle* vals,
2796 NDArrayHandle* vals,
2798 bool ignore_sparse);
2812 NDArrayHandle* vals,
2814 bool ignore_sparse);
2827 NDArrayHandle* vals,
2841 NDArrayHandle* vals,
2859 NDArrayHandle* vals,
2860 const NDArrayHandle* row_ids,
2877 NDArrayHandle* vals,
2878 const NDArrayHandle* row_ids,
2898 NDArrayHandle* vals,
2899 NDArrayHandle* outs,
2919 NDArrayHandle* vals,
2920 NDArrayHandle* outs,
2940 NDArrayHandle* vals,
2941 NDArrayHandle* outs,
2961 NDArrayHandle* vals,
2962 NDArrayHandle* outs,
2975 NDArrayHandle local,
2987 NDArrayHandle local,
2998 void *updater_handle);
3010 void *updater_handle);
3084 const int barrier_before_exit);
3094 void *controller_handle);
3105 void *controller_handle);
3116 const char* cmd_body);
3131 const int timeout_sec
DEFAULT(60));
3156 const char *buf,
size_t size);
3189 char const **buf,
size_t *size);
3211 char** input_names,
char** output_names,
3212 NDArrayHandle* inputs, NDArrayHandle* outputs,
3213 char* kernel, RtcHandle *out);
3219 NDArrayHandle* inputs, NDArrayHandle* outputs,
3225 uint32_t blockDimZ);
3246 int num_outputs, NDArrayHandle *outputs,
3258 const char** options,
int num_exports,
3259 const char** exports, CudaModuleHandle *out);
3276 int num_args,
int* is_ndarray,
int* is_const,
3277 int* arg_types, CudaKernelHandle *out);
3297 uint32_t grid_dim_x, uint32_t grid_dim_y,
3298 uint32_t grid_dim_z, uint32_t block_dim_x,
3299 uint32_t block_dim_y, uint32_t block_dim_z,
3300 uint32_t shared_mem);
3320 uint32_t ndim,
int dtype, NDArrayHandle *out);
3339 int ndim,
int dtype, NDArrayHandle *out);
3360 EngineVarHandle const_vars_handle,
int num_const_vars,
3361 EngineVarHandle mutable_vars_handle,
int num_mutable_vars,
3362 EngineFnPropertyHandle prop_handle
DEFAULT(NULL),
3383 EngineVarHandle const_vars_handle,
int num_const_vars,
3384 EngineVarHandle mutable_vars_handle,
int num_mutable_vars,
3385 EngineFnPropertyHandle prop_handle
DEFAULT(NULL),
3419 NDArrayHandle* const_nds_handle,
int num_const_nds,
3420 NDArrayHandle* mutable_nds_handle,
int num_mutable_nds,
3421 EngineFnPropertyHandle prop_handle
DEFAULT(NULL),
3442 NDArrayHandle* const_nds_handle,
int num_const_nds,
3443 NDArrayHandle* mutable_nds_handle,
int num_mutable_nds,
3444 EngineFnPropertyHandle prop_handle
DEFAULT(NULL),
3449 #endif // __cplusplus 3451 #endif // MXNET_C_API_H_ MXNET_DLL int MXKVStoreCreate(const char *type, KVStoreHandle *out)
Create a kvstore.
int(* CustomFunctionDelFunc)(void *)
Definition: c_api.h:216
MXNET_DLL int MXSymbolListAttr(SymbolHandle symbol, uint32_t *out_size, const char ***out)
Get all attributes from symbol, including all descendents.
MXNET_DLL int MXProfileAdjustCounter(ProfileHandle counter_handle, int64_t value)
Adjust a counter by the given amount, given its handle.
MXNET_DLL int MXProfileCreateFrame(ProfileHandle domain, const char *frame_name, ProfileHandle *out)
Create profile frame.
MXNET_DLL int MXSymbolRemoveAmpCast(SymbolHandle sym_handle, SymbolHandle *ret_sym_handle)
Remove the operators amp_cast and amp_multicast.
MXNET_DLL int MXDataIterGetLabel(DataIterHandle handle, NDArrayHandle *out)
Get the handle to the NDArray of underlying label.
MXNET_DLL int MXNDArrayCallDLPackDeleter(DLManagedTensorHandle dlpack)
Delete a dlpack tensor.
MXNET_DLL int MXNDArraySetGradState(NDArrayHandle handle, int state)
set the flag for gradient array state.
MXNET_DLL int MXNDArrayAt64(NDArrayHandle handle, int64_t idx, NDArrayHandle *out)
Index the NDArray along axis 0. This api is available when MXNet is built with flag USE_INT64_TENSOR_...
MXNET_DLL int MXSymbolListAuxiliaryStates(SymbolHandle symbol, uint32_t *out_size, const char ***out_str_array)
List auxiliary states in the symbol.
MXNET_DLL int MXSymbolListAttrShallow(SymbolHandle symbol, uint32_t *out_size, const char ***out)
Get all attributes from symbol, excluding descendents.
MXNET_DLL int MXProfilePause(int paused)
Pause profiler tuning collection for worker/current process.
MXNET_DLL int MXSetCalibTableToQuantizedSymbol(SymbolHandle qsym_handle, const uint32_t num_layers, const char **layer_names, const float *low_quantiles, const float *high_quantiles, SymbolHandle *ret_sym_handle)
Set calibration table to node attributes in the sym.
MXNET_DLL int MXAutogradComputeGradient(uint32_t num_output, NDArrayHandle *output_handles)
compute the gradient of outputs w.r.t variabels
MXNET_DLL int MXNDArrayGetShapeEx(NDArrayHandle handle, int *out_dim, const int **out_pdata)
get the shape of the array This api is available when MXNet is built with flag USE_INT64_TENSOR_SIZE=...
MXNET_DLL int MXRecordIOReaderSeek(RecordIOHandle handle, size_t pos)
Set the current reader pointer position.
void * DataIterCreator
handle a dataiter creator
Definition: c_api.h:80
int(* CustomOpInferShapeFunc)(int, int *, int **, void *)
Definition: c_api.h:188
MXNET_DLL int MXDataIterGetIndex(DataIterHandle handle, uint64_t **out_index, uint64_t *out_size)
Get the image index by array.
MXNET_DLL int MXRtcPush(RtcHandle handle, uint32_t num_input, uint32_t num_output, NDArrayHandle *inputs, NDArrayHandle *outputs, uint32_t gridDimX, uint32_t gridDimY, uint32_t gridDimZ, uint32_t blockDimX, uint32_t blockDimY, uint32_t blockDimZ)
Run cuda kernel.
MXNET_DLL int MXKVStoreIsServerNode(int *ret)
return whether or not this process is a server node.
void * ExecutorHandle
handle to an Executor
Definition: c_api.h:78
MXNET_DLL int MXProfileCreateEvent(const char *event_name, ProfileHandle *out)
Create profile event.
MXNET_DLL int MXKVStorePull(KVStoreHandle handle, uint32_t num, const int *keys, NDArrayHandle *vals, int priority)
pull a list of (key, value) pairs from the kvstore
void * p_infer_shape
Definition: c_api.h:144
MXNET_DLL int MXNDArraySaveRawBytes(NDArrayHandle handle, size_t *out_size, const char **out_buf)
save the NDArray into raw bytes.
PaddingExp< SrcExp, DType, ExpInfo< SrcExp >::kDim > pad(const Exp< SrcExp, DType, etype > &src, index_t pad)
padding expression, pad a image with zeros on boundaries, padding affects shape[0], and shape[1]
Definition: pad.h:71
MXNET_DLL int MXAutogradMarkVariables(uint32_t num_var, NDArrayHandle *var_handles, uint32_t *reqs_array, NDArrayHandle *grad_handles)
mark NDArrays as variables to compute gradient for autograd
void(* list_arguments)(char ***, void *)
Definition: c_api.h:124
void * SymbolHandle
handle to a symbol that can be bind as operator
Definition: c_api.h:74
MXNET_DLL int MXNDArrayGetDType(NDArrayHandle handle, int *out_dtype)
get the type of the data in NDArray
MXNET_DLL int MXRtcCudaKernelCall(CudaKernelHandle handle, int dev_id, void **args, uint32_t grid_dim_x, uint32_t grid_dim_y, uint32_t grid_dim_z, uint32_t block_dim_x, uint32_t block_dim_y, uint32_t block_dim_z, uint32_t shared_mem)
MXNET_DLL int MXAggregateProfileStatsPrint(const char **out_str, int reset)
Deprecated, use MXAggregateProfileStatsPrintEx instead.
MXNET_DLL int MXSymbolInferShapePartialEx64(SymbolHandle sym, uint32_t num_args, const char **keys, const int64_t *arg_ind_ptr, const int64_t *arg_shape_data, size_t *in_shape_size, const int **in_shape_ndim, const int64_t ***in_shape_data, size_t *out_shape_size, const int **out_shape_ndim, const int64_t ***out_shape_data, size_t *aux_shape_size, const int **aux_shape_ndim, const int64_t ***aux_shape_data, int *complete)
partially infer shape of unknown input shapes given the known one.
float mx_float
manually define float
Definition: c_api.h:59
MXNET_DLL int MXNDArraySlice(NDArrayHandle handle, uint32_t slice_begin, uint32_t slice_end, NDArrayHandle *out)
Slice the NDArray along axis 0. This api is available when MXNet is built with flag USE_INT64_TENSOR_...
MXNET_DLL int MXSetProfilerState(int state)
Set up state of profiler for current process.
int(* CustomOpBwdDepFunc)(const int *, const int *, const int *, int *, int **, void *)
Definition: c_api.h:196
MXNET_DLL int MXSymbolCutSubgraph(SymbolHandle sym, SymbolHandle **inputs, int *input_size)
Cut a subgraph whose nodes are marked with a subgraph attribute. The input graph will be modified...
MXNET_DLL int MXNDArraySyncCheckFormat(NDArrayHandle handle, const bool full_check)
check whether the NDArray format is valid
MXNET_DLL int MXGenBackendSubgraph(SymbolHandle sym_handle, const char *backend, SymbolHandle *ret_sym_handle)
Run subgraph pass based on the backend provided.
MXNET_DLL int MXNDArrayCreateEx64(const int64_t *shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, NDArrayHandle *out)
create a NDArray with specified shape and data type This api is available when MXNet is built with fl...
int(* CustomFunctionBwdFunc)(int, int, void **, const int *, const int, void *)
Definition: c_api.h:213
void * p_forward
Definition: c_api.h:142
void * ProfileHandle
handle to a Profile object (domain, duration, counter, etc.)
Definition: c_api.h:94
MXNET_DLL int MXAutogradBackward(uint32_t num_output, NDArrayHandle *output_handles, NDArrayHandle *ograd_handles, int retain_graph)
compute the gradient of outputs w.r.t variabels
MXNET_DLL int MXCachedOpRegisterOpHook(NDArrayHandle handle, CachedOpMonitorCallback callback, bool monitor_all)
cached op set monitor callback
MXNET_DLL int MXSymbolGetNumOutputs(SymbolHandle symbol, uint32_t *output_count)
Get number of outputs of the symbol.
MXNET_DLL int MXProfileDestroyHandle(ProfileHandle frame_handle)
Destroy a frame.
CustomFunctionCallbacks
Definition: c_api.h:208
MXNET_DLL int MXSymbolGetInputSymbols(SymbolHandle sym, SymbolHandle **inputs, int *input_size)
Get the input symbols of the graph.
MXNET_DLL int MXExecutorSetMonitorCallbackEX(ExecutorHandle handle, ExecutorMonitorCallback callback, void *callback_handle, bool monitor_all)
set a call back to notify the completion of operation
MXNET_DLL int MXExecutorBindEX(SymbolHandle symbol_handle, int dev_type, int dev_id, uint32_t num_map_keys, const char **map_keys, const int *map_dev_types, const int *map_dev_ids, uint32_t len, NDArrayHandle *in_args, NDArrayHandle *arg_grad_store, uint32_t *grad_req_type, uint32_t aux_states_len, NDArrayHandle *aux_states, ExecutorHandle shared_exec, ExecutorHandle *out)
Generate Executor from symbol, This is advanced function, allow specify group2ctx map...
MXNET_DLL int MXProfileSetMarker(ProfileHandle domain, const char *instant_marker_name, const char *scope)
Mark a single instant in time.
MXNET_DLL int MXKVStoreFree(KVStoreHandle handle)
Delete a KVStore handle.
MXNET_DLL int MXKVStoreBroadcastEx(KVStoreHandle handle, mx_uint vnum, const char **vkeys, mx_uint onum, const char **okeys, NDArrayHandle *vals, NDArrayHandle *outs, int priority)
broadcast a list of (key, value) pairs from the kvstore, where each key is a string ...
MXNET_DLL int MXShallowCopySymbol(SymbolHandle src, SymbolHandle *out)
Create an Symbol from source sharing the same graph structure.
MXNET_DLL int MXRtcCudaModuleFree(CudaModuleHandle handle)
MXNET_DLL int MXIsNumpyShape(int *curr)
get whether numpy compatibility is on
MXNET_DLL int MXKVStorePullRowSparse(KVStoreHandle handle, uint32_t num, const int *keys, NDArrayHandle *vals, const NDArrayHandle *row_ids, int priority)
pull a list of (key, value) pairs from the kvstore, where each key is an integer. The NDArray pulled ...
void * AtomicSymbolCreator
handle to a function that takes param and creates symbol
Definition: c_api.h:70
MXNET_DLL int MXNDArrayGetAuxType(NDArrayHandle handle, uint32_t i, int *out_type)
get the type of the ith aux data in NDArray This api is available when MXNet is built with flag USE_I...
MXNET_DLL int MXRtcFree(RtcHandle handle)
Delete a MXRtc object.
MXNET_DLL int MXFuncDescribe(FunctionHandle fun, uint32_t *num_use_vars, uint32_t *num_scalars, uint32_t *num_mutate_vars, int *type_mask)
get the argument requirements of the function
void * p_backward
Definition: c_api.h:127
MXNET_DLL int MXCreateCachedOpEx(SymbolHandle handle, int num_flags, const char **keys, const char **vals, CachedOpHandle *out)
create cached operator
void() MXKVStoreUpdater(int key, NDArrayHandle recv, NDArrayHandle local, void *handle)
user-defined updater for the kvstore It's this updater's responsibility to delete recv and local ...
Definition: c_api.h:2973
MXNET_DLL int MXSetIsNumpyShape(int is_np_shape, int *prev)
set numpy compatibility switch
MXNET_DLL int MXSymbolPrint(SymbolHandle symbol, const char **out_str)
Print the content of symbol, used for debug.
MXNET_DLL int MXGetGPUMemoryInformation64(int dev, uint64_t *free_mem, uint64_t *total_mem)
get the free and total available memory on a GPU
int(* CustomOpFBFunc)(int, void **, int *, const int *, const int, void *)
Definition: c_api.h:183
MXNET_DLL int MXKVStoreBarrier(KVStoreHandle handle)
global barrier among all worker machines
MXNET_DLL int MXExecutorSimpleBindEx(SymbolHandle symbol_handle, int dev_type, int dev_id, const uint32_t num_g2c_keys, const char **g2c_keys, const int *g2c_dev_types, const int *g2c_dev_ids, const uint32_t provided_grad_req_list_len, const char **provided_grad_req_names, const char **provided_grad_req_types, const uint32_t num_provided_arg_shapes, const char **provided_arg_shape_names, const int *provided_arg_shape_data, const uint32_t *provided_arg_shape_idx, const uint32_t num_provided_arg_dtypes, const char **provided_arg_dtype_names, const int *provided_arg_dtypes, const uint32_t num_provided_arg_stypes, const char **provided_arg_stype_names, const int *provided_arg_stypes, const uint32_t num_shared_arg_names, const char **shared_arg_name_list, int *shared_buffer_len, const char **shared_buffer_name_list, NDArrayHandle *shared_buffer_handle_list, const char ***updated_shared_buffer_name_list, NDArrayHandle **updated_shared_buffer_handle_list, uint32_t *num_in_args, NDArrayHandle **in_args, NDArrayHandle **arg_grads, uint32_t *num_aux_states, NDArrayHandle **aux_states, ExecutorHandle shared_exec_handle, ExecutorHandle *out)
MXNET_DLL int MXDataIterBeforeFirst(DataIterHandle handle)
Call iterator.Reset.
MXNET_DLL int MXNDArrayGetShape(NDArrayHandle handle, uint32_t *out_dim, const uint32_t **out_pdata)
DEPRECATED. Use MXNDArrayGetShapeEx instead. get the shape of the array.
MXNET_DLL int MXNDArrayGetAuxNDArray64(NDArrayHandle handle, int64_t i, NDArrayHandle *out)
Get a deep copy of the ith aux data blob This api is available when MXNet is built with flag USE_INT6...
MXNET_DLL int MXNDArrayCreateEx(const uint32_t *shape, uint32_t ndim, int dev_type, int dev_id, int delay_alloc, int dtype, NDArrayHandle *out)
create a NDArray with specified shape and data type This api is available when MXNet is built with fl...
MXNET_DLL int MXNDArrayDetach(NDArrayHandle handle, NDArrayHandle *out)
detach and ndarray from computation graph by clearing entry_
void(* CachedOpMonitorCallback)(const char *, const char *, NDArrayHandle)
Monitor callback called at operator level for cached op.
Definition: c_api.h:114
MXNET_DLL int MXAggregateProfileStatsPrintEx(const char **out_str, int reset, int format, int sort_by, int ascending)
Print sorted aggregate stats to the a string How aggregate stats are stored will not change...
bool enabled
Definition: c_api.h:160
MXNET_DLL int MXProfileDurationStop(ProfileHandle duration_handle)
Stop timing the duration of a profile duration object such as an event, task or frame.
MXNET_DLL int MXEngineSetBulkSize(int bulk_size, int *prev_bulk_size)
set bulk execution limit
MXNET_DLL int MXNDArrayFree(NDArrayHandle handle)
free the narray handle
MXNET_DLL int MXStorageEmptyCache(int dev_type, int dev_id)
Release all unreferenced memory from the devices storage managers memory pool.
MXNET_DLL int MXKVStoreSetUpdaterEx(KVStoreHandle handle, MXKVStoreUpdater updater, MXKVStoreStrUpdater str_updater, void *updater_handle)
register a push updater with int keys and one with string keys
void(* ExecutorMonitorCallback)(const char *, NDArrayHandle, void *)
Definition: c_api.h:110
MXNET_DLL int MXKVStoreGetGroupSize(KVStoreHandle handle, int *ret)
return The number of nodes in this group, which is
MXNET_DLL int MXSetNumOMPThreads(int thread_num)
Set the number of OMP threads to use.
MXNET_DLL int MXRandomSeed(int seed)
Seed all global random number generators in mxnet.
MXNET_DLL int MXExecutorOutputs(ExecutorHandle handle, uint32_t *out_size, NDArrayHandle **out)
Get executor's head NDArray.
MXNET_DLL int MXKVStorePullEx(KVStoreHandle handle, uint32_t num, const char **keys, NDArrayHandle *vals, int priority)
pull a list of (key, value) pairs from the kvstore, where each key is a string
MXNET_DLL int MXNDArrayAt(NDArrayHandle handle, uint32_t idx, NDArrayHandle *out)
Index the NDArray along axis 0. This api is available when MXNet is built with flag USE_INT64_TENSOR_...
MXNET_DLL int MXProfileCreateDomain(const char *domain, ProfileHandle *out)
Create profiling domain.
MXNET_DLL int MXSymbolFree(SymbolHandle symbol)
Free the symbol handle.
MXNET_DLL int MXSymbolGetAttr(SymbolHandle symbol, const char *key, const char **out, int *success)
Get string attribute from symbol.
MXNET_DLL int MXShallowCopyNDArray(NDArrayHandle src, NDArrayHandle *out)
Create an NDArray from source sharing the same data chunk.
void * p_forward
Definition: c_api.h:126
MXNET_DLL int MXExecutorBackward(ExecutorHandle handle, uint32_t len, NDArrayHandle *head_grads)
Excecutor run backward.
void * p_list_arguments
Definition: c_api.h:130
MXNET_DLL int MXKVStoreInitEx(KVStoreHandle handle, uint32_t num, const char **keys, NDArrayHandle *vals)
Init a list of (key,value) pairs in kvstore, where each key is a string.
MXNET_DLL int MXNDArrayCreate(const uint32_t *shape, uint32_t ndim, int dev_type, int dev_id, int delay_alloc, NDArrayHandle *out)
create a NDArray with specified shape
MXNET_DLL int MXKVStoreGetRank(KVStoreHandle handle, int *ret)
return The rank of this node in its group, which is in [0, GroupSize).
MXNET_DLL int MXNDArrayGetGrad(NDArrayHandle handle, NDArrayHandle *out)
return gradient buffer attached to this NDArray
MXNET_DLL int MXNDArrayGetGradState(NDArrayHandle handle, int *out)
set the flag for gradient array state.
MXNET_DLL int MXRecordIOWriterFree(RecordIOHandle handle)
Delete a RecordIO writer object.
MXNET_DLL int MXExecutorSimpleBindEx64(SymbolHandle symbol_handle, int dev_type, int dev_id, const uint32_t num_g2c_keys, const char **g2c_keys, const int *g2c_dev_types, const int *g2c_dev_ids, const uint32_t provided_grad_req_list_len, const char **provided_grad_req_names, const char **provided_grad_req_types, const uint32_t num_provided_arg_shapes, const char **provided_arg_shape_names, const int64_t *provided_arg_shape_data, const uint32_t *provided_arg_shape_idx, const uint32_t num_provided_arg_dtypes, const char **provided_arg_dtype_names, const int *provided_arg_dtypes, const uint32_t num_provided_arg_stypes, const char **provided_arg_stype_names, const int *provided_arg_stypes, const uint32_t num_shared_arg_names, const char **shared_arg_name_list, int *shared_buffer_len, const char **shared_buffer_name_list, NDArrayHandle *shared_buffer_handle_list, const char ***updated_shared_buffer_name_list, NDArrayHandle **updated_shared_buffer_handle_list, uint32_t *num_in_args, NDArrayHandle **in_args, NDArrayHandle **arg_grads, uint32_t *num_aux_states, NDArrayHandle **aux_states, ExecutorHandle shared_exec_handle, ExecutorHandle *out)
void * p_declare_backward_dependency
Definition: c_api.h:147
MXNET_DLL int MXNDArrayGetData(NDArrayHandle handle, void **out_pdata)
get the content of the data in NDArray
int(* CustomOpInferTypeFunc)(int, int *, void *)
Definition: c_api.h:195
MXNET_DLL int MXKVStoreSendCommmandToServers(KVStoreHandle handle, int cmd_id, const char *cmd_body)
Send a command to all server nodes.
MXNET_DLL int MXFuncInvoke(FunctionHandle fun, NDArrayHandle *use_vars, float *scalar_args, NDArrayHandle *mutate_vars)
invoke a function, the array size of passed in arguments must match the values in the ...
MXNET_DLL int MXExecutorReshapeEx(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, uint32_t num_map_keys, const char **map_keys, const int *map_dev_types, const int *map_dev_ids, const uint32_t num_provided_arg_shapes, const char **provided_arg_shape_names, const int *provided_arg_shape_data, const uint32_t *provided_arg_shape_idx, uint32_t *num_in_args, NDArrayHandle **in_args, NDArrayHandle **arg_grads, uint32_t *num_aux_states, NDArrayHandle **aux_states, ExecutorHandle shared_exec, ExecutorHandle *out)
Return a new executor with the same symbol and shared memory, but different input/output shapes...
void * CudaKernelHandle
handle to rtc cuda kernel
Definition: c_api.h:92
MXNET_DLL int MXRandomSeedContext(int seed, int dev_type, int dev_id)
Seed the global random number generator of the given device.
void * p_list_arguments
Definition: c_api.h:146
MXNET_DLL int MXSymbolGetChildren(SymbolHandle symbol, SymbolHandle *out)
Get a symbol that contains only direct children.
MXNET_DLL int MXKVStoreRunServer(KVStoreHandle handle, MXKVStoreServerController controller, void *controller_handle)
Run as server (or scheduler)
MXNET_DLL int MXInvokeCachedOpEx(CachedOpHandle handle, int num_inputs, NDArrayHandle *inputs, int *num_outputs, NDArrayHandle **outputs, const int **out_stypes)
invoke a cached op
MXNET_DLL int MXGenAtomicSymbolFromSymbol(SymbolHandle sym_handle, SymbolHandle *ret_sym_handle)
Generate atomic symbol (able to be composed) from a source symbol.
MXNET_DLL int MXAutogradIsRecording(bool *curr)
get whether autograd recording is on
MXNET_DLL int MXReducePrecisionSymbol(SymbolHandle sym_handle, SymbolHandle *ret_sym_handle, uint32_t num_args, const int *arg_type_data, uint32_t num_ind_ptr, const int *ind_ptr, const int *target_dtype, const int cast_optional_params, const uint32_t num_target_dtype_op_names, const uint32_t num_fp32_op_names, const uint32_t num_widest_dtype_op_names, const uint32_t num_conditional_fp32_op_names, const uint32_t num_excluded_symbols, const uint32_t num_model_params, const char **target_dtype_op_names, const char **fp32_op_names, const char **widest_dtype_op_names, const char **conditional_fp32_op_names, const char **excluded_symbols, const char **conditional_param_names, const char **conditional_param_vals, const char **model_param_names, const char **arg_names)
Convert a symbol into a mixed precision symbol with cast operators for target dtype casting...
MXNET_DLL int MXNDArrayToDLPack(NDArrayHandle handle, DLManagedTensorHandle *out_dlpack)
Create a reference view of NDArray that represents as DLManagedTensor Notice: MXNet uses asynchronous...
int(* CustomOpPropCreator)(const char *, const int, const char **, const char **, struct MXCallbackList *)
Definition: c_api.h:203
MXNET_DLL int MXFuncInvokeEx(FunctionHandle fun, NDArrayHandle *use_vars, float *scalar_args, NDArrayHandle *mutate_vars, int num_params, char **param_keys, char **param_vals)
invoke a function, the array size of passed in arguments must match the values in the ...
MXNET_DLL int MXLoadLib(const char *path, unsigned verbose)
Load library dynamically.
MXNET_DLL int MXExecutorSimpleBind(SymbolHandle symbol_handle, int dev_type, int dev_id, const uint32_t num_g2c_keys, const char **g2c_keys, const int *g2c_dev_types, const int *g2c_dev_ids, const uint32_t provided_grad_req_list_len, const char **provided_grad_req_names, const char **provided_grad_req_types, const uint32_t num_provided_arg_shapes, const char **provided_arg_shape_names, const uint32_t *provided_arg_shape_data, const uint32_t *provided_arg_shape_idx, const uint32_t num_provided_arg_dtypes, const char **provided_arg_dtype_names, const int *provided_arg_dtypes, const uint32_t num_provided_arg_stypes, const char **provided_arg_stype_names, const int *provided_arg_stypes, const uint32_t num_shared_arg_names, const char **shared_arg_name_list, int *shared_buffer_len, const char **shared_buffer_name_list, NDArrayHandle *shared_buffer_handle_list, const char ***updated_shared_buffer_name_list, NDArrayHandle **updated_shared_buffer_handle_list, uint32_t *num_in_args, NDArrayHandle **in_args, NDArrayHandle **arg_grads, uint32_t *num_aux_states, NDArrayHandle **aux_states, ExecutorHandle shared_exec_handle, ExecutorHandle *out)
DEPRECATED. Use MXExecutorSimpleBindEx instead.
MXNET_DLL int MXSymbolCreateFromFile(const char *fname, SymbolHandle *out)
Load a symbol from a json file.
MXNET_DLL int MXDataIterGetData(DataIterHandle handle, NDArrayHandle *out)
Get the handle to the NDArray of underlying data.
MXNET_DLL int MXSymbolCreateGroup(uint32_t num_symbols, SymbolHandle *symbols, SymbolHandle *out)
Create a Symbol by grouping list of symbols together.
int num_callbacks
Definition: c_api.h:153
MXNET_DLL int MXGetGPUCount(int *out)
Get the number of GPUs.
MXNET_DLL int MXListAllOpNames(uint32_t *out_size, const char ***out_array)
list all the available operator names, include entries.
MXNET_DLL int MXFreeCachedOp(CachedOpHandle handle)
free cached operator
MXNET_DLL int MXRecordIOReaderFree(RecordIOHandle handle)
Delete a RecordIO reader object.
MXNET_DLL int MXNotifyShutdown()
Notify the engine about a shutdown, This can help engine to print less messages into display...
void(* forward)(int, float **, int *, unsigned **, int *, void *)
Definition: c_api.h:120
MXNET_DLL int MXSymbolInferTypePartial(SymbolHandle sym, uint32_t num_args, const char **keys, const int *arg_type_data, uint32_t *in_type_size, const int **in_type_data, uint32_t *out_type_size, const int **out_type_data, uint32_t *aux_type_size, const int **aux_type_data, int *complete)
partially infer type of unknown input types given the known one.
MXNET_DLL int MXKVStorePushEx(KVStoreHandle handle, uint32_t num, const char **keys, NDArrayHandle *vals, int priority)
Push a list of (key,value) pairs to kvstore, where each key is a string.
MXNET_DLL int MXExecutorSetMonitorCallback(ExecutorHandle handle, ExecutorMonitorCallback callback, void *callback_handle)
set a call back to notify the completion of operation
MXNET_DLL int MXSymbolGetAtomicSymbolInfo(AtomicSymbolCreator creator, const char **name, const char **description, uint32_t *num_args, const char ***arg_names, const char ***arg_type_infos, const char ***arg_descriptions, const char **key_var_num_args, const char **return_type DEFAULT(NULL))
Get the detailed information about atomic symbol.
MXNET_DLL int MXEnginePushAsyncND(EngineAsyncFunc async_func, void *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 DEFAULT(NULL), int priority DEFAULT(0), const char *opr_name DEFAULT(NULL), bool wait DEFAULT(false))
Push an asynchronous operation to the engine.
int(* CustomOpDelFunc)(void *)
Definition: c_api.h:186
MXNET_DLL int MXProfileCreateTask(ProfileHandle domain, const char *task_name, ProfileHandle *out)
Create profile task.
const char * name
Definition: c_api.h:159
MXNET_DLL int MXAutogradSetIsRecording(int is_recording, int *prev)
set whether to record operator for autograd
MXNET_DLL int MXNDArrayLoad(const char *fname, uint32_t *out_size, NDArrayHandle **out_arr, uint32_t *out_name_size, const char ***out_names)
Load list of narray from the file.
MXNET_DLL int MXSymbolGetInternals(SymbolHandle symbol, SymbolHandle *out)
Get a symbol that contains all the internals.
MXNET_DLL int MXFuncGetInfo(FunctionHandle fun, const char **name, const char **description, uint32_t *num_args, const char ***arg_names, const char ***arg_type_infos, const char ***arg_descriptions, const char **return_type DEFAULT(NULL))
Get the information of the function handle.
MXNET_DLL int MXNDArrayGetAuxNDArray(NDArrayHandle handle, uint32_t i, NDArrayHandle *out)
Get a deep copy of the ith aux data blob This api is available when MXNet is built with flag USE_INT6...
MXNET_DLL int MXKVStoreGetType(KVStoreHandle handle, const char **type)
get the type of the kvstore
MXNET_DLL int MXImperativeInvoke(AtomicSymbolCreator creator, int num_inputs, NDArrayHandle *inputs, int *num_outputs, NDArrayHandle **outputs, int num_params, const char **param_keys, const char **param_vals)
invoke a nnvm op and imperative function
void(* EngineSyncFunc)(void *, void *)
Engine synchronous operation.
Definition: c_api.h:107
MXNET_DLL int MXQuantizeSymbol(SymbolHandle sym_handle, SymbolHandle *ret_sym_handle, const int *dev_type, const uint32_t num_excluded_sym_names, const char **excluded_sym_names, const uint32_t num_excluded_op_names, const char **excluded_op_names, const uint32_t num_offline, const char **offline_params, const char *quantized_dtype, const bool calib_quantize, const char *quantize_mode, const char *quantize_granularity, uint32_t *out_num_calib_names, const char ***out_calib_names)
Convert a symbol into a quantized symbol where FP32 operators are replaced with INT8.
MXNET_DLL int MXSymbolInferShapePartialEx(SymbolHandle sym, uint32_t num_args, const char **keys, const uint32_t *arg_ind_ptr, const int *arg_shape_data, uint32_t *in_shape_size, const int **in_shape_ndim, const int ***in_shape_data, uint32_t *out_shape_size, const int **out_shape_ndim, const int ***out_shape_data, uint32_t *aux_shape_size, const int **aux_shape_ndim, const int ***aux_shape_data, int *complete)
partially infer shape of unknown input shapes given the known one.
MXNET_DLL int MXNDArrayGetStorageType(NDArrayHandle handle, int *out_storage_type)
get the storage type of the array
MXNET_DLL int MXListFunctions(uint32_t *out_size, FunctionHandle **out_array)
list all the available functions handles most user can use it to list all the needed functions ...
MXNET_DLL int MXGetVersion(int *out)
get the MXNet library version as an integer
MXNET_DLL int MXNDArrayCreateSparseEx(int storage_type, const uint32_t *shape, uint32_t ndim, int dev_type, int dev_id, int delay_alloc, int dtype, uint32_t num_aux, int *aux_type, uint32_t *aux_ndims, const uint32_t *aux_shape, NDArrayHandle *out)
create an empty sparse NDArray with specified shape and data type This api is available when MXNet is...
int(* MXGenericCallback)(void)
Definition: c_api.h:150
void(* EngineFuncParamDeleter)(void *)
Callback to free the param for EngineAsyncFunc/EngineSyncFunc.
Definition: c_api.h:109
MXNET_DLL int MXAutogradIsTraining(bool *curr)
get whether training mode is on
MXNET_DLL int MXProcessProfilePause(int paused, int profile_process, KVStoreHandle kvStoreHandle)
Pause profiler tuning collection.
const void * ContextHandle
handle to Context
Definition: c_api.h:98
MXNET_DLL int MXRtcCudaKernelFree(CudaKernelHandle handle)
MXNET_DLL int MXKVStorePullWithSparse(KVStoreHandle handle, uint32_t num, const int *keys, NDArrayHandle *vals, int priority, bool ignore_sparse)
pull a list of (key, value) pairs from the kvstore
void * p_list_outputs
Definition: c_api.h:145
void(* EngineAsyncFunc)(void *, void *, void *)
Engine asynchronous operation.
Definition: c_api.h:105
MXNET_DLL int MXCreateCachedOp(SymbolHandle handle, CachedOpHandle *out)
create cached operator
MXNET_DLL int MXNDArrayReshape64(NDArrayHandle handle, int ndim, dim_t *dims, bool reverse, NDArrayHandle *out)
Reshape the NDArray.
MXNET_DLL int MXNDArrayWaitToRead(NDArrayHandle handle)
Wait until all the pending writes with respect NDArray are finished. Always call this before read dat...
MXNET_DLL int MXProfileDurationStart(ProfileHandle duration_handle)
Start timing the duration of a profile duration object such as an event, task or frame.
MXNET_DLL int MXEnginePushSyncND(EngineSyncFunc sync_func, void *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 DEFAULT(NULL), int priority DEFAULT(0), const char *opr_name DEFAULT(NULL))
Push a synchronous operation to the engine.
MXNET_DLL int MXSymbolListArguments(SymbolHandle symbol, uint32_t *out_size, const char ***out_str_array)
List arguments in the symbol.
void * RtcHandle
handle to MXRtc
Definition: c_api.h:88
MXNET_DLL int MXKVStorePushPull(KVStoreHandle handle, mx_uint vnum, const int *vkeys, mx_uint onum, const int *okeys, NDArrayHandle *vals, NDArrayHandle *outs, int priority)
push and pull a list of (key, value) pairs from the kvstore
MXNET_DLL int MXNDArraySlice64(NDArrayHandle handle, int64_t slice_begin, int64_t slice_end, NDArrayHandle *out)
Slice the NDArray along axis 0. This api is available when MXNet is built with flag USE_INT64_TENSOR_...
MXNET_DLL int MXSymbolInferType(SymbolHandle sym, uint32_t num_args, const char **keys, const int *arg_type_data, uint32_t *in_type_size, const int **in_type_data, uint32_t *out_type_size, const int **out_type_data, uint32_t *aux_type_size, const int **aux_type_data, int *complete)
infer type of unknown input types given the known one. The types are packed into a CSR matrix represe...
MXNET_DLL int MXImperativeInvokeEx(AtomicSymbolCreator creator, int num_inputs, NDArrayHandle *inputs, int *num_outputs, NDArrayHandle **outputs, int num_params, const char **param_keys, const char **param_vals, const int **out_stypes)
invoke a nnvm op and imperative function
#define DEFAULT(x)
Inhibit C++ name-mangling for MXNet functions.
Definition: c_api.h:36
MXNET_DLL int MXKVStoreGetNumDeadNode(KVStoreHandle handle, const int node_id, int *number, const int timeout_sec DEFAULT(60))
Get the number of ps dead node(s) specified by {node_id}.
MXNET_DLL int MXRtcCudaKernelCreate(CudaModuleHandle handle, const char *name, int num_args, int *is_ndarray, int *is_const, int *arg_types, CudaKernelHandle *out)
void() MXKVStoreStrUpdater(const char *key, NDArrayHandle recv, NDArrayHandle local, void *handle)
user-defined updater for the kvstore with string keys It's this updater's responsibility to delete re...
Definition: c_api.h:2985
void * RecordIOHandle
handle to RecordIO
Definition: c_api.h:86
MXNET_DLL int MXKVStoreSetBarrierBeforeExit(KVStoreHandle handle, const int barrier_before_exit)
whether to do barrier when finalize
MXNET_DLL int MXSetProcessProfilerConfig(int num_params, const char *const *keys, const char *const *vals, KVStoreHandle kvstoreHandle)
Set up configuration of profiler for the process passed as profile_process in keys.
MXNET_DLL int MXEnginePushAsync(EngineAsyncFunc async_func, void *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 DEFAULT(NULL), int priority DEFAULT(0), const char *opr_name DEFAULT(NULL), bool wait DEFAULT(false))
Push an asynchronous operation to the engine.
MXNET_DLL int MXExecutorFree(ExecutorHandle handle)
Delete the executor.
MXNET_DLL int MXSymbolSaveToFile(SymbolHandle symbol, const char *fname)
Save a symbol into a json file.
MXNET_DLL int MXRtcCudaModuleCreate(const char *source, int num_options, const char **options, int num_exports, const char **exports, CudaModuleHandle *out)
void * p_infer_shape
Definition: c_api.h:128
void * AtomicSymbolHandle
handle to a AtomicSymbol
Definition: c_api.h:76
const void * EngineFnPropertyHandle
handle to Engine FnProperty
Definition: c_api.h:100
MXNET_DLL int MXSymbolCreateAtomicSymbol(AtomicSymbolCreator creator, uint32_t num_param, const char **keys, const char **vals, SymbolHandle *out)
Create an AtomicSymbol.
MXNET_DLL int MXProfileSetCounter(ProfileHandle counter_handle, uint64_t value)
Set a counter, given its handle.
MXNET_DLL int MXNDArrayCreateNone(NDArrayHandle *out)
Load TVM operator from the binary library.
MXNET_DLL int MXKVStorePullWithSparseEx(KVStoreHandle handle, uint32_t num, const char **keys, NDArrayHandle *vals, int priority, bool ignore_sparse)
pull a list of (key, value) pairs from the kvstore, where each key is a string
MXNET_DLL int MXDumpProcessProfile(int finished, int profile_process, KVStoreHandle kvStoreHandle)
Save profile and stop profiler.
void() MXKVStoreServerController(int head, const char *body, void *controller_handle)
the prototype of a server controller
Definition: c_api.h:3092
MXNET_DLL int MXSetProcessProfilerState(int state, int profile_process, KVStoreHandle kvStoreHandle)
Set up state of profiler for either worker or server process.
CustomOpCallbacks
Definition: c_api.h:163
MXNET_DLL int MXNDArrayCreateFromSharedMemEx(int shared_pid, int shared_id, const int *shape, int ndim, int dtype, NDArrayHandle *out)
Reconstruct NDArray from shared memory handle.
void * p_backward
Definition: c_api.h:143
MXNET_DLL int MXNDArrayGetAuxType64(NDArrayHandle handle, int64_t i, int *out_type)
get the type of the ith aux data in NDArray This api is available when MXNet is built with flag USE_I...
void * NDArrayHandle
handle to NDArray
Definition: c_api.h:66
void * KVStoreHandle
handle to KVStore
Definition: c_api.h:84
void * CudaModuleHandle
handle to rtc cuda module
Definition: c_api.h:90
MXNET_DLL int MXSymbolGetName(SymbolHandle symbol, const char **out, int *success)
Get string name from symbol.
MXNET_DLL int MXNDArraySave(const char *fname, uint32_t num_args, NDArrayHandle *args, const char **keys)
Save list of narray into the file.
MXNET_DLL int MXKVStorePullRowSparseEx(KVStoreHandle handle, uint32_t num, const char **keys, NDArrayHandle *vals, const NDArrayHandle *row_ids, int priority)
pull a list of (key, value) pairs from the kvstore, where each key is a string. The NDArray pulled ba...
MXNET_DLL int MXSymbolSaveToJSON(SymbolHandle symbol, const char **out_json)
Save a symbol into a json string.
MXNET_DLL int MXLibInfoFeatures(const struct LibFeature **libFeature, size_t *size)
Get list of features supported on the runtime.
MXNET_DLL int MXKVStoreInit(KVStoreHandle handle, uint32_t num, const int *keys, NDArrayHandle *vals)
Init a list of (key,value) pairs in kvstore.
MXNET_DLL int MXKVStoreSetUpdater(KVStoreHandle handle, MXKVStoreUpdater updater, void *updater_handle)
register a push updater
MXNET_DLL int MXNDArrayWaitToWrite(NDArrayHandle handle)
Wait until all the pending read/write with respect NDArray are finished. Always call this before writ...
CustomOpPropCallbacks
Definition: c_api.h:169
MXNET_DLL const char * MXGetLastError()
return str message of the last error all function in this file will return 0 when success and -1 when...
MXNET_DLL int MXSymbolGetAtomicSymbolName(AtomicSymbolCreator creator, const char **name)
Get the name of an atomic symbol.
MXNET_DLL int MXAutogradGetSymbol(NDArrayHandle handle, SymbolHandle *out)
MXNET_DLL int MXCustomFunctionRecord(int num_inputs, NDArrayHandle *inputs, int num_outputs, NDArrayHandle *outputs, struct MXCallbackList *callbacks)
void * CachedOpHandle
handle to cached operator
Definition: c_api.h:72
MXNET_DLL int MXSymbolCompose(SymbolHandle sym, const char *name, uint32_t num_args, const char **keys, SymbolHandle *args)
Compose the symbol on other symbols.
MXNET_DLL int MXNDArrayFromDLPack(DLManagedTensorHandle dlpack, NDArrayHandle *out_handle)
DEPRECATED. Use MXNDArrayFromDLPackEx instead.
MXNET_DLL int MXSymbolListOutputs(SymbolHandle symbol, uint32_t *out_size, const char ***out_str_array)
List returns in the symbol.
MXNET_DLL int MXNDArrayWaitAll()
wait until all delayed operations in the system is completed
MXNET_DLL int MXSymbolInferShapeEx(SymbolHandle sym, uint32_t num_args, const char **keys, const uint32_t *arg_ind_ptr, const int *arg_shape_data, uint32_t *in_shape_size, const int **in_shape_ndim, const int ***in_shape_data, uint32_t *out_shape_size, const int **out_shape_ndim, const int ***out_shape_data, uint32_t *aux_shape_size, const int **aux_shape_ndim, const int ***aux_shape_data, int *complete)
infer shape of unknown input shapes given the known one. The shapes are packed into a CSR matrix repr...
MXNET_DLL int MXDataIterGetIterInfo(DataIterCreator creator, const char **name, const char **description, uint32_t *num_args, const char ***arg_names, const char ***arg_type_infos, const char ***arg_descriptions)
Get the detailed information about data iterator.
MXNET_DLL int MXInvokeCachedOp(CachedOpHandle handle, int num_inputs, NDArrayHandle *inputs, int *num_outputs, NDArrayHandle **outputs)
invoke cached operator
MXNET_DLL int MXRecordIOWriterTell(RecordIOHandle handle, size_t *pos)
Get the current writer pointer position.
#define MXNET_DLL
MXNET_DLL prefix for windows.
Definition: c_api.h:53
MXNET_DLL int MXNDArraySyncCopyFromCPU(NDArrayHandle handle, const void *data, size_t size)
Perform a synchronize copy from a contiguous CPU memory region.
MXNET_DLL int MXEnginePushSync(EngineSyncFunc sync_func, void *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 DEFAULT(NULL), int priority DEFAULT(0), const char *opr_name DEFAULT(NULL))
Push a synchronous operation to the engine.
MXNET_DLL int MXRtcCreate(char *name, uint32_t num_input, uint32_t num_output, char **input_names, char **output_names, NDArrayHandle *inputs, NDArrayHandle *outputs, char *kernel, RtcHandle *out)
Create a MXRtc object.
MXNET_DLL int MXNDArrayGetContext(NDArrayHandle handle, int *out_dev_type, int *out_dev_id)
get the context of the NDArray
MXNET_DLL int MXSymbolGrad(SymbolHandle sym, uint32_t num_wrt, const char **wrt, SymbolHandle *out)
Get the gradient graph of the symbol.
MXNET_DLL int MXKVStoreSetGradientCompression(KVStoreHandle handle, uint32_t num_params, const char **keys, const char **vals)
Set parameters to use low-bit compressed gradients.
MXNET_DLL int MXRecordIOWriterCreate(const char *uri, RecordIOHandle *out)
Create a RecordIO writer object.
MXNET_DLL int MXKVStoreIsSchedulerNode(int *ret)
return whether or not this process is a scheduler node.
MXNET_DLL int MXKVStorePush(KVStoreHandle handle, uint32_t num, const int *keys, NDArrayHandle *vals, int priority)
Push a list of (key,value) pairs to kvstore.
MXNET_DLL int MXSymbolSetAttr(SymbolHandle symbol, const char *key, const char *value)
Set string attribute from symbol. NOTE: Setting attribute to a symbol can affect the semantics(mutabl...
MXNET_DLL int MXSymbolCreateFromJSON(const char *json, SymbolHandle *out)
Load a symbol from a json string.
void * DLManagedTensorHandle
handle to DLManagedTensor
Definition: c_api.h:96
MXNET_DLL int MXDataIterCreateIter(DataIterCreator handle, uint32_t num_param, const char **keys, const char **vals, DataIterHandle *out)
Init an iterator, init with parameters the array size of passed in arguments.
MXNET_DLL int MXExecutorForward(ExecutorHandle handle, int is_train)
Executor forward method.
MXNET_DLL int MXExecutorBackwardEx(ExecutorHandle handle, uint32_t len, NDArrayHandle *head_grads, int is_train)
Excecutor run backward.
MXNET_DLL int MXSymbolGetOutput(SymbolHandle symbol, uint32_t index, SymbolHandle *out)
Get index-th outputs of the symbol.
MXNET_DLL int MXNDArrayReshape(NDArrayHandle handle, int ndim, int *dims, NDArrayHandle *out)
Reshape the NDArray.
MXNET_DLL int MXNDArrayGetSharedMemHandle(NDArrayHandle handle, int *shared_pid, int *shared_id)
Get shared memory handle from NDArray.
MXNET_DLL int MXKVStoreBroadcast(KVStoreHandle handle, mx_uint vnum, const int *vkeys, mx_uint onum, const int *okeys, NDArrayHandle *vals, NDArrayHandle *outs, int priority)
broadcast a list of (key, value) pairs from the kvstore
void ** contexts
Definition: c_api.h:155
MXNET_DLL int MXCustomOpRegister(const char *op_type, CustomOpPropCreator creator)
MXNET_DLL int MXNDArrayGetDataNDArray(NDArrayHandle handle, NDArrayHandle *out)
Get a deep copy of the data blob in the form of an NDArray of default storage type. This function blocks. Do not use it in performance critical code.
MXNET_DLL int MXNDArrayFromDLPackEx(DLManagedTensorHandle dlpack, const bool transient_handle, NDArrayHandle *out_handle)
Create a NDArray backed by a dlpack tensor.
MXNET_DLL int MXNDArrayCreateFromSharedMem(int shared_pid, int shared_id, const uint32_t *shape, uint32_t ndim, int dtype, NDArrayHandle *out)
DEPRECATED. Use MXNDArrayCreateFromSharedMemEx instead. Reconstruct NDArray from shared memory handle...
MXNET_DLL int MXExecutorGetOptimizedSymbol(ExecutorHandle handle, SymbolHandle *out)
get optimized graph from graph executor
MXNET_DLL int MXSetProfilerConfig(int num_params, const char *const *keys, const char *const *vals)
Set up configuration of profiler for worker/current process.
MXNET_DLL int MXDumpProfile(int finished)
Save profile and stop profiler for worker/current process.
MXNET_DLL int MXSymbolListAtomicSymbolCreators(uint32_t *out_size, AtomicSymbolCreator **out_array)
list all the available AtomicSymbolEntry
MXNET_DLL int MXDataIterNext(DataIterHandle handle, int *out)
Move iterator to next position.
MXNET_DLL int MXKVStoreIsWorkerNode(int *ret)
return whether or not this process is a worker node.
MXNET_DLL int MXSymbolInferShapePartial(SymbolHandle sym, uint32_t num_args, const char **keys, const uint32_t *arg_ind_ptr, const uint32_t *arg_shape_data, uint32_t *in_shape_size, const uint32_t **in_shape_ndim, const uint32_t ***in_shape_data, uint32_t *out_shape_size, const uint32_t **out_shape_ndim, const uint32_t ***out_shape_data, uint32_t *aux_shape_size, const uint32_t **aux_shape_ndim, const uint32_t ***aux_shape_data, int *complete)
DEPRECATED. Use MXSymbolInferShapePartialEx instead. partially infer shape of unknown input shapes gi...
MXNET_DLL int MXNDArraySyncCopyFromNDArray(NDArrayHandle handle_dst, const NDArrayHandle handle_src, const int i)
Copy src.data() to dst.data() if i = -1, else dst.aux_data(i) if i >= 0 This function blocks...
void(* infer_shape)(int, int *, unsigned **, void *)
Definition: c_api.h:122
int(* CustomOpListFunc)(char ***, void *)
Definition: c_api.h:187
MXNET_DLL int MXExecutorBindX(SymbolHandle symbol_handle, int dev_type, int dev_id, uint32_t num_map_keys, const char **map_keys, const int *map_dev_types, const int *map_dev_ids, uint32_t len, NDArrayHandle *in_args, NDArrayHandle *arg_grad_store, uint32_t *grad_req_type, uint32_t aux_states_len, NDArrayHandle *aux_states, ExecutorHandle *out)
Generate Executor from symbol, This is advanced function, allow specify group2ctx map...
MXNET_DLL int MXRecordIOReaderTell(RecordIOHandle handle, size_t *pos)
Get the current writer pointer position.
MXNET_DLL int MXRecordIOReaderCreate(const char *uri, RecordIOHandle *out)
Create a RecordIO reader object.
MXNET_DLL int MXRecordIOReaderReadRecord(RecordIOHandle handle, char const **buf, size_t *size)
Write a record to a RecordIO object.
void * p_list_outputs
Definition: c_api.h:129
MXNET_DLL int MXExecutorBind(SymbolHandle symbol_handle, int dev_type, int dev_id, uint32_t len, NDArrayHandle *in_args, NDArrayHandle *arg_grad_store, uint32_t *grad_req_type, uint32_t aux_states_len, NDArrayHandle *aux_states, ExecutorHandle *out)
Generate Executor from symbol.
MXNET_DLL int MXExecutorPrint(ExecutorHandle handle, const char **out_str)
Print the content of execution plan, used for debug.
void * EngineVarHandle
handle to Engine VarHandle
Definition: c_api.h:102
int64_t dim_t
data type to store dim size
Definition: c_api.h:61
MXNET_DLL int MXAutogradSetIsTraining(int is_training, int *prev)
set whether to record operator for autograd
MXNET_DLL int MXNDArrayLoadFromRawBytes(const void *buf, size_t size, NDArrayHandle *out)
create a NDArray handle that is loaded from raw bytes.
MXNET_DLL int MXListDataIters(uint32_t *out_size, DataIterCreator **out_array)
List all the available iterator entries.
MXNET_DLL int MXCreateCachedOpEX(SymbolHandle handle, int num_flags, const char **keys, const char **vals, CachedOpHandle *out, bool thread_safe DEFAULT(false))
create cached operator, allows to choose thread_safe version of cachedop
MXNET_DLL int MXAutogradBackwardEx(uint32_t num_output, NDArrayHandle *output_handles, NDArrayHandle *ograd_handles, uint32_t num_variables, NDArrayHandle *var_handles, int retain_graph, int create_graph, int is_train, NDArrayHandle **grad_handles, int **grad_stypes)
compute the gradient of outputs w.r.t variabels
const void * FunctionHandle
handle to a mxnet narray function that changes NDArray
Definition: c_api.h:68
MXNET_DLL int MXNDArrayGetShapeEx64(NDArrayHandle handle, int *out_dim, const int64_t **out_pdata)
get the shape of the array This api is available when MXNet is built with flag USE_INT64_TENSOR_SIZE=...
void(* backward)(int, float **, int *, unsigned **, int *, void *)
Definition: c_api.h:121
void * DataIterHandle
handle to a DataIterator
Definition: c_api.h:82
MXNET_DLL int MXProfileCreateCounter(ProfileHandle domain, const char *counter_name, ProfileHandle *out)
Create profile counter.
MXNET_DLL int MXGetGPUMemoryInformation(int dev, int *free_mem, int *total_mem)
get the free and total available memory on a GPU Note: Deprecated, use MXGetGPUMemoryInformation64 in...
MXNET_DLL int MXOptimizeForBackend(SymbolHandle sym_handle, const char *backend_name, const int dev_type, SymbolHandle *ret_sym_handle, const mx_uint args_len, NDArrayHandle *in_args_handle, const mx_uint aux_len, NDArrayHandle *in_aux_handle, const mx_uint num_options, const char **keys, const char **vals, const uint32_t num_input_shapes, const char **input_shape_names, const int64_t *input_shape_data, const uint32_t *input_shape_idx, const uint32_t num_input_dtypes, const char **input_dtype_names, const int *input_dtypes, const uint32_t num_input_stypes, const char **input_stype_names, const int *input_stypes, bool skip_infer, int *new_args_cnt, NDArrayHandle **new_args_handle, char ***new_arg_names_handle, int *new_aux_cnt, NDArrayHandle **new_aux_handle, char ***new_aux_names_handle)
Partitions symbol for given backend, potentially creating subgraphs.
int(* CustomOpInferStorageTypeFunc)(int, int *, void *)
Definition: c_api.h:190
MXNET_DLL int MXRecordIOWriterWriteRecord(RecordIOHandle handle, const char *buf, size_t size)
Write a record to a RecordIO object.
MXNET_DLL int MXKVStorePushPullEx(KVStoreHandle handle, mx_uint vnum, const char **vkeys, mx_uint onum, const char **okeys, NDArrayHandle *vals, NDArrayHandle *outs, int priority)
push and pull a list of (key, value) pairs from the kvstore, where each key is a string ...
MXNET_DLL int MXNDArrayCreateSparseEx64(int storage_type, const int64_t *shape, int ndim, int dev_type, int dev_id, int delay_alloc, int dtype, uint32_t num_aux, int *aux_type, int *aux_ndims, const int64_t *aux_shape, NDArrayHandle *out)
create an empty sparse NDArray with specified shape and data type This api is available when MXNet is...
MXNET_DLL int MXDataIterFree(DataIterHandle handle)
Free the handle to the IO module.
MXNET_DLL int MXSymbolCopy(SymbolHandle symbol, SymbolHandle *out)
Copy the symbol to another handle.
MXNET_DLL int MXExecutorReshape(int partial_shaping, int allow_up_sizing, int dev_type, int dev_id, uint32_t num_map_keys, const char **map_keys, const int *map_dev_types, const int *map_dev_ids, const uint32_t num_provided_arg_shapes, const char **provided_arg_shape_names, const uint32_t *provided_arg_shape_data, const uint32_t *provided_arg_shape_idx, uint32_t *num_in_args, NDArrayHandle **in_args, NDArrayHandle **arg_grads, uint32_t *num_aux_states, NDArrayHandle **aux_states, ExecutorHandle shared_exec, ExecutorHandle *out)
DEPRECATED. Use MXExecutorReshapeEx instead. Return a new executor with the same symbol and shared me...
MXNET_DLL int MXNDArrayLoadFromBuffer(const void *ndarray_buffer, size_t size, uint32_t *out_size, NDArrayHandle **out_arr, uint32_t *out_name_size, const char ***out_names)
Load list / dictionary of narrays from file content loaded into memory. This will load a list of ndar...
MXNET_DLL int MXDataIterGetPadNum(DataIterHandle handle, int *pad)
Get the padding number in current data batch.
void(* list_outputs)(char ***, void *)
Definition: c_api.h:123
MXNET_DLL int MXNDArraySyncCopyToCPU(NDArrayHandle handle, void *data, size_t size)
Perform a synchronize copyto a contiguous CPU memory region.
MXNET_DLL int MXInitPSEnv(uint32_t num_vars, const char **keys, const char **vals)
Initialized ps-lite environment variables.
MXNET_DLL int MXSymbolCreateVariable(const char *name, SymbolHandle *out)
Create a Variable Symbol.
MXNET_DLL int MXGetFunction(const char *name, FunctionHandle *out)
get the function handle by name
int(* CustomOpCreateFunc)(const char *, int, unsigned **, const int *, const int *, struct MXCallbackList *, void *)
Definition: c_api.h:199
MXNET_DLL int MXSymbolInferShapeEx64(SymbolHandle sym, uint32_t num_args, const char **keys, const int64_t *arg_ind_ptr, const int64_t *arg_shape_data, size_t *in_shape_size, const int **in_shape_ndim, const int64_t ***in_shape_data, size_t *out_shape_size, const int **out_shape_ndim, const int64_t ***out_shape_data, size_t *aux_shape_size, const int **aux_shape_ndim, const int64_t ***aux_shape_data, int *complete)
infer shape of unknown input shapes given the known one. The shapes are packed into a CSR matrix repr...
int(* CustomOpBackwardInferStorageTypeFunc)(int, int *, int *, void *)
Definition: c_api.h:191
MXNET_DLL int MXSymbolInferShape(SymbolHandle sym, uint32_t num_args, const char **keys, const uint32_t *arg_ind_ptr, const uint32_t *arg_shape_data, uint32_t *in_shape_size, const uint32_t **in_shape_ndim, const uint32_t ***in_shape_data, uint32_t *out_shape_size, const uint32_t **out_shape_ndim, const uint32_t ***out_shape_data, uint32_t *aux_shape_size, const uint32_t **aux_shape_ndim, const uint32_t ***aux_shape_data, int *complete)
DEPRECATED. Use MXSymbolInferShapeEx instead. infer shape of unknown input shapes given the known one...
uint32_t mx_uint
manually define unsigned int
Definition: c_api.h:57