ngraph/test/runtime/pass/opset0_downgrade.hpp

   1 //*****************************************************************************
   2 // Copyright 2017-2020 Intel Corporation
   3 //
   4 // Licensed under the Apache License, Version 2.0 (the "License");
   5 // you may not use this file except in compliance with the License.
   6 // You may obtain a copy of the License at
   7 //
   8 //     http://www.apache.org/licenses/LICENSE-2.0
   9 //
  10 // Unless required by applicable law or agreed to in writing, software
  11 // distributed under the License is distributed on an "AS IS" BASIS,
  12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 // See the License for the specific language governing permissions and
  14 // limitations under the License.
  15 //*****************************************************************************
  16
  17 #pragma once
  18
  19 #include "backend_visibility.hpp"
  20 #include "ngraph/pass/pass.hpp"
  21
  22 NGRAPH_SUPPRESS_DEPRECATED_START
  23
  24 namespace ngraph
  25 {
  26     namespace pass
  27     {
  28         class BACKEND_API Opset0Downgrade : public NodePass
  29         {
  30         public:
  31             ///
  32             /// \brief    Constructor for the Opv1 downgrade transformation pass.
  33             ///
  34             /// \details  This transformation pass iterates over all nodes in a graph
  35             /// and updates version 1 ops to their version 0 equivalents.
  36             /// All ops in the final graph have op version 0.
  37             Opset0Downgrade() = default;
  38             bool run_on_node(std::shared_ptr<ngraph::Node> node) override;
  39         };
  40     }
  41 }
  42
  43 NGRAPH_SUPPRESS_DEPRECATED_END