Mojo function

tile_shape

tile_shape[input_type: DType, repeats_type: DType, single_thread_blocking_override: Bool](input_buf: TileTensor[input_type, input_buf.LayoutType, input_buf.origin, address_space=input_buf.address_space, linear_idx_type=input_buf.linear_idx_type, element_shape_types=input_buf.element_shape_types], repeats_buf: TileTensor[repeats_type, repeats_buf.LayoutType, repeats_buf.origin, address_space=repeats_buf.address_space, linear_idx_type=repeats_buf.linear_idx_type, element_shape_types=repeats_buf.element_shape_types]) -> IndexList[TileTensor[input_type, input_buf.LayoutType, input_buf.origin, address_space=input_buf.address_space, linear_idx_type=input_buf.linear_idx_type, element_shape_types=input_buf.element_shape_types].rank]

Compute the output shape of a tile operation, and assert the inputs are compatible.

Parameters:

input_type (DType): Type of the input tensor.
repeats_type (DType): Type of the repeats tensor.
single_thread_blocking_override (Bool): If True, then the operation is run synchronously using a single thread.

Args:

input_buf (TileTensor): The input tensor.
repeats_buf (TileTensor): The repeats tensor.

Returns:

IndexList: The output shape.