List of all items
Structs
- Tensor
- backend::Cpu
- dispatch::Dispatch
- ops::ReduceSumArgs
- primitives::Add
- primitives::Broadcast
- primitives::Cos
- primitives::Div
- primitives::Full
- primitives::MatMul
- primitives::Mul
- primitives::Negative
- primitives::Normal
- primitives::ReduceSum
- primitives::Reshape
- primitives::Sin
- primitives::Square
- primitives::Sub
- primitives::Transpose
Enums
Traits
Functions
- dispatch::eval_rule
- dispatch::register
- dispatch::register_custom
- eval
- grad
- jvp
- ops::add
- ops::broadcast_to
- ops::cos
- ops::div
- ops::full
- ops::matmul
- ops::mul
- ops::negative
- ops::normal
- ops::reduce_sum
- ops::reshape
- ops::sin
- ops::square
- ops::sub
- ops::transpose
- utils::dot_graph
- value_and_grad
- vjp