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
• layers::blocks::cna::AbstractCNA2dConfig
• layers::blocks::cna::CNA2d
• layers::blocks::cna::CNA2dConfig
• layers::blocks::cna::CNA2dRecord
• layers::blocks::cna::CNA2dRecordItem
• layers::blocks::conv_norm::ConvNorm2d
• layers::blocks::conv_norm::ConvNorm2dConfig
• layers::blocks::conv_norm::ConvNorm2dRecord
• layers::blocks::conv_norm::ConvNorm2dRecordItem
• layers::drop::drop_block::DropBlock2d
• layers::drop::drop_block::DropBlock2dConfig
• layers::drop::drop_block::DropBlockOptions
• layers::drop::drop_path::DropPath
• layers::drop::drop_path::DropPathConfig
• layers::drop::rate_table::DropPathRateDepthTable
• layers::patching::patch_embed::PatchEmbed
• layers::patching::patch_embed::PatchEmbedConfig
• layers::patching::patch_embed::PatchEmbedRecord
• layers::patching::patch_embed::PatchEmbedRecordItem
• layers::pool::AvgPool2dSame
• layers::pool::AvgPool2dSameConfig
• models::resnet::basic_block::BasicBlock
• models::resnet::basic_block::BasicBlockConfig
• models::resnet::basic_block::BasicBlockRecord
• models::resnet::basic_block::BasicBlockRecordItem
• models::resnet::bottleneck_block::BottleneckBlock
• models::resnet::bottleneck_block::BottleneckBlockConfig
• models::resnet::bottleneck_block::BottleneckBlockRecord
• models::resnet::bottleneck_block::BottleneckBlockRecordItem
• models::resnet::bottleneck_block::BottleneckPolicyConfig
• models::resnet::downsample::ResNetDownsample
• models::resnet::downsample::ResNetDownsampleConfig
• models::resnet::downsample::ResNetDownsampleRecord
• models::resnet::downsample::ResNetDownsampleRecordItem
• models::resnet::layer_block::LayerBlock
• models::resnet::layer_block::LayerBlockContractConfig
• models::resnet::layer_block::LayerBlockRecord
• models::resnet::layer_block::LayerBlockRecordItem
• models::resnet::layer_block::LayerBlockStructureConfig
• models::resnet::residual_block::ResidualBlockContractConfig
• models::resnet::resnet_io::pytorch_stubs::BasicBlockStub
• models::resnet::resnet_io::pytorch_stubs::BasicBlockStubRecord
• models::resnet::resnet_io::pytorch_stubs::BasicBlockStubRecordItem
• models::resnet::resnet_io::pytorch_stubs::BottleneckStub
• models::resnet::resnet_io::pytorch_stubs::BottleneckStubRecord
• models::resnet::resnet_io::pytorch_stubs::BottleneckStubRecordItem
• models::resnet::resnet_io::pytorch_stubs::DownsampleStub
• models::resnet::resnet_io::pytorch_stubs::DownsampleStubRecord
• models::resnet::resnet_io::pytorch_stubs::DownsampleStubRecordItem
• models::resnet::resnet_io::pytorch_stubs::LayerBlockStub
• models::resnet::resnet_io::pytorch_stubs::LayerBlockStubRecord
• models::resnet::resnet_io::pytorch_stubs::LayerBlockStubRecordItem
• models::resnet::resnet_io::pytorch_stubs::ResNetStub
• models::resnet::resnet_io::pytorch_stubs::ResNetStubRecord
• models::resnet::resnet_io::pytorch_stubs::ResNetStubRecordItem
• models::resnet::resnet_model::ResNet
• models::resnet::resnet_model::ResNetContractConfig
• models::resnet::resnet_model::ResNetRecord
• models::resnet::resnet_model::ResNetRecordItem
• models::resnet::resnet_model::ResNetStructureConfig
• models::resnet::stems::ResNetStem
• models::resnet::stems::ResNetStemRecord
• models::resnet::stems::ResNetStemRecordItem
• models::resnet::stems::ResNetStemStructureConfig
• models::swin::v2::block_sequence::StochasticDepthTransformerBlockSequence
• models::swin::v2::block_sequence::StochasticDepthTransformerBlockSequenceConfig
• models::swin::v2::block_sequence::StochasticDepthTransformerBlockSequenceRecord
• models::swin::v2::block_sequence::StochasticDepthTransformerBlockSequenceRecordItem
• models::swin::v2::patch_merge::PatchMerging
• models::swin::v2::patch_merge::PatchMergingConfig
• models::swin::v2::patch_merge::PatchMergingRecord
• models::swin::v2::patch_merge::PatchMergingRecordItem
• models::swin::v2::swin_block::BlockMlp
• models::swin::v2::swin_block::BlockMlpConfig
• models::swin::v2::swin_block::BlockMlpRecord
• models::swin::v2::swin_block::BlockMlpRecordItem
• models::swin::v2::swin_block::ShiftedWindowTransformerBlock
• models::swin::v2::swin_block::ShiftedWindowTransformerBlockConfig
• models::swin::v2::swin_block::ShiftedWindowTransformerBlockRecord
• models::swin::v2::swin_block::ShiftedWindowTransformerBlockRecordItem
• models::swin::v2::swin_model::LayerConfig
• models::swin::v2::swin_model::SwinTransformerV2
• models::swin::v2::swin_model::SwinTransformerV2Config
• models::swin::v2::swin_model::SwinTransformerV2Plan
• models::swin::v2::swin_model::SwinTransformerV2Record
• models::swin::v2::swin_model::SwinTransformerV2RecordItem
• models::swin::v2::window_attention::ContinuousPositionBiasMlp
• models::swin::v2::window_attention::ContinuousPositionBiasMlpConfig
• models::swin::v2::window_attention::ContinuousPositionBiasMlpRecord
• models::swin::v2::window_attention::ContinuousPositionBiasMlpRecordItem
• models::swin::v2::window_attention::OffsetGridRelativePositionBias
• models::swin::v2::window_attention::OffsetGridRelativePositionBiasRecord
• models::swin::v2::window_attention::OffsetGridRelativePositionBiasRecordItem
• models::swin::v2::window_attention::RelativePositionBiasConfig
• models::swin::v2::window_attention::WindowAttention
• models::swin::v2::window_attention::WindowAttentionConfig
• models::swin::v2::window_attention::WindowAttentionRecord
• models::swin::v2::window_attention::WindowAttentionRecordItem
• utility::burn::clamp::ClampConfig
• utility::burn::distribution::DistributionDisplayAdapter
• utility::burn::noise::NoiseConfig

Enums

• models::resnet::residual_block::ResidualBlock
• models::resnet::residual_block::ResidualBlockRecord
• models::resnet::residual_block::ResidualBlockRecordItem
• models::resnet::residual_block::ResidualBlockStructureConfig
• models::resnet::resnet_io::pytorch_stubs::ResidualBlockStub
• models::resnet::resnet_io::pytorch_stubs::ResidualBlockStubRecord
• models::resnet::resnet_io::pytorch_stubs::ResidualBlockStubRecordItem
• models::resnet::stems::ResNetStemContractConfig

Traits

• layers::blocks::cna::CNA2dMeta
• layers::blocks::conv_norm::ConvNorm2dMeta
• layers::drop::drop_path::DropPathMeta
• layers::patching::patch_embed::PatchEmbedMeta
• models::resnet::basic_block::BasicBlockMeta
• models::resnet::bottleneck_block::BottleneckBlockMeta
• models::resnet::downsample::ResNetDownsampleMeta
• models::resnet::layer_block::LayerBlockMeta
• models::resnet::residual_block::ResidualBlockMeta
• models::swin::v2::block_sequence::StochasticDepthTransformerBlockSequenceMeta
• models::swin::v2::patch_merge::PatchMergingMeta
• models::swin::v2::swin_block::BlockMlpMeta
• models::swin::v2::swin_block::ShiftedWindowTransformerBlockMeta
• models::swin::v2::swin_model::SwinTransformerV2Meta
• models::swin::v2::window_attention::ContinuousPositionBiasMlpMeta
• models::swin::v2::window_attention::RelativePositionBiasMeta
• models::swin::v2::window_attention::WindowAttentionMeta

Functions

• 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
• layers::drop::drop_block::drop_block_2d
• layers::drop::drop_path::drop_path
• layers::drop::rate_table::progressive_dpr
• layers::pool::get_same_padding
• layers::pool::pad_same
• models::resnet::resnet_io::pytorch_stubs::copy_cna_weights
• models::resnet::resnet_io::pytorch_stubs::copy_conv_norm_weights
• models::resnet::resnet_io::pytorch_stubs::copy_downsample_weights
• models::resnet::resnet_io::pytorch_stubs::load_resnet_stub_record
• models::resnet::util::build_square_conv2d_padding_config
• models::resnet::util::get_square_conv2d_padding
• models::resnet::util::scalar_to_array
• models::resnet::util::stride_div_output_resolution
• models::swin::v2::patch_merge::collate_patches
• models::swin::v2::patch_merge::decollate_patches
• models::swin::v2::window_attention::apply_attention_mask
• models::swin::v2::window_attention::sw_attn_mask
• models::swin::v2::window_attention::window_attention_relative_position_index
• models::swin::v2::window_attention::window_index_offset_grid
• models::swin::v2::window_attention::window_log1p_relative_offset_grid
• models::swin::v2::window_attention::window_relative_offset_grid
• models::swin::v2::windowing::window_partition
• models::swin::v2::windowing::window_reverse
• 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

Statics

• models::resnet::pretrained::PREFAB_RESNET_MAP
• models::resnet::util::CONV_INTO_RELU_INITIALIZER

Constants

• models::resnet::bottleneck_block::BOTTLENECK_BLOCK_DEFAULT_PINCH_FACTOR
• models::resnet::resnet_model::RESNET101_BLOCKS
• models::resnet::resnet_model::RESNET152_BLOCKS
• models::resnet::resnet_model::RESNET18_BLOCKS
• models::resnet::resnet_model::RESNET34_BLOCKS
• models::resnet::resnet_model::RESNET50_BLOCKS
• models::swin::v2::window_attention::EPS