Browse Source

choose ai_core when input is 1D Tensor

tags/v1.1.0
buxue 5 years ago
parent
commit
76f341221e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc

+ 1
- 1
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(); << input->ToString();
} }
int base_number = 2; 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; return false;
} }
} }


Loading…
Cancel
Save