Searched refs:kEndMask (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | strided_slice_grad.cc | 45 (void)CheckAndConvertUtils::CheckInteger(kEndMask, end_mask, kGreaterEqual, 0, this->name()); in set_end_mask()
46 (void)this->AddAttr(kEndMask, MakeValue(end_mask)); in set_end_mask()
49 auto value_ptr = GetAttr(kEndMask); in get_end_mask()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | strided_slice.cc | 68 std::vector<int64_t> end_pos = TenToTwo(GetValue<int64_t>(primitive->GetAttr(kEndMask))); in EllipsisInferShape()
134 std::vector<int64_t> end_pos = TenToTwo(GetValue<int64_t>(primitive->GetAttr(kEndMask))); in ComputeInferShape()
240 (void)CheckAndConvertUtils::CheckInteger(kEndMask, end_mask, kGreaterEqual, 0, this->name()); in set_end_mask()
241 (void)this->AddAttr(kEndMask, MakeValue(end_mask)); in set_end_mask()
244 auto value_ptr = GetAttr(kEndMask); in get_end_mask()
|
| D | op_utils.h | 70 constexpr auto kEndMask = "end_mask"; variable
|