Unverified Commit 6e4ab0b7 authored by Tomasz Socha's avatar Tomasz Socha Committed by GitHub

Add attribute visitor to Gelu op (#4387)

Co-authored-by: 's avatarSang Ik Lee <sang.ik.lee@intel.com>
Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
parent 5414b7ad
......@@ -37,6 +37,11 @@ op::Gelu::Gelu(const Output<Node>& data)
constructor_validate_and_infer_types();
}
bool ngraph::op::v0::Gelu::visit_attributes(AttributeVisitor& visitor)
{
return true;
}
// f(x) = 0.5 * x * (1.0 + erf( x / sqrt(2.0) )
NodeVector op::Gelu::decompose_op() const
{
......
......@@ -39,6 +39,7 @@ namespace ngraph
/// \param data Input tensor
Gelu(const Output<Node>& data);
bool visit_attributes(AttributeVisitor& visitor) override;
virtual NodeVector decompose_op() const override;
void pre_validate_and_infer_types() override;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment