Searched refs:new_reduce_shapes (Results 1 – 1 of 1) sorted by relevance
45 absl::InlinedVector<Shape, 2> new_reduce_shapes; in HandleReduce() local138 new_reduce_shapes.push_back(new_reduce_shape); in HandleReduce()148 Shape new_reduce_shape = ShapeUtil::MakeMaybeTupleShape(new_reduce_shapes); in HandleReduce()