m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix bug

Browse Source
pull/1115/head
wangcong 5 years ago
parent 2f591cab79
commit c0772522d2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc
Unescape View File

@ -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"}, 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;
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()) { if (iter != buffer_fussion_op_map.end()) {
result = iter->second; result = iter->second;
} }

Write Preview
Loading…
Cancel
Save