List of all items

Structs

• BurnConfig
• Tensor
• backend::Autodiff
• backend::AutodiffDevice
• backend::BackTrace
• backend::Bytes
• backend::DeviceHandle
• backend::DeviceId
• backend::DeviceSettings
• backend::Dispatch
• backend::DispatchTensor
• backend::DistributionSampler
• backend::Flex
• backend::FlexDevice
• backend::HashMap
• backend::IndexWrap
• backend::MemoryPoolUsage
• backend::ProfileDuration
• backend::ProfileOptions
• backend::ProfileTicks
• backend::ProfileToken
• backend::ServiceId
• backend::Shape
• backend::Slice
• backend::SliceIter
• backend::SlicedPoolReport
• backend::SmallVec
• backend::StreamId
• backend::TensorData
• backend::Tolerance
• backend::autodiff::Autodiff
• backend::autodiff::NodeId
• backend::autodiff::checkpoint::base::Checkpointer
• backend::autodiff::checkpoint::state::BackwardStates
• backend::autodiff::checkpoint::strategy::BalancedCheckpointing
• backend::autodiff::checkpoint::strategy::NoCheckpointing
• backend::autodiff::grads::Gradients
• backend::autodiff::ops::ComputePropertyDone
• backend::autodiff::ops::Init
• backend::autodiff::ops::MemoryBound
• backend::autodiff::ops::MemoryBoundRetroForward
• backend::autodiff::ops::NodeGuard
• backend::autodiff::ops::Ops
• backend::autodiff::ops::OpsPrep
• backend::autodiff::ops::Tracked
• backend::autodiff::ops::UnTracked
• backend::backend::BackTrace
• backend::backend::DeviceId
• backend::backend::DeviceSettings
• backend::backend::HashMap
• backend::backend::MemoryPoolUsage
• backend::backend::ProfileDuration
• backend::backend::ProfileOptions
• backend::backend::ProfileTicks
• backend::backend::ProfileToken
• backend::backend::ServiceId
• backend::backend::SlicedPoolReport
• backend::backend::distributed::CollectiveTensor
• backend::backend::distributed::DistributedConfig
• backend::backend::distributed::DistributedParams
• backend::backend::distributed::TensorRef
• backend::backend::ops::AttentionModuleOptions
• backend::backend::ops::AvgPoolOptions
• backend::backend::ops::BatchNormTrain
• backend::backend::ops::BatchNormTrainBackward
• backend::backend::ops::Conv2dBackward
• backend::backend::ops::Conv3dBackward
• backend::backend::ops::ConvOptions
• backend::backend::ops::ConvTransposeOptions
• backend::backend::ops::DeformConv2dBackward
• backend::backend::ops::DeformConvOptions
• backend::backend::ops::GridSampleOptions
• backend::backend::ops::InterpolateBackward
• backend::backend::ops::InterpolateOptions
• backend::backend::ops::MaxPool1dBackward
• backend::backend::ops::MaxPool1dWithIndices
• backend::backend::ops::MaxPool2dBackward
• backend::backend::ops::MaxPool2dWithIndices
• backend::backend::ops::MaxPoolOptions
• backend::backend::ops::PaddedConvOptions
• backend::backend::ops::TransactionPrimitive
• backend::backend::ops::TransactionPrimitiveData
• backend::backend::ops::UnfoldOptions
• backend::bf16
• backend::distribution::DistributionSampler
• backend::f16
• backend::flex::Flex
• backend::flex::FlexDevice
• backend::flex::FlexQTensor
• backend::flex::FlexTensor
• backend::flex::Layout
• backend::fusion::FuserProperties
• backend::fusion::Fusion
• backend::fusion::FusionTensor
• backend::fusion::NoOp
• backend::fusion::UnfusedOp
• backend::fusion::client::GlobalFusionClient
• backend::fusion::custom::CustomOpIr
• backend::fusion::custom::HandleContainer
• backend::fusion::custom::StreamId
• backend::fusion::custom::TensorId
• backend::fusion::custom::TensorIr
• backend::fusion::custom::TensorSpec
• backend::fusion::observer::BlockOperations
• backend::fusion::observer::FusionObservation
• backend::fusion::stream::Context
• backend::fusion::stream::FallbackOp
• backend::fusion::stream::MultiStream
• backend::fusion::stream::OrderedExecution
• backend::fusion::stream::ScalarId
• backend::fusion::stream::StreamId
• backend::indexing::IndexWrap
• backend::ir::AdaptiveAvgPool1dBackwardOpIr
• backend::ir::AdaptiveAvgPool1dOpIr
• backend::ir::AdaptiveAvgPool2dBackwardOpIr
• backend::ir::AdaptiveAvgPool2dOpIr
• backend::ir::AdaptiveAvgPool3dBackwardOpIr
• backend::ir::AdaptiveAvgPool3dOpIr
• backend::ir::AllReduceOpIr
• backend::ir::AttentionOpIr
• backend::ir::AttentionOptionsIr
• backend::ir::AvgPool1dBackwardOpIr
• backend::ir::AvgPool1dOpIr
• backend::ir::AvgPool2dBackwardOpIr
• backend::ir::AvgPool2dOpIr
• backend::ir::BatchNormOpIr
• backend::ir::BinaryOpIr
• backend::ir::CastOpIr
• backend::ir::CatOpIr
• backend::ir::ClampOpIr
• backend::ir::Conv1dBiasBackwardOpIr
• backend::ir::Conv1dOpIr
• backend::ir::Conv1dOptionsIr
• backend::ir::Conv1dWeightBackwardOpIr
• backend::ir::Conv1dXBackwardOpIr
• backend::ir::Conv2dBiasBackwardOpIr
• backend::ir::Conv2dOpIr
• backend::ir::Conv2dOptionsIr
• backend::ir::Conv2dWeightBackwardOpIr
• backend::ir::Conv2dXBackwardOpIr
• backend::ir::Conv3dBiasBackwardOpIr
• backend::ir::Conv3dOpIr
• backend::ir::Conv3dOptionsIr
• backend::ir::Conv3dWeightBackwardOpIr
• backend::ir::Conv3dXBackwardOpIr
• backend::ir::ConvTranspose1dBiasBackwardOpIr
• backend::ir::ConvTranspose1dOpIr
• backend::ir::ConvTranspose1dOptionsIr
• backend::ir::ConvTranspose1dWeightBackwardOpIr
• backend::ir::ConvTranspose2dBiasBackwardOpIr
• backend::ir::ConvTranspose2dOpIr
• backend::ir::ConvTranspose2dOptionsIr
• backend::ir::ConvTranspose2dWeightBackwardOpIr
• backend::ir::ConvTranspose3dBiasBackwardOpIr
• backend::ir::ConvTranspose3dOpIr
• backend::ir::ConvTranspose3dOptionsIr
• backend::ir::ConvTranspose3dWeightBackwardOpIr
• backend::ir::CreationOpIr
• backend::ir::CrossOpIr
• backend::ir::CtcLossBackwardOpIr
• backend::ir::CtcLossOpIr
• backend::ir::CustomOpIr
• backend::ir::DeformConv2dBackwardOpIr
• backend::ir::DeformConv2dOpIr
• backend::ir::DeformableConv2dOptionsIr
• backend::ir::DequantizeOpIr
• backend::ir::DeviceIdIr
• backend::ir::DimOpIr
• backend::ir::EmbeddingBackwardOpIr
• backend::ir::EmbeddingOpIr
• backend::ir::FlipOpIr
• backend::ir::FullOpIr
• backend::ir::GatherNdOpIr
• backend::ir::GatherOpIr
• backend::ir::GraphBindings
• backend::ir::GraphBoundary
• backend::ir::GraphId
• backend::ir::GraphIr
• backend::ir::GridSample2dOpIr
• backend::ir::GridSampleOptionsIr
• backend::ir::HandleContainer
• backend::ir::HardSigmoidOpIr
• backend::ir::InitOperationIr
• backend::ir::InterpolateBackwardOpIr
• backend::ir::InterpolateOpIr
• backend::ir::InterpolateOptionsIr
• backend::ir::LayerNormOpIr
• backend::ir::LinearBiasBackwardOpIr
• backend::ir::LinearOpIr
• backend::ir::LinearWeightBackwardOpIr
• backend::ir::LinearXBackwardOpIr
• backend::ir::MaskFillOpIr
• backend::ir::MaskWhereOpIr
• backend::ir::MatmulOpIr
• backend::ir::MaxPool1dOpIr
• backend::ir::MaxPool1dWithIndicesBackwardOpIr
• backend::ir::MaxPool1dWithIndicesOpIr
• backend::ir::MaxPool2dOpIr
• backend::ir::MaxPool2dWithIndicesBackwardOpIr
• backend::ir::MaxPool2dWithIndicesOpIr
• backend::ir::PadOpIr
• backend::ir::PermuteOpIr
• backend::ir::QuantizationParametersIr
• backend::ir::QuantizeOpIr
• backend::ir::RandomOpIr
• backend::ir::ReduceDimOpIr
• backend::ir::ReduceDimWithIndicesOpIr
• backend::ir::ReduceDimsOpIr
• backend::ir::ReduceOpIr
• backend::ir::RepeatDimOpIr
• backend::ir::ScalarOpIr
• backend::ir::ScatterNdOpIr
• backend::ir::ScatterOpIr
• backend::ir::SelectAssignOpIr
• backend::ir::SelectOpIr
• backend::ir::ShapeOpIr
• backend::ir::SliceAssignOpIr
• backend::ir::SliceOpIr
• backend::ir::SortOpIr
• backend::ir::SortWithIndicesOpIr
• backend::ir::SwapDimsOpIr
• backend::ir::TensorError
• backend::ir::TensorHandle
• backend::ir::TensorId
• backend::ir::TensorIr
• backend::ir::TopKWithIndicesOpIr
• backend::ir::UnaryOpIr
• backend::ir::Unfold4dOpIr
• backend::ir::Unfold4dOptionsIr
• backend::ir::UnfoldOpIr
• backend::quantization::BlockLayout
• backend::quantization::BlockScale
• backend::quantization::BlockSize
• backend::quantization::QParamTensor
• backend::quantization::QParams
• backend::quantization::QuantScheme
• backend::quantization::QuantizationParametersPrimitive
• backend::quantization::QuantizedBytes
• backend::shape::Shape
• backend::shape::SmallVec
• backend::slice::Slice
• backend::slice::SliceIter
• data::dataloader::BatchDataLoader
• data::dataloader::DataLoaderBuilder
• data::dataloader::DatasetIterator
• data::dataloader::FixBatchStrategy
• data::dataloader::MultiThreadDataLoader
• data::dataloader::Progress
• data::dataset::DatasetError
• data::dataset::DatasetIterator
• data::dataset::HuggingfaceDatasetLoader
• data::dataset::InMemDataset
• data::dataset::SqliteDataset
• data::dataset::SqliteDatasetStorage
• data::dataset::SqliteDatasetWriter
• data::dataset::audio::SpeechCommandsDataset
• data::dataset::audio::SpeechItem
• data::dataset::audio::SpeechItemRaw
• data::dataset::source::huggingface::HuggingfaceDatasetLoader
• data::dataset::transform::ComposedDataset
• data::dataset::transform::MapperDataset
• data::dataset::transform::PartialDataset
• data::dataset::transform::SamplerDataset
• data::dataset::transform::SamplerDatasetOptions
• data::dataset::transform::SelectionDataset
• data::dataset::transform::ShuffledDataset
• data::dataset::transform::WindowsDataset
• data::dataset::transform::WindowsIterator
• data::dataset::vision::BoundingBox
• data::dataset::vision::ImageDatasetItem
• data::dataset::vision::ImageFolderDataset
• data::dataset::vision::MnistDataset
• data::dataset::vision::MnistItem
• data::dataset::vision::SegmentationMask
• lr_scheduler::DynLrScheduler
• lr_scheduler::LrSchedulerRecord
• lr_scheduler::composed::ComposedLrScheduler
• lr_scheduler::composed::ComposedLrSchedulerConfig
• lr_scheduler::constant::ConstantLr
• lr_scheduler::cosine::CosineAnnealingLrScheduler
• lr_scheduler::cosine::CosineAnnealingLrSchedulerConfig
• lr_scheduler::cosine::CosineAnnealingLrSchedulerState
• lr_scheduler::exponential::ExponentialLrScheduler
• lr_scheduler::exponential::ExponentialLrSchedulerConfig
• lr_scheduler::exponential::ExponentialLrSchedulerState
• lr_scheduler::linear::LinearLrScheduler
• lr_scheduler::linear::LinearLrSchedulerConfig
• lr_scheduler::linear::LinearLrSchedulerState
• lr_scheduler::module_lr_scheduler::ModuleLearningRate
• lr_scheduler::module_lr_scheduler::ModuleLrScheduler
• lr_scheduler::module_lr_scheduler::ModuleLrSchedulerConfig
• lr_scheduler::noam::NoamLrScheduler
• lr_scheduler::noam::NoamLrSchedulerConfig
• lr_scheduler::noam::NoamLrSchedulerState
• lr_scheduler::sequential::SequentialLrScheduler
• lr_scheduler::sequential::SequentialLrSchedulerConfig
• lr_scheduler::step::StepLrScheduler
• lr_scheduler::step::StepLrSchedulerConfig
• lr_scheduler::step::StepLrSchedulerState
• module::Attribute
• module::Content
• module::DisplaySettings
• module::Flag
• module::Ignored
• module::Lora
• module::LoraAdapter
• module::Param
• module::ParamGroup
• module::ParamId
• module::ParamMapper
• module::QLora
• module::Quantizer
• module::RunningState
• nn::BatchNorm
• nn::BatchNormConfig
• nn::BiGru
• nn::BiGruConfig
• nn::BiLstm
• nn::BiLstmConfig
• nn::BiRnn
• nn::BiRnnConfig
• nn::Celu
• nn::CeluConfig
• nn::CosineSimilarity
• nn::CosineSimilarityConfig
• nn::Dropout
• nn::DropoutConfig
• nn::Elu
• nn::EluConfig
• nn::Embedding
• nn::EmbeddingConfig
• nn::Fold4d
• nn::Fold4dConfig
• nn::GLU
• nn::GateController
• nn::GaussianNoise
• nn::GaussianNoiseConfig
• nn::Gelu
• nn::GroupNorm
• nn::GroupNormConfig
• nn::Gru
• nn::GruConfig
• nn::HardShrink
• nn::HardShrinkConfig
• nn::HardSigmoid
• nn::HardSigmoidConfig
• nn::Identity
• nn::InstanceNorm
• nn::InstanceNormConfig
• nn::LayerNorm
• nn::LayerNormConfig
• nn::LeakyRelu
• nn::LeakyReluConfig
• nn::Linear
• nn::LinearConfig
• nn::LocalResponseNorm
• nn::LocalResponseNormConfig
• nn::Lstm
• nn::LstmConfig
• nn::LstmState
• nn::PRelu
• nn::PReluConfig
• nn::PairwiseDistance
• nn::PairwiseDistanceConfig
• nn::PixelShuffle
• nn::PixelShuffleConfig
• nn::PixelUnshuffle
• nn::PixelUnshuffleConfig
• nn::PositionalEncoding
• nn::PositionalEncodingConfig
• nn::Relu
• nn::RmsNorm
• nn::RmsNormConfig
• nn::Rnn
• nn::RnnConfig
• nn::RnnState
• nn::RotaryEncoding
• nn::RotaryEncodingConfig
• nn::Selu
• nn::Shrink
• nn::ShrinkConfig
• nn::Sigmoid
• nn::SoftShrink
• nn::SoftShrinkConfig
• nn::Softplus
• nn::SoftplusConfig
• nn::Softsign
• nn::SwiGlu
• nn::SwiGluConfig
• nn::Tanh
• nn::ThresholdedRelu
• nn::ThresholdedReluConfig
• nn::Unfold4d
• nn::Unfold4dConfig
• nn::activation::Celu
• nn::activation::CeluConfig
• nn::activation::Elu
• nn::activation::EluConfig
• nn::activation::GLU
• nn::activation::Gelu
• nn::activation::HardShrink
• nn::activation::HardShrinkConfig
• nn::activation::HardSigmoid
• nn::activation::HardSigmoidConfig
• nn::activation::HardSwish
• nn::activation::Hardtanh
• nn::activation::HardtanhConfig
• nn::activation::LeakyRelu
• nn::activation::LeakyReluConfig
• nn::activation::LogSigmoid
• nn::activation::Mish
• nn::activation::PRelu
• nn::activation::PReluConfig
• nn::activation::RRelu
• nn::activation::RReluConfig
• nn::activation::Relu
• nn::activation::Relu6
• nn::activation::Selu
• nn::activation::Shrink
• nn::activation::ShrinkConfig
• nn::activation::SiLU
• nn::activation::Sigmoid
• nn::activation::SoftShrink
• nn::activation::SoftShrinkConfig
• nn::activation::Softplus
• nn::activation::SoftplusConfig
• nn::activation::Softsign
• nn::activation::SwiGlu
• nn::activation::SwiGluConfig
• nn::activation::Tanh
• nn::activation::Tanhshrink
• nn::activation::Threshold
• nn::activation::ThresholdConfig
• nn::activation::ThresholdedRelu
• nn::activation::ThresholdedReluConfig
• nn::loss::BinaryCrossEntropyLoss
• nn::loss::BinaryCrossEntropyLossConfig
• nn::loss::CTCLoss
• nn::loss::CTCLossConfig
• nn::loss::CosineEmbeddingLoss
• nn::loss::CosineEmbeddingLossConfig
• nn::loss::CrossEntropyLoss
• nn::loss::CrossEntropyLossConfig
• nn::loss::GaussianNLLLoss
• nn::loss::GaussianNLLLossConfig
• nn::loss::HingeEmbeddingLoss
• nn::loss::HingeEmbeddingLossConfig
• nn::loss::HuberLoss
• nn::loss::HuberLossConfig
• nn::loss::KLDivLoss
• nn::loss::KLDivLossConfig
• nn::loss::LpLoss
• nn::loss::LpLossConfig
• nn::loss::MarginRankingLoss
• nn::loss::MarginRankingLossConfig
• nn::loss::MseLoss
• nn::loss::MultiMarginLoss
• nn::loss::MultiMarginLossConfig
• nn::loss::PoissonNllLoss
• nn::loss::PoissonNllLossConfig
• nn::loss::RNNTLoss
• nn::loss::RNNTLossConfig
• nn::loss::SmoothL1Loss
• nn::loss::SmoothL1LossConfig
• nn::loss::SoftMarginLoss
• nn::loss::TripletMarginLoss
• nn::loss::TripletMarginLossConfig
• nn::modules::BatchNorm
• nn::modules::BatchNormConfig
• nn::modules::BiGru
• nn::modules::BiGruConfig
• nn::modules::BiLstm
• nn::modules::BiLstmConfig
• nn::modules::BiRnn
• nn::modules::BiRnnConfig
• nn::modules::CosineSimilarity
• nn::modules::CosineSimilarityConfig
• nn::modules::Dropout
• nn::modules::DropoutConfig
• nn::modules::Embedding
• nn::modules::EmbeddingConfig
• nn::modules::Fold4d
• nn::modules::Fold4dConfig
• nn::modules::GateController
• nn::modules::GaussianNoise
• nn::modules::GaussianNoiseConfig
• nn::modules::GroupNorm
• nn::modules::GroupNormConfig
• nn::modules::Gru
• nn::modules::GruConfig
• nn::modules::Identity
• nn::modules::InstanceNorm
• nn::modules::InstanceNormConfig
• nn::modules::LayerNorm
• nn::modules::LayerNormConfig
• nn::modules::Linear
• nn::modules::LinearConfig
• nn::modules::LocalResponseNorm
• nn::modules::LocalResponseNormConfig
• nn::modules::Lstm
• nn::modules::LstmConfig
• nn::modules::LstmState
• nn::modules::PairwiseDistance
• nn::modules::PairwiseDistanceConfig
• nn::modules::PixelShuffle
• nn::modules::PixelShuffleConfig
• nn::modules::PixelUnshuffle
• nn::modules::PixelUnshuffleConfig
• nn::modules::PositionalEncoding
• nn::modules::PositionalEncodingConfig
• nn::modules::RmsNorm
• nn::modules::RmsNormConfig
• nn::modules::Rnn
• nn::modules::RnnConfig
• nn::modules::RnnState
• nn::modules::RotaryEncoding
• nn::modules::RotaryEncodingConfig
• nn::modules::Unfold4d
• nn::modules::Unfold4dConfig
• nn::modules::attention::CrossAttention
• nn::modules::attention::CrossAttentionCache
• nn::modules::attention::CrossAttentionConfig
• nn::modules::attention::GeneratePaddingMask
• nn::modules::attention::MhaCache
• nn::modules::attention::MhaInput
• nn::modules::attention::MhaOutput
• nn::modules::attention::MultiHeadAttention
• nn::modules::attention::MultiHeadAttentionConfig
• nn::modules::basic::BiRnn
• nn::modules::basic::BiRnnConfig
• nn::modules::basic::Rnn
• nn::modules::basic::RnnConfig
• nn::modules::basic::RnnState
• nn::modules::cache::TensorCache
• nn::modules::conv::Conv1d
• nn::modules::conv::Conv1dConfig
• nn::modules::conv::Conv2d
• nn::modules::conv::Conv2dConfig
• nn::modules::conv::Conv3d
• nn::modules::conv::Conv3dConfig
• nn::modules::conv::ConvTranspose1d
• nn::modules::conv::ConvTranspose1dConfig
• nn::modules::conv::ConvTranspose2d
• nn::modules::conv::ConvTranspose2dConfig
• nn::modules::conv::ConvTranspose3d
• nn::modules::conv::ConvTranspose3dConfig
• nn::modules::conv::DeformConv2d
• nn::modules::conv::DeformConv2dConfig
• nn::modules::gru::BiGru
• nn::modules::gru::BiGruConfig
• nn::modules::gru::Gru
• nn::modules::gru::GruConfig
• nn::modules::interpolate::Interpolate1d
• nn::modules::interpolate::Interpolate1dConfig
• nn::modules::interpolate::Interpolate2d
• nn::modules::interpolate::Interpolate2dConfig
• nn::modules::lstm::BiLstm
• nn::modules::lstm::BiLstmConfig
• nn::modules::lstm::Lstm
• nn::modules::lstm::LstmConfig
• nn::modules::lstm::LstmState
• nn::modules::norm::BatchNorm
• nn::modules::norm::BatchNormConfig
• nn::modules::norm::GroupNorm
• nn::modules::norm::GroupNormConfig
• nn::modules::norm::InstanceNorm
• nn::modules::norm::InstanceNormConfig
• nn::modules::norm::LayerNorm
• nn::modules::norm::LayerNormConfig
• nn::modules::norm::LocalResponseNorm
• nn::modules::norm::LocalResponseNormConfig
• nn::modules::norm::RmsNorm
• nn::modules::norm::RmsNormConfig
• nn::modules::pool::AdaptiveAvgPool1d
• nn::modules::pool::AdaptiveAvgPool1dConfig
• nn::modules::pool::AdaptiveAvgPool2d
• nn::modules::pool::AdaptiveAvgPool2dConfig
• nn::modules::pool::AdaptiveAvgPool3d
• nn::modules::pool::AdaptiveAvgPool3dConfig
• nn::modules::pool::AvgPool1d
• nn::modules::pool::AvgPool1dConfig
• nn::modules::pool::AvgPool2d
• nn::modules::pool::AvgPool2dConfig
• nn::modules::pool::MaxPool1d
• nn::modules::pool::MaxPool1dConfig
• nn::modules::pool::MaxPool2d
• nn::modules::pool::MaxPool2dConfig
• nn::modules::transformer::PositionWiseFeedForward
• nn::modules::transformer::PositionWiseFeedForwardConfig
• nn::modules::transformer::TransformerDecoder
• nn::modules::transformer::TransformerDecoderAutoregressiveCache
• nn::modules::transformer::TransformerDecoderConfig
• nn::modules::transformer::TransformerDecoderInput
• nn::modules::transformer::TransformerDecoderLayer
• nn::modules::transformer::TransformerDecoderLayerAutoregressiveCache
• nn::modules::transformer::TransformerEncoder
• nn::modules::transformer::TransformerEncoderAutoregressiveCache
• nn::modules::transformer::TransformerEncoderConfig
• nn::modules::transformer::TransformerEncoderInput
• nn::modules::transformer::TransformerEncoderLayer
• nn::modules::transformer::TransformerEncoderLayerAutoregressiveCache
• optim::AdaGrad
• optim::AdaGradConfig
• optim::AdaGradState
• optim::Adafactor
• optim::AdafactorConfig
• optim::AdafactorState
• optim::Adam
• optim::AdamConfig
• optim::AdamState
• optim::AdamW
• optim::AdamWConfig
• optim::AdamWState
• optim::Adan
• optim::AdanConfig
• optim::AdanState
• optim::AdaptiveMomentumState
• optim::AdaptiveNesterovMomentumState
• optim::CenteredState
• optim::DynState
• optim::GradientsAccumulator
• optim::GradientsParams
• optim::LBFGS
• optim::LBFGSConfig
• optim::LBFGSState
• optim::Lamb
• optim::LambConfig
• optim::LambState
• optim::Lion
• optim::LionConfig
• optim::LionState
• optim::LrDecayState
• optim::ModuleOptimizer
• optim::MultiGradientsParams
• optim::Muon
• optim::MuonConfig
• optim::MuonState
• optim::OptimizerRecord
• optim::RmsProp
• optim::RmsPropConfig
• optim::RmsPropMomentum
• optim::RmsPropMomentumState
• optim::RmsPropState
• optim::Sgd
• optim::SgdConfig
• optim::SgdState
• optim::SquareAvgState
• optim::StateSink
• optim::StateSource
• optim::decay::WeightDecay
• optim::decay::WeightDecayConfig
• optim::decay::WeightDecayState
• optim::lr_scheduler::DynLrScheduler
• optim::lr_scheduler::LrSchedulerRecord
• optim::lr_scheduler::composed::ComposedLrScheduler
• optim::lr_scheduler::composed::ComposedLrSchedulerConfig
• optim::lr_scheduler::constant::ConstantLr
• optim::lr_scheduler::cosine::CosineAnnealingLrScheduler
• optim::lr_scheduler::cosine::CosineAnnealingLrSchedulerConfig
• optim::lr_scheduler::cosine::CosineAnnealingLrSchedulerState
• optim::lr_scheduler::exponential::ExponentialLrScheduler
• optim::lr_scheduler::exponential::ExponentialLrSchedulerConfig
• optim::lr_scheduler::exponential::ExponentialLrSchedulerState
• optim::lr_scheduler::linear::LinearLrScheduler
• optim::lr_scheduler::linear::LinearLrSchedulerConfig
• optim::lr_scheduler::linear::LinearLrSchedulerState
• optim::lr_scheduler::module_lr_scheduler::ModuleLearningRate
• optim::lr_scheduler::module_lr_scheduler::ModuleLrScheduler
• optim::lr_scheduler::module_lr_scheduler::ModuleLrSchedulerConfig
• optim::lr_scheduler::noam::NoamLrScheduler
• optim::lr_scheduler::noam::NoamLrSchedulerConfig
• optim::lr_scheduler::noam::NoamLrSchedulerState
• optim::lr_scheduler::sequential::SequentialLrScheduler
• optim::lr_scheduler::sequential::SequentialLrSchedulerConfig
• optim::lr_scheduler::step::StepLrScheduler
• optim::lr_scheduler::step::StepLrSchedulerConfig
• optim::lr_scheduler::step::StepLrSchedulerState
• optim::module_optimizer::ModuleOptimizer
• optim::momentum::Momentum
• optim::momentum::MomentumConfig
• optim::momentum::MomentumState
• optim::record::OptimizerRecord
• prelude::Bool
• prelude::Device
• prelude::Float
• prelude::Int
• prelude::Shape
• prelude::Tensor
• prelude::TensorData
• rl::ActionContext
• rl::AsyncPolicy
• rl::RLTrainOutput
• rl::StepResult
• rl::Transition
• rl::TransitionBatch
• rl::TransitionBuffer
• rl::environment::StepResult
• rl::policy::ActionContext
• rl::policy::AsyncPolicy
• rl::policy::RLTrainOutput
• rl::transition_buffer::Transition
• rl::transition_buffer::TransitionBatch
• rl::transition_buffer::TransitionBuffer
• serde::de::IgnoredAny
• serde::de::value::BoolDeserializer
• serde::de::value::BorrowedBytesDeserializer
• serde::de::value::BorrowedStrDeserializer
• serde::de::value::BytesDeserializer
• serde::de::value::CharDeserializer
• serde::de::value::CowStrDeserializer
• serde::de::value::EnumAccessDeserializer
• serde::de::value::Error
• serde::de::value::F32Deserializer
• serde::de::value::F64Deserializer
• serde::de::value::I128Deserializer
• serde::de::value::I16Deserializer
• serde::de::value::I32Deserializer
• serde::de::value::I64Deserializer
• serde::de::value::I8Deserializer
• serde::de::value::IsizeDeserializer
• serde::de::value::MapAccessDeserializer
• serde::de::value::MapDeserializer
• serde::de::value::SeqAccessDeserializer
• serde::de::value::SeqDeserializer
• serde::de::value::StrDeserializer
• serde::de::value::StringDeserializer
• serde::de::value::U128Deserializer
• serde::de::value::U16Deserializer
• serde::de::value::U32Deserializer
• serde::de::value::U64Deserializer
• serde::de::value::U8Deserializer
• serde::de::value::UnitDeserializer
• serde::de::value::UsizeDeserializer
• serde::ser::Impossible
• signal::StftOptions
• store::ModuleRecord
• tensor::BackTrace
• tensor::Bool
• tensor::Bytes
• tensor::Device
• tensor::DeviceConfig
• tensor::DeviceFilter
• tensor::DeviceId
• tensor::DeviceSettings
• tensor::Devices
• tensor::DimIter
• tensor::DispatchTensor
• tensor::DistributionSampler
• tensor::EinsumOperand
• tensor::Float
• tensor::Gradients
• tensor::Graph
• tensor::IndexWrap
• tensor::Int
• tensor::MemoryPoolUsage
• tensor::PrintOptions
• tensor::ProfileDuration
• tensor::ProfileOptions
• tensor::ProfileTicks
• tensor::Shape
• tensor::Slice
• tensor::SliceIter
• tensor::SlicedPoolReport
• tensor::SmallVec
• tensor::StreamId
• tensor::Tensor
• tensor::TensorCreationOptions
• tensor::TensorData
• tensor::Tolerance
• tensor::Transaction
• tensor::bf16
• tensor::container::TensorContainer
• tensor::distributed::CollectiveTensor
• tensor::distributed::DistributedConfig
• tensor::distributed::DistributedContext
• tensor::f16
• tensor::flex32
• tensor::grid::GridOptions
• tensor::kind::Bool
• tensor::kind::BridgeTensor
• tensor::kind::Float
• tensor::kind::Int
• tensor::module::BatchNormTrainOutput
• tensor::ops::AttentionModuleOptions
• tensor::ops::AvgPoolOptions
• tensor::ops::ConvOptions
• tensor::ops::ConvTransposeOptions
• tensor::ops::DeformConvOptions
• tensor::ops::GridSampleOptions
• tensor::ops::InterpolateOptions
• tensor::ops::MaxPoolOptions
• tensor::ops::PaddedConvOptions
• tensor::ops::UnfoldOptions
• tensor::quantization::BlockLayout
• tensor::quantization::BlockScale
• tensor::quantization::BlockSize
• tensor::quantization::CalibrationRange
• tensor::quantization::DecodedScales
• tensor::quantization::QParamTensor
• tensor::quantization::QParams
• tensor::quantization::QuantConfig
• tensor::quantization::QuantScheme
• tensor::quantization::QuantizedBytes
• tensor::signal::StftOptions
• train::AgentEnvAsyncLoop
• train::AgentEnvBaseLoop
• train::AsyncAgentEnvLoopConfig
• train::AsyncProcessorEvaluation
• train::AsyncProcessorTraining
• train::ClassificationOutput
• train::EpisodeSummary
• train::EvaluationItem
• train::EvaluationResult
• train::Evaluator
• train::EvaluatorBuilder
• train::EvaluatorComponentTypesMarker
• train::EventProcessorError
• train::FileApplicationLoggerInstaller
• train::Interrupter
• train::Interruption
• train::Learner
• train::LearnerSummary
• train::LearnerSummaryConfig
• train::LearningCheckpointer
• train::LearningResult
• train::MetricEarlyStoppingStrategy
• train::MetricEntry
• train::MetricError
• train::MetricSummary
• train::ModuleSharder
• train::MultiAgentEnvLoop
• train::MultiLabelClassificationOutput
• train::OffPolicyConfig
• train::OffPolicyStrategy
• train::RLCheckpointer
• train::RLComponents
• train::RLComponentsMarker
• train::RLEventProcessor
• train::RLResult
• train::RLTraining
• train::RegressionOutput
• train::SequenceOutput
• train::SummaryMetrics
• train::SupervisedTraining
• train::TimeStep
• train::TrainOutput
• train::TrainingComponents
• train::TrainingItem
• train::Trajectory
• train::checkpoint::AsyncCheckpointer
• train::checkpoint::ComposedCheckpointingStrategy
• train::checkpoint::ComposedCheckpointingStrategyBuilder
• train::checkpoint::FileCheckpointer
• train::checkpoint::KeepLastNCheckpoints
• train::checkpoint::MetricCheckpointingStrategy
• train::logger::AsyncLogger
• train::logger::FileLogger
• train::logger::FileMetricLogger
• train::logger::InMemoryLogger
• train::logger::InMemoryMetricLogger
• train::logger::ProgressSnapshot
• train::metric::AccuracyInput
• train::metric::AccuracyMetric
• train::metric::AucPrMetric
• train::metric::AurocMetric
• train::metric::BleuInput
• train::metric::BleuScore
• train::metric::CerInput
• train::metric::CharErrorRate
• train::metric::ConfusionStatsInput
• train::metric::CpuMemory
• train::metric::CpuTemperature
• train::metric::CpuUse
• train::metric::CudaMetric
• train::metric::CumulativeRewardInput
• train::metric::CumulativeRewardMetric
• train::metric::EpisodeLengthInput
• train::metric::EpisodeLengthMetric
• train::metric::ExplorationRateInput
• train::metric::ExplorationRateMetric
• train::metric::FBetaScoreMetric
• train::metric::HammingScore
• train::metric::HammingScoreInput
• train::metric::IterationSpeedMetric
• train::metric::LearningRateMetric
• train::metric::LossInput
• train::metric::LossMetric
• train::metric::MetricDefinition
• train::metric::MetricEntry
• train::metric::MetricId
• train::metric::MetricMetadata
• train::metric::NumericAttributes
• train::metric::PerplexityInput
• train::metric::PerplexityMetric
• train::metric::PrecisionMetric
• train::metric::RecallMetric
• train::metric::RougeLInput
• train::metric::RougeLScore
• train::metric::SerializedEntry
• train::metric::TopKAccuracyInput
• train::metric::TopKAccuracyMetric
• train::metric::WerInput
• train::metric::WordErrorRate
• train::metric::state::ConfusionStatsState
• train::metric::state::FormatOptions
• train::metric::state::NumericMetricState
• train::metric::state::PredictionAccumulatorState
• train::metric::store::EpochSummary
• train::metric::store::EventStoreClient
• train::metric::store::MetricsUpdate
• train::metric::store::NumericMetricUpdate
• train::renderer::CliMetricsRenderer
• train::renderer::EvaluationName
• train::renderer::tui::TuiMetricsRendererWrapper
• train::train::MultiDevicesTrainStep
• train::train::MultiTrainOutput
• train::train::WorkerFailure
• vision::BoxBlur
• vision::BoxBlurConfig
• vision::ConnectedStats
• vision::ConnectedStatsOptions
• vision::ConnectedStatsPrimitive
• vision::ConnectedStatsPrimitiveMetadata
• vision::MedianBlur
• vision::MedianBlurConfig
• vision::MorphOptions
• vision::MorphOptionsBuilder
• vision::NmsOptions
• vision::Point
• vision::Size
• vision::Transform2D
• vision::utils::TensorDisplayOptions

Enums

• backend::AllocationProperty
• backend::BackendTensor
• backend::BoolStore
• backend::DType
• backend::DTypeUsage
• backend::DataError
• backend::DeviceError
• backend::DeviceServiceStage
• backend::DispatchAutodiffContext
• backend::DispatchDevice
• backend::DispatchDeviceId
• backend::DispatchGraph
• backend::DispatchTensorKind
• backend::Distribution
• backend::DistributionSamplerKind
• backend::ExecutionError
• backend::ExpressionError
• backend::FloatDType
• backend::GradientCheckpointingStrategy
• backend::GraphUnsupported
• backend::IntDType
• backend::MetadataError
• backend::RouterDeviceType
• backend::Scalar
• backend::SplitPolicy
• backend::TensorPrimitive
• backend::TimingMethod
• backend::autodiff::checkpoint::strategy::CheckpointingError
• backend::autodiff::ops::OpsKind
• backend::backend::DTypeUsage
• backend::backend::DeviceError
• backend::backend::DeviceServiceStage
• backend::backend::ExecutionError
• backend::backend::GraphUnsupported
• backend::backend::RouterDeviceType
• backend::backend::TensorPrimitive
• backend::backend::TimingMethod
• backend::backend::distributed::ReduceOperation
• backend::backend::ops::GridSamplePaddingMode
• backend::backend::ops::InterpolateMode
• backend::backend::ops::PadMode
• backend::distribution::Distribution
• backend::distribution::DistributionSamplerKind
• backend::element::Scalar
• backend::fusion::ExecutionError
• backend::fusion::FuserStatus
• backend::fusion::OperationRan
• backend::fusion::custom::OperationIr
• backend::fusion::custom::ScalarIr
• backend::ir::ActivationOperationIr
• backend::ir::BaseOperationIr
• backend::ir::BoolOperationIr
• backend::ir::DistributedOperationIr
• backend::ir::FloatOperationIr
• backend::ir::GridSamplePaddingModeIr
• backend::ir::Handle
• backend::ir::HandleKind
• backend::ir::IntOperationIr
• backend::ir::InterpolateModeIr
• backend::ir::IrError
• backend::ir::ModuleOperationIr
• backend::ir::NumericOperationIr
• backend::ir::OperationIr
• backend::ir::PadModeIr
• backend::ir::ScalarIr
• backend::ir::TensorStatus
• backend::quantization::Calibration
• backend::quantization::QuantMode
• backend::quantization::QuantPropagation
• backend::quantization::QuantStore
• backend::quantization::QuantValue
• backend::quantization::ScaleDtype
• backend::shape::ExpressionError
• backend::shape::MetadataError
• backend::tensor::IndexingUpdateOp
• config::ConfigError
• data::dataloader::Message
• data::dataset::ImporterError
• data::dataset::RowError
• data::dataset::SqliteDatasetError
• data::dataset::audio::SpeechCommandClass
• data::dataset::source::huggingface::ImporterError
• data::dataset::transform::RngSource
• data::dataset::transform::SizeConfig
• data::dataset::vision::Annotation
• data::dataset::vision::ImageLoaderError
• data::dataset::vision::PixelData
• data::dataset::vision::PixelDepth
• grad_clipping::GradientClipping
• grad_clipping::GradientClippingConfig
• linalg::Norm
• lr_scheduler::LrSchedulerConfig
• lr_scheduler::composed::SchedulerReduction
• module::Initializer
• module::ParamGroupError
• nn::Initializer
• nn::LinearLayout
• nn::PaddingConfig1d
• nn::PaddingConfig2d
• nn::PaddingConfig3d
• nn::activation::Activation
• nn::activation::ActivationConfig
• nn::loss::Reduction
• nn::modules::LinearLayout
• nn::modules::attention::SeqLengthOption
• nn::modules::interpolate::InterpolateMode
• nn::modules::norm::Normalization
• nn::modules::norm::NormalizationConfig
• optim::AdjustLrFn
• optim::GradAdaptor
• optim::LineSearchFn
• optim::grad_clipping::GradientClipping
• optim::grad_clipping::GradientClippingConfig
• optim::lr_scheduler::LrSchedulerConfig
• optim::lr_scheduler::composed::SchedulerReduction
• optim::module_optimizer::GradAdaptor
• prelude::DeviceIndex
• prelude::DeviceKind
• serde::de::Unexpected
• store::DTypePolicy
• store::RecordError
• tensor::AllocationProperty
• tensor::BoolStore
• tensor::DType
• tensor::DataError
• tensor::DeviceError
• tensor::DeviceIndex
• tensor::DeviceKind
• tensor::DeviceType
• tensor::Distribution
• tensor::DistributionSamplerKind
• tensor::ExecutionError
• tensor::ExpressionError
• tensor::FloatDType
• tensor::GradientCheckpointingStrategy
• tensor::IndexingUpdateOp
• tensor::IntDType
• tensor::MetadataError
• tensor::PrimitiveConversionError
• tensor::Scalar
• tensor::TensorReadError
• tensor::TimingMethod
• tensor::distributed::ReduceOperation
• tensor::grid::GridIndexing
• tensor::grid::GridSparsity
• tensor::grid::IndexPos
• tensor::kind::Kind
• tensor::linalg::Norm
• tensor::ops::GridSamplePaddingMode
• tensor::ops::IndexingUpdateOp
• tensor::ops::InterpolateMode
• tensor::ops::PadMode
• tensor::quantization::Calibration
• tensor::quantization::QuantAcc
• tensor::quantization::QuantMode
• tensor::quantization::QuantPropagation
• tensor::quantization::QuantStore
• tensor::quantization::QuantValue
• tensor::quantization::ScaleDtype
• train::AgentEvaluationEvent
• train::EvaluatorEvent
• train::EventProcessorFailure
• train::ExecutionStrategy
• train::LearnerEvent
• train::MultiDeviceOptim
• train::RLEvent
• train::RLStrategies
• train::StoppingCondition
• train::TrainingError
• train::TrainingStrategy
• train::checkpoint::CheckpointerError
• train::checkpoint::CheckpointingAction
• train::metric::BleuSmoothing
• train::metric::ClassReduction
• train::metric::MetricAttributes
• train::metric::NumericEntry
• train::metric::store::Aggregate
• train::metric::store::Direction
• train::metric::store::Event
• train::metric::store::Split
• train::renderer::MetricState
• train::train::MultiDeviceStepError
• vision::BorderType
• vision::Connectivity
• vision::KernelShape
• vision::backends::KernelShape
• vision::utils::BatchDisplayOpts
• vision::utils::ColorDisplayOpts
• vision::utils::ImageDimOrder

Traits

• backend::AsIndex
• backend::AsSize
• backend::AutodiffBackend
• backend::Backend
• backend::BackendTypes
• backend::Device
• backend::DeviceOps
• backend::DeviceService
• backend::DispatchKindConversion
• backend::Element
• backend::ElementAdd
• backend::ElementConversion
• backend::ElementEq
• backend::ElementOrdered
• backend::ElementRandom
• backend::ExtensionType
• backend::SliceArg
• backend::SliceOps
• backend::TensorMetadata
• backend::autodiff::checkpoint::retro_forward::RetroForward
• backend::autodiff::checkpoint::strategy::CheckpointStrategy
• backend::autodiff::grads::DistributedRegistration
• backend::autodiff::ops::Backward
• backend::autodiff::ops::DifferentiableTransfer
• backend::backend::AutodiffBackend
• backend::backend::Backend
• backend::backend::BackendTypes
• backend::backend::Device
• backend::backend::DeviceOps
• backend::backend::DeviceService
• backend::backend::Element
• backend::backend::TensorMetadata
• backend::backend::distributed::DistributedOps
• backend::backend::ops::ActivationOps
• backend::backend::ops::BoolTensorOps
• backend::backend::ops::FloatTensorOps
• backend::backend::ops::IntTensorOps
• backend::backend::ops::ModuleOps
• backend::backend::ops::QTensorOps
• backend::backend::ops::TransactionOps
• backend::element::Element
• backend::element::ElementAdd
• backend::element::ElementComparison
• backend::element::ElementConversion
• backend::element::ElementEq
• backend::element::ElementLimits
• backend::element::ElementOrdered
• backend::element::ElementRandom
• backend::element::cast::ToElement
• backend::fusion::FusionBackend
• backend::fusion::FusionRuntime
• backend::fusion::NumOperations
• backend::fusion::OperationFuser
• backend::fusion::Optimization
• backend::fusion::custom::Operation
• backend::fusion::custom::OperationOutput
• backend::fusion::observer::FusionObserver
• backend::fusion::stream::Operation
• backend::indexing::AsIndex
• backend::indexing::AsSize
• backend::ir::BackendIr
• backend::ir::IrVisitorMut
• backend::ir::OperationOutput
• backend::shape::SliceOps
• backend::slice::SliceArg
• config::Config
• data::dataloader::BatchStrategy
• data::dataloader::DataLoader
• data::dataloader::DataLoaderIterator
• data::dataloader::Dataset
• data::dataloader::batcher::Batcher
• data::dataset::Dataset
• data::dataset::transform::Mapper
• data::dataset::transform::Window
• data::dataset::transform::Windows
• linalg::LinalgOps
• lr_scheduler::LrScheduler
• lr_scheduler::LrSchedulerClone
• module::Module
• module::ModuleDisplay
• module::ModuleDisplayDefault
• module::ModuleMapper
• module::ModuleVisitor
• module::Parameter
• module::ParameterValue
• module::Reparameterization
• module::Reparameterizer
• nn::OptionalInitialLstmState
• nn::modules::OptionalInitialLstmState
• nn::modules::lstm::OptionalInitialLstmState
• optim::DynOptimizer
• optim::Optimizer
• optim::RecordState
• optim::lr_scheduler::LrScheduler
• optim::lr_scheduler::LrSchedulerClone
• prelude::Config
• prelude::ElementConversion
• prelude::Module
• prelude::SliceArg
• prelude::ToElement
• rl::Batchable
• rl::Environment
• rl::EnvironmentInit
• rl::Policy
• rl::PolicyLearner
• rl::PolicyState
• rl::SliceAccess
• rl::ToAction
• rl::ToObservation
• rl::environment::Environment
• rl::environment::EnvironmentInit
• rl::policy::Batchable
• rl::policy::Policy
• rl::policy::PolicyLearner
• rl::policy::PolicyState
• rl::policy::ToAction
• rl::policy::ToObservation
• rl::transition_buffer::SliceAccess
• serde::Deserialize
• serde::Deserializer
• serde::Serialize
• serde::Serializer
• serde::de::Deserialize
• serde::de::DeserializeOwned
• serde::de::DeserializeSeed
• serde::de::Deserializer
• serde::de::EnumAccess
• serde::de::Error
• serde::de::Expected
• serde::de::IntoDeserializer
• serde::de::MapAccess
• serde::de::SeqAccess
• serde::de::VariantAccess
• serde::de::Visitor
• serde::ser::Error
• serde::ser::Serialize
• serde::ser::SerializeMap
• serde::ser::SerializeSeq
• serde::ser::SerializeStruct
• serde::ser::SerializeStructVariant
• serde::ser::SerializeTuple
• serde::ser::SerializeTupleStruct
• serde::ser::SerializeTupleVariant
• serde::ser::Serializer
• signal::SignalOps
• tensor::AsIndex
• tensor::AsSize
• tensor::BackendPrimitive
• tensor::BroadcastArgs
• tensor::Cast
• tensor::Element
• tensor::ElementAdd
• tensor::ElementComparison
• tensor::ElementConversion
• tensor::ElementEq
• tensor::ElementLimits
• tensor::ElementOrdered
• tensor::ElementRandom
• tensor::IntoPadding
• tensor::MovedimArgs
• tensor::ReshapeArgs
• tensor::SliceArg
• tensor::SliceOps
• tensor::cast::ToElement
• tensor::kind::Autodiff
• tensor::kind::Basic
• tensor::kind::FloatMath
• tensor::kind::Numeric
• tensor::kind::Ordered
• tensor::kind::TensorKind
• tensor::kind::Transaction
• tensor::linalg::LinalgOps
• tensor::signal::SignalOps
• train::AgentEnvLoop
• train::AgentMetricRegistration
• train::AgentTextMetricRegistration
• train::ApplicationLoggerInstaller
• train::CloneEarlyStoppingStrategy
• train::EarlyStoppingStrategy
• train::EpisodeMetricRegistration
• train::EpisodeTextMetricRegistration
• train::EvalMetricRegistration
• train::EvalTextMetricRegistration
• train::EvaluatorComponentTypes
• train::EventProcessorEvaluation
• train::EventProcessorTraining
• train::InferenceStep
• train::ItemLazy
• train::LearnerModel
• train::MetricRegistration
• train::RLComponentsTypes
• train::RLStrategy
• train::SupervisedLearningStrategy
• train::TextMetricRegistration
• train::TrainMetricRegistration
• train::TrainStep
• train::TrainTextMetricRegistration
• train::checkpoint::Checkpoint
• train::checkpoint::Checkpointer
• train::checkpoint::CheckpointingStrategy
• train::logger::EvaluationProgressLogger
• train::logger::Logger
• train::logger::MetricLogger
• train::logger::TrainingProgressLogger
• train::metric::Adaptor
• train::metric::ItemLazy
• train::metric::Metric
• train::metric::Numeric
• train::metric::store::EventStore
• train::renderer::MetricsRenderer
• train::renderer::MetricsRendererEvaluation
• train::renderer::MetricsRendererTraining
• vision::BoolVisionOps
• vision::ColorConversion
• vision::ConnectedComponents
• vision::FloatVisionOps
• vision::IntVisionOps
• vision::Morphology
• vision::Nms
• vision::VisionBackend

Macros

• backend::autodiff::retro_binary
• backend::autodiff::retro_unary
• backend::autodiff::retro_unary_scalar
• backend::dequant_op_flow
• backend::dequant_op_quant
• backend::doc_tensor
• backend::fusion::binary_float_cmp_ops
• backend::fusion::binary_float_ops
• backend::fusion::binary_int_cmp_ops
• backend::fusion::binary_int_ops
• backend::fusion::memory_checks
• backend::fusion::reduce_ops
• backend::fusion::scalar_float2int_ops
• backend::fusion::scalar_float_cmp_ops
• backend::fusion::scalar_float_ops
• backend::fusion::scalar_int_cmp_ops
• backend::fusion::scalar_int_ops
• backend::fusion::unary_float_ops
• backend::fusion::unary_int_ops
• backend::s
• backend::shape
• backend::shape::shape
• backend::slice::s
• empty
• prelude::assert_shape
• prelude::debug_assert_shape
• prelude::s
• serde::forward_to_deserialize_any
• tensor::assert_shape
• tensor::debug_assert_shape
• tensor::einsum
• tensor::s
• tensor::shape
• vision::dispatch_bool_dtype
• vision::dispatch_int_dtype

Attribute Macros

• backend::backend_extension

Derive Macros

• backend::ExtensionType
• config::Config
• module::Module
• optim::RecordState
• prelude::Config
• prelude::Module
• serde::Deserialize
• serde::Serialize

Functions

• backend::autodiff::ops::binary
• backend::autodiff::ops::broadcast_shape
• backend::autodiff::ops::unary
• backend::backend::distributed::close_distributed_sync_server
• backend::backend::distributed::get_distributed_sync_client
• backend::backend::distributed::start_distributed_sync_server
• backend::backend::get_device_settings
• backend::backend::get_or_init_device_settings
• backend::backend::ops::attention::attention_fallback
• backend::backend::ops::conv::calculate_conv_output_shape
• backend::backend::ops::conv::calculate_conv_output_size
• backend::backend::ops::conv::calculate_conv_output_size_asymmetric
• backend::backend::ops::conv::calculate_conv_output_sizes
• backend::backend::ops::conv::calculate_conv_padding
• backend::backend::ops::conv::calculate_conv_transpose_output_shape
• backend::backend::ops::conv::calculate_conv_transpose_output_size
• backend::backend::ops::conv::calculate_padding_out
• backend::backend::ops::conv::calculate_pool_output_shape
• backend::backend::ops::conv::calculate_pool_output_size
• backend::backend::ops::conv::pad_asymmetric_conv_input
• backend::backend::ops::ctc::ctc_grad_from_alpha_beta_default
• backend::backend::ops::ctc::ctc_loss_default
• backend::backend::ops::grid_sample::float_grid_sample_2d_ref
• backend::backend::ops::unfold::calculate_unfold_shape
• backend::backend::ops::unfold::calculate_unfold_windows
• backend::backend::profile_system_time
• backend::backend::profile_with_tokens
• backend::backend::set_default_dtypes
• backend::calculate_matmul_output
• backend::fusion::get_client
• backend::get_device_settings
• backend::get_or_init_device_settings
• backend::indexing::ravel_index
• backend::indexing::wrap_index
• backend::profile_system_time
• backend::profile_with_tokens
• backend::quantization::compute_q_params
• backend::quantization::compute_range
• backend::quantization::global_scale_dtype
• backend::quantization::params_shape
• backend::quantization::quantizable
• backend::quantization::scale_to_dtype
• backend::ravel_index
• backend::read_sync
• backend::set_default_dtypes
• backend::shape::calculate_matmul_output
• backend::try_read_sync
• backend::wrap_index
• config::config_to_json
• data::dataloader::split::split_dataloader
• data::dataset::network::downloader::download_file_as_bytes
• data::dataset::transform::iota
• data::dataset::transform::shuffled_indices
• linalg::cosine_similarity
• linalg::det
• linalg::diag
• linalg::l0_norm
• linalg::l0_norm_dims
• linalg::l1_norm
• linalg::l1_norm_dims
• linalg::l2_norm
• linalg::l2_norm_dims
• linalg::lp_norm
• linalg::lp_norm_dims
• linalg::lu
• linalg::matvec
• linalg::max_abs_norm
• linalg::max_abs_norm_dims
• linalg::min_abs_norm
• linalg::min_abs_norm_dims
• linalg::outer
• linalg::outer_dim
• linalg::qr
• linalg::svd
• linalg::trace
• linalg::vector_norm
• linalg::vector_norm_dims
• linalg::vector_normalize
• module::extract_type_name
• module::list_param_ids
• nn::generate_sinusoids
• nn::modules::attention::generate_autoregressive_mask
• nn::modules::attention::generate_padding_mask
• nn::modules::generate_sinusoids
• optim::join_index
• optim::join_path
• runtime_config
• signal::blackman_window
• signal::cfft
• signal::hamming_window
• signal::hann_window
• signal::irfft
• signal::istft
• signal::rfft
• signal::stft
• tensor::activation::celu
• tensor::activation::elu
• tensor::activation::gelu
• tensor::activation::gelu_approximate
• tensor::activation::glu
• tensor::activation::hard_shrink
• tensor::activation::hard_sigmoid
• tensor::activation::hard_swish
• tensor::activation::hardtanh
• tensor::activation::leaky_relu
• tensor::activation::log_sigmoid
• tensor::activation::log_softmax
• tensor::activation::mish
• tensor::activation::prelu
• tensor::activation::quiet_softmax
• tensor::activation::relu
• tensor::activation::relu6
• tensor::activation::selu
• tensor::activation::shrink
• tensor::activation::sigmoid
• tensor::activation::silu
• tensor::activation::soft_shrink
• tensor::activation::softmax
• tensor::activation::softmin
• tensor::activation::softplus
• tensor::activation::softplus_with_threshold
• tensor::activation::softsign
• tensor::activation::tanh
• tensor::activation::tanhshrink
• tensor::activation::threshold
• tensor::activation::thresholded_relu
• tensor::calculate_matmul_output
• tensor::capture
• tensor::cartesian_grid
• tensor::check_closeness
• tensor::distributed::all_reduce
• tensor::grid::affine_grid_2d
• tensor::grid::meshgrid
• tensor::grid::meshgrid_stack
• tensor::linalg::cosine_similarity
• tensor::linalg::det
• tensor::linalg::diag
• tensor::linalg::l0_norm
• tensor::linalg::l0_norm_dims
• tensor::linalg::l1_norm
• tensor::linalg::l1_norm_dims
• tensor::linalg::l2_norm
• tensor::linalg::l2_norm_dims
• tensor::linalg::lp_norm
• tensor::linalg::lp_norm_dims
• tensor::linalg::lu
• tensor::linalg::matvec
• tensor::linalg::max_abs_norm
• tensor::linalg::max_abs_norm_dims
• tensor::linalg::min_abs_norm
• tensor::linalg::min_abs_norm_dims
• tensor::linalg::outer
• tensor::linalg::outer_dim
• tensor::linalg::qr
• tensor::linalg::svd
• tensor::linalg::trace
• tensor::linalg::vector_norm
• tensor::linalg::vector_norm_dims
• tensor::linalg::vector_normalize
• tensor::loss::cross_entropy_with_logits
• tensor::module::adaptive_avg_pool1d
• tensor::module::adaptive_avg_pool2d
• tensor::module::adaptive_avg_pool3d
• tensor::module::attention
• tensor::module::attention_fallback
• tensor::module::avg_pool1d
• tensor::module::avg_pool2d
• tensor::module::avg_pool2d_backward
• tensor::module::batch_norm
• tensor::module::batch_norm_train
• tensor::module::conv1d
• tensor::module::conv2d
• tensor::module::conv2d_weight_backward
• tensor::module::conv3d
• tensor::module::conv_transpose1d
• tensor::module::conv_transpose2d
• tensor::module::conv_transpose3d
• tensor::module::ctc_loss
• tensor::module::deform_conv2d
• tensor::module::embedding
• tensor::module::fold4d
• tensor::module::interpolate
• tensor::module::layer_norm
• tensor::module::linear
• tensor::module::max_pool1d
• tensor::module::max_pool1d_with_indices
• tensor::module::max_pool2d
• tensor::module::max_pool2d_with_indices
• tensor::module::max_pool2d_with_indices_backward
• tensor::module::unfold4d
• tensor::quantization::compute_q_params
• tensor::quantization::compute_range
• tensor::quantization::global_scale_dtype
• tensor::quantization::global_scale_size
• tensor::quantization::pack_i8s_to_u32s
• tensor::quantization::params_shape
• tensor::quantization::quantizable
• tensor::quantization::quantized_data_len
• tensor::quantization::scale_size
• tensor::quantization::scale_to_dtype
• tensor::ravel_index
• tensor::read_sync
• tensor::set_print_options
• tensor::signal::blackman_window
• tensor::signal::cfft
• tensor::signal::hamming_window
• tensor::signal::hann_window
• tensor::signal::irfft
• tensor::signal::istft
• tensor::signal::rfft
• tensor::signal::stft
• tensor::try_read_sync
• tensor::wrap_index
• train::metric::format_float
• train::renderer::default_renderer
• vision::backends::create_structuring_element
• vision::create_structuring_element
• vision::filter2d
• vision::median
• vision::utils::save_tensor_as_image

Type Aliases

• backend::BackendGraph
• backend::BoolDType
• backend::DTypeUsageSet
• backend::ServerUtilitiesHandle
• backend::autodiff::grads::GradID
• backend::backend::BackendGraph
• backend::backend::DTypeUsageSet
• backend::backend::ServerUtilitiesHandle
• backend::backend::distributed::DistributedParamId
• backend::fusion::Client
• backend::fusion::FusionDevice
• backend::fusion::FusionHandle
• backend::tensor::BoolTensor
• backend::tensor::Device
• backend::tensor::FloatTensor
• backend::tensor::IntTensor
• backend::tensor::QuantizedTensor
• data::dataset::Result
• module::Devices
• optim::LearningRate
• rl::LearnerTransitionBatch
• rl::policy::LearnerTransitionBatch
• tensor::BoolDType
• tensor::quantization::QuantizationParameters
• train::CustomLearningStrategy
• train::CustomRLStrategy
• train::RLAgentRecord
• train::RLEventProcessorType
• train::RLPolicyRecord
• train::SupervisedTrainingEventProcessor
• train::TrainLoader
• train::ValidLoader
• train::metric::MetricName

Constants

• backend::quantization::QPARAM_ALIGN
• tensor::DEFAULT_ATOL
• tensor::DEFAULT_RTOL
• tensor::activation::DEFAULT_SOFTPLUS_THRESHOLD
• tensor::quantization::QPARAM_ALIGN