modified: ge/graph/common/transop_util.cc

4 years ago · 41e8e51bc9
parent 689b059f67
commit 41e8e51bc9
1 changed files with 4 additions and 1 deletions
--- a/ge/graph/common/transop_util.cc
+++ b/ge/graph/common/transop_util.cc
@ -23,7 +23,10 @@
 namespace {
 const int kInvalidTransopDataIndex = -1;
 const int kTransOpOutIndex = 0;
-std::map<ge::DataType, ge::DataType> precision_loss_transfer_map = {{ge::DT_FLOAT, ge::DT_BOOL}};
+std::map<ge::DataType, ge::DataType> precision_loss_transfer_map = {
+  {ge::DT_FLOAT, ge::DT_BOOL},
+  {ge::DT_INT64, ge::DT_BOOL}
+};
 }  // namespace

 namespace ge {