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

Go to the source code of this file.

Classes

class  lbann::argmin_layer< TensorDataType, Layout, Device >
 Get index of minimum-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 82 of file argmin.hpp.

◆ PROTO

#define PROTO (   T)
Value:
extern template class argmin_layer<T, \
data_layout::DATA_PARALLEL, \
El::Device::CPU>

Definition at line 77 of file argmin.hpp.