List of all items
Structs
- AutoGraphicsApi
- Dx11
- Dx12
- Metal
- OpenGl
- Vulkan
- WebGpu
- Wgpu
- compute::DynamicKernel
- compute::StaticKernel
- compute::WgpuResource
- compute::WgpuServer
- compute::WgpuStorage
- compute::WorkGroup
- kernel::BinaryElemwiseInplaceRaw
- kernel::BinaryElemwiseRaw
- kernel::CastRaw
- kernel::ContiguousRaw
- kernel::DynamicKernelSettings
- kernel::KernelSettings
- kernel::SourceTemplate
- kernel::UnaryInplaceRaw
- kernel::UnaryRaw
- kernel::UnaryScalarInplaceRaw
- kernel::UnaryScalarRaw
- kernel::conv::Conv2d
- kernel::conv::ConvTranspose2d
- kernel::matmul::MatmulAutotuneKey
- kernel::matmul::MatmulAutotuneOperationSet
- kernel::matmul::MatmulMemCoalescingRaw
- kernel::matmul::MatmulNaiveRaw
- kernel::matmul::unpadded::MatmulTiling2DUnpaddedRaw
- kernel::matmul::vec4::MatmulTiling2Dvec4Raw
- kernel::matmul::vec4_lhs::MatmulTiling2DVec4LhsRaw
- kernel::pool::AvgPool2dBackwardRaw
- kernel::pool::AvgPool2dRaw
- kernel::pool::MaxPool2d
- kernel::pool::MaxPool2dWithIndices
- kernel::pool::MaxPool2dWithIndicesBackward
- kernel::reduce::MeanDimAutotuneOperationSet
- kernel::reduce::RecursiveSumRaw
- kernel::reduce::ReduceAutotuneKey
- kernel::reduce::ReductionArgsRaw
- kernel::reduce::ReductionDimRaw
- kernel::reduce::ReductionDimSharedMemoryRaw
- kernel::reduce::SumDimAutotuneOperationSet
- tensor::WgpuTensor
Enums
Traits
- FloatElement
- GraphicsApi
- IntElement
- compute::Kernel
- kernel::DynamicKernelSource
- kernel::StaticKernelSource
Macros
- binary_elemwise
- binary_elemwise_inplace
- comparison
- comparison_elem
- comparison_elem_inplace
- comparison_inplace
- kernel::binary_elemwise
- kernel::binary_elemwise_inplace
- kernel::kernel_wgsl
- kernel::unary
- kernel::unary_inplace
- kernel::unary_scalar
- kernel::unary_scalar_inplace
- kernel_wgsl
- reduce_tune_ops
- unary
- unary_inplace
- unary_scalar
- unary_scalar_inplace
Functions
- compute::compute_client
- compute::init_async
- compute::select_device
- kernel::binary_elemwise
- kernel::binary_elemwise_default
- kernel::binary_elemwise_inplace
- kernel::binary_elemwise_inplace_default
- kernel::build_info
- kernel::build_info_dyn
- kernel::cast
- kernel::into_contiguous
- kernel::into_contiguous_dyn
- kernel::matmul::matmul_autotune
- kernel::matmul::matmul_mem_coalescing
- kernel::matmul::matmul_mem_coalescing_default
- kernel::matmul::matmul_naive
- kernel::matmul::matmul_naive_default
- kernel::matmul::unpadded::matmul_tiling_2d_unpadded
- kernel::matmul::utils::init_matmul_output
- kernel::matmul::vec4::matmul_tiling_2d_vec4
- kernel::matmul::vec4_lhs::matmul_tiling_2d_vec4_lhs
- kernel::prng::random_bernoulli
- kernel::prng::random_like_uniform
- kernel::prng::random_normal
- kernel::prng::random_uniform
- kernel::reduce::argmax
- kernel::reduce::argmin
- kernel::reduce::init_reduce_output
- kernel::reduce::mean_dim
- kernel::reduce::mean_dim_autotune
- kernel::reduce::mean_dim_shared_memory
- kernel::reduce::sum
- kernel::reduce::sum_dim
- kernel::reduce::sum_dim_autotune
- kernel::reduce::sum_dim_shared_memory
- kernel::unary
- kernel::unary_default
- kernel::unary_inplace
- kernel::unary_inplace_default
- kernel::unary_scalar
- kernel::unary_scalar_default
- kernel::unary_scalar_inplace
- kernel::unary_scalar_inplace_default