List of all items
Structs
- cache::disk::DiskCacheConfig
- cache::prefabs::PreFabConfig
- cache::prefabs::PreFabMap
- cache::prefabs::StaticPreFabConfig
- cache::prefabs::StaticPreFabMap
- cache::weights::PretrainedWeightsDescriptor
- cache::weights::PretrainedWeightsMap
- cache::weights::StaticPretrainedWeightsDescriptor
- cache::weights::StaticPretrainedWeightsMap
- nn::attention::csa::CausalSelfAttention
- nn::attention::csa::CausalSelfAttentionConfig
- nn::attention::csa::CausalSelfAttentionRecord
- nn::attention::csa::CausalSelfAttentionRecordItem
- nn::attention::kvcache::KVCache
- nn::attention::kvcache::KVCacheConfig
- nn::attention::kvcache::KVCacheRecord
- nn::attention::kvcache::KVCacheRecordItem
- nn::attention::sdpa::ScaledDotProductAttentionConfig
- nn::embedding::rotary::RotaryEmbedding
- nn::embedding::rotary::RotaryEmbeddingConfig
- nn::embedding::rotary::RotaryEmbeddingRecord
- nn::embedding::rotary::RotaryEmbeddingRecordItem
- nn::layers::blocks::cna::AbstractCNA2dConfig
- nn::layers::blocks::cna::CNA2d
- nn::layers::blocks::cna::CNA2dConfig
- nn::layers::blocks::cna::CNA2dRecord
- nn::layers::blocks::cna::CNA2dRecordItem
- nn::layers::blocks::conv_norm::ConvNorm2d
- nn::layers::blocks::conv_norm::ConvNorm2dConfig
- nn::layers::blocks::conv_norm::ConvNorm2dRecord
- nn::layers::blocks::conv_norm::ConvNorm2dRecordItem
- nn::layers::drop::drop_block::DropBlock2d
- nn::layers::drop::drop_block::DropBlock2dConfig
- nn::layers::drop::drop_block::DropBlockOptions
- nn::layers::drop::drop_path::DropPath
- nn::layers::drop::drop_path::DropPathConfig
- nn::layers::drop::rate_table::DropPathRateDepthTable
- nn::layers::patching::patch_embed::PatchEmbed
- nn::layers::patching::patch_embed::PatchEmbedConfig
- nn::layers::patching::patch_embed::PatchEmbedRecord
- nn::layers::patching::patch_embed::PatchEmbedRecordItem
- nn::layers::pool::AvgPool2dSame
- nn::layers::pool::AvgPool2dSameConfig
- nn::module_util::param_map::ModulePath
- nn::module_util::param_map::ModulePathNode
- nn::module_util::param_map::ParamMap
- nn::module_util::param_map::ParamTag
- utility::burn::clamp::ClampConfig
- utility::burn::distribution::DistributionDisplayAdapter
- utility::burn::noise::NoiseConfig
Enums
Traits
- nn::attention::csa::CausalSelfAttentionMeta
- nn::attention::kvcache::KVCacheMeta
- nn::embedding::rotary::RotaryEmbeddingMeta
- nn::layers::blocks::cna::CNA2dMeta
- nn::layers::blocks::conv_norm::ConvNorm2dMeta
- nn::layers::drop::drop_path::DropPathMeta
- nn::layers::patching::patch_embed::PatchEmbedMeta
Functions
- burn_ext::norm::rms_norm
- burn_ext::slice_util::slice_to_indices
- burn_ext::tensor::repeat_interleave
- cache::disk::try_cache_download_to_path
- cache::weights::fetch_model_weights
- cache::weights::pretrained_weights_resource_key
- cache::weights::url_to_cache_key
- compat::conv_shape::expect_conv1d_output_size
- compat::conv_shape::expect_conv_output_shape
- compat::conv_shape::expect_conv_output_shape_dyn
- compat::conv_shape::maybe_conv1d_output_size
- compat::conv_shape::maybe_conv_output_shape
- compat::conv_shape::maybe_conv_output_shape_dyn
- compat::ops::float_arange
- compat::ops::float_linspace
- compat::ops::float_vec_arange
- compat::ops::float_vec_linspace
- nn::attention::sdpa::scaled_dot_product_attention
- nn::attention::sdpa::sdpa_attn_weight
- nn::attention::sdpa::sdpa_bias
- nn::embedding::rotary::inverse_frequency_table
- nn::embedding::rotary::positional_frequency_table
- nn::functional::drop::dropout
- nn::layers::drop::drop_block::drop_block_2d
- nn::layers::drop::drop_path::drop_path
- nn::layers::drop::rate_table::progressive_dpr
- nn::layers::pool::get_same_padding
- nn::layers::pool::pad_same
- utility::burn::kernels::conv2d_kernel_midpoint_filter
- utility::burn::record::display_record
- utility::probability::expect_probability
- utility::probability::try_probability
- utility::results::expect_unwrap
- utility::zspace::expect_point_bounds_check
- utility::zspace::try_point_bounds_check
- utility::zspace::zspace_partial_cmp