Searched refs:epsilon_scalar (Results 1 – 2 of 2) sorted by relevance
481 auto epsilon_scalar = in HandleBatchNormGrad() local484 HloInstruction::CreateBroadcast(activation_shape, epsilon_scalar, {})); in HandleBatchNormGrad()486 add(HloInstruction::CreateBroadcast(feature_shape, epsilon_scalar, {})); in HandleBatchNormGrad()
859 const T epsilon_scalar = epsilon.scalar<T>()(); in DoCompute() local870 (accum_update_ + accum_update_.constant(epsilon_scalar)).sqrt() * in DoCompute()871 (accum_ + accum_.constant(epsilon_scalar)).rsqrt() * grad_; in DoCompute()3678 const T epsilon_scalar = epsilon.scalar<T>()(); in Compute() local3691 (ms_ + ms_.constant(epsilon_scalar)).rsqrt() * in Compute()3817 const T epsilon_scalar = epsilon.scalar<T>()(); in Compute() local3833 auto denom_ = ms_ + ms_.constant(epsilon_scalar) - mg_.square(); in Compute()