浏览代码

fix bug

tags/v0.3.0-alpha
wangcong 5 年前
父节点
当前提交
c0772522d2
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. +1
    -1
      mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc

+ 1
- 1
mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc 查看文件

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


正在加载...
取消
保存