inference-engine/src/inference_engine/transform/transformation.hpp

   1 // Copyright (C) 2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #pragma once
   6
   7 #include <transform/transform_network.hpp>
   8 #include <string>
   9 #include <vector>
  10 #include <map>
  11
  12 namespace InferenceEngine {
  13 namespace Transform {
  14
  15 class Transformation {
  16     std::string name;
  17 public:
  18     std::string getName() const;
  19     void setName(const std::string& name);
  20     virtual ~Transformation() = default;
  21     virtual void execute(Network& network) = 0;
  22 };
  23
  24 }  // namespace Transform
  25 }  // namespace InferenceEngine