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