Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/
Dcustom_gather_d_grad_v2_fp32.cc31 size_t get_element_num(const std::vector<int> &shape) { in get_element_num() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/
Dexpand_cpu_kernel.cc84 size_t ExpandCpuKernelMod::get_element_num(const std::vector<size_t> &shape) const { in get_element_num() function in mindspore::kernel::ExpandCpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dargmin_with_value_cpu_kernel.cc29 size_t get_element_num(const std::vector<size_t> &shape) { in get_element_num() function
Dargmax_with_value_cpu_kernel.cc25 size_t get_element_num(const std::vector<size_t> &shape) { in get_element_num() function
Dgather_d_grad_cpu_kernel.cc28 size_t get_element_num(const std::vector<size_t> &shape) { in get_element_num() function
Dargmin_cpu_kernel.cc28 int64_t get_element_num(const std::vector<int64_t> &shape) { in get_element_num() function
Dargmax_cpu_kernel.cc30 int64_t get_element_num(const std::vector<int64_t> &shape) { return SizeToLong(SizeOf(shape)); } in get_element_num() function
Dgather_d_cpu_kernel.cc28 int64_t get_element_num(const std::vector<size_t> &shape) { in get_element_num() function
Dlu_solve_cpu_kernel.cc27 int64_t get_element_num(const std::vector<int64_t> &shape) { return SizeToLong(SizeOf(shape)); } in get_element_num() function
Dfft_with_size_cpu_kernel.cc55 int64_t get_element_num(const std::vector<int64_t> &shape, size_t rank) { in get_element_num() function
Digamma_cpu_kernel.cc63 int64_t get_element_num(const std::vector<int64_t> &shape) { return SizeToLong(SizeOf(shape)); } in get_element_num() function
Digammac_cpu_kernel.cc66 int64_t get_element_num(const std::vector<int64_t> &shape) { return SizeToLong(SizeOf(shape)); } in get_element_num() function
Digammagrada_cpu_kernel.cc65 size_t get_element_num(const std::vector<int64_t> &shape) { in get_element_num() function
Ddct_cpu_kernel.cc72 int64_t get_element_num(const std::vector<int64_t> &shape, size_t rank) { in get_element_num() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Ddct.cc160 int64_t get_element_num(const std::vector<int64_t> &shape, size_t rank) { in get_element_num() function