Searched refs:mean_parameter (Results 1 – 1 of 1) sorted by relevance
811 auto mean_parameter = Parameter(&builder, 2, mean_literal.shape(), "mean"); in XLA_TEST_P() local827 BatchNormGrad(input_parameter, scale_parameter, mean_parameter, var_parameter, in XLA_TEST_P()