Mojo function
conv_miopen
conv_miopen[input_type: DType, filter_type: DType, output_type: DType, filter_is_fcrs: Bool = False](input: LayoutTensor[input_type, 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_type, 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], output: LayoutTensor[output_type, 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], stride: IndexList[2], dilation: IndexList[2], padding: IndexList[2], num_groups: Int, ctx: DeviceContext)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!