class ngraph::op::v0::Gelu

Overview

Gaussian Error Linear Unit f(x) = 0.5 * x * (1 + erf( x / sqrt(2) ) More…

#include <gelu.hpp>

class Gelu: public ngraph::op::util::FusedOp
{
public:
    // fields

    static constexpr NodeTypeInfo type_info {"Gelu", 0};

    // construction

    Gelu();
    Gelu(const Output<Node>& data);

    // methods

    const NodeTypeInfo& get_type_info() const;
    bool visit_attributes(AttributeVisitor& visitor);
    virtual OutputVector decompose_op() const;
    void pre_validate_and_infer_types();
    virtual std::shared_ptr<Node> clone_with_new_inputs(const OutputVector& new_args) const;
};

Detailed Documentation

Gaussian Error Linear Unit f(x) = 0.5 * x * (1 + erf( x / sqrt(2) )

Construction

Gelu(const Output<Node>& data)

Constructs a Gelu operation.

Parameters:

data

Input tensor