revise atanh
This commit is contained in:
parent
797fbb3366
commit
5db2f9a3fe
@ -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.
|
||||
|
@ -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)
|
||||
|
@ -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
|
||||
|
9
ngraph/test/type_prop/atanh.cpp
Normal file
9
ngraph/test/type_prop/atanh.cpp
Normal 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);
|
12
ngraph/test/visitors/op/atanh.cpp
Normal file
12
ngraph/test/visitors/op/atanh.cpp
Normal 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);
|
Loading…
Reference in New Issue
Block a user