Skip to main content

Module channel_chunk

teeny_kernels::nn::tensor

Module channel_chunk 

Source

Structs§

ChannelChunkBackward
Channel-wise chunk backward — propagates the gradient of one chunk output back into the full-width input gradient tensor.
ChannelChunkForward
Channel-wise chunk (split) forward — NC layout.
ChannelChunkOp
ChannelChunkRuntimeOp
Runtime op for channel_chunk forward + backward.

Functions§

channel_chunk_backward
Channel-wise chunk backward — propagates the gradient of one chunk output back into the full-width input gradient tensor.
channel_chunk_forward
Channel-wise chunk (split) forward — NC layout.