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 |