Home
last modified time | relevance | path

Searched defs:outer_dim (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/
Dfused_sparse_utils.cc32 auto outer_dim = param.outer_dim_; in WorkerForReduceSparseGradient() local
63 size_t first_dim, size_t outer_dim) { in ReduceSparseGradient()
130 size_t outer_dim) { in ReduceMultiSparseGradient()
158 size_t outer_dim) { in TwoLevelReduceSparseGradient()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/
Dgrad_array_ops.cc965 …auto outer_dim = std::accumulate(indices_shape.begin(), indices_shape.end() - 1, 1, std::multiplie… in __anond7f8849a0b02() local