parameters: (cls, dims, num_shards)
return_statement: return [shard_size + 1] * residual + [shard_size] * (num_shards - residual)
docstring: Helper function to split dims values into num_shards.
docstring_summary: Helper function to split dims values into num_shards.
docstring_tokens: Helper function to split dims values into num_shards .
function: def _shard_sizes(cls, dims, num_shards):
      """Helper function to split dims values into num_shards."""
      shard_size, residual = divmod(dims, num_shards)
      return [shard_size + 1] * residual + [shard_size] * (num_shards - residual)
function_tokens: def _shard_sizes ( cls , dims , num_shards ) : shard_size , residual = divmod ( dims , num_shards ) return [ shard_size + 1 ] * residual + [ shard_size ] * ( num_shards - residual )