Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore/core/ops/grad/
Dstrided_slice_grad.cc26 …SliceGrad::Init(int64_t begin_mask, int64_t end_mask, int64_t ellipsis_mask, int64_t new_axis_mask, in Init() argument
31 this->set_new_axis_mask(new_axis_mask); in Init()
68 void StridedSliceGrad::set_new_axis_mask(int64_t new_axis_mask) { in set_new_axis_mask() argument
69 …(void)CheckAndConvertUtils::CheckInteger(kNewAxisMask, new_axis_mask, kGreaterEqual, 0, this->name… in set_new_axis_mask()
70 (void)this->AddAttr(kNewAxisMask, MakeValue(new_axis_mask)); in set_new_axis_mask()
Dstrided_slice_grad.h36 …int64_t begin_mask = 0, int64_t end_mask = 0, int64_t ellipsis_mask = 0, int64_t new_axis_mask = 0,
41 void set_new_axis_mask(int64_t new_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);
50 void set_new_axis_mask(const int64_t new_axis_mask);
Dstrided_slice.cc261 void StridedSlice::set_new_axis_mask(const int64_t new_axis_mask) { in set_new_axis_mask() argument
262 …(void)CheckAndConvertUtils::CheckInteger(kNewAxisMask, new_axis_mask, kGreaterEqual, 0, this->name… in set_new_axis_mask()
263 (void)this->AddAttr(kNewAxisMask, MakeValue(new_axis_mask)); in set_new_axis_mask()
278 const int64_t new_axis_mask, const int64_t shrink_axis_mask) { in Init() argument
282 this->set_new_axis_mask(new_axis_mask); in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dstrided_slice_gpu_common.h99 auto new_axis_mask = Dec2Bin(new_axis_mask_int); in ParseMasks() local
100 for (size_t l = 0; l < new_axis_mask.size(); l++) { in ParseMasks()
101 if (new_axis_mask[l] && l < MAX_DIMS) { in ParseMasks()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconst_to_attr_strided_slice_grad.cc100 auto new_axis_mask = AnfAlgo::GetNodeAttr<int64_t>(strided_slice_grad, kAttrNewAxisMask); in CheckAttrs() local
102 if (new_axis_mask != 0 || shrink_axis_mask != 0) { in CheckAttrs()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dnorm_fusion.cc290 auto new_axis_mask = static_cast<size_t>(primitive.value.AsStridedSlice()->new_axis_mask); in StridedSliceShapeSizeInfer() local
292 while (new_axis_mask != 0) { in StridedSliceShapeSizeInfer()
293 new_axis_mask = (new_axis_mask - 1) & new_axis_mask; in StridedSliceShapeSizeInfer()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dselection_ops_declare.cc140 {"new_axis_mask", ATTR_DESC(new_axis_mask, AnyTraits<int64_t>())},
150 {"new_axis_mask", ATTR_DESC(new_axis_mask, AnyTraits<int64_t>())},
161 {"new_axis_mask", ATTR_DESC(new_axis_mask, AnyTraits<int64_t>())},
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dstrided_slice_populate.cc41 param->newAxisMask_ = value->new_axis_mask(); in PopulateStridedSliceParameter()
Dstrided_slice_grad_populate.cc42 param->newAxisMask_ = value->new_axis_mask(); in PopulateStridedSliceGradParameter()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_strided_slice_parser.cc42 prim->set_new_axis_mask(tflite_attr->new_axis_mask); in Parse()
Dschema.fbs758 new_axis_mask: int;
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dstrided_slice_npu.cc61 new_axis_mask_ = strided_slice_prim->new_axis_mask(); in Init()
/third_party/mindspore/mindspore/lite/mindir/include/
Dmindir.h384 int64_t new_axis_mask, int64_t shrink_axis_mask);
392 void MindIR_StridedSlice_SetNewAxisMask(PrimitivePtr *primitive, int64_t new_axis_mask);
/third_party/mindspore/mindspore/lite/schema/
Dops.fbs972 new_axis_mask: long;
1075 new_axis_mask: long;
Dops_generated.h10100 int64_t new_axis_mask() const {
10130 void add_new_axis_mask(int64_t new_axis_mask) {
10131 fbb_.AddElement<int64_t>(StridedSlice::VT_NEW_AXIS_MASK, new_axis_mask, 0);
10152 int64_t new_axis_mask = 0,
10156 builder_.add_new_axis_mask(new_axis_mask);
11152 int64_t new_axis_mask() const {
11182 void add_new_axis_mask(int64_t new_axis_mask) {
11183 fbb_.AddElement<int64_t>(StridedSliceGrad::VT_NEW_AXIS_MASK, new_axis_mask, 0);
11204 int64_t new_axis_mask = 0,
11208 builder_.add_new_axis_mask(new_axis_mask);
/third_party/mindspore/mindspore/lite/src/ops/
Dops_def.cc972 OP_ATTR(new_axis_mask, long)
1075 OP_ATTR(new_axis_mask, long)
/third_party/mindspore/tests/ut/python/ops/
Dtest_math_ops.py467 'block': (lambda x: P.StridedSlice(new_axis_mask="1.1"), {'exception': TypeError}),
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)
961 new_axis_mask=18)
/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 …schema::CreateStridedSlice(fbb, begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_ma… 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()
3979 return value->new_axis_mask(); in MindIR_StridedSlice_GetNewAxisMask()
3988 void MindIR_StridedSlice_SetNewAxisMask(PrimitivePtr *primitive, int64_t new_axis_mask) { in MindIR_StridedSlice_SetNewAxisMask() argument
3995 new_axis_mask, value->shrink_axis_mask()); in MindIR_StridedSlice_SetNewAxisMask()
4026 value->new_axis_mask(), shrink_axis_mask); in MindIR_StridedSlice_SetShrinkAxisMask()
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_populate_parameter.cc487 strided_slice_param->newAxisMask_ = value->new_axis_mask(); in PopulateStridedSliceGradParameter()
/third_party/mindspore/mindspore/ops/operations/
Darray_ops.py3210 new_axis_mask=0, argument
3219 validator.check_non_negative_int(new_axis_mask, 'new_axis_mask', self.name)
3282 new_axis_pos = bin(self.new_axis_mask)[-1:1:-1]
D_grad_ops.py1887 new_axis_mask=0, argument
1893 validator.check_value_type('new_axis_mask', new_axis_mask, [int], self.name)
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_array_ops.py655 self.new_axis_mask,
/third_party/mindspore/third_party/proto/tensorflow/lite/
Dschema.fbs758 new_axis_mask: int;

12