OpenNN
Open-source neural networks library
Loading...
Searching...
No Matches
opennn::opennn::Configuration Class Reference

#include <neural_network.h>

Classes

struct  Resolved
 

Public Member Functions

void set (Device new_device=Device::Auto, Type new_training_type=Type::Auto, Type new_inference_type=Type::Auto)
 
Device get_device () const
 
Type get_training_type () const
 
Type get_inference_type () const
 
const Resolvedresolve () const
 
bool is_gpu () const
 
bool is_cpu () const
 
bool is_bf16_training () const
 
bool is_bf16_inference () const
 

Static Public Member Functions

static Configurationinstance ()
 

Member Function Documentation

◆ get_device()

Device opennn::opennn::Configuration::get_device ( ) const
inline

◆ get_inference_type()

Type opennn::opennn::Configuration::get_inference_type ( ) const
inline

◆ get_training_type()

Type opennn::opennn::Configuration::get_training_type ( ) const
inline

◆ instance()

static Configuration & opennn::opennn::Configuration::instance ( )
inlinestatic

◆ is_bf16_inference()

bool opennn::opennn::Configuration::is_bf16_inference ( ) const
inline

◆ is_bf16_training()

bool opennn::opennn::Configuration::is_bf16_training ( ) const
inline

◆ is_cpu()

bool opennn::opennn::Configuration::is_cpu ( ) const
inline

◆ is_gpu()

bool opennn::opennn::Configuration::is_gpu ( ) const
inline

◆ resolve()

const Resolved & opennn::opennn::Configuration::resolve ( ) const
inline

◆ set()

void opennn::opennn::Configuration::set ( Device new_device = Device::Auto,
Type new_training_type = Type::Auto,
Type new_inference_type = Type::Auto )