inference-engine/tests/ngraph_functions/include/ngraph_functions/low_precision_transformations/round_function.hpp

   1 // Copyright (C) 2020 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #pragma once
   6
   7 #include <memory>
   8 #include <ngraph/ngraph.hpp>
   9
  10 #include "ngraph_functions/low_precision_transformations/common/dequantization_operations.hpp"
  11 #include "ngraph_functions/subgraph_builders.hpp"
  12
  13 namespace ngraph {
  14 namespace builder {
  15 namespace subgraph {
  16
  17 class RoundWithToleranceFunction {
  18 public:
  19     static std::shared_ptr<ngraph::Function> getOriginal(
  20         const ngraph::element::Type precision,
  21         const ngraph::Shape& inputShape,
  22         const ngraph::builder::subgraph::DequantizationOperations dequantization);
  23
  24     static std::shared_ptr<ngraph::Function> getReference(
  25         const ngraph::element::Type precision,
  26         const ngraph::Shape& inputShape,
  27         const ngraph::builder::subgraph::DequantizationOperations dequantization);
  28 };
  29
  30 }  // namespace subgraph
  31 }  // namespace builder
  32 }  // namespace ngraph