List of all items

Structs

• Tensor
• data::dataloader::BatchDataLoader
• data::dataloader::DataLoaderBuilder
• data::dataloader::DatasetIterator
• data::dataloader::FixBatchStrategy
• data::dataloader::MultiThreadDataLoader
• data::dataloader::Progress
• 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
• module::Attribute
• module::ConstantRecord
• module::Content
• module::DisplaySettings
• module::Ignored
• module::Param
• module::ParamId
• module::ParamMapper
• module::Quantizer
• module::Reinitializer
• module::RunningState
• prelude::Bool
• prelude::Float
• prelude::Int
• prelude::Shape
• prelude::Tensor
• prelude::TensorData
• record::BinBytesRecorder
• record::BinFileRecorder
• record::BinGzFileRecorder
• record::BurnMetadata
• record::BurnRecord
• record::BurnRecordNoItem
• record::DoublePrecisionSettings
• record::FullPrecisionSettings
• record::HalfPrecisionSettings
• record::JsonGzFileRecorder
• record::NamedMpkBytesRecorder
• record::NamedMpkFileRecorder
• record::NamedMpkGzFileRecorder
• record::ParamSerde
• record::PrettyJsonFileRecorder
• tensor::Bool
• tensor::Bytes
• tensor::DimIter
• tensor::DistributionSampler
• tensor::Float
• tensor::IndexWrap
• tensor::Int
• tensor::PrintOptions
• tensor::Shape
• tensor::Slice
• tensor::SliceIter
• tensor::StreamId
• tensor::Tensor
• tensor::TensorCreationOptions
• tensor::TensorData
• tensor::Tolerance
• tensor::Transaction
• tensor::TransactionPrimitive
• tensor::backend::BackTrace
• tensor::backend::DeviceContext
• tensor::backend::DeviceGuard
• tensor::backend::DeviceId
• tensor::backend::DeviceStateGuard
• tensor::backend::ops::Conv2dBackward
• tensor::backend::ops::Conv3dBackward
• tensor::backend::ops::ConvOptions
• tensor::backend::ops::ConvTransposeOptions
• tensor::backend::ops::DeformConv2dBackward
• tensor::backend::ops::DeformConvOptions
• tensor::backend::ops::GridSampleOptions
• tensor::backend::ops::InterpolateBackward
• tensor::backend::ops::InterpolateOptions
• tensor::backend::ops::MaxPool1dBackward
• tensor::backend::ops::MaxPool1dWithIndices
• tensor::backend::ops::MaxPool2dBackward
• tensor::backend::ops::MaxPool2dWithIndices
• tensor::backend::ops::TransactionPrimitive
• tensor::backend::ops::TransactionPrimitiveData
• tensor::backend::ops::UnfoldOptions
• tensor::bf16
• tensor::container::TensorContainer
• tensor::f16
• tensor::grid::GridOptions
• tensor::ops::Conv2dBackward
• tensor::ops::Conv3dBackward
• tensor::ops::ConvOptions
• tensor::ops::ConvTransposeOptions
• tensor::ops::DeformConv2dBackward
• tensor::ops::DeformConvOptions
• tensor::ops::GridSampleOptions
• tensor::ops::InterpolateBackward
• tensor::ops::InterpolateOptions
• tensor::ops::MaxPool1dBackward
• tensor::ops::MaxPool1dWithIndices
• tensor::ops::MaxPool2dBackward
• tensor::ops::MaxPool2dWithIndices
• tensor::ops::TransactionPrimitive
• tensor::ops::TransactionPrimitiveData
• tensor::ops::UnfoldOptions
• tensor::quantization::BlockSize
• tensor::quantization::CalibrationRange
• tensor::quantization::QParamTensor
• tensor::quantization::QParams
• tensor::quantization::QuantScheme
• tensor::quantization::QuantizationParametersPrimitive
• tensor::quantization::QuantizedBytes

Enums

• config::ConfigError
• data::dataloader::Message
• data::dataset::ImporterError
• data::dataset::SqliteDatasetError
• data::dataset::audio::SpeechCommandClass
• data::dataset::source::huggingface::ImporterError
• data::dataset::transform::RngSource
• data::dataset::transform::SizeConfig
• module::Initializer
• record::RecorderError
• tensor::AllocationProperty
• tensor::BoundsError
• tensor::DType
• tensor::DataError
• tensor::Distribution
• tensor::DistributionSamplerKind
• tensor::ExpressionError
• tensor::FloatDType
• tensor::IndexingUpdateOp
• tensor::IntDType
• tensor::ShapeError
• tensor::TensorPrimitive
• tensor::backend::ExecutionError
• tensor::backend::TensorPrimitive
• tensor::backend::ops::GridSamplePaddingMode
• tensor::backend::ops::InterpolateMode
• tensor::backend::ops::PadMode
• tensor::grid::GridIndexing
• tensor::grid::GridSparsity
• tensor::grid::IndexPos
• tensor::linalg::Norm
• tensor::ops::GridSamplePaddingMode
• tensor::ops::InterpolateMode
• tensor::ops::PadMode
• tensor::quantization::Calibration
• tensor::quantization::QuantLevel
• tensor::quantization::QuantMode
• tensor::quantization::QuantParam
• tensor::quantization::QuantPropagation
• tensor::quantization::QuantStore
• tensor::quantization::QuantValue

Traits

• 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
• module::AutodiffModule
• module::Module
• module::ModuleDisplay
• module::ModuleDisplayDefault
• module::ModuleMapper
• module::ModuleVisitor
• module::Parameter
• prelude::Backend
• prelude::DeviceOps
• prelude::ElementConversion
• prelude::SliceArg
• prelude::ToElement
• record::BytesRecorder
• record::FileRecorder
• record::PrecisionSettings
• record::Record
• record::Recorder
• tensor::AsIndex
• tensor::AsSize
• tensor::BasicAutodiffOps
• tensor::BasicOps
• tensor::BroadcastArgs
• tensor::Element
• tensor::ElementComparison
• tensor::ElementConversion
• tensor::ElementLimits
• tensor::ElementRandom
• tensor::MovedimArgs
• tensor::Numeric
• tensor::ReshapeArgs
• tensor::SliceArg
• tensor::TensorKind
• tensor::TensorMetadata
• tensor::backend::AutodiffBackend
• tensor::backend::Backend
• tensor::backend::Device
• tensor::backend::DeviceOps
• tensor::backend::DeviceState
• tensor::backend::QTensorPrimitive
• tensor::backend::TensorMetadata
• tensor::backend::ops::ActivationOps
• tensor::backend::ops::BoolTensorOps
• tensor::backend::ops::FloatTensorOps
• tensor::backend::ops::IntTensorOps
• tensor::backend::ops::ModuleOps
• tensor::backend::ops::QTensorOps
• tensor::backend::ops::TransactionOps
• tensor::cast::ToElement
• tensor::ops::ActivationOps
• tensor::ops::BoolTensorOps
• tensor::ops::FloatTensorOps
• tensor::ops::IntTensorOps
• tensor::ops::ModuleOps
• tensor::ops::QTensorOps
• tensor::ops::TransactionOps
• tensor::quantization::QTensorPrimitive

Macros

• constant
• module::constant
• prelude::s
• tensor::s

Derive Macros

• config::Config
• module::Module
• prelude::Config
• prelude::Module
• record::Record

Functions

• config::config_to_json
• data::dataloader::split::split_dataloader
• data::dataset::transform::iota
• data::dataset::transform::shuffled_indices
• module::extract_type_name
• module::list_param_ids
• tensor::activation::gelu
• tensor::activation::glu
• tensor::activation::hard_sigmoid
• tensor::activation::hard_swish
• 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::sigmoid
• tensor::activation::silu
• tensor::activation::softmax
• tensor::activation::softmin
• tensor::activation::softplus
• tensor::activation::tanh
• tensor::backend::ops::attention::naive_attention
• tensor::backend::ops::conv::calculate_conv_output_shape
• tensor::backend::ops::conv::calculate_conv_output_size
• tensor::backend::ops::conv::calculate_conv_output_sizes
• tensor::backend::ops::conv::calculate_conv_padding
• tensor::backend::ops::conv::calculate_conv_transpose_output_shape
• tensor::backend::ops::conv::calculate_conv_transpose_output_size
• tensor::backend::ops::conv::calculate_pool_output_shape
• tensor::backend::ops::conv::calculate_pool_output_size
• tensor::backend::ops::grid_sample::float_grid_sample_2d_ref
• tensor::backend::ops::unfold::calculate_unfold_shape
• tensor::backend::ops::unfold::calculate_unfold_windows
• tensor::calculate_matmul_output
• tensor::cartesian_grid
• tensor::check_closeness
• tensor::expect_wrap
• tensor::grid::affine_grid_2d
• tensor::grid::meshgrid
• tensor::grid::meshgrid_stack
• tensor::linalg::cosine_similarity
• tensor::linalg::diag
• tensor::linalg::l0_norm
• tensor::linalg::l1_norm
• tensor::linalg::l2_norm
• tensor::linalg::lp_norm
• tensor::linalg::lu_decomposition
• tensor::linalg::matvec
• tensor::linalg::max_abs_norm
• tensor::linalg::min_abs_norm
• tensor::linalg::outer
• tensor::linalg::outer_dim
• tensor::linalg::trace
• tensor::linalg::vector_norm
• tensor::linalg::vector_normalize
• tensor::loss::cross_entropy_with_logits
• tensor::module::adaptive_avg_pool1d
• tensor::module::adaptive_avg_pool2d
• tensor::module::attention
• tensor::module::avg_pool1d
• tensor::module::avg_pool2d
• tensor::module::conv1d
• tensor::module::conv2d
• tensor::module::conv3d
• tensor::module::conv_transpose1d
• tensor::module::conv_transpose2d
• tensor::module::conv_transpose3d
• tensor::module::deform_conv2d
• tensor::module::embedding
• tensor::module::interpolate
• 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::naive_attention
• tensor::module::unfold4d
• tensor::quantization::compute_q_params
• tensor::quantization::compute_range
• tensor::quantization::params_shape
• tensor::ravel_index
• tensor::read_sync
• tensor::set_print_options
• tensor::try_read_sync
• tensor::try_wrap
• tensor::wrap_index

Type Aliases

• data::dataset::Result
• module::Devices
• prelude::Device
• record::CompactRecorder
• record::DebugRecordSettings
• record::DefaultFileRecorder
• record::DefaultRecorder
• record::NoStdInferenceRecorder
• record::NoStdTrainingRecorder
• record::SensitiveCompactRecorder
• tensor::Device
• tensor::ops::BoolElem
• tensor::ops::BoolTensor
• tensor::ops::Device
• tensor::ops::FloatElem
• tensor::ops::FloatTensor
• tensor::ops::IntElem
• tensor::ops::IntTensor
• tensor::ops::QuantizedTensor
• tensor::quantization::QuantizationParameters

Constants

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