Wilson Seok
8262aed30c
* Remove fp16 of Convert layer test from skip_tests.config.cpp as it works now * update repo * add sin asin template plugin reference test * add cos template plugin reference test and remove asin/sin/cos ngraph backend test * update CMakeList.txt * add template plugin reference test for asinh, cosh, sinh, tanh * remove ngraph backend test for asinh, cosh, sinh, tanh * update CMakeList.txt * add ngraph type prop test for tanh * add ngraph visitor api test for asin, sin * add atanh type in activation SLT * remove boolean from sin/cos/tan/asin/sinh/cosh evaluate
80 lines
2.5 KiB
C++
80 lines
2.5 KiB
C++
// Copyright (C) 2018-2021 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#include "ngraph/op/cosh.hpp"
|
|
|
|
#include "itt.hpp"
|
|
#include "ngraph/runtime/host_tensor.hpp"
|
|
#include "ngraph/runtime/reference/cosh.hpp"
|
|
#include "ngraph/validation_util.hpp"
|
|
|
|
using namespace std;
|
|
using namespace ngraph;
|
|
|
|
BWDCMP_RTTI_DEFINITION(op::v0::Cosh);
|
|
|
|
op::Cosh::Cosh(const Output<Node>& arg) : UnaryElementwiseArithmetic(arg) {
|
|
constructor_validate_and_infer_types();
|
|
}
|
|
|
|
bool op::Cosh::visit_attributes(AttributeVisitor& visitor) {
|
|
NGRAPH_OP_SCOPE(v0_Cosh_visit_attributes);
|
|
return true;
|
|
}
|
|
|
|
shared_ptr<Node> op::Cosh::clone_with_new_inputs(const OutputVector& new_args) const {
|
|
NGRAPH_OP_SCOPE(v0_Cosh_clone_with_new_inputs);
|
|
check_new_args_count(this, new_args);
|
|
return make_shared<Cosh>(new_args.at(0));
|
|
}
|
|
|
|
namespace coshop {
|
|
template <element::Type_t ET>
|
|
inline bool evaluate(const HostTensorPtr& arg0, const HostTensorPtr& out, const size_t count) {
|
|
using T = typename element_type_traits<ET>::value_type;
|
|
runtime::reference::cosh<T>(arg0->get_data_ptr<ET>(), out->get_data_ptr<ET>(), count);
|
|
return true;
|
|
}
|
|
|
|
bool evaluate_cosh(const HostTensorPtr& arg0, const HostTensorPtr& out, const size_t count) {
|
|
bool rc = true;
|
|
out->set_unary(arg0);
|
|
|
|
switch (arg0->get_element_type()) {
|
|
NGRAPH_TYPE_CASE(evaluate_cosh, i32, arg0, out, count);
|
|
NGRAPH_TYPE_CASE(evaluate_cosh, i64, arg0, out, count);
|
|
NGRAPH_TYPE_CASE(evaluate_cosh, u32, arg0, out, count);
|
|
NGRAPH_TYPE_CASE(evaluate_cosh, u64, arg0, out, count);
|
|
NGRAPH_TYPE_CASE(evaluate_cosh, f16, arg0, out, count);
|
|
NGRAPH_TYPE_CASE(evaluate_cosh, f32, arg0, out, count);
|
|
default:
|
|
rc = false;
|
|
break;
|
|
}
|
|
return rc;
|
|
}
|
|
} // namespace coshop
|
|
|
|
bool op::Cosh::evaluate(const HostTensorVector& outputs, const HostTensorVector& inputs) const {
|
|
NGRAPH_OP_SCOPE(v0_Cosh_evaluate);
|
|
NGRAPH_CHECK(validate_host_tensor_vector(outputs, 1) && validate_host_tensor_vector(inputs, 1));
|
|
return coshop::evaluate_cosh(inputs[0], outputs[0], shape_size(get_output_shape(0)));
|
|
}
|
|
|
|
bool op::Cosh::has_evaluate() const {
|
|
NGRAPH_OP_SCOPE(v0_Cosh_has_evaluate);
|
|
switch (get_input_element_type(0)) {
|
|
case ngraph::element::i32:
|
|
case ngraph::element::i64:
|
|
case ngraph::element::u32:
|
|
case ngraph::element::u64:
|
|
case ngraph::element::f16:
|
|
case ngraph::element::f32:
|
|
return true;
|
|
default:
|
|
break;
|
|
}
|
|
return false;
|
|
}
|