List of all items

Structs

• Bool
• Bytes
• DimIter
• DistributionSampler
• Float
• IndexWrap
• Int
• PrintOptions
• Shape
• Slice
• SliceIter
• StreamId
• Tensor
• TensorCreationOptions
• TensorData
• Tolerance
• Transaction
• TransactionPrimitive
• 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
• container::TensorContainer
• f16
• grid::GridOptions
• ops::Conv2dBackward
• ops::Conv3dBackward
• ops::ConvOptions
• ops::ConvTransposeOptions
• ops::DeformConv2dBackward
• ops::DeformConvOptions
• ops::GridSampleOptions
• ops::InterpolateBackward
• ops::InterpolateOptions
• ops::MaxPool1dBackward
• ops::MaxPool1dWithIndices
• ops::MaxPool2dBackward
• ops::MaxPool2dWithIndices
• ops::TransactionPrimitive
• ops::TransactionPrimitiveData
• ops::UnfoldOptions
• quantization::BlockSize
• quantization::CalibrationRange
• quantization::QParamTensor
• quantization::QParams
• quantization::QuantScheme
• quantization::QuantizationParametersPrimitive
• quantization::QuantizedBytes

Enums

• AllocationProperty
• BoundsError
• DType
• DataError
• Distribution
• DistributionSamplerKind
• ExpressionError
• FloatDType
• IndexingUpdateOp
• IntDType
• ShapeError
• TensorPrimitive
• backend::ExecutionError
• backend::TensorPrimitive
• backend::ops::GridSamplePaddingMode
• backend::ops::InterpolateMode
• backend::ops::PadMode
• grid::GridIndexing
• grid::GridSparsity
• grid::IndexPos
• linalg::Norm
• ops::GridSamplePaddingMode
• ops::InterpolateMode
• ops::PadMode
• quantization::Calibration
• quantization::QuantLevel
• quantization::QuantMode
• quantization::QuantParam
• quantization::QuantPropagation
• quantization::QuantStore
• quantization::QuantValue

Traits

• AsIndex
• AsSize
• BasicAutodiffOps
• BasicOps
• BroadcastArgs
• Element
• ElementComparison
• ElementConversion
• ElementLimits
• ElementRandom
• MovedimArgs
• Numeric
• ReshapeArgs
• SliceArg
• TensorKind
• TensorMetadata
• 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
• cast::ToElement
• ops::ActivationOps
• ops::BoolTensorOps
• ops::FloatTensorOps
• ops::IntTensorOps
• ops::ModuleOps
• ops::QTensorOps
• ops::TransactionOps
• quantization::QTensorPrimitive

Macros

• s

Functions

• activation::gelu
• activation::glu
• activation::hard_sigmoid
• activation::hard_swish
• 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
• 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
• calculate_matmul_output
• cartesian_grid
• check_closeness
• expect_wrap
• grid::affine_grid_2d
• grid::meshgrid
• grid::meshgrid_stack
• linalg::cosine_similarity
• linalg::diag
• linalg::l0_norm
• linalg::l1_norm
• linalg::l2_norm
• linalg::lp_norm
• linalg::lu_decomposition
• linalg::matvec
• linalg::max_abs_norm
• linalg::min_abs_norm
• linalg::outer
• linalg::outer_dim
• linalg::trace
• linalg::vector_norm
• linalg::vector_normalize
• loss::cross_entropy_with_logits
• module::adaptive_avg_pool1d
• module::adaptive_avg_pool2d
• module::attention
• 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::naive_attention
• module::unfold4d
• ops::attention::naive_attention
• 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_pool_output_shape
• ops::conv::calculate_pool_output_size
• ops::grid_sample::float_grid_sample_2d_ref
• ops::unfold::calculate_unfold_shape
• ops::unfold::calculate_unfold_windows
• quantization::compute_q_params
• quantization::compute_range
• quantization::params_shape
• ravel_index
• read_sync
• set_print_options
• try_read_sync
• try_wrap
• wrap_index

Type Aliases

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

Constants

• DEFAULT_ATOL
• DEFAULT_RTOL
• linalg::DEFAULT_EPSILON
• quantization::QPARAM_ALIGN