LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
Elemental_extensions.hpp File Reference
#include "El.hpp"
Include dependency graph for Elemental_extensions.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

 El
 

Macros

#define LBANN_PROTO(F)
 
#define LBANN_PROTO_REAL(T)   LBANN_PROTO(T)
 
#define LBANN_PROTO_FLOAT   LBANN_PROTO_REAL(float)
 
#define LBANN_PROTO_DOUBLE   LBANN_PROTO_REAL(double)
 

Functions

template<typename F >
void El::ColumnSum (const Matrix< F > &X, Matrix< F > &sums)
 
template<typename F >
void El::ColumnSum (const AbstractMatrix< F > &X, AbstractMatrix< F > &sums)
 
template<typename F >
void El::ColumnSum (const AbstractDistMatrix< F > &A, AbstractDistMatrix< F > &sums)
 
template<typename F >
void El::RowSum (const Matrix< F > &X, Matrix< F > &sums)
 
template<typename F >
void El::RowSum (const AbstractMatrix< F > &X, AbstractMatrix< F > &sums)
 
template<typename F >
void El::RowSum (const AbstractDistMatrix< F > &A, AbstractDistMatrix< F > &sums)
 
template<typename F >
void El::ColumnSummaryStats (const Matrix< F > &X, F &sum, F &min, F &max, F &mean)
 

Macro Definition Documentation

◆ LBANN_PROTO

#define LBANN_PROTO (   F)
Value:
template void ColumnSum(const Matrix<F>& X, Matrix<F>& norms); \
template void RowSum(const Matrix<F>& X, Matrix<F>& norms); \
template void ColumnSummaryStats(const Matrix<F>& X, \
F& sum, \
F& min, \
F& max, \
F& mean); \
template void ColumnSum(const AbstractMatrix<F>& X, \
AbstractMatrix<F>& norms); \
template void RowSum(const AbstractMatrix<F>& X, AbstractMatrix<F>& norms); \
template void ColumnSum(const AbstractDistMatrix<F>& X, \
AbstractDistMatrix<F>& sums); \
template void RowSum(const AbstractDistMatrix<F>& X, \
AbstractDistMatrix<F>& sums);
void RowSum(const AbstractDistMatrix< F > &A, AbstractDistMatrix< F > &sums)
void ColumnSummaryStats(const Matrix< F > &X, F &sum, F &min, F &max, F &mean)
void ColumnSum(const AbstractDistMatrix< F > &A, AbstractDistMatrix< F > &sums)

Definition at line 52 of file Elemental_extensions.hpp.

◆ LBANN_PROTO_DOUBLE

#define LBANN_PROTO_DOUBLE   LBANN_PROTO_REAL(double)

Definition at line 80 of file Elemental_extensions.hpp.

◆ LBANN_PROTO_FLOAT

#define LBANN_PROTO_FLOAT   LBANN_PROTO_REAL(float)

Definition at line 77 of file Elemental_extensions.hpp.

◆ LBANN_PROTO_REAL

#define LBANN_PROTO_REAL (   T)    LBANN_PROTO(T)

Definition at line 73 of file Elemental_extensions.hpp.