LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
binary.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 
27 #ifndef LBANN_INCLUDE_LBANN_OPERATORS_MATH_BINARY_HPP_INCLUDED
28 #define LBANN_INCLUDE_LBANN_OPERATORS_MATH_BINARY_HPP_INCLUDED
29 
30 #include "lbann/operators/declare_stateless_op.hpp"
31 
32 namespace lbann {
33 
34 // Arithmetic operations
35 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Add, "add", false);
36 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Subtract, "subtract", false);
37 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Multiply, "multiply", true);
38 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Divide, "divide", true);
39 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Mod, "modulo", true);
40 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Pow, "power", true);
41 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(SafeDivide, "safe divide", true);
42 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(SquaredDifference,
43  "squared difference",
44  true);
45 
46 // Comparison operations
47 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Max, "maximum", true);
48 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Min, "minimum", true);
49 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Equal, "equal", false);
50 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(NotEqual, "not equal", false);
51 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Less, "less than", false);
52 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(LessEqual,
53  "less than or equal",
54  false);
55 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(Greater, "greater than", false);
56 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(GreaterEqual,
57  "greater than or equal",
58  false);
59 
60 // Logical operations
61 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(LogicalAnd, "logical and", false);
62 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(LogicalOr, "logical or", false);
63 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(LogicalXor, "logical xor", false);
64 
65 } // namespace lbann
66 
67 #endif // LBANN_INCLUDE_LBANN_OPERATORS_MATH_BINARY_HPP_INCLUDED
lbann::LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR
LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(LogSigmoid, "log sigmoid", true)
declare_stateless_op.hpp