Mojo function

pack_filter_shape

pack_filter_shape[filter_type: DType, input_shape: DimList[input_shape.values], filter_shape: DimList[filter_shape.values], output_shape: DimList[output_shape.values], strides: DimList[strides.values], dilations: DimList[dilations.values], paddings: DimList[paddings.values], num_groups: Int, single_thread_blocking_override: Bool](filter: LayoutTensor[filter.dtype, filter.layout, filter.origin, address_space=filter.address_space, element_layout=filter.element_layout, layout_int_type=filter.layout_int_type, linear_idx_type=filter.linear_idx_type, masked=filter.masked, alignment=filter.alignment]) -> IndexList[(filter.rank + 1)]

Compute the shape of packed filter. The packed layout is FRSCf. shape_ref should be allocated with size 5 outside this kernel.

Returns:

IndexList: The output shape.