Searched refs:reduce_output_shape (Results 1 – 2 of 2) sorted by relevance
83 Shape* reduce_output_shape = reduce_instruction->mutable_shape(); in XLA_TEST_P() local84 *reduce_output_shape->mutable_layout() = in XLA_TEST_P()
1756 auto reduce_output_shape = input_tensor.shape; in Parse() local1758 reduce_output_shape.set(axis, 1); in Parse()1768 reduce_result->tensor.shape = reduce_output_shape; in Parse()