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?
Thank you! We'll create more content like this.
Thank you for helping us improve!