Home
last modified time | relevance | path

Searched refs:end_mask (Results 1 – 25 of 33) sorted by relevance

12

/third_party/mindspore/mindspore/core/ops/grad/
Dstrided_slice_grad.cc26 void StridedSliceGrad::Init(int64_t begin_mask, int64_t end_mask, int64_t ellipsis_mask, int64_t ne… in Init() argument
29 this->set_end_mask(end_mask); in Init()
44 void StridedSliceGrad::set_end_mask(int64_t end_mask) { in set_end_mask() argument
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()
Dstrided_slice_grad.h36 …void Init(int64_t begin_mask = 0, int64_t end_mask = 0, int64_t ellipsis_mask = 0, int64_t new_axi…
39 void set_end_mask(int64_t end_mask);
/third_party/mindspore/mindspore/core/ops/
Dstrided_slice.h41 …void Init(const int64_t begin_mask = 0, const int64_t end_mask = 0, const int64_t ellipsis_mask = …
46 void set_end_mask(const int64_t end_mask);
Dstrided_slice.cc239 void StridedSlice::set_end_mask(const int64_t end_mask) { in set_end_mask() argument
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()
277 void StridedSlice::Init(const int64_t begin_mask, const int64_t end_mask, const int64_t ellipsis_ma… in Init() argument
280 this->set_end_mask(end_mask); in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dstrided_slice_gpu_common.h81 auto end_mask = Dec2Bin(end_mask_int); in ParseMasks() local
82 for (size_t j = 0; j < end_mask.size(); j++) { in ParseMasks()
83 if (end_mask[j] && j < MAX_DIMS) { in ParseMasks()
/third_party/mindspore/tests/st/ops/gpu/
Dtest_stridedslice_grad_op.py26 def __init__(self, begin, end, stride, begin_mask=0, end_mask=0, ellipsis_mask=0): argument
31 self.slice = P.StridedSlice(begin_mask, end_mask, ellipsis_mask)
155 begin_mask=0b1000, end_mask=0b0010, ellipsis_mask=0b0100)
Dtest_stridedslice_op.py67 op = P.StridedSlice(begin_mask=0b1000, end_mask=0b0010, ellipsis_mask=0b0100)
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dselection_ops_declare.cc138 {"end_mask", ATTR_DESC(end_mask, AnyTraits<int64_t>())},
148 {"end_mask", ATTR_DESC(end_mask, AnyTraits<int64_t>())},
159 {"end_mask", ATTR_DESC(end_mask, AnyTraits<int64_t>())},
/third_party/mindspore/mindspore/lite/test/ut/tools/converter/parser/tflite/
Dtflite_strided_slice_parser_test.cc38 ASSERT_EQ(val->end_mask, 0); in TEST_F()
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dstrided_slice_grad_populate.cc40 param->ends_mask_ = value->end_mask(); in PopulateStridedSliceGradParameter()
Dstrided_slice_populate.cc39 param->ends_mask_ = value->end_mask(); in PopulateStridedSliceParameter()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_strided_slice_parser.cc40 prim->set_end_mask(tflite_attr->end_mask); in Parse()
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dstrided_slice_npu.cc59 ends_mask_ = strided_slice_prim->end_mask(); in Init()
/third_party/skia/third_party/externals/libpng/
Dpngrutil.c3208 unsigned int end_mask; in png_combine_row() local
3234 end_mask = (pixel_depth * row_width) & 7; in png_combine_row()
3235 if (end_mask != 0) in png_combine_row()
3243 end_mask = (unsigned int)(0xff << end_mask); in png_combine_row()
3247 end_mask = 0xff >> end_mask; in png_combine_row()
3679 *end_ptr = (png_byte)((end_byte & end_mask) | (*end_ptr & ~end_mask)); in png_combine_row()
/third_party/flutter/skia/third_party/externals/libpng/
Dpngrutil.c3208 unsigned int end_mask; in png_combine_row() local
3234 end_mask = (pixel_depth * row_width) & 7; in png_combine_row()
3235 if (end_mask != 0) in png_combine_row()
3243 end_mask = (unsigned int)(0xff << end_mask); in png_combine_row()
3247 end_mask = 0xff >> end_mask; in png_combine_row()
3679 *end_ptr = (png_byte)((end_byte & end_mask) | (*end_ptr & ~end_mask)); in png_combine_row()
/third_party/libpng/
Dpngrutil.c3208 unsigned int end_mask; in png_combine_row() local
3234 end_mask = (pixel_depth * row_width) & 7; in png_combine_row()
3235 if (end_mask != 0) in png_combine_row()
3243 end_mask = (unsigned int)(0xff << end_mask); in png_combine_row()
3247 end_mask = 0xff >> end_mask; in png_combine_row()
3679 *end_ptr = (png_byte)((end_byte & end_mask) | (*end_ptr & ~end_mask)); in png_combine_row()
/third_party/mindspore/mindspore/lite/mindir/include/
Dmindir.h383 PrimitivePtr MindIR_StridedSlice_CreatePrimitive(int64_t begin_mask, int64_t end_mask, int64_t elli…
388 void MindIR_StridedSlice_SetEndMask(PrimitivePtr *primitive, int64_t end_mask);
/third_party/mindspore/mindspore/lite/schema/
Dops.fbs970 end_mask: long;
1073 end_mask: long;
/third_party/mindspore/mindspore/lite/src/ops/
Dops_def.cc970 OP_ATTR(end_mask, long)
1073 OP_ATTR(end_mask, long)
/third_party/mindspore/tests/ut/python/ops/
Dtest_math_ops.py461 'block': (lambda x: P.StridedSlice(end_mask="1"), {'exception': TypeError}),
Dtest_ops.py927 self.strided_slice_0 = P.StridedSlice(begin_mask=3, end_mask=5, ellipsis_mask=4,
929 self.strided_slice_1 = P.StridedSlice(begin_mask=5, end_mask=2, ellipsis_mask=2,
931 self.strided_slice_2 = P.StridedSlice(begin_mask=3, end_mask=3, ellipsis_mask=4,
933 self.strided_slice_3 = P.StridedSlice(begin_mask=0, end_mask=0, ellipsis_mask=4,
958 end_mask=6,
/third_party/mindspore/mindspore/lite/mindir/src/
Dmindir.cc3869 PrimitivePtr MindIR_StridedSlice_CreatePrimitive(int64_t begin_mask, int64_t end_mask, int64_t elli… 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()
3901 …auto ops_offset = schema::CreateStridedSlice(fbb, begin_mask, value->end_mask(), value->ellipsis_m… in MindIR_StridedSlice_SetBeginMask()
3917 return value->end_mask(); in MindIR_StridedSlice_GetEndMask()
3926 void MindIR_StridedSlice_SetEndMask(PrimitivePtr *primitive, int64_t end_mask) { in MindIR_StridedSlice_SetEndMask() argument
3932 …auto ops_offset = schema::CreateStridedSlice(fbb, value->begin_mask(), end_mask, value->ellipsis_m… in MindIR_StridedSlice_SetEndMask()
3963 …auto ops_offset = schema::CreateStridedSlice(fbb, value->begin_mask(), value->end_mask(), ellipsis… in MindIR_StridedSlice_SetEllipsisMask()
3994 …auto ops_offset = schema::CreateStridedSlice(fbb, value->begin_mask(), value->end_mask(), value->e… in MindIR_StridedSlice_SetNewAxisMask()
4025 …auto ops_offset = schema::CreateStridedSlice(fbb, value->begin_mask(), value->end_mask(), value->e… in MindIR_StridedSlice_SetShrinkAxisMask()
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_populate_parameter.cc485 strided_slice_param->ends_mask_ = value->end_mask(); in PopulateStridedSliceGradParameter()
/third_party/mindspore/mindspore/ops/operations/
Darray_ops.py3208 end_mask=0, argument
3215 validator.check_non_negative_int(end_mask, 'end_mask', self.name)
3280 end_pos = bin(self.end_mask)[-1:1:-1]
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_array_ops.py653 self.end_mask,

12