Searched refs:kShrinkAxisMask (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | strided_slice_grad.cc | 78 …(void)CheckAndConvertUtils::CheckInteger(kShrinkAxisMask, shrink_axis_mask, kGreaterEqual, 0, this… in set_shrink_axis_mask()
79 (void)this->AddAttr(kShrinkAxisMask, MakeValue(shrink_axis_mask)); in set_shrink_axis_mask()
82 auto value_ptr = GetAttr(kShrinkAxisMask); in get_shrink_axis_mask()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | strided_slice.cc | 70 …vector<int64_t> shrink_axis_pos = TenToTwo(GetValue<int64_t>(primitive->GetAttr(kShrinkAxisMask))); in EllipsisInferShape()
137 …vector<int64_t> shrink_axis_pos = TenToTwo(GetValue<int64_t>(primitive->GetAttr(kShrinkAxisMask))); in ComputeInferShape()
270 …(void)CheckAndConvertUtils::CheckInteger(kShrinkAxisMask, shrink_axis_mask, kGreaterEqual, 0, this… in set_shrink_axis_mask()
271 (void)this->AddAttr(kShrinkAxisMask, MakeValue(shrink_axis_mask)); in set_shrink_axis_mask()
274 auto value_ptr = GetAttr(kShrinkAxisMask); in get_shrink_axis_mask()
|
| D | op_utils.h | 183 constexpr auto kShrinkAxisMask = "shrink_axis_mask"; variable
|