mxnet
Functions
c_api_test.h File Reference

C API of mxnet for ease of testing backend in Python. More...

#include <mxnet/c_api.h>
Include dependency graph for c_api_test.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...
 

Detailed Description

C API of mxnet for ease of testing backend in Python.

Copyright (c) 2018 by Contributors

Function Documentation

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.