Browse Source

fix bug

tags/v0.3.0-alpha
wangcong 5 years ago
parent
commit
2f591cab79
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc

+ 3
- 2
mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc View File

@@ -725,8 +725,9 @@ std::string TbeKernelBuild::GetRealOpType(const std::string &origin_type) {
static std::map<std::string, std::string> buffer_fussion_op_map = {{"DepthwiseConv2dNative", "DepthwiseConv2D"}, static std::map<std::string, std::string> buffer_fussion_op_map = {{"DepthwiseConv2dNative", "DepthwiseConv2D"},
{"TensorAdd", "Add"}}; {"TensorAdd", "Add"}};
string result = origin_type; string result = origin_type;
if (buffer_fussion_op_map.find(origin_type) != buffer_fussion_op_map.end()) {
result = buffer_fussion_op_map[origin_type];
std::map<std::string, std::string> iter = buffer_fussion_op_map.find(origin_type);
if (iter != buffer_fussion_op_map.end()) {
result = iter->second;
} }
return result; return result;
} }


Loading…
Cancel
Save