Caffe2 - C++ API
A deep learning, cross platform ML framework
module_test_dynamic.cc
1 
17 #include "caffe2/core/module.h"
18 #include "caffe2/core/operator.h"
19 
20 // An explicitly defined module, testing correctness when we dynamically link a
21 // module
22 CAFFE2_MODULE(caffe2_module_test_dynamic, "Dynamic module for testing.");
23 
24 namespace caffe2 {
25 
27  public:
28  using OperatorBase::OperatorBase;
29  bool Run(int /* unused */ /*stream_id*/) override {
30  return true;
31  }
32  virtual string type() {
33  return "base";
34  }
35 };
36 
37 REGISTER_CPU_OPERATOR(
38  Caffe2ModuleTestDynamicDummy, Caffe2ModuleTestDynamicDummyOp);
39 OPERATOR_SCHEMA(Caffe2ModuleTestDynamicDummy);
40 
41 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.