List of all items
Structs
- Aux
- BF16
- CandleBackend
- F16
- F32
- F64
- I64
- Tensor
- U32
- U8
- device::Cpu
- device::Cuda
- device::Metal
- dispatch::BackendWrapper
- primitives::Abs
- primitives::Add
- primitives::Arange
- primitives::ArgMax
- primitives::ArgMin
- primitives::AvgPool1d
- primitives::AvgPool2d
- primitives::Broadcast
- primitives::Concatenate
- primitives::Conv1d
- primitives::Conv2d
- primitives::ConvTranspose1d
- primitives::ConvTranspose2d
- primitives::Cos
- primitives::Div
- primitives::Equal
- primitives::Erf
- primitives::Exp
- primitives::FlashAttention
- primitives::FromArray
- primitives::Full
- primitives::Gather
- primitives::Greater
- primitives::GreaterEqual
- primitives::IndexAdd
- primitives::IndexSelect
- primitives::Less
- primitives::LessEqual
- primitives::Log
- primitives::Log10
- primitives::Log2
- primitives::LogSoftmax
- primitives::MatMul
- primitives::MaxPool1d
- primitives::MaxPool2d
- primitives::Maximum
- primitives::Minimum
- primitives::Mul
- primitives::Narrow
- primitives::Negative
- primitives::Normal
- primitives::NotEqual
- primitives::Permute
- primitives::PowerFloat
- primitives::Random
- primitives::ReduceMax
- primitives::ReduceMin
- primitives::ReduceSum
- primitives::Reshape
- primitives::Rsqrt
- primitives::ScatterAdd
- primitives::Sign
- primitives::Sin
- primitives::Softmax
- primitives::Sqrt
- primitives::Square
- primitives::Sub
- primitives::Tanh
- primitives::ToContiguous
- primitives::ToDType
- primitives::ToDevice
- primitives::Transpose
- primitives::UpsampleNearest1d
- primitives::UpsampleNearest2d
- primitives::Where
- ty_kind::Array
- ty_kind::Basic
- ty_kind::Module
- ty_kind::Tuple
Enums
Traits
- AsDType
- Backend
- DType
- Dim
- Dims
- ElemType
- Eval
- FloatElemType
- Func
- GenericValue
- Shape
- TensorIter
- Type
- Value
- ValueSpec
- device::AsDevice
- device::Device
- nn::ApplyModule
- nn::Module
- nn::NonTrainableModule
- nn::ToApplyArg
- nn::TrainableModule
- nn::WithParams
- ops::ArangeArgs
- ops::ArgReduceArgs
- ops::AvgPool1dArgs
- ops::AvgPool2dArgs
- ops::ClampBound
- ops::FlashAttentionOpts
- ops::FlattenArgs
- ops::MaxPool1dArgs
- ops::MaxPool2dArgs
- ops::ReduceArgs
- ops::ToPair
- ops::VarArgs
- primitives::Primitive
Macros
Functions
- device::cuda_if_available
- dispatch::eval_rule
- dispatch::register
- eval
- grad
- jvp
- linearize
- ops::abs
- ops::add
- ops::arange
- ops::argmax
- ops::argmin
- ops::avg_pool1d
- ops::avg_pool2d
- ops::broadcast_left
- ops::broadcast_right
- ops::broadcast_to
- ops::broadcast_to_unchecked
- ops::cat
- ops::chunk
- ops::clamp
- ops::conv1d
- ops::conv2d
- ops::conv_transpose1d
- ops::conv_transpose2d
- ops::cos
- ops::div
- ops::dropout
- ops::eq
- ops::erf
- ops::exp
- ops::flash_attention
- ops::flatten
- ops::from_array
- ops::from_safetensor
- ops::full
- ops::full_like
- ops::gather
- ops::ge
- ops::gelu
- ops::gt
- ops::index_add
- ops::index_select
- ops::le
- ops::linspace
- ops::log
- ops::log10
- ops::log2
- ops::log_softmax
- ops::lt
- ops::matmul
- ops::max
- ops::max_pool1d
- ops::max_pool2d
- ops::maximum
- ops::mean
- ops::min
- ops::minimum
- ops::mul
- ops::narrow
- ops::ne
- ops::neg
- ops::new_gelu
- ops::ones
- ops::ones_like
- ops::permute
- ops::powf
- ops::rand
- ops::rand_like
- ops::rand_with
- ops::randn
- ops::randn_like
- ops::randn_with
- ops::relu
- ops::relu2
- ops::relu6
- ops::reshape
- ops::rsqrt
- ops::scatter_add
- ops::sign
- ops::silu
- ops::sin
- ops::softmax
- ops::sqrt
- ops::square
- ops::squeeze
- ops::sub
- ops::sum
- ops::tanh
- ops::to_contiguous
- ops::to_device
- ops::to_dtype
- ops::transpose
- ops::unsqueeze
- ops::upsample_nearest1d
- ops::upsample_nearest2d
- ops::var
- ops::where_cond
- ops::zeros
- ops::zeros_like
- optimize
- raiexpr
- utils::accelerate_enabled
- utils::assert_all_close
- utils::check_grad
- utils::check_vjp
- utils::cuda_enabled
- utils::dot_graph
- utils::dprint
- utils::metal_enabled
- utils::mkl_enabled
- utils::numerical_jvp
- value_and_grad
- vjp