This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
choose ai_core when input is 1D Tensor
tags/v1.1.0
buxue
5 years ago
parent
9a5f9d137d
commit
76f341221e
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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();
}
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;
}
}
Write
Preview
Loading…
Cancel
Save