revise atanh

This commit is contained in:
Yan, Xiping 2021-07-12 11:13:06 +08:00
parent 797fbb3366
commit 5db2f9a3fe
5 changed files with 26 additions and 3 deletions

View File

@ -19,8 +19,8 @@ namespace ngraph
class NGRAPH_API Atanh : public util::UnaryElementwiseArithmetic
{
public:
static constexpr NodeTypeInfo type_info{"Atanh", 3};
const NodeTypeInfo& get_type_info() const override { return type_info; }
NGRAPH_RTTI_DECLARATION;
/// \brief Constructs an Atanh operation.
Atanh() = default;
/// \brief Constructs an Atanh operation.

View File

@ -14,7 +14,7 @@
using namespace std;
using namespace ngraph;
constexpr NodeTypeInfo op::v3::Atanh::type_info;
NGRAPH_RTTI_DEFINITION(op::Atanh, "Atanh", 0, util::UnaryElementwiseArithmetic);
op::v3::Atanh::Atanh(const Output<Node>& arg)
: UnaryElementwiseArithmetic(arg)

View File

@ -95,6 +95,7 @@ set(SRC
type_prop/asinh.cpp
type_prop/assign.cpp
type_prop/atan.cpp
type_prop/atanh.cpp
type_prop/avg_pool.cpp
type_prop/batch_norm.cpp
type_prop/batch_to_space.cpp
@ -232,6 +233,7 @@ set(SRC
visitors/op/adaptive_max_pool.cpp
visitors/op/asinh.cpp
visitors/op/atan.cpp
visitors/op/atanh.cpp
visitors/op/batch_norm.cpp
visitors/op/batch_to_space.cpp
visitors/op/broadcast.cpp

View File

@ -0,0 +1,9 @@
// Copyright (C) 2021 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#include "unary_ops.hpp"
using Type = ::testing::Types<ngraph::op::Atanh>;
INSTANTIATE_TYPED_TEST_SUITE_P(type_prop_atanh, UnaryOperator, Type);

View File

@ -0,0 +1,12 @@
// Copyright (C) 2021 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#include "unary_ops.hpp"
using Types = ::testing::Types<UnaryOperatorType<ngraph::op::v3::Atanh, element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute,
UnaryOperatorVisitor,
Types,
UnaryOperatorTypeName);