Mojo function
get_conv2d_shape
get_conv2d_shape[output_layout: Layout, input_layout: Layout, filter_layout_param: Layout, dtype: DType, data_layout: Image2DLayout, filter_layout: Image2DLayout](output: LayoutTensor[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[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[dtype, filter_layout_param, 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], pad_h: IndexList[2], pad_w: IndexList[2], stride: IndexList[2], dilation: IndexList[2], num_groups: Int) -> ConvShape[2]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!