Searched refs:input_parameter (Results 1 – 1 of 1) sorted by relevance
872 auto input_parameter = Parameter(&builder, 0, input_literal.shape(), "input"); in XLA_TEST_P() local890 BatchNormGrad(input_parameter, scale_parameter, mean_parameter, var_parameter, in XLA_TEST_P()