Mojo function
conv_transposed_cpu
conv_transposed_cpu[filter_packed: Bool, filter_is_cfrs: Bool, lambdas_have_fusion: Bool, elementwise_lambda: def[dtype: DType, rank: Int, width: Int](IndexList[rank], SIMD[dtype, width]) capturing -> None](output: TileTensor[output.dtype, output.LayoutType, output.origin, linear_idx_type=output.linear_idx_type, element_size=output.element_size], input: TileTensor[input.dtype, input.LayoutType, input.origin, linear_idx_type=input.linear_idx_type, element_size=input.element_size], filter: TileTensor[filter.dtype, filter.LayoutType, filter.origin, linear_idx_type=filter.linear_idx_type, element_size=filter.element_size], stride: IndexList[(TileTensor[input.dtype, input.LayoutType, input.origin, linear_idx_type=input.linear_idx_type, element_size=input.element_size].rank - 2)], dilation: IndexList[(TileTensor[input.dtype, input.LayoutType, input.origin, linear_idx_type=input.linear_idx_type, element_size=input.element_size].rank - 2)], pad_d: IndexList[2], pad_h: IndexList[2], pad_w: IndexList[2])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!