mxnet
|
C API of mxnet for ease of testing backend in Python. More...
#include <mxnet/c_api.h>
Go to the source code of this file.
Functions | |
MXNET_DLL int | MXBuildSubgraphByOpNames (SymbolHandle sym_handle, const char *prop_name, const mx_uint num_ops, const char **op_names, SymbolHandle *ret_sym_handle) |
Inhibit C++ name-mangling for MXNet functions. More... | |
MXNET_DLL int | MXSetSubgraphPropertyOpNames (const char *prop_name, const mx_uint num_ops, const char **op_names) |
Given a subgraph property name, use the provided op names as the op_names attribute for that subgraph property, instead of the predefined one. This is only for the purpose of testing. More... | |
MXNET_DLL int | MXRemoveSubgraphPropertyOpNames (const char *prop_name) |
Given a subgraph property name, delete the op name set in the SubgraphPropertyOpNameSet. More... | |
C API of mxnet for ease of testing backend in Python.
Copyright (c) 2018 by Contributors
MXNET_DLL int MXBuildSubgraphByOpNames | ( | SymbolHandle | sym_handle, |
const char * | prop_name, | ||
const mx_uint | num_ops, | ||
const char ** | op_names, | ||
SymbolHandle * | ret_sym_handle | ||
) |
Inhibit C++ name-mangling for MXNet functions.
This API partitions a graph only by the operator names provided by users. This will attach a DefaultSubgraphProperty to the input graph for partitioning. This function should be used only for the testing purpose.
MXNET_DLL int MXRemoveSubgraphPropertyOpNames | ( | const char * | prop_name | ) |
Given a subgraph property name, delete the op name set in the SubgraphPropertyOpNameSet.
MXNET_DLL int MXSetSubgraphPropertyOpNames | ( | const char * | prop_name, |
const mx_uint | num_ops, | ||
const char ** | op_names | ||
) |
Given a subgraph property name, use the provided op names as the op_names attribute for that subgraph property, instead of the predefined one. This is only for the purpose of testing.