Skip to main content

Mojo function

transpose_3d_swap_inner

transpose_3d_swap_inner[rank: Int, dtype: DType, //](output: NDBuffer[dtype, output.origin, output.shape], input: NDBuffer[dtype, input.origin, input.shape], perms: UnsafePointer[Scalar[DType.int], perms.origin], simplified_input_shape: IndexList[rank], simplified_rank: Int)

Was this page helpful?