Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNormalization.cuh1378 __global__ void batch_norm_backward_elemt_channels_last_kernel( in batch_norm_backward_elemt_channels_last_kernel() function
1408 __global__ void batch_norm_backward_elemt_channels_last_kernel( in batch_norm_backward_elemt_channels_last_kernel() function
1638 batch_norm_backward_elemt_channels_last_kernel<ELEMENTS_PER_ITER> in batch_norm_backward_elemt_channels_last_cuda_template()
1661 batch_norm_backward_elemt_channels_last_kernel<ELEMENTS_PER_ITER> in batch_norm_backward_elemt_channels_last_cuda_template()
1707 batch_norm_backward_elemt_channels_last_kernel<ELEMENTS_PER_ITER> in batch_norm_backward_elemt_channels_last_cuda_template()
1722 batch_norm_backward_elemt_channels_last_kernel<ELEMENTS_PER_ITER> in batch_norm_backward_elemt_channels_last_cuda_template()