Searched defs:gamma_v (Results 1 – 6 of 6) sorted by relevance
91 const CTYPE gamma_v = gamma_null ? CTYPE(1) : gamma_data[j]; in layer_norm() local
60 const T gamma_v = gamma_null ? T(1) : gamma_data[j]; in LayerNormKernelImplInternal() local130 const float gamma_v = gamma_null ? float(1) : float(gamma_data[d]); in layer_norm_kernel_mixed_type() local
688 const opmath_t gamma_v = gamma_null ? opmath_t(1) : opmath_t(gamma[g * D + j]); in GroupNormInputBackward() local
121 const T_ACC gamma_v = in Compute1dBackwardFusedParamsCUDAKernel() local331 const T_ACC gamma_v = in ComputeBackwardFusedParamsCUDAKernel() local
104 const T_ACC gamma_v = in LayerNormForwardCUDAKernel() local
2909 const float gamma_v = gamma_null ? 1.0f : gamma_data[remIdx]; local