Modules§
- channel_
bias_ add - Channel bias add Triton kernels.
- channel_
cat - channel_
chunk - elemwise_
add - elemwise_
binary - elemwise_
unary - reduction
- Reduction kernels — each CTA handles one output element (one “row” of the
flattened [outer, inner] view). The caller is responsible for reshaping
the input to
[n_outer, n_inner]before invoking these kernels. - upsample_
nearest2d