Searched refs:scale_activations (Results 1 – 1 of 1) sorted by relevance
521 auto scale_activations = in XLA_TEST_P() local537 BatchNormTraining(input_activations, scale_activations, offset_activations, in XLA_TEST_P()621 auto scale_activations = in XLA_TEST_P() local642 BatchNormInference(input_activations, scale_activations, offset_activations, in XLA_TEST_P()