Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

use isdynamicrank func replace origin code

This commit is contained in:
greatpan 2023-02-17 17:34:37 +08:00
parent ac916ed2cd
commit bcdc572d9c
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/common/pass/reduce_optimizer.cc
View File

@ -154,7 +154,7 @@ AnfNodePtr ReduceOptimizer::NewAssistValueNode(const CNodePtr &cnode, const Kern
// case 2: contain value less 0;
for (auto &iter : value_tuple->value()) {
auto item = GetValue<int64_t>(iter->cast<ScalarPtr>());
if (item < 0 && !(x_shape->shape().size() == 1 && x_shape->shape()[0] == -2)) { // -2 : dynamic rank
if (item < 0 && !(IsDynamicRank(x_shape->shape()))) {
(void)axis_value.emplace_back(item + static_cast<int64_t>(x_shape->shape().size()));
} else {
(void)axis_value.emplace_back(item);