LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
layers/regularizers/dropout.hpp File Reference
Include dependency graph for layers/regularizers/dropout.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  lbann::dropout< TensorDataType, T_layout, Dev >
 Probabilistically drop layer outputs. More...
 

Namespaces

 lbann
 

Macros

#define PROTO_DEVICE(T, Device)
 

Typedefs

template<typename T , data_layout L, El::Device D>
using lbann::dropout_layer = dropout< T, L, D >
 

Functions

 lbann::LBANN_DEFINE_LAYER_BUILDER (dropout)
 
 lbann::PROTO (float)
 
 lbann::PROTO (double)
 

Macro Definition Documentation

◆ PROTO_DEVICE

#define PROTO_DEVICE (   T,
  Device 
)
Value:
extern template class dropout<T, data_layout::DATA_PARALLEL, Device>; \
extern template class dropout<T, data_layout::MODEL_PARALLEL, Device>

Definition at line 252 of file layers/regularizers/dropout.hpp.