mxnet
|
An abstract class for graph passes. More...
#include <lib_api.h>
Public Member Functions | |
CustomPass () | |
CustomPass (const char *pass_name) | |
CustomPass & | setBody (graphPass_t fn) |
Public Attributes | |
const char * | name |
pass name More... | |
graphPass_t | pass |
pass function More... | |
An abstract class for graph passes.
mxnet::ext::CustomPass::CustomPass | ( | ) |
|
explicit |
CustomPass& mxnet::ext::CustomPass::setBody | ( | graphPass_t | fn | ) |
const char* mxnet::ext::CustomPass::name |
pass name
graphPass_t mxnet::ext::CustomPass::pass |
pass function