Mojo function
get_conv_shape
get_conv_shape[rank: Int, filter_packed: Bool](output: LayoutTensor[output.dtype, output.layout, output.origin, address_space=output.address_space, element_layout=output.element_layout, layout_int_type=output.layout_int_type, linear_idx_type=output.linear_idx_type, masked=output.masked, alignment=output.alignment], input: LayoutTensor[input.dtype, input.layout, input.origin, address_space=input.address_space, element_layout=input.element_layout, layout_int_type=input.layout_int_type, linear_idx_type=input.linear_idx_type, masked=input.masked, alignment=input.alignment], 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], stride: IndexList[rank], dilation: IndexList[rank], pad_d: IndexList[2], pad_h: IndexList[2], pad_w: IndexList[2], num_groups: Int) -> ConvShape[rank]
Returns:
get_conv_shape[rank: Int, filter_packed: Bool](output: TileTensor[output.dtype, output.LayoutType, output.origin, address_space=output.address_space, linear_idx_type=output.linear_idx_type, element_size=output.element_size], input: TileTensor[input.dtype, input.LayoutType, input.origin, address_space=input.address_space, linear_idx_type=input.linear_idx_type, element_size=input.element_size], filter: TileTensor[filter.dtype, filter.LayoutType, filter.origin, address_space=filter.address_space, linear_idx_type=filter.linear_idx_type, element_size=filter.element_size], stride: IndexList[rank], dilation: IndexList[rank], pad_d: IndexList[2], pad_h: IndexList[2], pad_w: IndexList[2], num_groups: Int) -> ConvShape[rank]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!