initializer.hpp File Reference

#include "lbann/base.hpp"
#include "lbann/utils/cloneable.hpp"
#include "lbann/utils/description.hpp"
#include <google/protobuf/message.h>
#include "lbann/macros/instantiate.hpp"

Include dependency graph for initializer.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	lbann::weights_initializer
	Scheme for initializing weight values. More...
class	lbann::data_type_weights_initializer< TensorDataType >
	Scheme for initializing weight values. More...
class	lbann::constant_initializer< TensorDataType >
	Fill weights with a single constant value. More...
class	lbann::value_initializer< TensorDataType >
	Fill weights with values from a list. More...
class	lbann::numpy_initializer< TensorDataType >
	Fill weights with values from a NumPy file. More...
class	lbann::uniform_initializer< TensorDataType >
	Draw weights values from a uniform random distribution. More...
class	lbann::normal_initializer< TensorDataType >
	Draw weights values from a normal random distribution. More...

Namespaces
	lbann_data
	lbann

Macros
#define	PROTO(T)
#define	LBANN_INSTANTIATE_CPU_HALF
#define	LBANN_INSTANTIATE_GPU_HALF

Functions
template<typename TensorDataType >
std::unique_ptr< weights_initializer >	lbann::build_constant_initializer_from_pbuf (google::protobuf::Message const &msg)
template<typename TensorDataType >
std::unique_ptr< weights_initializer >	lbann::build_value_initializer_from_pbuf (google::protobuf::Message const &msg)
template<typename TensorDataType >
std::unique_ptr< weights_initializer >	lbann::build_numpy_initializer_from_pbuf (google::protobuf::Message const &msg)
template<typename TensorDataType >
std::unique_ptr< weights_initializer >	lbann::build_uniform_initializer_from_pbuf (google::protobuf::Message const &msg)
template<typename TensorDataType >
std::unique_ptr< weights_initializer >	lbann::build_normal_initializer_from_pbuf (google::protobuf::Message const &msg)
	lbann::PROTO (float)
	lbann::PROTO (double)

Macro Definition Documentation

LBANN_INSTANTIATE_CPU_HALF

#define LBANN_INSTANTIATE_CPU_HALF

Definition at line 283 of file initializer.hpp.

LBANN_INSTANTIATE_GPU_HALF

#define LBANN_INSTANTIATE_GPU_HALF

Definition at line 284 of file initializer.hpp.

PROTO

#define PROTO

(

T

)

Value:

extern template class data_type_weights_initializer<T>;                      \
  extern template class constant_initializer<T>;                               \
  extern template class value_initializer<T>;                                  \
  extern template class numpy_initializer<T>;                                  \
  extern template class uniform_initializer<T>;                                \
  extern template class normal_initializer<T>

Definition at line 275 of file initializer.hpp.