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

fix avgpoolgrad

This commit is contained in:
hbhu_bin 2023-02-02 11:59:53 +08:00
parent 3b12ea8520
commit ca5ae9904b
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/plugin/device/ascend/optimizer/ge/avg_pool_grad_for_ge.cc
View File

@ -60,7 +60,7 @@ const AnfNodePtr AvgPoolGradForGE::Process(const FuncGraphPtr &graph, const AnfN
std::string pad_mode;
if (pad_mode_type == TypeId::kNumberTypeInt64) {
auto pad_value = GetValue<int64_t>(pad_mode_value);
pad_mode = pad_value != 0 ? "SAME" : "VALID";
pad_mode = pad_value == 1 ? "SAME" : "VALID";
} else {
pad_mode = GetValue<std::string>(pad_mode_value);
}