Home
last modified time | relevance | path

Searched defs:outer_index (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dcumprod.cc122 for (size_t outer_index = 0; outer_index < outer; ++outer_index) { in CumprodCompute() local
156 for (size_t outer_index = 0; outer_index < outer; ++outer_index) { in CumprodCompute() local
Dargmin_with_value.cc84 int64_t outer_index = i * dim_on * num_inner; in ExecArgMinWithValue() local
Dreduce_sum.cc147 for (int64_t outer_index = 0; outer_index < outer; ++outer_index) { in ReduceSumOneAxes() local
224 for (int64_t outer_index = 0; outer_index < outer; outer_index++) { in ReduceSumOneAxes2() local
Dargmax_with_value.cc79 int64_t outer_index = i * dim_on * num_inner; in ExecArgMaxWithValue() local
Dcumulativelogsumexp.cc95 for (size_t outer_index = 0; outer_index < outer; ++outer_index) { in CumulativeProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcumulative_logsumexp_cpu_kernel.cc79 for (size_t outer_index = 0; outer_index < outer; ++outer_index) { in CumulativeProcess() local
Dcpu_kernel.h334 size_t outer_index = index / inner_size_; in SetOffset() local
339 inline void SetOffset(size_t outer_index, size_t inner_index) { in SetOffset()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dsoftmax_impl.cu140 for (uint32_t outer_index = blockIdx.x; outer_index < outer_size; outer_index += gridDim.x) { in SpatialSoftMaxForward() local