Searched refs:new_operand_shapes (Results 1 – 1 of 1) sorted by relevance
2735 std::vector<Shape*> new_operand_shapes(input_count * 2); in HandleReduce() local2737 new_operand_shapes[i] = inputs[i].hlo()->mutable_shape(); in HandleReduce()2738 new_operand_shapes[i + input_count] = inits[i]->mutable_shape(); in HandleReduce()2743 ShapeInference::InferReduceShape(new_operand_shapes, hlo->dimensions(), in HandleReduce()