Home
last modified time | relevance | path

Searched defs:dims_ (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dgathernd_cpu_kernel.h43 std::vector<size_t> dims_; variable
Drandom_choice_with_mask_cpu_kernel.h51 std::vector<int32_t> dims_; variable
Drandom_choice_with_mask_cpu_kernel.cc60 void GetInputTotalCount(const std::vector<int32_t> &dims_, int32_t *input_total_count, const int32_… in GetInputTotalCount()
66 void UpdateOutput(const std::vector<int32_t> &dims_, const int32_t &non_zero_num, const int32_t &co… in UpdateOutput()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dfill_parameter.h24 int dims_[COMM_SHAPE_SIZE]; member
Dunsqueeze_parameter.h34 int dims_[COMM_SHAPE_SIZE]; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dtile_base.h26 int dims_[MAX_TILE_DIM_SIZE]; member
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dparallel_fusion.h61 std::vector<DimInfoPtr> dims_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dcumprod_gpu_kernel.h117 size_t dims_[kMaxDimsSize] = {}; variable
Dcumsum_gpu_kernel.h117 size_t dims_[kMaxDimsSize] = {}; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dgather_grad_gpu_kernel.h130 size_t dims_[4] = {}; variable
Dgather_gpu_kernel.h129 size_t dims_[4] = {}; variable
Dgatherv2_gpu_kernel.h152 size_t dims_[3] = {}; variable
Dembedding_lookup_gpu_kernel.h152 size_t dims_[3] = {}; variable
Dgathernd_gpu_kernel.h158 std::vector<size_t> dims_; variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/
Dlite_mat.h344 int dims_; variable