Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DMultiLabelMarginCriterion.cu117 accscalar_t total_sum = cuda_utils::BlockReduceSum(sum, smem); in C10_LAUNCH_BOUNDS_1() local
189 accscalar_t total_sum = cuda_utils::BlockReduceSum(sum, smem); in C10_LAUNCH_BOUNDS_1() local