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

!32295 [bugfix] SparseGatherV2 support on ascend platform 

Merge pull request !32295 from zyli2020/master
This commit is contained in:
i-robot 2022-03-31 10:44:42 +00:00 committed by Gitee
parent 39de9aa01d a3fb313392
commit 5a8568474a
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/core/abstract/primitive_infer_map.cc
View File

@ -61,6 +61,7 @@ std::set<int64_t> GetDependsFormMap(const CNodePtr &cnode) {
static const auto &kUnsortedSegmentMax = prim::kPrimUnsortedSegmentMax->name();
static const auto &kGather = prim::kPrimGather->name();
static const auto &kGatherV2 = prim::kPrimGatherV2->name();
static const auto &kSparseGatherV2 = prim::kPrimSparseGatherV2->name();
static const auto &kRange = prim::kPrimRange->name();
static const auto &kConv2DBackpropFilter = prim::kPrimConv2DBackpropFilter->name();
static const auto &kConv2DBackpropInput = prim::kPrimConv2DBackpropInput->name();
@ -76,6 +77,7 @@ std::set<int64_t> GetDependsFormMap(const CNodePtr &cnode) {
{kUnsortedSegmentMax, ShapeSet{2}},
{kGather, ShapeSet{2}},
{kGatherV2, ShapeSet{2}},
{kSparseGatherV2, ShapeSet{2}},
{kRange, ShapeSet{0, 1, 2}},
{kConv2DBackpropFilter, ShapeSet{2}},
{kConv2DBackpropInput, ShapeSet{2}},