To fix caffe2 model with Copy OP cannot export to onnx model (pytorch#37144)

mpjlu · facebook-github-bot · commit 6dd1beaaa8b8 · 2020-05-04T11:34:09.000-07:00
Summary: To fix caffe2 model with Copy OP cannot export to onnx model Pull Request resolved: pytorch#37144 Reviewed By: houseroad Differential Revision: D21252421 Pulled By: yinghai fbshipit-source-id: 4f1077188f36b0691d199e418880bbb27f11032d
diff --git a/caffe2/onnx/onnx_exporter.cc b/caffe2/onnx/onnx_exporter.cc
@@ -371,7 +371,8 @@ OnnxExporter::get_renamed_operators() const {
       {"MaxPool3D", "MaxPool"},
       {"AveragePool1D", "AveragePool"},
       {"AveragePool2D", "AveragePool"},
-      {"AveragePool3D", "AveragePool"}};
+      {"AveragePool3D", "AveragePool"},
+      {"Copy", "Identity"}};
   return kRenamedOperators;
 }
 
diff --git a/caffe2/operators/copy_op.cc b/caffe2/operators/copy_op.cc
@@ -16,6 +16,7 @@ OPERATOR_SCHEMA(Copy)
     .NumOutputs(1)
     .IdenticalTypeAndShape()
     .InputsCanCrossDevices()
+    .InheritOnnxSchema("Identity")
     .SetDoc(R"DOC(
 Copy input tensor into output, potentially across devices.
 

Original file line number	Diff line number	Diff line change
`@@ -371,7 +371,8 @@ OnnxExporter::get_renamed_operators() const {`
`371`	`371`	`{"MaxPool3D", "MaxPool"},`
`372`	`372`	`{"AveragePool1D", "AveragePool"},`
`373`	`373`	`{"AveragePool2D", "AveragePool"},`
`374`		`- {"AveragePool3D", "AveragePool"}};`
	`374`	`+ {"AveragePool3D", "AveragePool"},`
	`375`	`+ {"Copy", "Identity"}};`
`375`	`376`	`return kRenamedOperators;`
`376`	`377`	`}`
`377`	`378`