Searched refs:sqrt_2 (Results 1 – 3 of 3) sorted by relevance
47 xla::XlaOp sqrt_2 = xla::ScalarLike(uniform, std::sqrt(2.0)); in ParameterizedTruncatedNormal() local50 return (one + xla::Erf(x / sqrt_2)) / two; in ParameterizedTruncatedNormal()75 xla::XlaOp x = mu + sigma * sqrt_2 * xla::ErfInv(v); in ParameterizedTruncatedNormal()
768 let sqrt_2 = bf16::from_f32(core::f32::consts::SQRT_2); in test_bf16_consts_from_f32() localVariable787 assert_eq!(bf16::SQRT_2, sqrt_2); in test_bf16_consts_from_f32()826 let sqrt_2 = bf16::from_f64(core::f64::consts::SQRT_2); in test_bf16_consts_from_f64() localVariable845 assert_eq!(bf16::SQRT_2, sqrt_2); in test_bf16_consts_from_f64()
1057 let sqrt_2 = f16::from_f32(core::f32::consts::SQRT_2); in test_f16_consts_from_f32() localVariable1076 assert_eq!(f16::SQRT_2, sqrt_2); in test_f16_consts_from_f32()1117 let sqrt_2 = f16::from_f64(core::f64::consts::SQRT_2); in test_f16_consts_from_f64() localVariable1136 assert_eq!(f16::SQRT_2, sqrt_2); in test_f16_consts_from_f64()