mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
65bf181ddf
Summary: zrphercule Pull Request resolved: https://github.com/pytorch/pytorch/pull/12157 Differential Revision: D10100799 Pulled By: bddppq fbshipit-source-id: 76fdd126e0b52c54276752b3b0174735355a7d2f
18 lines
469 B
C++
18 lines
469 B
C++
#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
|