fix optimizer does not set ir_version bug (onnx#1098)

aipul · Jun 8, 2018 · 39e4668 · 39e4668
1 parent 2508156
commit 39e4668
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnx/optimizer/optimize.cc b/onnx/optimizer/optimize.cc
@@ -8,7 +8,7 @@ namespace ONNX_NAMESPACE { namespace optimization {
 ONNX_NAMESPACE::ModelProto PrepareOutput(const ONNX_NAMESPACE::ModelProto& mp_in) {
   ONNX_NAMESPACE::ModelProto mp_out{};
 
-  if (mp_in.has_producer_name()) {
+  if (mp_in.has_ir_version()) {
     mp_out.set_ir_version(mp_in.ir_version());
   }
   if (mp_in.has_producer_name()) {