Skip to main content
Log in

Mojo function

pack_filter_shape_impl

pack_filter_shape_impl[filter_type: DType](Q: Int, R: Int, S: Int, C: Int, F: Int, num_groups: Int) -> Index[6]

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

Args:

  • Q (Int): Original Q filter dimension.
  • R (Int): Original R filter dimension.
  • S (Int): Original S filter dimension.
  • C (Int): Original C filter dimension.
  • F (Int): Original F filter dimension.
  • num_groups (Int): Number of groups in the convolution.

Returns:

The output shape.

Was this page helpful?