LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
Public Member Functions | List of all members
lbann::lbann_summary Class Reference

#include <summary.hpp>

Public Member Functions

 lbann_summary (std::string logdir, lbann_comm *comm)
 
void report_image (std::string const &tag, std::string const &img_format, CPUMat const &image, std::vector< int > const &dims, int step)
 
template<typename TensorDataType >
void reduce_mean (const std::string tag, const El::AbstractDistMatrix< TensorDataType > &mat, int step)
 
template<typename TensorDataType >
void reduce_min (const std::string tag, const El::AbstractDistMatrix< TensorDataType > &mat, int step)
 
template<typename TensorDataType >
void reduce_max (const std::string tag, const El::AbstractDistMatrix< TensorDataType > &mat, int step)
 
template<typename TensorDataType >
void reduce_stdev (const std::string tag, const El::AbstractDistMatrix< TensorDataType > &mat, int step)
 
template<typename TensorDataType >
void reduce_scalar (const std::string tag, TensorDataType s, int step)
 
template<typename TensorDataType >
void sum_reduce_scalar (const std::string tag, TensorDataType s, int step)
 
template<typename TensorDataType >
void reduce_scalar_all (const std::string tag, TensorDataType s, int step)
 
template<typename TensorDataType >
void reduce_histogram (const std::string tag, const El::AbstractDistMatrix< TensorDataType > &mat, int step)
 
template<typename TensorDataType >
void reduce_2norm (const std::string tag, const El::AbstractDistMatrix< TensorDataType > &mat, int step)
 
void flush ()
 

Detailed Description

Dummy class when TBinf is not present.

Definition at line 262 of file utils/summary.hpp.

Constructor & Destructor Documentation

◆ lbann_summary()

lbann::lbann_summary::lbann_summary ( std::string  logdir,
lbann_comm *  comm 
)
inline

Definition at line 265 of file utils/summary.hpp.

Member Function Documentation

◆ flush()

void lbann::lbann_summary::flush ( )
inline

Definition at line 313 of file utils/summary.hpp.

◆ reduce_2norm()

template<typename TensorDataType >
void lbann::lbann_summary::reduce_2norm ( const std::string  tag,
const El::AbstractDistMatrix< TensorDataType > &  mat,
int  step 
)
inline

Definition at line 309 of file utils/summary.hpp.

◆ reduce_histogram()

template<typename TensorDataType >
void lbann::lbann_summary::reduce_histogram ( const std::string  tag,
const El::AbstractDistMatrix< TensorDataType > &  mat,
int  step 
)
inline

Definition at line 304 of file utils/summary.hpp.

◆ reduce_max()

template<typename TensorDataType >
void lbann::lbann_summary::reduce_max ( const std::string  tag,
const El::AbstractDistMatrix< TensorDataType > &  mat,
int  step 
)
inline

Definition at line 285 of file utils/summary.hpp.

◆ reduce_mean()

template<typename TensorDataType >
void lbann::lbann_summary::reduce_mean ( const std::string  tag,
const El::AbstractDistMatrix< TensorDataType > &  mat,
int  step 
)
inline

Definition at line 275 of file utils/summary.hpp.

◆ reduce_min()

template<typename TensorDataType >
void lbann::lbann_summary::reduce_min ( const std::string  tag,
const El::AbstractDistMatrix< TensorDataType > &  mat,
int  step 
)
inline

Definition at line 280 of file utils/summary.hpp.

◆ reduce_scalar()

template<typename TensorDataType >
void lbann::lbann_summary::reduce_scalar ( const std::string  tag,
TensorDataType  s,
int  step 
)
inline

Definition at line 295 of file utils/summary.hpp.

◆ reduce_scalar_all()

template<typename TensorDataType >
void lbann::lbann_summary::reduce_scalar_all ( const std::string  tag,
TensorDataType  s,
int  step 
)
inline

Definition at line 301 of file utils/summary.hpp.

◆ reduce_stdev()

template<typename TensorDataType >
void lbann::lbann_summary::reduce_stdev ( const std::string  tag,
const El::AbstractDistMatrix< TensorDataType > &  mat,
int  step 
)
inline

Definition at line 290 of file utils/summary.hpp.

◆ report_image()

void lbann::lbann_summary::report_image ( std::string const &  tag,
std::string const &  img_format,
CPUMat const &  image,
std::vector< int > const &  dims,
int  step 
)
inline

Definition at line 267 of file utils/summary.hpp.

◆ sum_reduce_scalar()

template<typename TensorDataType >
void lbann::lbann_summary::sum_reduce_scalar ( const std::string  tag,
TensorDataType  s,
int  step 
)
inline

Definition at line 298 of file utils/summary.hpp.

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