List of all items
Structs
- device::Cpu
- device::Cuda
- fusion::ComputeGraph
- fusion::FusionEngine
- fusion::FusionOpportunity
- fusion::GraphNode
- hardware::HardwareDevice
- layout::DeviceInfo
- layout::LayoutOptimizer
- memory::AllocationStats
- memory::MemoryLayoutOptimizer
- memory::MemoryPool
- memory::MemoryStats
- memory::OptimizedLayout
- memory::TrackedAllocator
- metal::MetalBuffer
- metal::MetalDevice
- metal::MetalPipeline
- profiler::Benchmark
- profiler::BenchmarkResult
- profiler::OpStats
- profiler::ProfileRecord
- profiler::ProfileScope
- profiler::ProfileSummary
- profiler::Profiler
- rocm::RocmBuffer
- rocm::RocmDevice
- rocm::RocmKernel
- shape::Shape
- shape::Strides
- sparse::SparseTensorCOO
- sparse::SparseTensorCSC
- sparse::SparseTensorCSR
- storage::Storage
- storage::StorageReadGuard
- storage::StorageWriteGuard
- tensor::Tensor
- tpu::TpuBuffer
- tpu::TpuDevice
- tpu::TpuPod
- tpu::xla::CompiledXla
- tpu::xla::XlaComputation
Enums
- dtype::DType
- error::GhostError
- fusion::FusionPattern
- hardware::ElementwiseOp
- hardware::HardwareBackend
- layout::MemoryLayout
- layout::OperationType
- tpu::PodTopology
- tpu::TpuVersion
- tpu::xla::XlaOp
Traits
Macros
Functions
- hardware::list_devices
- metal::mps::batch_norm_mps
- metal::mps::conv2d_mps
- metal::mps::matmul_mps
- metal::mps::relu_mps
- neon::add_neon
- neon::conv2d_neon
- neon::dot_neon
- neon::is_neon_available
- neon::matmul_neon
- neon::mul_neon
- neon::relu_neon
- neon::sigmoid_neon
- ops::conv::conv2d_optimized
- ops::simd::add_simd
- ops::simd::gelu_simd
- ops::simd::relu_simd
- ops::simd::sigmoid_simd
- profiler::global_profiler
- rocm::ops::batch_norm_rocm
- rocm::ops::conv2d_rocm
- rocm::ops::matmul_rocm
- rocm::ops::relu_rocm
- serialize::load_state_dict
- serialize::safetensors::load
- serialize::safetensors::save
- serialize::save_state_dict
- simd_ops::simd_add_f32
- simd_ops::simd_dot_f32
- simd_ops::simd_mul_f32
- simd_ops::simd_relu_f32
- tpu::ops::attention_tpu
- tpu::ops::batch_matmul_tpu
- tpu::ops::conv2d_tpu
- tpu::ops::matmul_tpu