|
LBANN
0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
|
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) | |
| #define LBANN_INSTANTIATE_CPU_HALF |
Definition at line 82 of file argmin.hpp.
| #define PROTO | ( | T | ) |
Definition at line 77 of file argmin.hpp.