ttnn.sharded_to_interleaved_partial

ttnn.sharded_to_interleaved_partial() → None

Converts a partial tensor from sharded_to_interleaved memory layout

:param * input_tensor: input tensor :type * input_tensor: ttnn.Tensor :param * cache_tensor: cache tensor :type * cache_tensor: ttnn.Tensor :param * num_slices: Number of slices. :type * num_slices: int :param * slice_index: Slice index. :type * slice_index: int

:keyword * memory_config: Memory configuration for the operation. :kwtype * memory_config: Optional[ttnn.MemoryConfig] :keyword * output_dtype: Output data type, defaults to same as input. :kwtype * output_dtype: Optional[ttnn.DataType]

Example

>>> interleaved_tensor = ttnn.sharded_to_interleaved(tensor, cache_tensor, 4, 2)