List of all items

Structs

• Bool
• Bytes
• DimIter
• DistributionSampler
• Float
• Int
• PrintOptions
• Shape
• Slice
• StreamId
• Tensor
• TensorData
• Tolerance
• Transaction
• backend::DeviceContext
• backend::DeviceGuard
• backend::DeviceId
• backend::DeviceStateGuard
• bf16
• container::TensorContainer
• f16
• grid::GridOptions
• ops::Conv2dBackward
• ops::Conv3dBackward
• ops::ConvOptions
• ops::ConvTransposeOptions
• ops::DeformConv2dBackward
• ops::DeformConvOptions
• ops::InterpolateBackward
• ops::InterpolateOptions
• ops::MaxPool1dBackward
• ops::MaxPool1dWithIndices
• ops::MaxPool2dBackward
• ops::MaxPool2dWithIndices
• ops::TransactionPrimitive
• ops::TransactionPrimitiveResult
• ops::UnfoldOptions
• quantization::BlockSize
• quantization::CalibrationRange
• quantization::QParamTensor
• quantization::QParams
• quantization::QuantScheme
• quantization::QuantizationParametersPrimitive
• quantization::QuantizedBytes
• quantization::SymmetricQuantization

Enums

• AllocationError
• DType
• DataError
• Distribution
• DistributionSamplerKind
• FloatDType
• IntDType
• ShapeError
• TensorPrimitive
• grid::GridIndexing
• grid::GridSparsity
• grid::IndexPos
• linalg::Norm
• ops::InterpolateMode
• quantization::Calibration
• quantization::QuantAcc
• quantization::QuantLevel
• quantization::QuantMode
• quantization::QuantParam
• quantization::QuantPropagation
• quantization::QuantStore
• quantization::QuantValue
• quantization::QuantizationStrategy

Traits

• AllocationController
• AsIndex
• BasicAutodiffOps
• BasicOps
• BroadcastArgs
• Element
• ElementComparison
• ElementConversion
• ElementLimits
• ElementRandom
• MovedimArgs
• Numeric
• ReshapeArgs
• SliceArg
• TensorKind
• TensorMetadata
• backend::AutodiffBackend
• backend::Backend
• backend::Device
• backend::DeviceOps
• backend::DeviceState
• cast::ToElement
• indexing::AsIndex
• ops::ActivationOps
• ops::BoolTensorOps
• ops::FloatTensorOps
• ops::IntTensorOps
• ops::ModuleOps
• ops::QTensorOps
• ops::TransactionOps
• quantization::QTensorPrimitive
• quantization::Quantization

Macros

• dequant_op_flow
• dequant_op_quant
• make_element
• ops::dequant_op_flow
• ops::dequant_op_quant
• s

Functions

• activation::gelu
• activation::glu
• activation::hard_sigmoid
• activation::leaky_relu
• activation::log_sigmoid
• activation::log_softmax
• activation::mish
• activation::prelu
• activation::quiet_softmax
• activation::relu
• activation::sigmoid
• activation::silu
• activation::softmax
• activation::softmin
• activation::softplus
• activation::tanh
• argsort
• argwhere_data
• cartesian_grid
• check_closeness
• grid::affine_grid_2d
• grid::meshgrid
• grid::meshgrid_stack
• indexing::canonicalize_dim
• indexing::canonicalize_index
• indexing::wrap_index
• linalg::cosine_similarity
• linalg::diag
• linalg::l0_norm
• linalg::l1_norm
• linalg::l2_norm
• linalg::lp_norm
• linalg::lu_decomposition
• linalg::max_abs_norm
• linalg::min_abs_norm
• linalg::outer
• linalg::trace
• linalg::vector_norm
• linalg::vector_normalize
• loss::cross_entropy_with_logits
• module::adaptive_avg_pool1d
• module::adaptive_avg_pool2d
• module::avg_pool1d
• module::avg_pool2d
• module::conv1d
• module::conv2d
• module::conv3d
• module::conv_transpose1d
• module::conv_transpose2d
• module::conv_transpose3d
• module::deform_conv2d
• module::embedding
• module::interpolate
• module::linear
• module::max_pool1d
• module::max_pool1d_with_indices
• module::max_pool2d
• module::max_pool2d_with_indices
• module::unfold4d
• ops::conv::calculate_conv_output_size
• ops::conv::calculate_conv_output_sizes
• ops::conv::calculate_conv_padding
• ops::conv::calculate_conv_transpose_output_size
• ops::conv::calculate_pool_output_size
• ops::grid_sample::float_grid_sample_2d_bilinear
• ops::unfold::calculate_unfold_shape
• ops::unfold::calculate_unfold_windows
• quantization::compute_q_params
• quantization::compute_range
• quantization::pack_i8s_to_u32s
• quantization::params_shape
• read_sync
• set_print_options
• sort
• sort_with_indices
• try_read_sync

Type Aliases

• Device
• ops::BoolTensor
• ops::Device
• ops::FloatElem
• ops::FloatTensor
• ops::IntElem
• ops::IntTensor
• ops::QuantizedTensor
• quantization::QuantizationParameters

Constants

• DEFAULT_ATOL
• DEFAULT_RTOL
• linalg::DEFAULT_EPSILON