Searched refs:sharded_subshapes (Results 1 – 1 of 1) sorted by relevance
46 std::vector<Shape> sharded_subshapes; in GetShardedShape() local51 sharded_subshapes.emplace_back(std::move(sharded_subshape)); in GetShardedShape()53 return ShapeUtil::MakeTupleShape(sharded_subshapes); in GetShardedShape()