Searched defs:GetShardedShape (Results 1 – 1 of 1) sorted by relevance
31 StatusOr<Shape> GetShardedShape(const Shape& shape, in GetShardedShape() function60 StatusOr<Shape> GetShardedShape(const HloInstructionProto& instr) { in GetShardedShape() function