evaluation.hpp File Reference

#include "lbann/layers/data_type_layer.hpp"
#include "lbann/macros/instantiate.hpp"
#include "lbann/macros/instantiate_device.hpp"

Include dependency graph for evaluation.hpp:

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

Go to the source code of this file.

Classes
class	lbann::abstract_evaluation_layer< TensorDataType >
	Interface with objective function and metrics. More...
class	lbann::evaluation_layer< TensorDataType, T_layout, Dev >

Namespaces
	lbann

Macros
#define	PROTO(T)   extern template class abstract_evaluation_layer<T>
#define	LBANN_INSTANTIATE_CPU_HALF
#define	LBANN_INSTANTIATE_GPU_HALF
#define	PROTO_DEVICE(T, Device)

Functions
	lbann::PROTO (float)
	lbann::PROTO (double)

Macro Definition Documentation

LBANN_INSTANTIATE_CPU_HALF

#define LBANN_INSTANTIATE_CPU_HALF

Definition at line 170 of file evaluation.hpp.

LBANN_INSTANTIATE_GPU_HALF

#define LBANN_INSTANTIATE_GPU_HALF

Definition at line 171 of file evaluation.hpp.

PROTO

#define PROTO

(

T

)

extern template class abstract_evaluation_layer<T>

Definition at line 168 of file evaluation.hpp.

PROTO_DEVICE

#define PROTO_DEVICE	(		T,
			Device
	)

Value:

extern template class evaluation_layer<T,                                    \
                                         data_layout::DATA_PARALLEL,           \
                                         Device>;                              \
  extern template class evaluation_layer<T, data_layout::MODEL_PARALLEL, Device>

Definition at line 177 of file evaluation.hpp.