List of all items

Structs

• JitBackend
• JitTuneId
• fusion::FusionJitRuntime
• fusion::JitFusionHandle
• kernel::matmul::MatmulAutotuneKey
• kernel::prng::tests_utils::BinStats
• kernel::quantization::QParams
• kernel::quantization::QParamsCompilationArg
• kernel::quantization::QParamsExpand
• kernel::quantization::QParamsLaunch
• kernel::reduce::ArgMax
• kernel::reduce::ArgMin
• kernel::reduce::Mean
• kernel::reduce::Prod
• kernel::reduce::Reduce
• kernel::reduce::ReduceAutotuneKey
• kernel::reduce::ReduceOps
• kernel::reduce::ReducePlane
• kernel::reduce::ReduceShared
• kernel::reduce::ReduceSharedPlane
• kernel::reduce::Sum
• tensor::JitTensor

Enums

• JitAutotuneKey
• fusion::JitOptimization
• fusion::JitOptimizationState
• kernel::MaskFillStrategy
• kernel::MaskWhereStrategy
• kernel::conv::Conv2dStrategy
• kernel::conv::ConvTranspose2dStrategy
• kernel::matmul::MatmulStrategy
• kernel::reduce::ReduceStrategy

Traits

• JitRuntime
• element::BoolElement
• element::FloatElement
• element::IntElement
• element::JitElement
• kernel::Kernel

Macros

• execute_with_dtype
• tensor::execute_with_dtype

Functions

• kernel::bool_cast
• kernel::cast
• kernel::conv::conv2d
• kernel::conv::conv_transpose2d
• kernel::conv::nchw_to_nhwc
• kernel::interpolate::interpolate
• kernel::interpolate::interpolate_backward
• kernel::into_contiguous
• kernel::mask_fill
• kernel::mask_where
• kernel::matmul::matmul
• kernel::matmul::matmul_autotune
• kernel::matmul::utils::init_matmul_output
• kernel::prng::random_bernoulli
• kernel::prng::random_like_uniform
• kernel::prng::random_normal
• kernel::prng::random_uniform
• kernel::prng::tests_utils::calculate_bin_stats
• kernel::quantization::dequantize
• kernel::quantization::quantize
• kernel::reduce::autotune_reduce
• kernel::reduce::init_reduce_output
• kernel::reduce::reduce
• kernel::reduce::reduce_dim

Type Aliases

• kernel::quantization::QTensor

Constants

• kernel::PLANE_DIM_APPROX