layers/activations/softmax.hpp File Reference

#include "lbann/layers/data_type_layer.hpp"
#include "lbann/proto/datatype_helpers.hpp"
#include "lbann/utils/dnn_enums.hpp"
#include "lbann/proto/layers.pb.h"
#include "lbann/utils/dnn_lib/softmax.hpp"
#include "lbann/macros/instantiate_device.hpp"

Include dependency graph for layers/activations/softmax.hpp:

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

Go to the source code of this file.

Classes
class	lbann::softmax_layer< TensorDataType, Layout, Device >

Namespaces
	lbann

Macros
#define	LBANN_ENABLE_SOFTMAX_THRESHOLD
#define	PROTO_DEVICE(T, Device)

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

Macro Definition Documentation

LBANN_ENABLE_SOFTMAX_THRESHOLD

#define LBANN_ENABLE_SOFTMAX_THRESHOLD

Definition at line 52 of file layers/activations/softmax.hpp.

PROTO_DEVICE

#define PROTO_DEVICE	(		T,
			Device
	)

Value:

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

Definition at line 265 of file layers/activations/softmax.hpp.