Home
last modified time | relevance | path

Searched refs:shrink_axis_mask (Results 1 – 25 of 29) sorted by relevance

12

/third_party/mindspore/mindspore/core/ops/grad/
Dstrided_slice_grad.cc27 int64_t shrink_axis_mask) { in Init() argument
32 this->set_shrink_axis_mask(shrink_axis_mask); in Init()
77 void StridedSliceGrad::set_shrink_axis_mask(int64_t shrink_axis_mask) { in set_shrink_axis_mask() argument
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()
Dstrided_slice_grad.h37 int64_t shrink_axis_mask = 0);
42 void set_shrink_axis_mask(int64_t shrink_axis_mask);
/third_party/mindspore/mindspore/core/ops/
Dstrided_slice.h42 const int64_t new_axis_mask = 0, const int64_t shrink_axis_mask = 0);
52 void set_shrink_axis_mask(const int64_t shrink_axis_mask);
Dstrided_slice.cc269 void StridedSlice::set_shrink_axis_mask(const int64_t shrink_axis_mask) { in set_shrink_axis_mask() argument
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()
278 const int64_t new_axis_mask, const int64_t shrink_axis_mask) { in Init() argument
283 this->set_shrink_axis_mask(shrink_axis_mask); in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dstrided_slice_gpu_common.h109 auto shrink_axis_mask = Dec2Bin(shrink_axis_mask_int); in ParseMasks() local
110 for (size_t m = 0; m < shrink_axis_mask.size(); m++) { in ParseMasks()
111 if (shrink_axis_mask[m] && m < MAX_DIMS) { in ParseMasks()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_property_checker.cc45 …auto shrink_axis_mask = static_cast<int>(AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrShrinkAxisMask)… in CheckStridedSlice() local
70 if (shrink_axis_mask >= std::pow<int, int>(base_number, input_dims - 1) && input_dims > 1) { in CheckStridedSlice()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconst_to_attr_strided_slice_grad.cc101 auto shrink_axis_mask = AnfAlgo::GetNodeAttr<int64_t>(strided_slice_grad, kAttrShrinkAxisMask); in CheckAttrs() local
102 if (new_axis_mask != 0 || shrink_axis_mask != 0) { in CheckAttrs()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dselection_ops_declare.cc141 … {"shrink_axis_mask", ATTR_DESC(shrink_axis_mask, AnyTraits<int64_t>())}};
151 {"shrink_axis_mask", ATTR_DESC(shrink_axis_mask, AnyTraits<int64_t>())}};
162 … {"shrink_axis_mask", ATTR_DESC(shrink_axis_mask, AnyTraits<int64_t>())}};
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dstrided_slice_populate.cc42 param->shrinkAxisMask_ = value->shrink_axis_mask(); in PopulateStridedSliceParameter()
Dstrided_slice_grad_populate.cc43 param->shrinkAxisMask_ = value->shrink_axis_mask(); in PopulateStridedSliceGradParameter()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_strided_slice_parser.cc43 prim->set_shrink_axis_mask(tflite_attr->shrink_axis_mask); in Parse()
Dschema.fbs759 shrink_axis_mask: int;
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dstrided_slice_npu.cc62 shrink_axis_mask_ = strided_slice_prim->shrink_axis_mask(); in Init()
/third_party/mindspore/mindspore/ops/composite/multitype_ops/
D_compile_utils.py321 shrink_axis_mask = 1
322 …return P.StridedSlice(0, 0, 0, 0, shrink_axis_mask)(data, begin_strides, end_strides, step_strides)
403 …begin_strides, end_strides, step_strides, shrink_axis_mask = const_utils.get_stride_info_from_tupl…
405 …return P.StridedSlice(0, 0, 0, 0, shrink_axis_mask)(data, begin_strides, end_strides, step_strides)
/third_party/mindspore/mindspore/lite/mindir/include/
Dmindir.h384 int64_t new_axis_mask, int64_t shrink_axis_mask);
394 void MindIR_StridedSlice_SetShrinkAxisMask(PrimitivePtr *primitive, int64_t shrink_axis_mask);
/third_party/mindspore/mindspore/lite/schema/
Dops.fbs973 shrink_axis_mask: long;
1076 shrink_axis_mask: long;
Dops_generated.h10103 int64_t shrink_axis_mask() const {
10133 void add_shrink_axis_mask(int64_t shrink_axis_mask) {
10134 fbb_.AddElement<int64_t>(StridedSlice::VT_SHRINK_AXIS_MASK, shrink_axis_mask, 0);
10153 int64_t shrink_axis_mask = 0) {
10155 builder_.add_shrink_axis_mask(shrink_axis_mask);
11155 int64_t shrink_axis_mask() const {
11185 void add_shrink_axis_mask(int64_t shrink_axis_mask) {
11186 fbb_.AddElement<int64_t>(StridedSliceGrad::VT_SHRINK_AXIS_MASK, shrink_axis_mask, 0);
11205 int64_t shrink_axis_mask = 0) {
11207 builder_.add_shrink_axis_mask(shrink_axis_mask);
/third_party/mindspore/tests/ut/python/ops/
Dtest_ops.py928 shrink_axis_mask=2, new_axis_mask=8)
930 shrink_axis_mask=6, new_axis_mask=10)
932 shrink_axis_mask=5, new_axis_mask=13)
934 shrink_axis_mask=12, new_axis_mask=15)
960 shrink_axis_mask=6,
1237 'block': P.StridedSlice(shrink_axis_mask=0),
/third_party/mindspore/mindspore/lite/src/ops/
Dops_def.cc973 OP_ATTR(shrink_axis_mask, long)
1076 OP_ATTR(shrink_axis_mask, long)
/third_party/mindspore/mindspore/lite/mindir/src/
Dmindir.cc3870 int64_t new_axis_mask, int64_t shrink_axis_mask) { in MindIR_StridedSlice_CreatePrimitive() argument
3873 …ema::CreateStridedSlice(fbb, begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask); in MindIR_StridedSlice_CreatePrimitive()
3902 … value->new_axis_mask(), value->shrink_axis_mask()); in MindIR_StridedSlice_SetBeginMask()
3933 … value->new_axis_mask(), value->shrink_axis_mask()); in MindIR_StridedSlice_SetEndMask()
3964 … value->new_axis_mask(), value->shrink_axis_mask()); in MindIR_StridedSlice_SetEllipsisMask()
3995 new_axis_mask, value->shrink_axis_mask()); in MindIR_StridedSlice_SetNewAxisMask()
4010 return value->shrink_axis_mask(); in MindIR_StridedSlice_GetShrinkAxisMask()
4019 void MindIR_StridedSlice_SetShrinkAxisMask(PrimitivePtr *primitive, int64_t shrink_axis_mask) { in MindIR_StridedSlice_SetShrinkAxisMask() argument
4026 value->new_axis_mask(), shrink_axis_mask); in MindIR_StridedSlice_SetShrinkAxisMask()
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_populate_parameter.cc488 strided_slice_param->shrinkAxisMask_ = value->shrink_axis_mask(); in PopulateStridedSliceGradParameter()
/third_party/mindspore/mindspore/ops/operations/
Darray_ops.py3211 shrink_axis_mask=0): argument
3220 validator.check_non_negative_int(shrink_axis_mask, 'shrink_axis_mask', self.name)
3283 shrink_axis_pos = bin(self.shrink_axis_mask)[-1:1:-1]
D_grad_ops.py1888 shrink_axis_mask=0): argument
1894 validator.check_value_type('shrink_axis_mask', shrink_axis_mask, [int], self.name)
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_array_ops.py656 self.shrink_axis_mask)
/third_party/mindspore/third_party/proto/tensorflow/lite/
Dschema.fbs759 shrink_axis_mask: int;

12