List of all items

Structs

• Bool
• Data
• DataSerialize
• DimIter
• DistributionSampler
• Float
• Int
• PrintOptions
• Shape
• T
• Tensor
• TensorData
• backend::DeviceId
• bf16
• container::TensorContainer
• f16
• 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::UnfoldOptions
• quantization::AffineQuantization
• quantization::CalibrationRange
• quantization::MinMaxCalibration
• quantization::QuantizationParameters
• quantization::QuantizationParametersPrimitive
• quantization::SymmetricQuantization
• repr::AdaptiveAvgPool1dBackwardDescription
• repr::AdaptiveAvgPool1dDescription
• repr::AdaptiveAvgPool2dBackwardDescription
• repr::AdaptiveAvgPool2dDescription
• repr::AvgPool1dBackwardDescription
• repr::AvgPool1dDescription
• repr::AvgPool2dBackwardDescription
• repr::AvgPool2dDescription
• repr::BinaryOperationDescription
• repr::CatOperationDescription
• repr::ClampOperationDescription
• repr::Conv1dDescription
• repr::Conv1dOptionsDescription
• repr::Conv2dDescription
• repr::Conv2dOptionsDescription
• repr::Conv3dDescription
• repr::Conv3dOptionsDescription
• repr::ConvTranspose1dDescription
• repr::ConvTranspose1dOptionsDescription
• repr::ConvTranspose2dDescription
• repr::ConvTranspose2dOptionsDescription
• repr::ConvTranspose3dDescription
• repr::ConvTranspose3dOptionsDescription
• repr::DeformConv2dBackwardDescription
• repr::DeformConv2dDescription
• repr::DeformableConv2dOptionsDescription
• repr::DequantizeOperationDescription
• repr::EmbeddingBackwardDescription
• repr::EmbeddingDescription
• repr::ExpandDescription
• repr::ExpandOperationDescription
• repr::FlipOperationDescription
• repr::GatherOperationDescription
• repr::HandleContainer
• repr::InterpolateBackwardDescription
• repr::InterpolateDescription
• repr::InterpolateOptionsDescription
• repr::MaskFillOperationDescription
• repr::MaskWhereOperationDescription
• repr::MaxPool1dDescription
• repr::MaxPool1dWithIndicesBackwardDescription
• repr::MaxPool1dWithIndicesDescription
• repr::MaxPool2dDescription
• repr::MaxPool2dWithIndicesBackwardDescription
• repr::MaxPool2dWithIndicesDescription
• repr::PermuteOperationDescription
• repr::QuantizationParametersDescription
• repr::QuantizeOperationDescription
• repr::QuantizedKind
• repr::QuantizedTensorDescription
• repr::RandomOperationDescription
• repr::ReduceDimWithIndicesDescription
• repr::RepeatDimOperationDescription
• repr::ReshapeDescription
• repr::ScalarOperationDescription
• repr::ScatterOperationDescription
• repr::SelectAssignOperationDescription
• repr::SelectOperationDescription
• repr::SliceAssignOperationDescription
• repr::SliceOperationDescription
• repr::SwapDimsDescription
• repr::TensorDescription
• repr::TensorHandle
• repr::TensorId
• repr::UnaryOperationDescription

Enums

• DType
• DataError
• Distribution
• DistributionSamplerKind
• Precision
• TensorPrimitive
• ops::InterpolateMode
• quantization::QuantizationScheme
• quantization::QuantizationStrategy
• quantization::QuantizationType
• repr::BaseOperationDescription
• repr::BoolOperationDescription
• repr::FloatOperationDescription
• repr::Handle
• repr::HandleKind
• repr::IntOperationDescription
• repr::InterpolateModeDescription
• repr::ModuleOperationDescription
• repr::NumericOperationDescription
• repr::OperationDescription
• repr::TensorStatus

Traits

• BasicAutodiffOps
• BasicOps
• BroadcastArgs
• Element
• ElementComparison
• ElementConversion
• ElementPrecision
• ElementRandom
• MovedimArgs
• Numeric
• RangesArg
• ReshapeArgs
• TensorKind
• backend::AutodiffBackend
• backend::Backend
• backend::BackendBridge
• backend::DeviceOps
• cast::ToElement
• ops::ActivationOps
• ops::BoolTensorOps
• ops::FloatTensorOps
• ops::IntTensorOps
• ops::ModuleOps
• ops::QTensorOps
• quantization::Calibration
• quantization::QTensorPrimitive
• quantization::Quantization
• repr::ReprBackend

Macros

• dequant_op_quant
• make_element
• ops::dequant_op_quant

Functions

• activation::gelu
• 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
• chunk
• 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::max_pool1d
• module::max_pool1d_with_indices
• module::max_pool2d
• module::max_pool2d_with_indices
• module::unfold4d
• narrow
• ops::binary_ops_shape
• ops::conv::calculate_conv_output_size
• ops::conv::calculate_conv_padding
• ops::conv::calculate_conv_transpose_output_size
• ops::conv::calculate_pool_output_size
• read_sync
• set_print_options
• sort
• sort_with_indices
• try_read_sync

Type Aliases

• Device
• ops::BoolTensor
• ops::Device
• ops::FloatElem
• ops::FloatTensor
• ops::FullPrecisionBackend
• ops::IntElem
• ops::IntTensor
• ops::QuantizedTensor