Searched defs:new_axis_mask (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | strided_slice_grad.cc | 26 …SliceGrad::Init(int64_t begin_mask, int64_t end_mask, int64_t ellipsis_mask, int64_t new_axis_mask, in Init()
68 void StridedSliceGrad::set_new_axis_mask(int64_t new_axis_mask) { in set_new_axis_mask()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | const_to_attr_strided_slice_grad.cc | 100 auto new_axis_mask = AnfAlgo::GetNodeAttr<int64_t>(strided_slice_grad, kAttrNewAxisMask); in CheckAttrs() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | strided_slice_gpu_common.h | 99 auto new_axis_mask = Dec2Bin(new_axis_mask_int); in ParseMasks() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | strided_slice.cc | 261 void StridedSlice::set_new_axis_mask(const int64_t new_axis_mask) { in set_new_axis_mask()
278 const int64_t new_axis_mask, const int64_t shrink_axis_mask) { in Init()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | norm_fusion.cc | 290 auto new_axis_mask = static_cast<size_t>(primitive.value.AsStridedSlice()->new_axis_mask); in StridedSliceShapeSizeInfer() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 3210 new_axis_mask=0, argument
|
| D | _grad_ops.py | 1887 new_axis_mask=0, argument
|
| /third_party/mindspore/mindspore/lite/mindir/src/ |
| D | mindir.cc | 3870 int64_t new_axis_mask, int64_t shrink_axis_mask) { in MindIR_StridedSlice_CreatePrimitive()
3988 void MindIR_StridedSlice_SetNewAxisMask(PrimitivePtr *primitive, int64_t new_axis_mask) { in MindIR_StridedSlice_SetNewAxisMask()
|