|
LBANN
0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
|
#include "lbann/layers/data_type_layer.hpp"#include "lbann/models/model.hpp"#include "lbann/utils/random.hpp"#include "lbann/macros/instantiate.hpp"Go to the source code of this file.
Classes | |
| class | lbann::categorical_random_layer< TensorDataType, T_layout, Dev > |
| Random categorical outputs. 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 95 of file categorical_random.hpp.
| #define PROTO | ( | T | ) |
Definition at line 90 of file categorical_random.hpp.