List of all items

Structs

• Bytes
• StreamId
• TensorData
• Tolerance
• backend::BackTrace
• backend::DeviceContext
• backend::DeviceGuard
• backend::DeviceId
• backend::DeviceStateGuard
• backend::ops::Conv2dBackward
• backend::ops::Conv3dBackward
• backend::ops::ConvOptions
• backend::ops::ConvTransposeOptions
• backend::ops::DeformConv2dBackward
• backend::ops::DeformConvOptions
• backend::ops::GridSampleOptions
• backend::ops::InterpolateBackward
• backend::ops::InterpolateOptions
• backend::ops::MaxPool1dBackward
• backend::ops::MaxPool1dWithIndices
• backend::ops::MaxPool2dBackward
• backend::ops::MaxPool2dWithIndices
• backend::ops::TransactionPrimitive
• backend::ops::TransactionPrimitiveData
• backend::ops::UnfoldOptions
• bf16
• distribution::DistributionSampler
• f16
• indexing::IndexWrap
• quantization::BlockSize
• quantization::QuantScheme
• quantization::QuantizedBytes
• shape::Shape
• slice::Slice
• slice::SliceIter
• tensor::Bool
• tensor::Float
• tensor::Int
• tensor::TensorContainer
• tensor::quantization::QParamTensor
• tensor::quantization::QParams
• tensor::quantization::QuantizationParametersPrimitive

Enums

• AllocationProperty
• DType
• DataError
• FloatDType
• IntDType
• backend::ExecutionError
• backend::TensorPrimitive
• backend::ops::GridSamplePaddingMode
• backend::ops::InterpolateMode
• backend::ops::PadMode
• distribution::Distribution
• distribution::DistributionSamplerKind
• indexing::BoundsError
• quantization::QuantLevel
• quantization::QuantMode
• quantization::QuantParam
• quantization::QuantPropagation
• quantization::QuantStore
• quantization::QuantValue
• shape::ShapeError
• tensor::IndexingUpdateOp
• tensor::quantization::Calibration

Traits

• backend::AutodiffBackend
• backend::Backend
• backend::Device
• backend::DeviceOps
• backend::DeviceState
• backend::QTensorPrimitive
• backend::TensorMetadata
• backend::ops::ActivationOps
• backend::ops::BoolTensorOps
• backend::ops::FloatTensorOps
• backend::ops::IntTensorOps
• backend::ops::ModuleOps
• backend::ops::QTensorOps
• backend::ops::TransactionOps
• element::Element
• element::ElementComparison
• element::ElementConversion
• element::ElementLimits
• element::ElementRandom
• element::cast::ToElement
• indexing::AsIndex
• slice::SliceArg
• tensor::BasicAutodiffOps
• tensor::BasicOps
• tensor::Numeric
• tensor::TensorKind

Macros

• backend::ops::dequant_op_flow
• backend::ops::dequant_op_quant
• dequant_op_flow
• dequant_op_quant
• doc_tensor
• element::make_element
• make_element
• slice::s

Functions

• backend::ops::attention::naive_attention
• backend::ops::conv::calculate_conv_output_shape
• backend::ops::conv::calculate_conv_output_size
• backend::ops::conv::calculate_conv_output_sizes
• backend::ops::conv::calculate_conv_padding
• backend::ops::conv::calculate_conv_transpose_output_shape
• backend::ops::conv::calculate_conv_transpose_output_size
• backend::ops::conv::calculate_pool_output_shape
• backend::ops::conv::calculate_pool_output_size
• backend::ops::grid_sample::float_grid_sample_2d_ref
• backend::ops::unfold::calculate_unfold_shape
• backend::ops::unfold::calculate_unfold_windows
• indexing::expect_wrap
• indexing::ravel_index
• indexing::try_wrap
• indexing::wrap_index
• read_sync
• shape::calculate_matmul_output
• tensor::quantization::compute_q_params
• tensor::quantization::compute_range
• tensor::quantization::params_shape
• try_read_sync

Type Aliases

• tensor::BoolElem
• tensor::BoolTensor
• tensor::Device
• tensor::FloatElem
• tensor::FloatTensor
• tensor::IntElem
• tensor::IntTensor
• tensor::QuantizedTensor

Constants

• tensor::quantization::QPARAM_ALIGN