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