Home
last modified time | relevance | path

Searched refs:preCompiledLayer (Results 1 – 5 of 5) sorted by relevance

/external/armnn/src/armnn/test/
DCloneTests.cpp66 …armnn::Layer* const preCompiledLayer = graph1.AddLayer<armnn::PreCompiledLayer>(descriptor, "preCo… variable
67 …::PreCompiledLayer* layer = armnn::PolymorphicDowncast<armnn::PreCompiledLayer*>(preCompiledLayer);
89 …armnn::Layer* const preCompiledLayer = graph1.AddLayer<armnn::PreCompiledLayer>(descriptor, "preCo… variable
90 …::PreCompiledLayer* layer = armnn::PolymorphicDowncast<armnn::PreCompiledLayer*>(preCompiledLayer);
DSubgraphViewTests.cpp370 IConnectableLayer* const preCompiledLayer = variable
374 graph.SubstituteSubgraph(*subgraph, preCompiledLayer);
377 CHECK_EQ(preCompiledLayer->GetInputSlot(0).GetConnection(), subgraphInputConn);
378 CHECK_EQ(preCompiledLayer->GetOutputSlot(0).GetConnection(0), subgraphOutputConn);
413 IConnectableLayer* preCompiledLayer = network->AddPrecompiledLayer(preCompiledDescriptor, variable
418 graph.SubstituteSubgraph(*subgraph, preCompiledLayer);
421 CHECK_EQ(preCompiledLayer->GetInputSlot(0).GetConnection(), subgraphInputConn);
422 CHECK_EQ(preCompiledLayer->GetOutputSlot(0).GetConnection(0), subgraphOutputConn);
457 IConnectableLayer* preCompiledLayer = network->AddPrecompiledLayer(preCompiledDescriptor, variable
460 SubgraphView substituteSubgraph(preCompiledLayer);
[all …]
/external/armnn/src/backends/tosaReference/
DTosaRefBackend.cpp148 IConnectableLayer* preCompiledLayer = optimizationViews.GetINetwork()->AddPrecompiledLayer( in OptimizeSubgraphView() local
157preCompiledLayer->GetOutputSlot(i).SetTensorInfo(subgraph.GetIOutputSlot(i)->GetTensorInfo()); in OptimizeSubgraphView()
160 optimizationViews.AddSubstitution({ std::move(subgraph), SubgraphView(preCompiledLayer) }); in OptimizeSubgraphView()
/external/armnn/src/armnnTestUtils/
DMockBackend.cpp228 IConnectableLayer* preCompiledLayer = in OptimizeSubgraphView() local
237 SubgraphView replacementSubgraph(preCompiledLayer); in OptimizeSubgraphView()
DCreateWorkload.hpp2191 Layer* preCompiledLayer = nullptr; in CreatePreCompiledWorkloadTest() local
2196 preCompiledLayer = layer; in CreatePreCompiledWorkloadTest()
2199 CHECK(preCompiledLayer != nullptr); in CreatePreCompiledWorkloadTest()
2205 auto workload = MakeAndCheckWorkload<PreCompiledWorkload>(*preCompiledLayer, factory); in CreatePreCompiledWorkloadTest()