Searched refs:reduce_input_shape (Results 1 – 1 of 1) sorted by relevance
88 Shape* reduce_input_shape = in XLA_TEST_P() local90 *reduce_input_shape->mutable_layout() = in XLA_TEST_P()110 reduce_input.Relayout(reduce_input_shape->layout()); in XLA_TEST_P()