pytorch

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
18