List of all items

Structs

• SparseMatrix
• base::ModuleBase
• compile_time::FusedLinearActivation
• compile_time::StaticActivation
• compile_time::StaticLinear
• compile_time::StaticMLP
• composition::ComposedModule
• composition::ConditionalModule
• composition::ParallelModule
• composition::ResidualModule
• construction::ModuleConfig
• container::basic::FunctionModule
• container::basic::ModuleDict
• container::basic::ModuleList
• container::basic::Sequential
• container::dynamic_graph::DynamicGraph
• container::lazy::LazyModuleDict
• container::lazy::LazyModuleList
• container::lazy::LazySequential
• container::parameters::ParameterDict
• container::parameters::ParameterList
• conversion::ConversionConfig
• conversion::ConvertedModel
• conversion::LayerInfo
• conversion::MigratedModel
• conversion::MigrationHelper
• conversion::ModelConverter
• conversion::ModelMetadata
• core::module_ext::ParameterStats
• core::module_ext::ValidationReport
• cuda_kernels::CudaKernelRegistry
• cuda_kernels::CudaNeuralOps
• cuda_kernels::CudaOptimizations
• cuda_kernels::CustomActivations
• cuda_kernels::KernelBenchmarkResult
• cuda_kernels::MemoryProfile
• diagnostics::ModuleDiagnostics
• diagnostics::ModuleInfo
• export::DeploymentOptimizer
• export::ExportConfig
• export::ModelExporter
• export::OptimizedModel
• export::benchmarks::BenchmarkResults
• export::benchmarks::BenchmarkSummary
• export::benchmarks::ConversionBenchmarker
• export::benchmarks::ConversionMetrics
• export::benchmarks::ExportBenchmarker
• export::benchmarks::ExportMetrics
• functional::SparseMatrix
• functional::activation::ELU
• functional::activation::GELU
• functional::activation::LeakyReLU
• functional::activation::LogSoftmax
• functional::activation::Mish
• functional::activation::ReLU
• functional::activation::SELU
• functional::activation::Sigmoid
• functional::activation::Softmax
• functional::activation::Swish
• functional::activation::Tanh
• functional::core::ActivationConfig
• functional::core::FunctionalBuilder
• functional::core::FunctionalConfig
• functional::loss_advanced::AdaptiveLoss
• functional::loss_advanced::BinaryCrossEntropy
• functional::loss_advanced::CategoricalCrossEntropy
• functional::loss_advanced::CombinedLoss
• functional::loss_advanced::CosineEmbeddingLoss
• functional::loss_advanced::DiceLoss
• functional::loss_advanced::FocalLoss
• functional::loss_advanced::HingeLoss
• functional::loss_advanced::HuberLoss
• functional::loss_advanced::IoULoss
• functional::loss_advanced::KLDivLoss
• functional::loss_advanced::L1Loss
• functional::loss_advanced::LossBuilder
• functional::loss_advanced::LossFactory
• functional::loss_advanced::MSELoss
• functional::loss_advanced::NLLLoss
• functional::loss_advanced::SmoothL1Loss
• functional::loss_advanced::TripletMarginLoss
• functional::loss_advanced::WeightedLoss
• functional::norm::BatchNormOutput
• gradcheck::GradCheckConfig
• gradcheck::GradCheckResult
• gradcheck::GradChecker
• gradcheck::ParameterGradCheckResult
• hardware_opts::HardwareCapabilities
• hardware_opts::HardwareContext
• hardware_opts::HardwareLinear
• hooks::HookHandle
• hooks::HookRegistry
• layers::activation::basic::LeakyReLU
• layers::activation::basic::PReLU
• layers::activation::basic::ReLU
• layers::activation::basic::ReLU6
• layers::activation::basic::Sigmoid
• layers::activation::basic::Tanh
• layers::activation::gated::GEGLU
• layers::activation::gated::GLU
• layers::activation::gated::ReGLU
• layers::activation::gated::SwiGLU
• layers::activation::modern::ELU
• layers::activation::modern::GELU
• layers::activation::modern::Hardswish
• layers::activation::modern::Mish
• layers::activation::modern::SELU
• layers::activation::modern::SiLU
• layers::activation::smooth::Hardsigmoid
• layers::activation::smooth::Softplus
• layers::activation::smooth::Softsign
• layers::activation::softmax::LogSigmoid
• layers::activation::softmax::LogSoftmax
• layers::activation::softmax::Softmax
• layers::activation::threshold::Hardshrink
• layers::activation::threshold::Hardtanh
• layers::activation::threshold::Softshrink
• layers::activation::threshold::Tanhshrink
• layers::activation::threshold::Threshold
• layers::advanced::AdvancedLayerNorm
• layers::advanced::MultiHeadAttention
• layers::advanced::PositionalEncoding
• layers::attention::AdditiveAttention
• layers::attention::FlashMultiHeadAttention
• layers::attention::FlexAttention
• layers::attention::LocationBasedAttention
• layers::attention::MultiheadAttention
• layers::attention::MultiplicativeAttention
• layers::blocks::BasicBlock
• layers::blocks::BottleneckBlock
• layers::blocks::DenseBlock
• layers::blocks::DenseLayer
• layers::blocks::MBConvBlock
• layers::blocks::SEBlock
• layers::blocks::TransitionLayer
• layers::conv::Conv1d
• layers::conv::Conv2d
• layers::conv::Conv3d
• layers::conv::ConvTranspose1d
• layers::conv::ConvTranspose2d
• layers::conv::ConvTranspose3d
• layers::conv::DepthwiseSeparableConv
• layers::cross_attention::CrossAttention
• layers::efficientnet::BlockConfig
• layers::efficientnet::EfficientNet
• layers::efficientnet::EfficientNetBackbone
• layers::efficientnet::EfficientNetConfig
• layers::efficientnet::MBConvWithStochasticDepth
• layers::efficientnet::TrainingConfig
• layers::embedding::AlibiPositionalBias
• layers::embedding::Embedding
• layers::embedding::LearnablePositionalEncoding
• layers::embedding::PositionInterpolation
• layers::embedding::RelativePositionalEncoding
• layers::embedding::RotaryPositionalEmbedding
• layers::embedding::SinusoidalPositionEmbedding
• layers::embedding::SinusoidalPositionalEncoding
• layers::ip_adapter::IPAdapterCrossAttention
• layers::ip_adapter::IPAdapterCrossAttentionConfig
• layers::lazy::LazyConv1d
• layers::lazy::LazyConv2d
• layers::lazy::LazyLinear
• layers::linear::Flatten
• layers::linear::Linear
• layers::mobilenet::ComparisonResult
• layers::mobilenet::DeploymentConfig
• layers::mobilenet::DepthwiseSeparableConv
• layers::mobilenet::EfficiencyMetrics
• layers::mobilenet::InvertedResidual
• layers::mobilenet::MobileNetConfig
• layers::mobilenet::MobileNetV1
• layers::mobilenet::MobileNetV1Block
• layers::mobilenet::MobileNetV2
• layers::mobilenet::MobileNetV2Block
• layers::normalization::NormalizationFactory
• layers::normalization::NormalizationPresets
• layers::normalization::advanced::SwitchableNorm2d
• layers::normalization::batch::BatchNorm1d
• layers::normalization::batch::BatchNorm2d
• layers::normalization::batch::BatchNorm3d
• layers::normalization::batch::BatchRenorm2d
• layers::normalization::batch::BatchRenormSchedule
• layers::normalization::batch::SyncBatchNorm2d
• layers::normalization::batch::VirtualBatchNorm2d
• layers::normalization::common::NormalizationConfig
• layers::normalization::common::NormalizationStats
• layers::normalization::common::RunningStats
• layers::normalization::instance::InstanceNorm1d
• layers::normalization::instance::InstanceNorm2d
• layers::normalization::instance::InstanceNorm3d
• layers::normalization::layer_group::GroupNorm
• layers::normalization::layer_group::LayerNorm
• layers::normalization::layer_group::RMSNorm
• layers::normalization::weight_based::SpectralNorm
• layers::normalization::weight_based::WeightNorm
• layers::normalization::weight_based::WeightStandardization
• layers::pooling::AdaptiveAvgPool1d
• layers::pooling::AdaptiveAvgPool2d
• layers::pooling::AdaptiveAvgPool3d
• layers::pooling::AdaptiveMaxPool1d
• layers::pooling::AdaptiveMaxPool2d
• layers::pooling::AdaptiveMaxPool3d
• layers::pooling::AvgPool2d
• layers::pooling::FractionalMaxPool1d
• layers::pooling::FractionalMaxPool2d
• layers::pooling::FractionalMaxPool3d
• layers::pooling::GlobalPool
• layers::pooling::LPPool1d
• layers::pooling::LPPool2d
• layers::pooling::MaxPool1d
• layers::pooling::MaxPool2d
• layers::pooling::MaxPool3d
• layers::recurrent::CustomRNNCell
• layers::recurrent::GRU
• layers::recurrent::GRUCell
• layers::recurrent::LSTM
• layers::recurrent::LSTMCell
• layers::recurrent::RNN
• layers::regularization::AlphaDropout
• layers::regularization::CutMix
• layers::regularization::Cutout
• layers::regularization::DropBlock2d
• layers::regularization::DropConnect
• layers::regularization::Dropout
• layers::regularization::Dropout2d
• layers::regularization::Mixup
• layers::regularization::StochasticDepth
• layers::transformer::Transformer
• layers::transformer::TransformerEncoder
• layers::transformer::TransformerEncoderLayer
• layers::upsampling::PixelShuffle
• layers::upsampling::PixelShuffle1d
• layers::upsampling::PixelUnshuffle
• layers::upsampling::PixelUnshuffle1d
• lazy::LazyLinear
• lazy::LazyWrapper
• mixed_precision::AutocastModel
• mixed_precision::GradScaler
• mixed_precision::MixedPrecisionConfig
• mixed_precision::MixedPrecisionTrainer
• mixed_precision::bf16::BF16Config
• mixed_precision::bf16::BF16Trainer
• mixed_precision::loss_scaling::DynamicLossScaler
• model_zoo::ModelConfig
• model_zoo::ModelMetadata
• model_zoo::ModelZoo
• model_zoo::PretrainedWeights
• model_zoo::WeightInfo
• numerical_stability::StabilityConfig
• numerical_stability::StabilityResults
• numerical_stability::StabilityTester
• numerical_stability::TestResult
• optimization::ComputationGraph
• optimization::MemoryProfiler
• optimization::NetworkOptimizer
• optimization::OpNode
• optimization::OptimizationApplied
• optimization::OptimizationReport
• parameter::Parameter
• parameter::ParameterCollection
• parameter::ParameterDiagnostics
• parameter::ParameterStats
• parameter::parameter_ext::ParameterAnalysis
• parameter::parameter_ext::ParameterGroup
• parameter_updates::LayerSpecificOptimizers
• parameter_updates::ParameterUpdater
• parameter_updates::UpdateConfig
• parameter_updates::UpdateStatistics
• pruning::Pruner
• pruning::PruningConfig
• pruning::PruningMask
• quantization::BackendQuantConfig
• quantization::CalibrationConfig
• quantization::CalibrationMetrics
• quantization::CalibrationStats
• quantization::QuantizationConfig
• quantization::QuantizationParams
• quantization::QuantizedModel
• quantization::calibration::Calibrator
• quantization::qat::FakeQuantize
• quantization::qat::QATConfig
• quantization::qat::QATEvaluationMetrics
• quantization::qat::QATLinear
• quantization::qat::QATModel
• quantization::qat::QATScheduler
• quantization::qat::QuantizedInferenceModel
• quantization::schemes::BlockWiseQuantization
• quantization::schemes::LayerQuantConfig
• quantization::schemes::LearnableQuantParams
• quantization::schemes::MixedPrecisionQuantization
• quantization::schemes::PostTrainingQuantization
• quantization::schemes::QuantizationAwareTraining
• quantization::utils::QuantizationError
• quantization::utils::SizeReduction
• quantization::utils::TensorDistributionStats
• research::CapsuleLayer
• research::DARTSCell
• research::GraphAttentionLayer
• research::GraphConvLayer
• research::MAMLModule
• research::NeuralODE
• scirs2_neural_integration::GELU
• scirs2_neural_integration::LayerNorm
• scirs2_neural_integration::MemoryEfficientSequential
• scirs2_neural_integration::Mish
• scirs2_neural_integration::MultiHeadAttention
• scirs2_neural_integration::NeuralConfig
• scirs2_neural_integration::SciRS2NeuralProcessor
• scirs2_neural_integration::Swish
• scirs2_neural_integration::TransformerDecoderLayer
• scirs2_neural_integration::TransformerEncoderLayer
• sparse::GradualPruningScheduler
• sparse::SparseConv2d
• sparse::SparseLinear
• sparse::SparseMask
• sparse::SparseTrainingConfig
• summary::LayerInfo
• summary::ModelProfiler
• summary::ModelSummary
• summary::ProfileResult
• summary::SummaryConfig
• summary::profiling::AnalysisConfig
• summary::profiling::AnalysisReport
• summary::profiling::BatchProfiler
• summary::profiling::BatchProfilingConfig
• summary::profiling::BatchProfilingResult
• summary::profiling::FLOPSAnalysis
• summary::profiling::FLOPSCounter
• summary::profiling::MemoryAnalysis
• summary::profiling::ModelAnalyzer
• utils::ModuleParameterStats
• visualization::AsciiRenderer
• visualization::DotRenderer
• visualization::GraphEdge
• visualization::GraphNode
• visualization::GraphStatistics
• visualization::NetworkGraph
• visualization::TextRenderer
• visualization::VisualizationConfig

Enums

• compile_time::ActivationKind
• container::dynamic_graph::GraphNode
• conversion::FrameworkSource
• export::ExportFormat
• export::OptimizationLevel
• export::TargetDevice
• functional::activation::GeluApproximation
• functional::core::MemoryOptLevel
• functional::loss_advanced::Reduction
• hooks::HookType
• init::ActivationHint
• init::ArchitectureHint
• init::Distribution
• init::FanMode
• init::InitMethod
• init::Nonlinearity
• layers::attention::AttentionPattern
• layers::attention::LuongAttentionType
• layers::embedding::PositionalEncodingType
• layers::recurrent::CustomCellType
• lazy::LazyState
• numerical_stability::StabilityIssue
• optimization::FusionPattern
• optimization::MemoryHint
• optimization::OptimizationStrategy
• parameter::LayerType
• parameter::parameter_ext::ParameterConstraint
• pruning::PruningScope
• pruning::PruningStrategy
• quantization::CalibrationMethod
• quantization::DeploymentPlatform
• quantization::QuantizationScheme
• research::ODESolver
• sparse::SparsityPattern
• visualization::ColorScheme
• visualization::EdgeStyle
• visualization::LayoutAlgorithm

Traits

• compile_time::CompileTimeActivation
• compile_time::ShapeValidator
• composition::ModuleBuilder
• composition::ModuleComposition
• construction::ModuleConstruct
• core::Module
• core::module_ext::ModuleExt
• functional::core::Activation
• functional::loss_advanced::CustomLoss
• init::Initializable
• init::Initializer
• layers::lazy::LazyModule
• layers::recurrent::RNNCell
• lazy::LazyModule
• parameter::parameter_ext::ParameterCollectionExt
• parameter::parameter_ext::ParameterExt
• utils::ModuleApply

Macros

• func_error
• impl_module_constructor
• lazy_module
• validate_inputs

Functions

• compile_time::cold_path_hint
• compile_time::inline_always_hint
• conversion::pytorch_compat::convert_state_dict
• conversion::pytorch_compat::generate_pytorch_names
• conversion::pytorch_compat::layer_type_mapping
• conversion::tensorflow_compat::convert_saved_model
• conversion::tensorflow_compat::operation_mapping
• cuda_kernels::global_kernel_registry
• cuda_kernels::utils::compare_with_reference
• cuda_kernels::utils::generate_test_tensors
• cuda_kernels::utils::validate_kernel_args
• export::benchmarks::utils::compare_benchmarks
• export::benchmarks::utils::create_benchmark_report
• functional::activation::configured::gelu_configured
• functional::activation::configured::mish_configured
• functional::activation::configured::relu_configured
• functional::activation::configured::sigmoid_configured
• functional::activation::configured::softmax_configured
• functional::activation::configured::swish_configured
• functional::activation::configured::tanh_configured
• functional::activation::dropout
• functional::activation::elu
• functional::activation::gelu
• functional::activation::gelu_with_approximation
• functional::activation::leaky_relu
• functional::activation::log_softmax
• functional::activation::mish
• functional::activation::relu
• functional::activation::relu_inplace
• functional::activation::selu
• functional::activation::sigmoid
• functional::activation::softmax
• functional::activation::swish
• functional::activation::tanh
• functional::conv::conv1d
• functional::conv::conv2d
• functional::conv::conv3d
• functional::conv::conv_output_size
• functional::conv::conv_transpose1d
• functional::conv::conv_transpose2d
• functional::conv::conv_transpose3d
• functional::conv::conv_transpose_output_size
• functional::conv::validate_conv_params
• functional::core::default_config
• functional::core::numerics::epsilon_tensor
• functional::core::numerics::safe_clamp
• functional::core::numerics::safe_div
• functional::core::numerics::safe_rsqrt
• functional::core::numerics::safe_sqrt
• functional::core::optimized
• functional::core::performance::should_use_inplace
• functional::core::performance::with_memory_opt
• functional::core::safe
• functional::core::validation::validate_compatible_shapes
• functional::core::validation::validate_min_ndim
• functional::core::validation::validate_ndim
• functional::core::validation::validate_not_empty
• functional::core::validation::validate_positive
• functional::core::validation::validate_range
• functional::linear::apply_attention_mask
• functional::linear::bilinear
• functional::linear::create_causal_mask
• functional::linear::create_padding_mask
• functional::linear::embedding
• functional::linear::embedding_bag
• functional::linear::geglu
• functional::linear::glu
• functional::linear::grouped_query_attention
• functional::linear::learnable_positional_encoding
• functional::linear::linear
• functional::linear::multi_head_attention
• functional::linear::multi_query_attention
• functional::linear::one_hot
• functional::linear::post_norm_layer_norm
• functional::linear::pre_norm_layer_norm
• functional::linear::rms_norm
• functional::linear::rotary_positional_encoding
• functional::linear::scaled_dot_product_attention
• functional::linear::sinusoidal_positional_encoding
• functional::linear::swiglu
• functional::loss::binary_cross_entropy
• functional::loss::binary_cross_entropy_with_logits
• functional::loss::center_loss
• functional::loss::contrastive_loss
• functional::loss::cosine_embedding_loss
• functional::loss::cross_entropy
• functional::loss::dice_loss
• functional::loss::focal_loss
• functional::loss::huber_loss
• functional::loss::infonce_loss
• functional::loss::kl_div
• functional::loss::l1_loss
• functional::loss::mse_loss
• functional::loss::multi_margin_loss
• functional::loss::multilabel_margin_loss
• functional::loss::nll_loss
• functional::loss::smooth_l1_loss
• functional::loss::triplet_margin_loss
• functional::loss::tversky_loss
• functional::loss::wing_loss
• functional::loss_advanced::validation::check_classification_targets
• functional::loss_advanced::validation::check_probabilities
• functional::loss_advanced::validation::check_shapes
• functional::losses::cross_entropy_configured
• functional::losses::l1_loss_configured
• functional::losses::mse_loss_configured
• functional::norm::batch_norm
• functional::norm::batch_norm_1d
• functional::norm::batch_norm_2d
• functional::norm::batch_norm_2d_with_config
• functional::norm::batch_norm_3d
• functional::norm::batch_norm_nd_with_running_stats
• functional::norm::configured::batch_norm_configured
• functional::norm::create_affine_params
• functional::norm::get_norm_features
• functional::norm::group_norm
• functional::norm::instance_norm
• functional::norm::layer_norm
• functional::norm::layer_norm_configured
• functional::norm::layer_norm_enhanced
• functional::norm::local_response_norm
• functional::norm::rms_norm
• functional::norm::spectral_norm
• functional::norm::validate_norm_params
• functional::norm::weight_norm
• functional::pooling::functions::adaptive_avg_pool1d
• functional::pooling::functions::adaptive_avg_pool2d
• functional::pooling::functions::adaptive_avg_pool3d
• functional::pooling::functions::adaptive_max_pool1d
• functional::pooling::functions::adaptive_max_pool2d
• functional::pooling::functions::adaptive_max_pool3d
• functional::pooling::functions::adaptive_pool_params
• functional::pooling::functions::avg_pool1d
• functional::pooling::functions::avg_pool2d
• functional::pooling::functions::avg_pool3d
• functional::pooling::functions::global_avg_pool1d
• functional::pooling::functions::global_avg_pool2d
• functional::pooling::functions::global_avg_pool3d
• functional::pooling::functions::global_max_pool1d
• functional::pooling::functions::global_max_pool2d
• functional::pooling::functions::global_max_pool3d
• functional::pooling::functions::max_pool1d
• functional::pooling::functions::max_pool2d
• functional::pooling::functions::max_pool3d
• functional::pooling::functions::pad
• functional::pooling::functions::pool_output_size
• functional::pooling::functions::reflection_pad1d
• functional::pooling::functions::reflection_pad2d
• functional::pooling::functions::replication_pad1d
• functional::pooling::functions::replication_pad2d
• functional::pooling::functions::zero_pad2d
• gradcheck::fast_gradcheck
• gradcheck::fast_gradcheck_function
• gradcheck::gradcheck
• gradcheck::gradcheck_function
• gradcheck::precise_gradcheck
• hardware_opts::print_hardware_info
• init::auto_init
• init::calculate_fan
• init::calculate_fan_in_fan_out
• init::constant
• init::coordinate_mlp_init
• init::delta_orthogonal_init
• init::dirac_init
• init::eye_init
• init::eye_init_tensor
• init::fixup_init
• init::gan_balanced_init
• init::init
• init::init_tensor
• init::kaiming_normal
• init::kaiming_normal_with_nonlinearity
• init::kaiming_uniform
• init::kaiming_uniform_with_nonlinearity
• init::lecun_normal
• init::lecun_uniform
• init::lsuv_init
• init::metainit
• init::normal
• init::orthogonal_init
• init::recommend_init_method
• init::rezero_alpha_init
• init::rezero_init
• init::siren_init
• init::sparse_init
• init::truncated_normal
• init::uniform
• init::variance_scaling
• init::xavier_normal
• init::xavier_normal_with_gain
• init::xavier_uniform
• init::xavier_uniform_with_gain
• init::zero_centered_variance_init
• layers::activation::factory::geglu
• layers::activation::factory::gelu
• layers::activation::factory::hardswish
• layers::activation::factory::leaky_relu
• layers::activation::factory::leaky_relu_with_slope
• layers::activation::factory::log_softmax
• layers::activation::factory::mish
• layers::activation::factory::prelu
• layers::activation::factory::relu
• layers::activation::factory::relu6
• layers::activation::factory::selu
• layers::activation::factory::silu
• layers::activation::factory::softmax
• layers::activation::factory::swiglu
• layers::attention::flash_attention
• layers::attention::scaled_dot_product_attention
• layers::efficientnet::utils::calculate_model_params
• layers::efficientnet::utils::create_custom_efficientnet
• layers::efficientnet::utils::get_training_config
• layers::mobilenet::utils::calculate_efficiency_metrics
• layers::mobilenet::utils::compare_configurations
• layers::mobilenet::utils::create_custom_mobilenet_v2
• layers::mobilenet::utils::get_deployment_config
• layers::normalization::common::unbiased_variance
• layers::normalization::common::utils::apply_channel_affine
• layers::normalization::common::utils::apply_channel_normalization
• layers::normalization::common::utils::apply_normalization
• layers::normalization::common::utils::channel_broadcast_shape
• layers::normalization::common::utils::compute_channel_mean
• layers::normalization::common::utils::compute_channel_variance
• layers::normalization::common::utils::detached_statistic
• layers::normalization::common::utils::instance_normalize
• layers::upsampling::utils::pixel_shuffle_input_channels
• layers::upsampling::utils::pixel_shuffle_output_size
• layers::upsampling::utils::pixel_unshuffle_output_channels
• layers::upsampling::utils::pixel_unshuffle_output_size
• lazy::lazy_linear
• lazy::lazy_linear_no_bias
• mixed_precision::prelude::aggressive_config
• mixed_precision::prelude::bf16_config
• mixed_precision::prelude::conservative_config
• mixed_precision::prelude::fp16_config
• model_zoo::get_pretrained_weights
• model_zoo::is_pretrained_available
• model_zoo::load_pretrained_weights
• numerical_stability::utils::comprehensive_stability_analysis
• numerical_stability::utils::generate_pathological_inputs
• numerical_stability::utils::generate_test_inputs
• numerical_stability::utils::quick_stability_check
• optimization::optimize_for_inference
• optimization::optimize_module
• parameter_updates::utils::are_parameters_vectorizable
• parameter_updates::utils::calculate_memory_usage
• parameter_updates::utils::estimate_batch_size
• parameter_updates::utils::group_parameters_by_device
• quantization::calibration::calculate_optimal_scale
• quantization::ops::dequantize_tensor
• quantization::ops::per_channel_quantize_weights
• quantization::ops::quantize_tensor
• quantization::ops::quantized_conv2d_int8
• quantization::ops::quantized_matmul_int8
• quantization::ops::quantized_relu_int8
• quantization::prelude::dynamic_quantization
• quantization::prelude::int8_asymmetric
• quantization::prelude::int8_symmetric
• quantization::prelude::qat_aggressive_config
• quantization::prelude::qat_conservative_config
• quantization::prelude::qat_int8_config
• quantization::qat::utils::calibrate_qat_model
• quantization::qat::utils::convert_model_to_qat
• quantization::qat::utils::evaluate_qat_quality
• quantization::qat::utils::prepare_qat_model
• quantization::qat::utils::progressive_qat_training
• quantization::qat::utils::qat_training_loop
• quantization::schemes::utils::dynamic_int8
• quantization::schemes::utils::int8_symmetric
• quantization::schemes::utils::kl_divergence_int8
• quantization::schemes::utils::percentile_int8
• quantization::schemes::utils::uint8_asymmetric
• quantization::utils::analyze_tensor_distribution
• quantization::utils::batch_analyze_tensors
• quantization::utils::calculate_quantization_error
• quantization::utils::create_deployment_config
• quantization::utils::create_mixed_precision_config
• quantization::utils::estimate_size_reduction
• quantization::utils::find_optimal_bitwidth
• summary::estimate_tensor_memory
• summary::summarize
• summary::utils::check_memory_budget
• summary::utils::count_model_parameters
• summary::utils::get_model_size_mb
• summary::utils::quick_summary
• utils::analysis::count_parameters
• utils::analysis::count_trainable_parameters
• utils::analysis::find_parameters_by_pattern
• utils::analysis::is_training
• utils::analysis::parameter_names
• utils::analysis::parameter_statistics
• utils::introspection::health_check
• utils::introspection::print_parameter_summary
• visualization::create_graph_from_model
• visualization::utils::generate_dot
• visualization::utils::print_model_viz
• visualization::utils::quick_ascii_viz
• visualization::utils::quick_text_viz

Type Aliases

• cuda_kernels::CudaKernelFn
• functional::core::FuncResult
• hooks::HookCallback
• layers::activation::modern::Swish
• layers::recurrent::ActivationFn

Constants

• VERSION
• VERSION_MAJOR
• VERSION_MINOR
• VERSION_PATCH