text: // Copyright (C) 2018-2021 Intel Corporation
  // SPDX-License-Identifier: Apache-2.0
  //
  
  #include "ngraph/op/hswish.hpp"
  #include "itt.hpp"
  #include "ngraph/attribute_visitor.hpp"
  
  #include <ngraph/validation_util.hpp>
  #include "ngraph/runtime/host_tensor.hpp"
  #include "ngraph/runtime/reference/hswish.hpp"
  
  using namespace std;
  using namespace ngraph;
  
  NGRAPH_RTTI_DEFINITION(op::v4::HSwish, "HSwish", 4);
  
  op::v4::HSwish::HSwish(const Output<Node>& arg)
      : UnaryElementwiseArithmetic(arg)
  {
      constructor_validate_and_infer_types();
  }
  
  bool op::v4::HSwish::visit_attributes(AttributeVisitor& visitor)
  {
      NGRAPH_OP_SCOPE(v4_HSwish_visit_attributes);
      return true;
  }
  
  shared_ptr<Node> op::v4::HSwish::clone_with_new_inputs(const OutputVector& new_args) const
  {
      NGRAPH_OP_SCOPE(v4_HSwish_clone_with_new_inputs);
      return make_shared<op::v4::HSwish>(new_args.at(0));
  }
  
  namespace hswish
  {
      template <element::Type_t ET>
      inline bool evaluate(const HostTensorPtr& arg, const HostTensorPtr& out, const size_t count)
      {
          using T = typename element_type_traits<ET>::value_type;
  
          runtime::reference::hswish<T>(arg->get_data_ptr<ET>(), out->get_data_ptr<ET>(), count);
          return true;
      }
  
      bool evaluate_hswish(const HostTensorPtr& arg, const HostTensorPtr& out)
      {
          bool rc = true;
          size_t count = shape_size(arg->get_shape());
          out->set_unary(arg);
  
          switch (arg->get_element_type())
          {
              NGRAPH_TYPE_CASE(evaluate_hswish, bf16, arg, out, count);
              NGRAPH_TYPE_CASE(evaluate_hswish, f16, arg, out, count);
              NGRAPH_TYPE_CASE(evaluate_hswish, f32, arg, out, count);
          default: rc = false; break;
          }
          return rc;
      }
  }
  
  bool op::v4::HSwish::evaluate(const HostTensorVector& outputs, const HostTensorVector& inputs) const
  {
      NGRAPH_OP_SCOPE(v4_HSwish_evaluate);
      NGRAPH_CHECK(validate_host_tensor_vector(outputs, 1) && validate_host_tensor_vector(inputs, 1));
      return hswish::evaluate_hswish(inputs[0], outputs[0]);
  }
  