diff --git a/ge/graph/common/transop_util.cc b/ge/graph/common/transop_util.cc index 9b513fe6..62b4c4e4 100755 --- 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 precision_loss_transfer_map = {{ge::DT_FLOAT, ge::DT_BOOL}}; +std::map precision_loss_transfer_map = { + {ge::DT_FLOAT, ge::DT_BOOL}, + {ge::DT_INT64, ge::DT_BOOL} +}; } // namespace namespace ge {