Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dgroup_norm_kernel.cpp700 T* dX_ptr = dX + (i * D + j) * HxW; in GroupNormInputBackward() local
1074 T* dX_ptr = dX_data + m * C; in ApplyInputGradientsChannelsLastColMov() local
1089 T* dX_ptr = dX_data + m * C; in ApplyInputGradientsChannelsLastColMov() local
1125 T* dX_ptr = dX_data + m * C; in ApplyInputGradientsChannelsLastColMov() local
1144 T* dX_ptr = dX_data + m * C; in ApplyInputGradientsChannelsLastColMov() local
1429 T* dX_ptr = dX_data + n * HxW * C + g * D; in GroupNormBackwardKernelImplChannelsLastInternal() local
1500 T* dX_ptr = dX_data + i * C + g * D; in GroupNormBackwardKernelImplChannelsLastInternal() local
H A Dlayer_norm_kernel.cpp234 T* dX_ptr = dX_data + i * N; in layer_norm_backward_frame() local
350 T* dX_ptr = dX_data + i * N; in layer_norm_backward_frame() local