ttnn.reshard

ttnn.reshard() → None

Converts a tensor from one sharded layout to another sharded layout

:param * input_tensor: input tensor :type * input_tensor: ttnn.Tensor :param * output_memory_config: Memory config with shard spec of output tensor :type * output_memory_config: MemoryConfig

Example

>>> sharded_memory_config_dict = dict(
    core_grid=ttnn.CoreRangeSet(
        {
            ttnn.CoreRange(
                ttnn.CoreCoord(0, 0), ttnn.CoreCoord(1, 1)
            ),
        }
    ),
    strategy=ttnn.ShardStrategy.BLOCK,
),
>>> shard_memory_config = ttnn.create_sharded_memory_config(input_shape, **input_sharded_memory_config_args)
>>> sharded_tensor = ttnn.reshard(tensor, shard_memory_config)