xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!6725 choose ai_core when input is 1D Tensor 

Merge pull request !6725 from zhangbuxue/choose_aicore_when_input_is_1D
This commit is contained in:
mindspore-ci-bot 2020-09-22 19:27:46 +08:00 committed by Gitee
parent 343dd3e341 76f341221e
commit 0846fad846
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc
View File

@ -63,7 +63,7 @@ static bool CheckStridedSlice(const CNodePtr &cnode) {
<< input->ToString();
}
int base_number = 2;
if (shrink_axis_mask >= std::pow<int, int>(base_number, input_dims - 1)) {
if (shrink_axis_mask >= std::pow<int, int>(base_number, input_dims - 1) && input_dims > 1) {
return false;
}
}