For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

get_partition

def get_partition(task_id: Int, num_partitions: IndexList[4], conv_shape: ConvShape, micro_kernel_height: Int, micro_kernel_f_size: Int) -> ConvPartition

Returns:

ConvPartition