argmax.hpp File Reference

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

Include dependency graph for argmax.hpp:

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

Go to the source code of this file.

Classes
class	lbann::argmax_layer< TensorDataType, Layout, Device >
	Get index of maximum-value tensor entry. More...

Namespaces
	lbann

Macros
#define	PROTO(T)
#define	LBANN_INSTANTIATE_CPU_HALF

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

Macro Definition Documentation

LBANN_INSTANTIATE_CPU_HALF

#define LBANN_INSTANTIATE_CPU_HALF

Definition at line 83 of file argmax.hpp.

PROTO

#define PROTO

(

T

)

Value:

extern template class argmax_layer<T,                                        \
                                     data_layout::DATA_PARALLEL,               \
                                     El::Device::CPU>

Definition at line 78 of file argmax.hpp.