saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-19 19:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Register PytorchAtenDomain in RegisterOrtOpSchemas (#14567)

Browse source
This commit is contained in:
Nhat Nguyen 2023-04-04 20:34:13 -04:00 committed by GitHub
parent 5294cd0c55
commit 198994d01d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
orttraining/orttraining/core/graph/training_op_defs.cc
View file

@ -4583,6 +4583,7 @@ void RegisterOrtOpSchemas() {
}
domainToVersionRangeInstance.AddDomainToVersion(onnxruntime::kMSExperimentalDomain, 1, 1);
domainToVersionRangeInstance.AddDomainToVersion(onnxruntime::kMSNchwcDomain, 1, 1);
domainToVersionRangeInstance.AddDomainToVersion(onnxruntime::kPytorchAtenDomain, 1, 1);
onnxruntime::contrib::RegisterContribSchemas();
onnxruntime::training::RegisterTrainingOpSchemas();