List of all items

Structs

• Bytes
• DeviceHandle
• DistributionSampler
• StreamId
• TensorData
• Tolerance
• api::Bytes
• api::DeviceSettings
• api::DimIter
• api::DistributionSampler
• api::PrintOptions
• api::StreamId
• api::Tensor
• api::TensorCreationOptions
• api::TensorData
• api::Tolerance
• api::Transaction
• api::bf16
• api::f16
• api::grid::GridOptions
• api::quantization::CalibrationRange
• api::signal::StftOptions
• api::sparse::CsrTensor
• backend::BackTrace
• bf16
• device::DeviceId
• device::DeviceSettings
• device::HashMap
• distribution::DistributionSampler
• f16
• indexing::IndexWrap
• ops::AttentionModuleOptions
• ops::Conv2dBackward
• ops::Conv3dBackward
• ops::ConvOptions
• ops::ConvTransposeOptions
• ops::CsrAddition
• ops::DeformConv2dBackward
• ops::DeformConvOptions
• ops::GridSampleOptions
• ops::InterpolateBackward
• ops::InterpolateOptions
• ops::MaxPool1dBackward
• ops::MaxPool1dWithIndices
• ops::MaxPool2dBackward
• ops::MaxPool2dWithIndices
• ops::PaddedConvOptions
• ops::TransactionPrimitive
• ops::TransactionPrimitiveData
• ops::UnfoldOptions
• quantization::BlockSize
• quantization::QuantScheme
• quantization::QuantizedBytes
• shape::Shape
• shape::SmallVec
• slice::Slice
• slice::SliceIter
• tensor::Bool
• tensor::Float
• tensor::Int
• tensor::TensorContainer
• tensor::quantization::QParamTensor
• tensor::quantization::QParams
• tensor::quantization::QuantizationParametersPrimitive

Enums

• AllocationProperty
• BoolStore
• DType
• DataError
• Distribution
• DistributionSamplerKind
• FloatDType
• IntDType
• Scalar
• api::AllocationProperty
• api::BoolStore
• api::DType
• api::DataError
• api::Distribution
• api::DistributionSamplerKind
• api::FloatDType
• api::IndexingUpdateOp
• api::IntDType
• api::Scalar
• api::grid::GridIndexing
• api::grid::GridSparsity
• api::grid::IndexPos
• api::linalg::Norm
• backend::DTypeUsage
• backend::ExecutionError
• device::DeviceError
• device::DeviceServiceStage
• distribution::Distribution
• distribution::DistributionSamplerKind
• element::Scalar
• indexing::IndexingUpdateOp
• ops::GridSamplePaddingMode
• ops::InterpolateMode
• ops::PadMode
• primitive::TensorPrimitive
• quantization::QuantLevel
• quantization::QuantMode
• quantization::QuantParam
• quantization::QuantPropagation
• quantization::QuantStore
• quantization::QuantValue
• shape::ExpressionError
• shape::MetadataError
• tensor::IndexingUpdateOp
• tensor::quantization::Calibration

Traits

• Element
• ElementComparison
• ElementConversion
• ElementEq
• ElementLimits
• ElementOrdered
• ElementRandom
• api::BroadcastArgs
• api::Cast
• api::Element
• api::ElementComparison
• api::ElementConversion
• api::ElementEq
• api::ElementLimits
• api::ElementOrdered
• api::ElementRandom
• api::IntoPadding
• api::MovedimArgs
• api::ReshapeArgs
• api::TensorMetadata
• api::cast::ToElement
• api::quantization::QTensorPrimitive
• backend::AutodiffBackend
• backend::Backend
• backend::BackendTypes
• cast::ToElement
• device::Device
• device::DeviceOps
• device::DeviceService
• element::Element
• element::ElementComparison
• element::ElementConversion
• element::ElementEq
• element::ElementLimits
• element::ElementOrdered
• element::ElementRandom
• element::cast::ToElement
• indexing::AsIndex
• indexing::AsSize
• ops::ActivationOps
• ops::BoolTensorOps
• ops::FloatTensorOps
• ops::IntTensorOps
• ops::ModuleOps
• ops::QTensorOps
• ops::SparseOps
• ops::TransactionOps
• primitive::QTensorPrimitive
• primitive::TensorMetadata
• shape::SliceOps
• slice::SliceArg
• slice::SliceOps
• tensor::BasicAutodiffOps
• tensor::BasicOps
• tensor::FloatMathOps
• tensor::Numeric
• tensor::Ordered
• tensor::TensorKind
• tensor::TransactionOp

Macros

• dequant_op_flow
• dequant_op_quant
• doc_tensor
• make_element
• ops::dequant_op_flow
• ops::dequant_op_quant
• shape::shape
• slice::s

Functions

• api::activation::celu
• api::activation::elu
• api::activation::gelu
• api::activation::gelu_approximate
• api::activation::glu
• api::activation::hard_shrink
• api::activation::hard_sigmoid
• api::activation::hard_swish
• api::activation::leaky_relu
• api::activation::log_sigmoid
• api::activation::log_softmax
• api::activation::mish
• api::activation::prelu
• api::activation::quiet_softmax
• api::activation::relu
• api::activation::selu
• api::activation::shrink
• api::activation::sigmoid
• api::activation::silu
• api::activation::soft_shrink
• api::activation::softmax
• api::activation::softmin
• api::activation::softplus
• api::activation::softsign
• api::activation::tanh
• api::activation::thresholded_relu
• api::cartesian_grid
• api::check_closeness
• api::grid::affine_grid_2d
• api::grid::meshgrid
• api::grid::meshgrid_stack
• api::linalg::cosine_similarity
• api::linalg::det
• api::linalg::diag
• api::linalg::l0_norm
• api::linalg::l1_norm
• api::linalg::l2_norm
• api::linalg::lp_norm
• api::linalg::lu
• api::linalg::matvec
• api::linalg::max_abs_norm
• api::linalg::min_abs_norm
• api::linalg::outer
• api::linalg::outer_dim
• api::linalg::trace
• api::linalg::vector_norm
• api::linalg::vector_normalize
• api::loss::cross_entropy_with_logits
• api::module::adaptive_avg_pool1d
• api::module::adaptive_avg_pool2d
• api::module::attention
• api::module::attention_fallback
• api::module::avg_pool1d
• api::module::avg_pool2d
• api::module::conv1d
• api::module::conv2d
• api::module::conv3d
• api::module::conv_transpose1d
• api::module::conv_transpose2d
• api::module::conv_transpose3d
• api::module::ctc_loss
• api::module::deform_conv2d
• api::module::embedding
• api::module::interpolate
• api::module::linear
• api::module::max_pool1d
• api::module::max_pool1d_with_indices
• api::module::max_pool2d
• api::module::max_pool2d_with_indices
• api::module::unfold4d
• api::quantization::compute_q_params
• api::quantization::compute_range
• api::read_sync
• api::set_print_options
• api::signal::blackman_window
• api::signal::cfft
• api::signal::hamming_window
• api::signal::hann_window
• api::signal::irfft
• api::signal::istft
• api::signal::rfft
• api::signal::stft
• api::try_read_sync
• device::get_device_settings
• device::set_default_dtypes
• device::set_default_float_dtype
• device::set_default_int_dtype
• indexing::ravel_index
• indexing::wrap_index
• ops::attention::attention_fallback
• ops::conv::calculate_conv_output_shape
• ops::conv::calculate_conv_output_size
• ops::conv::calculate_conv_output_sizes
• ops::conv::calculate_conv_padding
• ops::conv::calculate_conv_transpose_output_shape
• ops::conv::calculate_conv_transpose_output_size
• ops::conv::calculate_padding_out
• ops::conv::calculate_pool_output_shape
• ops::conv::calculate_pool_output_size
• ops::ctc::ctc_grad_from_alpha_beta_default
• ops::ctc::ctc_loss_default
• ops::grid_sample::float_grid_sample_2d_ref
• ops::unfold::calculate_unfold_shape
• ops::unfold::calculate_unfold_windows
• read_sync
• shape::calculate_matmul_output
• tensor::quantization::compute_q_params
• tensor::quantization::compute_range
• tensor::quantization::params_shape
• try_read_sync

Type Aliases

• BoolDType
• api::BoolDType
• api::quantization::QuantizationParameters
• backend::DTypeUsageSet
• device::ServerUtilitiesHandle
• tensor::BoolElem
• tensor::BoolTensor
• tensor::Device
• tensor::FloatElem
• tensor::FloatTensor
• tensor::IntElem
• tensor::IntTensor
• tensor::QuantizedTensor

Constants

• api::DEFAULT_ATOL
• api::DEFAULT_RTOL
• tensor::quantization::QPARAM_ALIGN