Loading...
Searching...
No Matches
configuration.h File Reference
#include <atomic>
#include <mutex>

Go to the source code of this file.

Classes

struct  opennn::PrecisionPlan
 
struct  opennn::EffectiveConfig
 
class  opennn::Configuration
 

Namespaces

namespace  opennn
 

Enumerations

enum class  opennn::Device { opennn::Auto , opennn::CPU , opennn::CUDA }
 
enum class  opennn::Type { opennn::Auto , opennn::FP32 , opennn::BF16 , opennn::INT8 }
 
enum class  opennn::Blas { opennn::Eigen , opennn::Mkl }
 
enum class  opennn::ActivationFunction {
  opennn::Identity , opennn::Sigmoid , opennn::Tanh , opennn::ReLU ,
  opennn::Softmax , opennn::LeakyReLU , opennn::GELU , opennn::GELUTanh ,
  opennn::SiLU
}
 
enum class  opennn::LinearEpilogue {
  opennn::Default , opennn::Relu , opennn::Bias , opennn::ReluBias ,
  opennn::ReluAuxBias , opennn::DRelu , opennn::GeluAuxBias , opennn::BiasGradient
}
 

Functions

constexpr PrecisionPlan opennn::make_precision_plan (const Type type) noexcept
 
Type opennn::activation_dtype (Type type)
 

Variables

constexpr float opennn::LEAKY_RELU_SLOPE = 0.1f