• Home
  • Raw
  • Download

Lines Matching full:layer

14 #include "Layer.hpp"
62 bool IsLayerSupported(const armnn::Layer* layer) in IsLayerSupported() argument
64 ARMNN_ASSERT(layer != nullptr); in IsLayerSupported()
66 armnn::LayerType layerType = layer->GetType(); in IsLayerSupported()
75 // Layer supported in IsLayerSupported()
78 // Layer unsupported in IsLayerSupported()
83 bool IsLayerSupported(const armnn::Layer& layer) in IsLayerSupported() argument
85 return IsLayerSupported(&layer); in IsLayerSupported()
88 bool IsLayerOptimizable(const armnn::Layer* layer) in IsLayerOptimizable() argument
90 ARMNN_ASSERT(layer != nullptr); in IsLayerOptimizable()
92 // A Layer is not optimizable if its name contains "unoptimizable" in IsLayerOptimizable()
93 const std::string layerName(layer->GetName()); in IsLayerOptimizable()
99 bool IsLayerOptimizable(const armnn::Layer& layer) in IsLayerOptimizable() argument
101 return IsLayerOptimizable(&layer); in IsLayerOptimizable()
176 [&](IConnectableLayer* layer) in OptimizeSubgraphView()
178 bool supported = IsLayerSupported(PolymorphicDowncast<Layer*>(layer)); in OptimizeSubgraphView()
181 // Layer supported, check if it's optimizable in OptimizeSubgraphView()
182 bool optimizable = IsLayerOptimizable(PolymorphicDowncast<Layer*>(layer)); in OptimizeSubgraphView()
185 // Layer fully supported in OptimizeSubgraphView()
186 supportedLayers.push_back(layer); in OptimizeSubgraphView()
190 // Layer supported but not optimizable in OptimizeSubgraphView()
191 untouchedLayers.push_back(layer); in OptimizeSubgraphView()
196 // Layer unsupported in OptimizeSubgraphView()
197 unsupportedLayers.push_back(layer); in OptimizeSubgraphView()
205 auto supportedSubgraphSelector = [](const Layer& layer) in OptimizeSubgraphView()
207 return layer.GetType() != LayerType::Input && in OptimizeSubgraphView()
208 layer.GetType() != LayerType::Output && in OptimizeSubgraphView()
209 IsLayerSupported(layer) && in OptimizeSubgraphView()
210 IsLayerOptimizable(layer); in OptimizeSubgraphView()
247 auto unsupportedSubgraphSelector = [](const Layer& layer) in OptimizeSubgraphView()
249 return layer.GetType() != LayerType::Input && in OptimizeSubgraphView()
250 layer.GetType() != LayerType::Output && in OptimizeSubgraphView()
251 !IsLayerSupported(layer); in OptimizeSubgraphView()
274 auto untouchedSubgraphSelector = [](const Layer& layer) in OptimizeSubgraphView()
276 return layer.GetType() != LayerType::Input && in OptimizeSubgraphView()
277 layer.GetType() != LayerType::Output && in OptimizeSubgraphView()
278 IsLayerSupported(layer) && in OptimizeSubgraphView()
279 !IsLayerOptimizable(layer); in OptimizeSubgraphView()