List of all items

Structs

• backend::wgpu::MemoryPool
• backend::wgpu::WgpuContext
• broadcast::ExpandBackward
• graph::Graph
• graph::Node
• graph::TracerContext
• jit::Executor
• jit::Graph
• jit::Node
• jit::Optimizer
• ops::AddBackward
• ops::FusedMatmulBackward
• ops::MatmulBackward
• ops::MeanBackward
• ops::MulBackward
• ops::ReluBackward
• ops::SubBackward
• ops::SumBackward
• ops::VarBackward
• ops::activations::SigmoidBackward
• ops::activations::SoftmaxBackward
• ops::activations::TanhBackward
• ops::conv::Conv2dBackward
• ops::embedding::EmbeddingBackward
• ops::norm::BatchNorm2dBackwardFull
• ops::norm::LayerNormBackward
• ops::pool::MaxPool2dBackward
• ops::view::ContiguousBackward
• ops::view::PermuteBackward
• ops::view::ReshapeBackward
• optimizer::Adam
• storage::PooledBuffer
• storage::Storage
• tensor::Tensor

Enums

• backend::Activation
• backend::wgpu::ElementwiseOp
• graph::NodeOp
• jit::NodeType
• storage::Device

Traits

• autograd::BackwardOp

Functions

• autograd::backward
• backend::wgpu::adam_step_wgpu
• backend::wgpu::clear_memory_pool
• backend::wgpu::contiguous_wgpu
• backend::wgpu::elementwise_wgpu_buffer
• backend::wgpu::flush_queue
• backend::wgpu::get_context
• backend::wgpu::get_memory_pool
• backend::wgpu::matmul_fused_wgpu_buffer
• backend::wgpu::matmul_wgpu
• backend::wgpu::matmul_wgpu_buffer
• backend::wgpu::record_commands
• backend::wgpu::reduce_sum_all_wgpu
• backend::wgpu::reduce_sum_dim0_wgpu
• backend::wgpu::reduce_sum_dim_wgpu
• broadcast::broadcast_shapes
• graph::is_tracing
• graph::record_op
• graph::register_input
• graph::start_tracing
• graph::stop_tracing
• ops::activations::sigmoid
• ops::activations::softmax
• ops::activations::tanh
• ops::add
• ops::conv::conv2d
• ops::div
• ops::embedding::embedding
• ops::get_fused_pipeline_stats
• ops::linear_mse_grads
• ops::matmul
• ops::matmul_bias_norm_activation
• ops::matmul_fused
• ops::mean
• ops::mul
• ops::neg
• ops::norm::batch_norm2d
• ops::norm::layer_norm
• ops::pool::max_pool2d
• ops::relu
• ops::sgd_step
• ops::sub
• ops::sum
• ops::sum_to
• ops::var
• ops::view::contiguous
• ops::view::permute
• ops::view::sum_to
• ops::view::transpose