List of all items
Structs
- device::Device
- device::DeviceInfo
- device::async_executor::AsyncExecutor
- device::async_executor::AsyncExecutorConfig
- device::async_executor::AsyncOperation
- device::async_executor::ExecutionStream
- device::async_executor::ExecutorStats
- device::async_executor::OperationId
- device::async_executor::StreamId
- device::async_executor::StreamStats
- device::cpu::CpuBackend
- device::cpu::CpuFeatures
- device::cpu::CpuKernel
- device::cpu::CpuMemory
- device::gpu::VulkanBackend
- device::gpu::VulkanBuffer
- device::gpu::VulkanKernel
- device::kernel_fusion::BufferId
- device::kernel_fusion::FusedKernel
- device::kernel_fusion::FusionConfig
- device::kernel_fusion::KernelFusionEngine
- device::kernel_fusion::MatMulDims
- device::memory_pool::BucketStats
- device::memory_pool::DetailedPoolStats
- device::memory_pool::GpuMemoryPool
- device::memory_pool::PoolConfig
- device::memory_pool::PoolStats
- device::memory_pool::ScopedBuffer
- io::DatasetInfo
- io::LayerInfo
- io::ModelArchitecture
- io::ModelMetadata
- io::SerializableModel
- io::TrainingInfo
- layers::conv::Conv2DLayer
- layers::dense::DenseLayer
- layers::dropout::DropoutLayer
- layers::dropout::SpatialDropoutLayer
- layers::normalization::BatchNormLayer
- layers::normalization::LayerNormLayer
- layers::pooling::AvgPool2DLayer
- layers::pooling::FlattenLayer
- layers::pooling::MaxPool2DLayer
- layers::pooling::ReshapeLayer
- network::LayerInfo
- network::Network
- network::NetworkMetrics
- network::NetworkSummary
- network::ProgressCallback
- network::builder::NetworkBuilder
- network::training::MovingAverage
- network::training::TrainingConfig
- network::training::TrainingHistory
- network::training::TrainingMetrics
- network::training::TrainingSummary
- optimizers::AdaGrad
- optimizers::Adam
- optimizers::SGD
- prelude::HashMap
- tensor::SerializableTensor
- tensor::Tensor
- tensor::view::AxisIterator
- tensor::view::TensorView
- tensor::view::TensorViewIterator
- utils::perf::Profiler
Enums
- activations::Activation
- device::DeviceType
- device::async_executor::Priority
- device::async_executor::StreamSelection
- device::cpu::ActivationType
- device::cpu::CpuOperation
- device::cpu::ReductionType
- device::kernel_fusion::FusableOp
- error::NnlError
- io::ModelFormat
- layers::LayerConfig
- layers::TrainingMode
- layers::WeightInit
- losses::LossFunction
- network::training::LearningRateSchedule
- optimizers::OptimizerConfig
- tensor::TensorData
- tensor::ops::TensorOp
Traits
- device::Backend
- device::DeviceMemory
- device::Kernel
- error::IntoNnlError
- layers::Layer
- network::TrainingCallback
- optimizers::Optimizer
Functions
- device::clear_default_device
- device::kernel_fusion::patterns::add_relu
- device::kernel_fusion::patterns::matmul_bias_relu
- device::kernel_fusion::patterns::scale_shift_activate
- device::set_default_device
- device::utils::benchmark_devices
- device::utils::list_devices
- io::checkpoint::cleanup_checkpoints
- io::checkpoint::load_latest_checkpoint
- io::checkpoint::save_checkpoint
- io::export::export_onnx
- io::load_model
- io::load_model_auto
- io::load_network
- io::load_network_auto
- io::save_model
- io::validation::validate_model
- layers::create_layer
- layers::utils::conv_output_size
- layers::utils::count_parameters
- layers::utils::pool_output_size
- layers::utils::same_padding
- layers::utils::validate_shapes
- network::builder::presets::lenet5
- network::builder::presets::mnist_classifier
- network::builder::presets::xor_network
- network::training::utils::cosine_annealing_lr
- network::training::utils::ema
- network::training::utils::polynomial_lr
- network::training::utils::warmup_lr
- optimizers::create_optimizer
- tensor::ops::activation
- tensor::ops::binary_op
- tensor::ops::broadcast_to
- tensor::ops::conv2d
- tensor::ops::is_broadcastable
- tensor::ops::matmul
- tensor::ops::reduce_max
- tensor::ops::reduce_min
- tensor::ops::reduce_sum
- tensor::ops::scalar_op
- tensor::ops::sqrt
- tensor::ops::transpose
- utils::data::create_batches
- utils::data::min_max_normalize
- utils::data::one_hot_decode
- utils::data::one_hot_encode
- utils::data::shuffle_data
- utils::data::standardize
- utils::data::train_test_split
- utils::io::load_from_csv
- utils::io::save_to_csv
- utils::math::binary_cross_entropy_loss
- utils::math::classification_accuracy
- utils::math::clip_gradients_by_norm
- utils::math::clip_gradients_by_value
- utils::math::cross_entropy_loss
- utils::math::mean_absolute_error
- utils::math::mean_squared_error
- utils::math::root_mean_squared_error
- utils::math::stable_softmax
- utils::math::top_k_accuracy
- utils::random::rand_tensor
- utils::random::randn_tensor
- utils::random::set_seed
- utils::tensor_utils::chw_to_hwc
- utils::tensor_utils::conv_output_size
- utils::tensor_utils::hwc_to_chw
- utils::tensor_utils::pad_2d