Skip to main content

Mojo function

tile_shape

tile_shape[input_type: DType, repeats_type: DType, single_thread_blocking_override: Bool](input_buf: TileTensor[input_type, LayoutType, origin, address_space=address_space, linear_idx_type=linear_idx_type, element_shape_types=element_shape_types], repeats_buf: TileTensor[repeats_type, LayoutType, origin, address_space=address_space, linear_idx_type=linear_idx_type, element_shape_types=element_shape_types]) -> IndexList[TileTensor[input_type, LayoutType, origin, address_space=address_space, linear_idx_type=linear_idx_type, element_shape_types=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:

Returns:

IndexList: The output shape.

Was this page helpful?