sigmoid.hpp
1 // Copyright (C) 2018-2021 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
3 //
4 
5 #pragma once
6 
7 #include "ngraph/op/op.hpp"
8 #include "ngraph/op/util/binary_elementwise_arithmetic.hpp"
9 #include "ngraph/op/util/unary_elementwise_arithmetic.hpp"
10 #include "ngraph/util.hpp"
11 
12 namespace ngraph
13 {
14  namespace op
15  {
16  namespace v0
17  {
18  class NGRAPH_API Sigmoid : public util::UnaryElementwiseArithmetic
19  {
20  public:
21  static constexpr NodeTypeInfo type_info{"Sigmoid", 0};
22  const NodeTypeInfo& get_type_info() const override { return type_info; }
23  Sigmoid(const Output<Node>& arg);
24  Sigmoid() = default;
25  virtual std::shared_ptr<Node>
26  clone_with_new_inputs(const OutputVector& new_args) const override;
27  bool evaluate(const HostTensorVector& outputs,
28  const HostTensorVector& inputs) const override;
29  bool has_evaluate() const override;
30  };
31  } // namespace v0
32  using v0::Sigmoid;
33  } // namespace op
34 } // namespace ngraph
A handle for one of a node's outputs.
Definition: node_output.hpp:33
Abstract base class for elementwise unary arithmetic operations, i.e., operations where the same scal...
Definition: unary_elementwise_arithmetic.hpp:37
Definition: sigmoid.hpp:19
const NodeTypeInfo & get_type_info() const override
Definition: sigmoid.hpp:22
bool has_evaluate() const override
Allows to get information about availability of evaluate method for the current operation.
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:16
Definition: type.hpp:27