Searched refs:reduce_output_shape (Results 1 – 1 of 1) sorted by relevance
84 Shape* reduce_output_shape = reduce_instruction->mutable_shape(); in XLA_TEST_P() local85 *reduce_output_shape->mutable_layout() = in XLA_TEST_P()