Searched defs:begin_mask (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | strided_slice_grad.cc | 26 void StridedSliceGrad::Init(int64_t begin_mask, int64_t end_mask, int64_t ellipsis_mask, int64_t ne… in Init()
35 void StridedSliceGrad::set_begin_mask(int64_t begin_mask) { in set_begin_mask()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_stridedslice_grad_op.py | 26 def __init__(self, begin, end, stride, begin_mask=0, end_mask=0, ellipsis_mask=0): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | strided_slice_gpu_common.h | 73 auto begin_mask = Dec2Bin(begin_mask_int); in ParseMasks() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | strided_slice.cc | 231 void StridedSlice::set_begin_mask(const int64_t begin_mask) { in set_begin_mask()
277 void StridedSlice::Init(const int64_t begin_mask, const int64_t end_mask, const int64_t ellipsis_ma… in Init()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 3207 begin_mask=0, argument
|
| D | _grad_ops.py | 1884 begin_mask=0, argument
|
| /third_party/mindspore/mindspore/lite/mindir/src/ |
| D | mindir.cc | 3869 PrimitivePtr MindIR_StridedSlice_CreatePrimitive(int64_t begin_mask, int64_t end_mask, int64_t elli… in MindIR_StridedSlice_CreatePrimitive()
3895 void MindIR_StridedSlice_SetBeginMask(PrimitivePtr *primitive, int64_t begin_mask) { in MindIR_StridedSlice_SetBeginMask()
|