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, alignment: Int = 1](IndexList[rank], SIMD[dtype, width]) capturing -> None](output: TileTensor[linear_idx_type=output.linear_idx_type, element_size=output.element_size], input: TileTensor[linear_idx_type=input.linear_idx_type, element_size=input.element_size], filter: TileTensor[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], ctx: Optional[DeviceContext] = None)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!