colossalai.zero.shard_utils.tensor_shard_strategy

class colossalai.zero.shard_utils.tensor_shard_strategy.TensorShardStrategy[source]

A naive implementation which shard each tensor evenly over all ranks