Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dgroup_norm_kernel.cu757 T* dgamma_data = dgamma.defined() ? dgamma.mutable_data_ptr<T>() : nullptr; in GroupNorm1dBackward() local
928 T* dgamma_data = dgamma.defined() ? dgamma.mutable_data_ptr<T>() : nullptr; in GroupNormBackwardKernelImplInternal() local
H A Dlayer_norm_kernel.cu1212 T* dgamma_data = in LayerNormBackwardKernelImplInternal() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dgroup_norm_kernel.cpp906 PT* dgamma_data = dgamma.defined() ? dgamma.data_ptr<PT>() : nullptr; in GroupNormBackwardKernelImplInternal() local
1386 PT* dgamma_data = dgamma.defined() ? dgamma.data_ptr<PT>() : nullptr; in GroupNormBackwardKernelImplChannelsLastInternal() local
H A Dlayer_norm_kernel.cpp515 T2* dgamma_data = dgamma->defined() ? dgamma->template data_ptr<T2>() : nullptr; in LayerNormBackwardKernelImplInternal() local