19 #include "ngraph/node.hpp"
20 #include "ngraph/op/op.hpp"
35 static constexpr NodeTypeInfo type_info{
"Swish", 4};
36 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
44 Swish(
const Output<Node>& arg,
const Output<Node>& beta);
45 explicit Swish(
const Output<Node>& arg);
50 virtual std::shared_ptr<Node>
51 clone_with_new_inputs(
const OutputVector& new_args)
const override;
52 bool evaluate(
const HostTensorVector& outputs,
53 const HostTensorVector& inputs)
const override;
const NodeTypeInfo & get_type_info() const override
Definition: swish.hpp:36
Swish(const Output< Node > &arg, const Output< Node > &beta)
Constructs an Swish operation.
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
A Swish Activation Function f(x) = x / (1.0 + exp(-beta * x)) or f(x) = x * sigmoid(beta * x)
Definition: swish.hpp:33
Root of all actual ops.
Definition: op.hpp:29