An abstract class for graph passes. More...
#include <lib_api.h>
Collaboration diagram for mxnet::ext::CustomPass:
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... | |
Detailed Description
An abstract class for graph passes.
Constructor & Destructor Documentation
◆ CustomPass() [1/2]
| mxnet::ext::CustomPass::CustomPass | ( | ) |
◆ CustomPass() [2/2]
|
explicit |
Member Function Documentation
◆ setBody()
| CustomPass& mxnet::ext::CustomPass::setBody | ( | graphPass_t | fn | ) |
Member Data Documentation
◆ name
| const char* mxnet::ext::CustomPass::name |
pass name
◆ pass
| graphPass_t mxnet::ext::CustomPass::pass |
pass function
The documentation for this class was generated from the following file:
- /work/mxnet/include/mxnet/lib_api.h
