Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNormalization.cuh525 stat_accscalar_t m_dy_c = sum_dy[plane] * norm_fct; in batch_norm_backward_elemt_kernel_impl() local
1349 auto m_dy_c = sum_dy[c_offset] * norm_fct; in batch_norm_backward_elemt_channels_last_kernel_impl() local