BlackCat_Tensors
A GPU-supported autograd and linear algebra library, designed for neural network construction
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
bc::nn::Layer_Loader Struct Reference

#include <layer_loader.h>

Collaboration diagram for bc::nn::Layer_Loader:
Collaboration graph
[legend]

Public Member Functions

 Layer_Loader (string root_directory)
 
void set_current_layer_name (string current_layername)
 
void set_current_layer_index (int layer_index)
 
string current_layer_subdir () const
 
template<class T >
void save_variable (const T &tensor, string variable_name)
 
template<class T >
void load_variable (T &tensor, string variable_name)
 
template<class Tensor >
void load_variable (Tensor &tensor, string variable_name, Integer< 1 >)
 
template<class T >
void load_variable (T &tensor, string variable_name, Integer< 2 >)
 
template<class T , int X>
void load_variable (T &tensor, string variable_name, Integer< X >)
 
void make_current_directory ()
 
bool file_exists (int dim, string filename)
 

Static Public Member Functions

static string dim_to_extension (int dim)
 

Public Attributes

string root_directory
 
string current_layername
 
int current_layer_index = -1
 

Constructor & Destructor Documentation

◆ Layer_Loader()

bc::nn::Layer_Loader::Layer_Loader ( string  root_directory)
inline

Member Function Documentation

◆ current_layer_subdir()

string bc::nn::Layer_Loader::current_layer_subdir ( ) const
inline

◆ dim_to_extension()

static string bc::nn::Layer_Loader::dim_to_extension ( int  dim)
inlinestatic

◆ file_exists()

bool bc::nn::Layer_Loader::file_exists ( int  dim,
string  filename 
)
inline

◆ load_variable() [1/4]

template<class T >
void bc::nn::Layer_Loader::load_variable ( T &  tensor,
string  variable_name 
)
inline

◆ load_variable() [2/4]

template<class Tensor >
void bc::nn::Layer_Loader::load_variable ( Tensor tensor,
string  variable_name,
Integer< 1 >   
)
inline

◆ load_variable() [3/4]

template<class T >
void bc::nn::Layer_Loader::load_variable ( T &  tensor,
string  variable_name,
Integer< 2 >   
)
inline

◆ load_variable() [4/4]

template<class T , int X>
void bc::nn::Layer_Loader::load_variable ( T &  tensor,
string  variable_name,
Integer< X >   
)
inline

◆ make_current_directory()

void bc::nn::Layer_Loader::make_current_directory ( )
inline

◆ save_variable()

template<class T >
void bc::nn::Layer_Loader::save_variable ( const T &  tensor,
string  variable_name 
)
inline

◆ set_current_layer_index()

void bc::nn::Layer_Loader::set_current_layer_index ( int  layer_index)
inline

◆ set_current_layer_name()

void bc::nn::Layer_Loader::set_current_layer_name ( string  current_layername)
inline

Member Data Documentation

◆ current_layer_index

int bc::nn::Layer_Loader::current_layer_index = -1

◆ current_layername

string bc::nn::Layer_Loader::current_layername

◆ root_directory

string bc::nn::Layer_Loader::root_directory

The documentation for this struct was generated from the following file: