LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
scatter.hpp File Reference
Include dependency graph for scatter.hpp:

Go to the source code of this file.

Classes

class  lbann::scatter_layer< TensorDataType, Layout, Device >
 Scatter values to specified tensor indices. More...
 

Namespaces

 lbann
 

Macros

#define PROTO_DEVICE(T, Device)   extern template class scatter_layer<T, data_layout::DATA_PARALLEL, Device>;
 

Functions

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

Macro Definition Documentation

◆ PROTO_DEVICE

#define PROTO_DEVICE (   T,
  Device 
)    extern template class scatter_layer<T, data_layout::DATA_PARALLEL, Device>;

Definition at line 519 of file scatter.hpp.