Home
last modified time | relevance | path

Searched refs:batch_norm_backward_reduce_channels_last_kernel (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNormalization.cuh1167 __global__ void batch_norm_backward_reduce_channels_last_kernel( in batch_norm_backward_reduce_channels_last_kernel() function
1568 batch_norm_backward_reduce_channels_last_kernel<ELEMENTS_PER_ITER> in batch_norm_backward_reduce_cuda_channels_last_template()
1593 batch_norm_backward_reduce_channels_last_kernel<ELEMENTS_PER_ITER> in batch_norm_backward_reduce_cuda_channels_last_template()