src/armnn/test/GraphUtils.hpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5 #pragma once
   6
   7 #include <armnn/Graph.hpp>
   8
   9 #include <string>
  10
  11 namespace
  12 {
  13
  14 bool GraphHasNamedLayer(const armnn::Graph& graph, const std::string& name)
  15 {
  16     for (auto&& layer : graph)
  17     {
  18         if (layer->GetName() == name)
  19         {
  20             return true;
  21         }
  22     }
  23     return false;
  24 }
  25 }