MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle / reduce

Method reduce

paddle/phi/kernels/cpu/cum_kernel.cc:194–197  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

192template <typename T>
193struct LogSumExpReducer {
194 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const T t, T* accum) const {
195 LogSumExp<T> logsumexp;
196 *accum = logsumexp(*accum, t);
197 }
198
199 template <typename Packet>
200 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reducePacket(const Packet& p,

Callers 15

atomicReduceFunction · 0.45
FullReductionKernelFunction · 0.45
FullReductionKernelHalfFloatFunction · 0.45
ReductionCleanupKernelHalfFloatFunction · 0.45
InnerReductionKernelFunction · 0.45
InnerReductionKernelHalfFloatFunction · 0.45
OuterReductionKernelFunction · 0.45
get_modelMethod · 0.45
test_create_process_group_bkclMethod · 0.45
get_modelMethod · 0.45
try_reduceFunction · 0.45
try_reduce_iterFunction · 0.45

Calls

no outgoing calls

Tested by 15

test_create_process_group_bkclMethod · 0.36
try_reduceFunction · 0.36
try_reduce_iterFunction · 0.36
try_reduce_iter_failedFunction · 0.36
test_reduceFunction · 0.36
test_create_process_group_glooMethod · 0.36
test_reduce_sumFunction · 0.36
test_reduce_maxFunction · 0.36
test_reduce_minFunction · 0.36
test_reduce_prodFunction · 0.36
test_create_process_group_ncclMethod · 0.36
test_reduce_with_zero_dimFunction · 0.36