Home
last modified time | relevance | path

Searched refs:EdgeStrategy (Results 1 – 7 of 7) sorted by relevance

/external/armnn/src/armnn/
DGraph.cpp317 auto IsCompatibilityStrategy = [](EdgeStrategy strategy) in AddCompatibilityLayers()
319 return strategy == EdgeStrategy::CopyToTarget || in AddCompatibilityLayers()
320 strategy == EdgeStrategy::ExportToTarget; in AddCompatibilityLayers()
338 const std::vector<EdgeStrategy> srcEdgeStrategies = srcOutputSlot.GetEdgeStrategies(); in AddCompatibilityLayers()
344 EdgeStrategy strategy = srcEdgeStrategies[srcConnectionIndex]; in AddCompatibilityLayers()
345 ARMNN_ASSERT_MSG(strategy != EdgeStrategy::Undefined, in AddCompatibilityLayers()
361 if (strategy == EdgeStrategy::CopyToTarget) in AddCompatibilityLayers()
367 … ARMNN_ASSERT_MSG(strategy == EdgeStrategy::ExportToTarget, "Invalid edge strategy found."); in AddCompatibilityLayers()
417 compOutputSlot.SetEdgeStrategy(0, EdgeStrategy::DirectCompatibility); in AddCompatibilityLayers()
428 EdgeStrategy::DirectCompatibility); in AddCompatibilityLayers()
DLayer.cpp63 prevSlot->SetEdgeStrategy(armnn::numeric_cast<unsigned int>(idx), EdgeStrategy::Undefined); in Insert()
72 layer.GetOutputSlot(0).SetEdgeStrategy(0, EdgeStrategy::Undefined); in Insert()
116 m_EdgeStrategies.push_back(EdgeStrategy::Undefined); in Connect()
149 ARMNN_ASSERT_MSG(m_EdgeStrategies[0] == EdgeStrategy::Undefined, in MoveAllConnections()
210 void OutputSlot::SetEdgeStrategy(unsigned int connectionIndex, EdgeStrategy strategy) in SetEdgeStrategy()
215 EdgeStrategy OutputSlot::GetEdgeStrategyForConnection(unsigned int connectionIdx) const in GetEdgeStrategyForConnection()
DLayer.hpp133 const std::vector<EdgeStrategy>& GetEdgeStrategies() const { return m_EdgeStrategies; } in GetEdgeStrategies()
170 void SetEdgeStrategy(unsigned int connectionIndex, EdgeStrategy strategy);
171 EdgeStrategy GetEdgeStrategyForConnection(unsigned int connectionIdx) const;
181 std::vector<EdgeStrategy> m_EdgeStrategies;
DNetwork.cpp1613 EdgeStrategy CalculateEdgeStrategy(BackendsMap& backends, in CalculateEdgeStrategy()
1630 return EdgeStrategy::CopyToTarget; in CalculateEdgeStrategy()
1634 return EdgeStrategy::DirectCompatibility; in CalculateEdgeStrategy()
1642 return EdgeStrategy::DirectCompatibility; in CalculateEdgeStrategy()
1650 return EdgeStrategy::DirectCompatibility; in CalculateEdgeStrategy()
1679 return EdgeStrategy::ExportToTarget; in CalculateEdgeStrategy()
1693 return EdgeStrategy::CopyToTarget; in CalculateEdgeStrategy()
1698 return EdgeStrategy::Undefined; in CalculateEdgeStrategy()
1748 … EdgeStrategy strategy = CalculateEdgeStrategy(backends, slotOption, *layer, connectedLayer, in SelectTensorHandleStrategy()
1751 if (strategy == EdgeStrategy::Undefined) in SelectTensorHandleStrategy()
/external/armnn/src/armnn/test/
DGraphTests.cpp486 inputLayer->GetOutputSlot(0).SetEdgeStrategy(0, EdgeStrategy::DirectCompatibility); in InitialiseTestGraph()
487 convLayer1->GetOutputSlot(0).SetEdgeStrategy(0, EdgeStrategy::CopyToTarget); in InitialiseTestGraph()
488 convLayer1->GetOutputSlot(0).SetEdgeStrategy(1, EdgeStrategy::DirectCompatibility); in InitialiseTestGraph()
489 convLayer2->GetOutputSlot(0).SetEdgeStrategy(0, EdgeStrategy::CopyToTarget); in InitialiseTestGraph()
490 concatLayer->GetOutputSlot(0).SetEdgeStrategy(0, EdgeStrategy::DirectCompatibility); in InitialiseTestGraph()
491 actLayer->GetOutputSlot(0).SetEdgeStrategy(0, EdgeStrategy::DirectCompatibility); in InitialiseTestGraph()
492 softmaxLayer->GetOutputSlot(0).SetEdgeStrategy(0, EdgeStrategy::CopyToTarget); in InitialiseTestGraph()
562 inputLayer->GetOutputSlot(0).SetEdgeStrategy(0, armnn::EdgeStrategy::DirectCompatibility);
563 splitterLayer->GetOutputSlot(0).SetEdgeStrategy(0, armnn::EdgeStrategy::CopyToTarget);
564 splitterLayer->GetOutputSlot(1).SetEdgeStrategy(0, armnn::EdgeStrategy::CopyToTarget);
[all …]
DTensorHandleStrategyTest.cpp363 CHECK((inputLayerOut.GetEdgeStrategyForConnection(0) == EdgeStrategy::DirectCompatibility));
364 CHECK((softmaxLayer1Out.GetEdgeStrategyForConnection(0) == EdgeStrategy::DirectCompatibility));
365 CHECK((softmaxLayer2Out.GetEdgeStrategyForConnection(0) == EdgeStrategy::CopyToTarget));
366 CHECK((softmaxLayer3Out.GetEdgeStrategyForConnection(0) == EdgeStrategy::ExportToTarget));
367 CHECK((softmaxLayer4Out.GetEdgeStrategyForConnection(0) == EdgeStrategy::DirectCompatibility));
/external/armnn/include/armnn/backends/
DITensorHandleFactory.hpp104 enum class EdgeStrategy enum