#include <layer_loader.h>
|
| 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) |
|
◆ Layer_Loader()
bc::nn::Layer_Loader::Layer_Loader |
( |
string |
root_directory | ) |
|
|
inline |
◆ 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 |
◆ 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: