Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cpu/
Dlayer_norm_kernel.cpp46 const bool gamma_null = gamma_data == nullptr; in LayerNormKernelImplInternal() local
106 const bool gamma_null = gamma_data == nullptr; in layer_norm_kernel_mixed_type() local
195 const bool gamma_null, in layer_norm_backward_frame()
310 const bool gamma_null, in layer_norm_backward_frame()
518 const bool gamma_null = gamma_data == nullptr; in LayerNormBackwardKernelImplInternal() local
Dgroup_norm_kernel.cpp52 const bool gamma_null = (gamma_data == nullptr); in GroupNormKernelImplInternal() local
311 const bool gamma_null = (gamma_data == nullptr); in GroupNormKernelImplChannelsLastInternal() local
671 const bool gamma_null = (gamma == nullptr); in GroupNormInputBackward() local
1064 const bool gamma_null = (gamma == nullptr); in ApplyInputGradientsChannelsLastColMov() local
1114 const bool gamma_null = (gamma == nullptr); in ApplyInputGradientsChannelsLastColMov() local
1169 const bool gamma_null = (gamma == nullptr); in ApplyInputGradientsChannelsLastRowMov() local
1207 const bool gamma_null = (gamma == nullptr); in ApplyInputGradientsChannelsLastRowMov() local
1251 const bool gamma_null = (gamma_ptr == nullptr); in CalcInternalGradientsChannelsLast() local
1307 const bool gamma_null = (gamma_ptr == nullptr); in CalcInternalGradientsChannelsLast() local
1388 const bool gamma_null = (gamma_data == nullptr); in GroupNormBackwardKernelImplChannelsLastInternal() local
/external/executorch/kernels/optimized/cpu/
Dop_native_layer_norm.cpp74 const bool gamma_null = gamma_data == nullptr; in layer_norm() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
DQuantizedOpKernels.cpp2788 const bool gamma_null = gamma_data == nullptr; local
3076 const bool gamma_null = gamma_data == nullptr; local