LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
transform_builders.hpp
Go to the documentation of this file.
1 // Copyright (c) 2014-2023, Lawrence Livermore National Security, LLC.
3 // Produced at the Lawrence Livermore National Laboratory.
4 // Written by the LBANN Research Team (B. Van Essen, et al.) listed in
5 // the CONTRIBUTORS file. <lbann-dev@llnl.gov>
6 //
7 // LLNL-CODE-697807.
8 // All rights reserved.
9 //
10 // This file is part of LBANN: Livermore Big Artificial Neural Network
11 // Toolkit. For details, see http://software.llnl.gov/LBANN or
12 // https://github.com/LLNL/LBANN.
13 //
14 // Licensed under the Apache License, Version 2.0 (the "Licensee"); you
15 // may not use this file except in compliance with the License. You may
16 // obtain a copy of the License at:
17 //
18 // http://www.apache.org/licenses/LICENSE-2.0
19 //
20 // Unless required by applicable law or agreed to in writing, software
21 // distributed under the License is distributed on an "AS IS" BASIS,
22 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
23 // implied. See the License for the specific language governing
24 // permissions and limitations under the license.
26 #ifndef LBANN_LAYERS_TRANSFORM_TRANSFORM_BUILDERS_HPP_INCLUDED
27 #define LBANN_LAYERS_TRANSFORM_TRANSFORM_BUILDERS_HPP_INCLUDED
28 
29 #include "lbann/layers/layer.hpp"
30 
31 namespace lbann {
32 
33 LBANN_DEFINE_LAYER_BUILDER(batchwise_reduce_sum);
34 LBANN_DEFINE_LAYER_BUILDER(bernoulli);
35 LBANN_DEFINE_LAYER_BUILDER(categorical_random);
36 LBANN_DEFINE_LAYER_BUILDER(concatenate);
37 LBANN_DEFINE_LAYER_BUILDER(constant);
38 LBANN_DEFINE_LAYER_BUILDER(crop);
39 LBANN_DEFINE_LAYER_BUILDER(cross_grid_sum);
40 LBANN_DEFINE_LAYER_BUILDER(cross_grid_sum_slice);
41 LBANN_DEFINE_LAYER_BUILDER(discrete_random);
42 LBANN_DEFINE_LAYER_BUILDER(dummy);
43 LBANN_DEFINE_LAYER_BUILDER(evaluation);
44 LBANN_DEFINE_LAYER_BUILDER(gather);
45 LBANN_DEFINE_LAYER_BUILDER(gaussian);
46 LBANN_DEFINE_LAYER_BUILDER(hadamard);
47 LBANN_DEFINE_LAYER_BUILDER(identity_zero);
48 LBANN_DEFINE_LAYER_BUILDER(in_top_k);
49 LBANN_DEFINE_LAYER_BUILDER(permute);
50 LBANN_DEFINE_LAYER_BUILDER(pooling);
51 LBANN_DEFINE_LAYER_BUILDER(reduction);
52 LBANN_DEFINE_LAYER_BUILDER(scatter);
53 LBANN_DEFINE_LAYER_BUILDER(slice);
54 LBANN_DEFINE_LAYER_BUILDER(sort);
55 LBANN_DEFINE_LAYER_BUILDER(split);
56 LBANN_DEFINE_LAYER_BUILDER(stop_gradient);
57 LBANN_DEFINE_LAYER_BUILDER(sum);
58 LBANN_DEFINE_LAYER_BUILDER(tessellate);
59 LBANN_DEFINE_LAYER_BUILDER(uniform);
60 LBANN_DEFINE_LAYER_BUILDER(unpooling);
61 LBANN_DEFINE_LAYER_BUILDER(weighted_sum);
62 LBANN_DEFINE_LAYER_BUILDER(weights);
63 
64 } // namespace lbann
65 #endif // LBANN_LAYERS_TRANSFORM_TRANSFORM_BUILDERS_HPP_INCLUDED
lbann::LBANN_DEFINE_LAYER_BUILDER
LBANN_DEFINE_LAYER_BUILDER(elu)