| #include "caffe2/opt/passes.h" | |
| namespace caffe2 { | |
| C10_DEFINE_REGISTRY( | |
| WorkspaceOptimizationPassRegistry, | |
| WorkspaceOptimizationPass, | |
| NNModule*, | |
| Workspace*); | |
| C10_DEFINE_REGISTRY(OptimizationPassRegistry, OptimizationPass, NNModule*); | |
| } // namespace caffe2 |