List of all items
Structs
- Tensor
- backend::Cpu
- dispatch::Dispatch
- primitives::Abs
- primitives::Add
- primitives::Arange
- primitives::Broadcast
- primitives::Cos
- primitives::Div
- primitives::Exp
- primitives::Full
- primitives::MatMul
- primitives::Mul
- primitives::Negative
- primitives::Normal
- primitives::ReduceSum
- primitives::Reshape
- primitives::Rsqrt
- primitives::Sign
- primitives::Sin
- primitives::Sqrt
- primitives::Square
- primitives::Sub
- primitives::Transpose
Enums
Traits
- ElemType
- FromTensorMap
- Module
- Shape
- WithTensors
- backend::Backend
- dispatch::Eval
- ops::ArangeArgs
- ops::Axes
- ops::ReduceSumArgs
- primitives::Primitive
- utils::TensorIter
Functions
- dispatch::eval_rule
- dispatch::register
- dispatch::register_custom
- eval
- grad
- jvp
- ops::abs
- ops::add
- ops::arange
- ops::broadcast_to
- ops::cos
- ops::div
- ops::exp
- ops::full
- ops::matmul
- ops::mul
- ops::negative
- ops::normal
- ops::reduce_sum
- ops::reshape
- ops::rsqrt
- ops::sign
- ops::sin
- ops::sqrt
- ops::square
- ops::sub
- ops::transpose
- utils::dot_graph
- value_and_grad
- vjp