Searched refs:new_operand_shapes (Results 1 – 1 of 1) sorted by relevance
3507 std::vector<const Shape*> new_operand_shapes(input_count * 2); in HandleReduce() local3509 new_operand_shapes[i] = &inputs[i].hlo()->shape(); in HandleReduce()3510 new_operand_shapes[i + input_count] = &inits[i]->shape(); in HandleReduce()3515 ShapeInference::InferReduceShape(new_operand_shapes, hlo->dimensions(), in HandleReduce()