List of all items
Structs
- Bool
- Bytes
- DimIter
- DistributionSampler
- Float
- Int
- PrintOptions
- Shape
- Slice
- T
- Tensor
- TensorData
- Tolerance
- Transaction
- 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::TransactionPrimitive
- ops::TransactionPrimitiveResult
- ops::UnfoldOptions
- quantization::CalibrationRange
- quantization::QParams
- quantization::QuantizationParametersPrimitive
- quantization::QuantizedBytes
- quantization::SymmetricQuantization
Enums
- DType
- DataError
- Distribution
- DistributionSamplerKind
- FloatDType
- Precision
- TensorPrimitive
- ops::InterpolateMode
- quantization::Calibration
- quantization::QuantizationMode
- quantization::QuantizationScheme
- quantization::QuantizationStrategy
- quantization::QuantizationType
Traits
- BasicAutodiffOps
- BasicOps
- BroadcastArgs
- Element
- ElementComparison
- ElementConversion
- ElementLimits
- ElementPrecision
- ElementRandom
- IndexConversion
- MovedimArgs
- Numeric
- RangesArg
- ReshapeArgs
- TensorKind
- TensorMetadata
- backend::AutodiffBackend
- backend::Backend
- backend::DeviceOps
- cast::ToElement
- ops::ActivationOps
- ops::BoolTensorOps
- ops::FloatTensorOps
- ops::IntTensorOps
- ops::ModuleOps
- ops::QTensorOps
- ops::TransactionOps
- quantization::QTensorPrimitive
- quantization::Quantization
Macros
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
- quantization::pack_i8s_to_u32s
- quantization::unpack_u32s_to_i8s
- read_sync
- set_print_options
- sort
- sort_with_indices
- split
- split_with_sizes
- try_read_sync
Type Aliases
- Device
- ops::BoolTensor
- ops::Device
- ops::FloatElem
- ops::FloatTensor
- ops::IntElem
- ops::IntTensor
- ops::QuantizedTensor
- quantization::QuantizationParameters