Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dgroup_norm_kernel.cpp544 const T* dY_ptr = dY + i * HxW; in ComputeInternalGradients() local
578 const T* dY_ptr = dY + i * HxW; in ComputeInternalGradients() local
698 const T* dY_ptr = dY + (i * D + j) * HxW; in GroupNormInputBackward() local
942 const T* dY_ptr, in DsDbRowwiseMomentsChannelsLast()
977 const T* dY_ptr, in DsDbRowwiseMomentsChannelsLast()
1073 const T* dY_ptr = dY_data + m * C; in ApplyInputGradientsChannelsLastColMov() local
1088 const T* dY_ptr = dY_data + m * C; in ApplyInputGradientsChannelsLastColMov() local
1124 const T* dY_ptr = dY_data + m * C; in ApplyInputGradientsChannelsLastColMov() local
1143 const T* dY_ptr = dY_data + m * C; in ApplyInputGradientsChannelsLastColMov() local
1260 const T* dY_ptr = dY_data + m * C; in CalcInternalGradientsChannelsLast() local
[all …]
H A Dlayer_norm_kernel.cpp202 const T* dY_ptr = dY_data + i * N; in layer_norm_backward_frame() local
318 const T* dY_ptr = dY_data + i * N; in layer_norm_backward_frame() local