List of all items
Structs
- Aux
- BasicValue
- F16
- F32
- F64
- ModuleValue
- Tensor
- U32
- U8
- backend::Cpu
- dispatch::Dispatch
- nn::Chain
- primitives::Abs
- primitives::Add
- primitives::Arange
- primitives::ArgMax
- primitives::ArgMin
- primitives::AsType
- primitives::Broadcast
- primitives::Concatenate
- primitives::Cos
- primitives::Div
- primitives::Equal
- primitives::Erf
- primitives::Exp
- primitives::FromArray
- primitives::FromSafetensor
- primitives::Full
- primitives::Gather
- primitives::Greater
- primitives::GreaterEqual
- primitives::IndexSelect
- primitives::Less
- primitives::LessEqual
- primitives::Log
- primitives::Log10
- primitives::Log2
- primitives::LogSoftmax
- primitives::MatMul
- primitives::Maximum
- primitives::Mul
- primitives::Narrow
- primitives::Negative
- primitives::Normal
- primitives::NotEqual
- primitives::PowerFloat
- primitives::ReduceMax
- primitives::ReduceMin
- primitives::ReduceSum
- primitives::Reshape
- primitives::Rsqrt
- primitives::Sign
- primitives::Sin
- primitives::Softmax
- primitives::Sqrt
- primitives::Square
- primitives::Sub
- primitives::Tanh
- primitives::ToContiguous
- primitives::Transpose
- primitives::Where
Enums
Traits
- DType
- Dim
- Dims
- DynDType
- ElemType
- Func
- GenericValue
- Shape
- TensorIter
- Value
- ValueSpec
- backend::Backend
- dispatch::Eval
- nn::Module
- nn::NonTrainableModule
- nn::TrainableModule
- ops::ArangeArgs
- ops::ArgReduceArgs
- ops::FlattenArgs
- ops::ReduceArgs
- ops::VarArgs
- primitives::Primitive
Macros
Functions
- dispatch::eval_rule
- dispatch::register
- dispatch::register_custom
- eval
- grad
- jvp
- ops::abs
- ops::add
- ops::arange
- ops::argmax
- ops::argmin
- ops::as_type
- ops::as_type_of
- ops::broadcast_left
- ops::broadcast_to
- ops::cat
- ops::chunk
- ops::cos
- ops::div
- ops::eq
- ops::erf
- ops::exp
- ops::flatten
- ops::from_array
- ops::full
- ops::full_like
- ops::gather
- ops::ge
- ops::gelu
- ops::gt
- ops::index_select
- ops::le
- ops::log
- ops::log10
- ops::log2
- ops::log_softmax
- ops::lt
- ops::matmul
- ops::max
- ops::maximum
- ops::mean
- ops::min
- ops::mul
- ops::narrow
- ops::ne
- ops::neg
- ops::new_gelu
- ops::normal
- ops::ones_like
- ops::powf
- ops::relu
- ops::reshape
- ops::rsqrt
- ops::sign
- ops::sin
- ops::softmax
- ops::sqrt
- ops::square
- ops::squeeze
- ops::sub
- ops::sum
- ops::tanh
- ops::to_contiguous
- ops::transpose
- ops::unsqueeze
- ops::var
- ops::where_cond
- ops::zeros_like
- raiexpr
- utils::dot_graph
- value_and_grad
- vjp