Skip to main content
Log in

Mojo function

pack_conv_filter_shape

pack_conv_filter_shape[single_thread_blocking_override: Bool](filter: NDBuffer[type, rank, origin, shape, strides, alignment=alignment, address_space=address_space, exclusive=exclusive], num_groups: Int) -> Index[(rank + 1)]

Compute the output shape of convolution filter packing.

Parameters:

  • single_thread_blocking_override (Bool): If True, then the operation is run synchronously using a single thread.

Args:

  • filter (NDBuffer[type, rank, origin, shape, strides, alignment=alignment, address_space=address_space, exclusive=exclusive]): The filter to be packed.
  • num_groups (Int): The number of groups in the convolution.

Returns:

The output shape.

Was this page helpful?