Searched refs:activation_deviations (Results 1 – 1 of 1) sorted by relevance
151 auto activation_deviations = Sub(input_activations, set_means, in XLA_TEST_P() local154 auto dev_squares = Square(activation_deviations); in XLA_TEST_P()197 auto activation_deviations = Sub(input_activations, set_means, in XLA_TEST_P() local199 auto dev_squares = Square(activation_deviations); in XLA_TEST_P()214 Mul(activation_deviations, normalization_factors, in XLA_TEST_P()