Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dcumulativelogsumexp_impl.cu61 T pre_sum = expT(output[read_index]); in CumulativeLogsumexpKernelReverse() local
83 T pre_sum = expT(output[read_index]); in CumulativeLogsumexpKernel() local
105 T pre_sum = expT(output[read_index]); in CumulativeLogsumexpKernelExclusive() local
127 T pre_sum = expT(output[read_index]); in CumulativeLogsumexpKernelReverseExclusive() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dconcat.c143 void ComputeConcatUnitBoundary(ConcatStruct *concat, int64_t *pre_sum, int offset, int *input, int6… in ComputeConcatUnitBoundary()
163 int64_t *pre_sum = in ChooseConcatThreadCuttingStrategy() local