List of all items
Structs
- BackTrace
- Bool
- Bytes
- Device
- DeviceConfig
- DeviceFilter
- DeviceId
- DeviceSettings
- Devices
- DimIter
- DistributionSampler
- Float
- Graph
- IndexWrap
- Int
- PrintOptions
- Shape
- Slice
- SliceIter
- SmallVec
- StreamId
- Tensor
- TensorCreationOptions
- TensorData
- Tolerance
- Transaction
- bf16
- container::TensorContainer
- distributed::CollectiveTensor
- distributed::DistributedConfig
- distributed::DistributedContext
- f16
- grid::GridOptions
- kind::Bool
- kind::Float
- kind::Int
- ops::AttentionModuleOptions
- ops::ConvOptions
- ops::ConvTransposeOptions
- ops::DeformConvOptions
- ops::GridSampleOptions
- ops::InterpolateOptions
- ops::PaddedConvOptions
- ops::UnfoldOptions
- quantization::BlockSize
- quantization::CalibrationRange
- quantization::QParamTensor
- quantization::QParams
- quantization::QuantConfig
- quantization::QuantScheme
- quantization::QuantizedBytes
- signal::StftOptions
Enums
- AllocationProperty
- BoolStore
- DType
- DataError
- DeviceError
- DeviceIndex
- DeviceKind
- DeviceType
- Distribution
- DistributionSamplerKind
- ExecutionError
- ExpressionError
- FloatDType
- IndexingUpdateOp
- IntDType
- MetadataError
- Scalar
- distributed::ReduceOperation
- grid::GridIndexing
- grid::GridSparsity
- grid::IndexPos
- kind::Kind
- linalg::Norm
- ops::GridSamplePaddingMode
- ops::IndexingUpdateOp
- ops::InterpolateMode
- ops::PadMode
- quantization::Calibration
- quantization::QuantAcc
- quantization::QuantLevel
- quantization::QuantMode
- quantization::QuantParam
- quantization::QuantPropagation
- quantization::QuantStore
- quantization::QuantValue
Traits
- AsIndex
- AsSize
- BroadcastArgs
- Cast
- Element
- ElementAdd
- ElementComparison
- ElementConversion
- ElementEq
- ElementLimits
- ElementOrdered
- ElementRandom
- IntoPadding
- MovedimArgs
- ReshapeArgs
- SliceArg
- SliceOps
- cast::ToElement
- kind::Autodiff
- kind::Basic
- kind::FloatMath
- kind::Numeric
- kind::Ordered
- kind::TensorKind
- kind::Transaction
Macros
Functions
- activation::celu
- activation::elu
- activation::gelu
- activation::gelu_approximate
- activation::glu
- activation::hard_shrink
- activation::hard_sigmoid
- activation::hard_swish
- activation::hardtanh
- activation::leaky_relu
- activation::log_sigmoid
- activation::log_softmax
- activation::mish
- activation::prelu
- activation::quiet_softmax
- activation::relu
- activation::relu6
- activation::selu
- activation::shrink
- activation::sigmoid
- activation::silu
- activation::soft_shrink
- activation::softmax
- activation::softmin
- activation::softplus
- activation::softsign
- activation::tanh
- activation::tanhshrink
- activation::threshold
- activation::thresholded_relu
- calculate_matmul_output
- capture
- cartesian_grid
- check_closeness
- distributed::all_reduce
- grid::affine_grid_2d
- grid::meshgrid
- grid::meshgrid_stack
- linalg::cosine_similarity
- linalg::det
- linalg::diag
- linalg::l0_norm
- linalg::l1_norm
- linalg::l2_norm
- linalg::lp_norm
- linalg::lu
- linalg::matvec
- linalg::max_abs_norm
- linalg::min_abs_norm
- linalg::outer
- linalg::outer_dim
- linalg::qr
- linalg::trace
- linalg::vector_norm
- linalg::vector_normalize
- loss::cross_entropy_with_logits
- module::adaptive_avg_pool1d
- module::adaptive_avg_pool2d
- module::adaptive_avg_pool3d
- module::attention
- module::attention_fallback
- module::avg_pool1d
- module::avg_pool2d
- module::avg_pool2d_backward
- module::conv1d
- module::conv2d
- module::conv2d_weight_backward
- module::conv3d
- module::conv_transpose1d
- module::conv_transpose2d
- module::conv_transpose3d
- module::ctc_loss
- module::deform_conv2d
- module::embedding
- module::fold4d
- module::interpolate
- module::layer_norm
- module::linear
- module::max_pool1d
- module::max_pool1d_with_indices
- module::max_pool2d
- module::max_pool2d_with_indices
- module::max_pool2d_with_indices_backward
- module::unfold4d
- quantization::compute_q_params
- quantization::compute_range
- quantization::pack_i8s_to_u32s
- quantization::params_shape
- quantization::scale_to_param
- ravel_index
- read_sync
- set_print_options
- signal::blackman_window
- signal::cfft
- signal::hamming_window
- signal::hann_window
- signal::irfft
- signal::istft
- signal::rfft
- signal::stft
- try_read_sync
- wrap_index