Home
last modified time | relevance | path

Searched defs:out_strides_ (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dscatter_nd_cpu_kernel.h55 std::vector<int> out_strides_; variable
Dscatter_nd_update_cpu_kernel.h57 std::vector<int> out_strides_; variable
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dscatter_nd_fp32.h45 std::vector<int> out_strides_; variable
Dscatter_nd_update_fp32.h44 std::vector<int> out_strides_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dtranspose.h35 int out_strides_[MAX_TRANSPOSE_DIM_SIZE]; member
Darithmetic.h40 int out_strides_[ARITHMETIC_SUPPORT_DIMS_NUM]; member
Darg_min_max_parameter.h52 int32_t out_strides_[COMM_SHAPE_SIZE]; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dtile_base.h34 int out_strides_[MAX_TILE_DIM_SIZE]; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Droi_pooling_fp32.h30 int out_strides_[DIMENSION_4D]; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dscatter_nd_functor_gpu_kernel.h171 std::vector<S> out_strides_; variable