Skip to main content

Module flatten

teeny_kernels::nn::mlp

Module flatten 

Source

Structs§

FlattenBackward
Copy a contiguous row-major [B, N] gradient back to an output buffer with arbitrary strides.
FlattenForward
Copy a [B, N] tensor with arbitrary input strides to a contiguous row-major [B, N] output.
FlattenOp

Functions§

flatten_backward
Copy a contiguous row-major [B, N] gradient back to an output buffer with arbitrary strides.
flatten_forward
Copy a [B, N] tensor with arbitrary input strides to a contiguous row-major [B, N] output.