Browse Source

!2494 fixed StridedSliceGrad

Merge pull request !2494 from jiangjinsheng/issue_fix4
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
edeba61c05
1 changed files with 11 additions and 1 deletions
  1. +11
    -1
      mindspore/ops/operations/_grad_ops.py

+ 11
- 1
mindspore/ops/operations/_grad_ops.py View File

@@ -1093,8 +1093,18 @@ class StridedSliceGrad(PrimitiveWithInfer):
self.init_prim_io_names(inputs=['dy', 'shapex', 'begin', 'end', 'strides'], outputs=['output'])

def __infer__(self, dy, shapex, begin, end, strides):
args = {"shapex": shapex['dtype'],"begin": begin['dtype'],"end": end['dtype'],"strides": strides['dtype']}
args = {"dy": dy['dtype']}
validator.check_tensor_type_same(args, mstype.number_type, self.name)

for idx, item in enumerate(shapex['value']):
validator.check_value_type("shapex[%d]" % idx, item, [int], self.name)
for idx, item in enumerate(begin['value']):
validator.check_value_type("begin[%d]" % idx, item, [int], self.name)
for idx, item in enumerate(end['value']):
validator.check_value_type("end[%d]" % idx, item, [int], self.name)
for idx, item in enumerate(strides['value']):
validator.check_value_type("strides[%d]" % idx, item, [int], self.name)

return {'shape': shapex['value'],
'dtype': dy['dtype'],
'value': None}


Loading…
Cancel
Save