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};