Browse Source

!12308 fix SliceGradUnifyMindIR

From: @yuchaojie
Reviewed-by: @jjfeing,@kisnwang
Signed-off-by: @kisnwang
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
243055d983
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ccsrc/backend/optimizer/ascend/mindir/slice_grad_unify_mindir.cc

+ 2
- 2
mindspore/ccsrc/backend/optimizer/ascend/mindir/slice_grad_unify_mindir.cc View File

@@ -30,7 +30,7 @@
namespace mindspore { namespace mindspore {
namespace opt { namespace opt {
namespace { namespace {
constexpr size_t kSliceGradInputNum = 4;
constexpr size_t kSliceGradInputTensorNum = 4;


std::vector<int64_t> GetInputXShape(const AnfNodePtr &node) { std::vector<int64_t> GetInputXShape(const AnfNodePtr &node) {
MS_EXCEPTION_IF_NULL(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(graph);
MS_EXCEPTION_IF_NULL(node); 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)}; std::vector<AnfNodePtr> pad_inputs = {NewValueNode(std::make_shared<Primitive>(kPadOpName)), slice_grad->input(1)};
auto pad = graph->NewCNode(pad_inputs); auto pad = graph->NewCNode(pad_inputs);
MS_EXCEPTION_IF_NULL(pad); MS_EXCEPTION_IF_NULL(pad);


Loading…
Cancel
Save