Caffe2 - C++ API
A deep learning, cross platform ML framework
schema.cc
1 #include "./schema.h"
2 #include "./operator_sets.h"
3 
4 namespace {
5 using namespace ONNX_NAMESPACE;
6 class PyTorchSchemasRegisterer {
7  public:
8  PyTorchSchemasRegisterer() {
9  OpSchemaRegistry::DomainToVersionRange::Instance().AddDomainToVersion(
10  AI_ONNX_PYTORCH_DOMAIN,
11  AI_ONNX_PYTORCH_DOMAIN_MIN_OPSET,
12  AI_ONNX_PYTORCH_DOMAIN_MAX_OPSET);
13  RegisterPyTorchOperatorSetSchema();
14  }
15 };
16 static PyTorchSchemasRegisterer registerer{};
17 } // namespace