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

fix SliceGradUnifyMindIR

This commit is contained in:
yuchaojie 2021-02-09 15:32:37 +08:00
parent b7b553ae48
commit ad93497838
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/backend/optimizer/ascend/mindir/slice_grad_unify_mindir.cc
View File

@ -30,7 +30,7 @@
namespace mindspore {
namespace opt {
namespace {
constexpr size_t kSliceGradInputNum = 4;
constexpr size_t kSliceGradInputTensorNum = 4;
std::vector<int64_t> GetInputXShape(const AnfNodePtr &node) {
MS_EXCEPTION_IF_NULL(node);
@ -63,7 +63,7 @@ const AnfNodePtr SliceGradUnifyMindIR::Process(const FuncGraphPtr &graph, const
MS_EXCEPTION_IF_NULL(graph);
MS_EXCEPTION_IF_NULL(node);
auto slice_grad = CheckAnfNodeIfCNodeAndInputSize(node, kSliceGradInputNum + 1);
auto slice_grad = CheckAnfNodeIfCNodeAndInputSize(node, kSliceGradInputTensorNum);
std::vector<AnfNodePtr> pad_inputs = {NewValueNode(std::make_shared<Primitive>(kPadOpName)), slice_grad->input(1)};
auto pad = graph->NewCNode(pad_inputs);
MS_EXCEPTION_IF_NULL(pad);