Skip to main content

Module channel_cat

teeny_kernels::nn::tensor

Module channel_cat 

Source

Structs§

ChannelCatBackward
Channel-wise cat backward — extracts the gradient slice for one input from the combined upstream gradient tensor.
ChannelCatForward
Channel-wise concatenation forward — NC layout.
ChannelCatOp
ChannelCatRuntimeOp
Combined runtime op for channel-cat forward + backward.

Functions§

channel_cat_backward
Channel-wise cat backward — extracts the gradient slice for one input from the combined upstream gradient tensor.
channel_cat_forward
Channel-wise concatenation forward — NC layout.