Searched refs:unsharded_shape (Results 1 – 1 of 1) sorted by relevance
61 const Shape unsharded_shape(instr.shape()); in GetShardedShape() local65 GetShardedShape(unsharded_shape, instr.sharding())); in GetShardedShape()67 sharded_shape = unsharded_shape; in GetShardedShape()