List of all items
Structs
- DTypeParseError
- MetalDevice
- MetalStorage
- StridedIndex
- Tensor
- TensorId
- UgIOp1
- Var
- backprop::GradStore
- conv::ParamsConv1D
- conv::ParamsConv2D
- conv::ParamsConvTranspose1D
- conv::ParamsConvTranspose2D
- cpu_backend::CpuDevice
- display::PrinterOptions
- dummy_cuda_backend::CudaDevice
- dummy_cuda_backend::CudaStorage
- error::MatMulUnexpectedStriding
- layout::ContiguousOffsetsWithBroadcast
- layout::Layout
- npy::NpzTensors
- op::BackpropOp
- pickle::PthTensors
- pickle::Stack
- pickle::TensorInfo
- quantized::QTensor
- quantized::ggml_file::Content
- quantized::ggml_file::HParams
- quantized::ggml_file::Vocab
- quantized::gguf_file::Content
- quantized::gguf_file::TensorInfo
- quantized::k_quants::BlockQ2K
- quantized::k_quants::BlockQ3K
- quantized::k_quants::BlockQ4K
- quantized::k_quants::BlockQ4_0
- quantized::k_quants::BlockQ4_1
- quantized::k_quants::BlockQ5K
- quantized::k_quants::BlockQ5_0
- quantized::k_quants::BlockQ5_1
- quantized::k_quants::BlockQ6K
- quantized::k_quants::BlockQ8K
- quantized::k_quants::BlockQ8_0
- quantized::k_quants::BlockQ8_1
- safetensors::BufferedSafetensors
- safetensors::MmapedFile
- safetensors::MmapedSafetensors
- safetensors::SliceSafetensors
- shape::Shape
- streaming::Map
- streaming::StreamTensor
- streaming::StreamingBinOp
Enums
- DType
- Device
- DeviceLocation
- MetalError
- Storage
- StridedBlocks
- TensorIndexer
- conv::CudnnFwdAlgo
- cpu_backend::CpuStorage
- cpu_backend::CpuStorageRef
- error::Error
- op::BinaryOp
- op::CmpOp
- op::Op
- op::ReduceOp
- op::UnaryOp
- pickle::Object
- pickle::OpCode
- quantized::GgmlDType
- quantized::QMatMul
- quantized::QStorage
- quantized::ggml_file::VersionedMagic
- quantized::gguf_file::Value
- quantized::gguf_file::ValueType
- quantized::gguf_file::VersionedMagic
- scalar::TensorScalar
- shape::D
- streaming::BinOp
Traits
- CustomOp1
- CustomOp2
- CustomOp3
- FloatDType
- IndexOp
- InplaceOp1
- InplaceOp2
- InplaceOp3
- IntDType
- Module
- ModuleT
- NdArray
- ToUsize2
- WithDType
- backend::BackendDevice
- backend::BackendStorage
- cpu::kernels::VecOps
- cpu_backend::Map1
- cpu_backend::Map1Any
- cpu_backend::Map2
- cpu_backend::Map2U8
- op::BinaryOpT
- op::UnaryOpT
- quantized::QuantizedType
- quantized::k_quants::GgmlType
- safetensors::Load
- scalar::TensorOrScalar
- shape::Dim
- shape::Dims
- shape::ShapeWithOneHole
- streaming::Dim
- streaming::StreamingModule
Macros
Functions
- cpu::erf::erf
- cpu::erf::erf_inv
- cpu::erf::erfc
- cpu::erf::erfc_inv
- cpu::kernels::par_for_each
- cpu::kernels::par_range
- cpu_backend::binary_map
- cpu_backend::binary_map_vec
- cpu_backend::unary_map
- cpu_backend::unary_map_vec
- display::print_options
- display::set_edge_items
- display::set_line_width
- display::set_precision
- display::set_print_options
- display::set_print_options_default
- display::set_print_options_full
- display::set_print_options_short
- display::set_sci_mode
- display::set_threshold
- dummy_cuda_backend::gemm_reduced_precision_bf16
- dummy_cuda_backend::gemm_reduced_precision_f16
- dummy_cuda_backend::gemm_reduced_precision_f32
- dummy_cuda_backend::set_gemm_reduced_precision_bf16
- dummy_cuda_backend::set_gemm_reduced_precision_f16
- dummy_cuda_backend::set_gemm_reduced_precision_f32
- error::zip
- pickle::read_all
- pickle::read_all_with_key
- pickle::read_pth_tensor_info
- quantized::ggml_file::qtensor_from_ggml
- quantized::gguf_file::write
- quantized::k_quants::matmul
- safetensors::load
- safetensors::load_buffer
- safetensors::save
- shape::dims0
- shape::dims1
- shape::dims2
- shape::dims3
- shape::dims4
- shape::dims5
- test_utils::to_vec0_round
- test_utils::to_vec1_round
- test_utils::to_vec2_round
- test_utils::to_vec3_round
- utils::cuda_is_available
- utils::get_num_threads
- utils::has_accelerate
- utils::has_mkl
- utils::metal_is_available
- utils::with_avx
- utils::with_f16c
- utils::with_neon
- utils::with_simd128