#include "./schema.h" | |
#include "./operator_sets.h" | |
namespace { | |
using namespace ONNX_NAMESPACE; | |
class PyTorchSchemasRegisterer { | |
public: | |
PyTorchSchemasRegisterer() { | |
OpSchemaRegistry::DomainToVersionRange::Instance().AddDomainToVersion( | |
AI_ONNX_PYTORCH_DOMAIN, | |
AI_ONNX_PYTORCH_DOMAIN_MIN_OPSET, | |
AI_ONNX_PYTORCH_DOMAIN_MAX_OPSET); | |
RegisterPyTorchOperatorSetSchema(); | |
} | |
}; | |
static PyTorchSchemasRegisterer registerer{}; | |
} // namespace |