Searched refs:begin_v (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | strided_slice.cc | 56 const std::vector<int64_t> &begin_v, const std::vector<int64_t> &end_v, in EllipsisInferShape() argument
66 size_t slice_len = begin_v.size(); in EllipsisInferShape()
89 int64_t start = begin_v[j]; in EllipsisInferShape()
130 …ctor<int64_t> ComputeInferShape(const PrimitivePtr &primitive, const std::vector<int64_t> &begin_v, in ComputeInferShape() argument
146 size_t slice_len = begin_v.size(); in ComputeInferShape()
155 start = begin_v[j]; in ComputeInferShape()
193 …EllipsisInferShape(primitive, x_shape, begin_v, end_v, strides_v, &infer_shape, i, j, has_ellipsis… in ComputeInferShape()
204 auto begin_v = GetValue<std::vector<int64_t>>(temp_begin_v); in StridedSliceInferShape() local
216 auto ret_in_shape = ComputeInferShape(primitive, begin_v, end_v, x_shape, strides_v); in StridedSliceInferShape()
220 auto ret_min_shape = ComputeInferShape(primitive, begin_v, end_v, min_shape, strides_v); in StridedSliceInferShape()
[all …]
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | slice_fusion.cc | 39 auto begin_v = input_args[kInputIndex1]->BuildValue(); in SliceFusionInfer() local
43 MS_EXCEPTION_IF_NULL(begin_v); in SliceFusionInfer()
49 if (begin_v == kAnyValue || size_v == kAnyValue) { in SliceFusionInfer()
52 auto begin = GetValue<std::vector<int64_t>>(begin_v); in SliceFusionInfer()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 2795 begin_v, size_v = begin['value'], size['value']
2796 if begin_v is None or size_v is None:
2800 validator.check_value_type("input begin", begin_v, [tuple, list], self.name)
2802 for key, value in zip(('begin', 'size'), (begin_v, size_v)):
2807 validator.check_non_negative_int(begin_v[i], f'input begin[{i}]')
2808 if x_shape[i] < begin_v[i] + size_v[i]:
2809 y = begin_v[i] + size_v[i]
3223 begin_v, end_v, strides_v = begin['value'], end['value'], strides['value']
3224 validator.check_value_type("begin", begin_v, [tuple], self.name)
3228 if tuple(filter(lambda x: not isinstance(x, int), begin_v + end_v + strides_v)):
[all …]
|