LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
entrywise.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_LOSS_ENTRYWISE_HPP_INCLUDED
28 #define LBANN_INCLUDE_LBANN_OPERATORS_LOSS_ENTRYWISE_HPP_INCLUDED
29 
31 
32 namespace lbann {
33 
34 // Cross entropy loss
36  "binary cross entropy",
37  true);
38 LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(SigmoidBinaryCrossEntropy,
39  "sigmoid binary cross entropy",
40  true);
41 
42 // Boolean loss functions
44  "Boolean accuracy",
45  false);
47  "Boolean false negative rate",
48  false);
50  "Boolean false positive rate",
51  false);
52 
53 } // namespace lbann
54 
55 #endif // LBANN_INCLUDE_LBANN_OPERATORS_LOSS_ENTRYWISE_HPP_INCLUDED
LBANN_DECLARE_STATELESS_ELEMENTWISE_OPERATOR(LogSigmoid, "log sigmoid", true)