Skip to main content

cubecl_tile/ops/softmax/
mod.rs

1//! The softmax reading of a [`Tile`](crate::Tile): [`OnlineSoftmax`] takes a score where its
2//! holder keeps it, the ownership read off the score.
3
4pub(crate) mod online;
5
6pub use online::{
7    OnlineSoftmax, OnlineSoftmaxExpand, OnlineSoftmaxState, OnlineSoftmaxStateExpand,
8    RelayedFactors, RelayedFactorsExpand,
9};