inference-engine/tests/ngraph_functions/include/ngraph_functions/low_precision_transformations/max_pool_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 #include "common/fake_quantize_on_data.hpp"
  10 #include "low_precision/layer_transformation.hpp"
  11 #include "ngraph_functions/low_precision_transformations/common/dequantization_operations.hpp"
  12
  13 namespace ngraph {
  14 namespace builder {
  15 namespace subgraph {
  16
  17 class MaxPoolFunction {
  18 public:
  19     static std::shared_ptr<ngraph::Function> getOriginal(
  20         const ngraph::element::Type originalFunctionPrecision,
  21         const ngraph::Shape& inputShape,
  22         const FakeQuantizeOnData& fakeQuantizeOnData);
  23
  24     static std::shared_ptr<ngraph::Function> get(
  25         const ngraph::Shape& inputShape,
  26         const ngraph::element::Type precisionBeforeDequantization,
  27         const ngraph::builder::subgraph::DequantizationOperations& dequantizationBefore,
  28         const ngraph::builder::subgraph::DequantizationOperations& dequantizationAfter);
  29 };
  30
  31 }  // namespace subgraph
  32 }  // namespace builder
  33 }  // namespace ngraph