List of all items
Structs
- dag::DAGraph
- dag::format::StringOperation
- dag::manager::DAGraphManager
- dag::manager::DAGraphManagerError
- dag::node::Node
- dag::node::NodeId
- device::cpu::CpuBuffer
- device::cpu::CpuError
- device::cpu::CpuMarker
- device::cpu::CpuThread
- device::function::AbsPowerError
- device::function::AbsPowerErrorBackward
- device::function::BackpropSparseAffineActivate
- device::function::CopyOrAddStrided
- device::function::CrossEntropy
- device::function::DeviceFunction
- device::function::LinearCombination
- device::function::LinearCombinationSplat
- device::function::Matmul
- device::function::MaybeUpdateBatchSize
- device::function::PairwiseMul
- device::function::PairwiseMulBackward
- device::function::ReduceAcrossBatch
- device::function::Select
- device::function::SelectBackprop
- device::function::Set
- device::function::Softmax
- device::function::SoftmaxCrossEntropyBackward
- device::function::SparseAffineActivate
- device::function::SparseToDense
- device::function::SplatAcrossBatch
- device::function::Unary
- device::function::UnaryBackward
- device::operation::AdamConfig
- device::operation::GemmConfig
- device::tensor::DenseMatrix
- device::tensor::Shape
- device::tensor::SparseMatrix
- device::tensor::Tensor
- device::tensor::TensorRef
- graph::Graph
- graph::GraphNodeId
- graph::Node
- graph::builder::Affine
- graph::builder::GraphBuilder
- graph::builder::GraphBuilderNode
- graph::ir::GraphIRManager
- graph::ir::GraphIROperation
- graph::ir::GraphIRType
- graph::ir::compile::GraphIRCompileError
- graph::ir::node::AnnotatedNode
- graph::ir::node::NodeInfo
- graph::ir::operation::GraphIRLeaf
- graph::ir::operation::affine::Affine
- graph::ir::operation::affine::Matmul
- graph::ir::operation::binary::AbsPowerError
- graph::ir::operation::binary::Concat
- graph::ir::operation::binary::FusedPairwiseMulConcat
- graph::ir::operation::binary::Select
- graph::ir::operation::binary::SoftmaxCrossEntropy
- graph::ir::operation::nary::LinearCombination
- graph::ir::operation::sparse::SparseAffineActivate
- graph::ir::operation::unary::Copy
- graph::ir::operation::unary::PairwiseMul
- graph::ir::operation::unary::ReduceAcrossBatch
- graph::ir::operation::unary::Slice
- graph::ir::operation::unary::ToDense
- graph::ir::operation::unary::Unary
- graph::ir::passes::ExchangeConcatAndUnary
- graph::ir::passes::ExchangeElementwiseAndSelect
- graph::ir::passes::ExchangeMatmulAndConcatWithSliceAndMatmul
- graph::ir::passes::FusePairwiseMulWithConcat
- graph::ir::passes::FuseSparseAffineWithDiffableFromOutput
- graph::ir::passes::FuseSparseMatmulWithAdd
- graph::ir::passes::LowPriorityFusions
- trainer::Trainer
- trainer::dataloader::HostDenseMatrix
- trainer::dataloader::HostSparseMatrix
- trainer::dataloader::PreparedBatchDevice
- trainer::dataloader::PreparedBatchHost
- trainer::optimiser::Optimiser
- trainer::optimiser::WrapOptimiser
- trainer::optimiser::adam::AdamW
- trainer::optimiser::adam::AdamWParams
- trainer::optimiser::clip::WeightClipping
- trainer::optimiser::clip::WeightClippingParams
- trainer::optimiser::decay::WeightDecay
- trainer::optimiser::decay::WeightDecayParams
- trainer::optimiser::radam::RAdam
- trainer::optimiser::radam::RAdamParams
- trainer::optimiser::ranger::RangerLookahead
- trainer::optimiser::ranger::RangerLookaheadParams
- trainer::optimiser::ranger::RangerParams
- trainer::schedule::TrainingSchedule
- trainer::schedule::TrainingSteps
Enums
- dag::DAGraphError
- device::OperationError
- device::function::MatmulType
- device::function::Reduce
- device::function::UnaryOp
- device::operation::DiffableFromOutput
- device::tensor::Matrix
- graph::GraphError
- graph::GraphNodeIdTy
- graph::builder::InitSettings
- graph::ir::GraphIRError
- graph::ir::node::AnnotatedNodeError
- graph::ir::operation::GraphIROperationError
- trainer::DataLoadingError
- trainer::TrainerError
- trainer::dataloader::HostMatrix
- trainer::optimiser::utils::Placement
Traits
- dag::Operation
- dag::manager::DAGraphType
- device::Device
- device::DeviceBuffer
- device::function::DeviceOperation
- device::operation::BaseOperations
- device::operation::BlasOperations
- device::operation::CoreDeviceOps
- device::operation::SparseAffineOps
- graph::ir::BackendMarker
- graph::ir::GraphIRMethods
- graph::ir::operation::GraphIROperationBase
- graph::ir::operation::GraphIROperationCompilable
- graph::ir::operation::sparse::SparseAffineImpl
- graph::ir::passes::GraphIRPass
- graph::ir::passes::GraphIRSimplePass
- trainer::dataloader::DataLoader
- trainer::optimiser::OptimiserState
Functions
- dag::topo::topo_order
- device::cpu::sparse::affine_bwd
- device::cpu::sparse::affine_fwd
- device::cpu::sparse::by_chunks_32_1
- device::cpu::sparse::by_chunks_32_2
- device::cpu::sparse::by_chunks_32_3
- device::tensor::read_from_byte_buffer
- device::tensor::rng::vec_f32
- graph::ir::operation::util::batch_size_node
- graph::ir::operation::util::check_dense_eq
- graph::ir::operation::util::check_has_grad
- graph::ir::operation::util::check_matmul
- graph::ir::operation::util::check_no_grad
- graph::ir::operation::util::check_not_batched
- graph::ir::operation::util::check_same_batching
- graph::ir::passes::downcast
- trainer::logger::ansi
- trainer::logger::clear_colours
- trainer::logger::num_cs
- trainer::logger::report_superbatch_finished
- trainer::logger::report_superbatch_progress
- trainer::logger::report_time_left
- trainer::logger::seconds_to_hms
- trainer::logger::set_cbcs
- trainer::logger::set_colour
- trainer::optimiser::utils::load_weights_from_file
- trainer::optimiser::utils::write_weights_to_file