commit 234e6b3797b055f9474593fc5bb89b360709abf6
author harrysummer <harrysummerxr@gmail.com> Thu Oct 11 00:21:12 2018 -0700
committer Facebook Github Bot <facebook-github-bot@users.noreply.github.com> Thu Oct 11 00:25:20 2018 -0700
tree 2d1aad993e4797e25eb540b73ce977c99099c6d6
parent 1f7cbea9848bdf451c7bb99ee197b6dbafc96896

Bugfix in onnx exporter (#10607)

Summary:
Incorrect processing for int and float arguments. Possibly a typo.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/10607

Differential Revision: D9376040

Pulled By: bddppq

fbshipit-source-id: e3665e7bbb26842d1d7eed50442993cfdbf55a80

caffe2/onnx/onnx_exporter.cc

diff --git a/caffe2/onnx/onnx_exporter.cc b/caffe2/onnx/onnx_exporter.cc
index 9dc4ea3..fbdf51d 100644
--- a/caffe2/onnx/onnx_exporter.cc
+++ b/caffe2/onnx/onnx_exporter.cc
@@ -428,8 +428,9 @@
     std::transform(
         c2_dtype.begin(), c2_dtype.end(), c2_dtype.begin(), ::toupper);
     if (c2_dtype == "FLOAT") {
-    } else if (c2_dtype == "INT32") {
       onnx_dtype = ::ONNX_NAMESPACE::TensorProto::FLOAT;
+    } else if (c2_dtype == "INT32") {
+      onnx_dtype = ::ONNX_NAMESPACE::TensorProto::INT32;
     } else if (c2_dtype == "BOOL") {
       onnx_dtype = ::ONNX_NAMESPACE::TensorProto::BOOL;
     } else if (c2_dtype == "UINT8") {