Home
last modified time | relevance | path

Searched refs:GetNumInputSlots (Results 1 – 25 of 38) sorted by relevance

12

/external/armnn/python/pyarmnn/test/
Dtest_iconnectable.py118 assert input1.GetNumInputSlots() == 0
122 assert input2.GetNumInputSlots() == 0
126 assert add.GetNumInputSlots() == 2
130 assert output.GetNumInputSlots() == 1
/external/armnn/src/armnn/
DLayer.cpp36 ARMNN_ASSERT(layer.GetNumInputSlots() == 2 || in AssertNumberOfInputSlots()
37 layer.GetNumInputSlots() == 3); in AssertNumberOfInputSlots()
42 ARMNN_ASSERT(layer.GetNumInputSlots() == 1); in AssertNumberOfInputSlots()
315 if (GetNumInputSlots() > 0) // Ignore the input layer. in GetDataType()
380 ARMNN_ASSERT(GetNumInputSlots() == expectedConnections); in VerifyLayerConnections()
399 ARMNN_ASSERT(GetNumInputSlots() != 0); in InferOutputShapes()
407 if (GetNumInputSlots() != GetNumOutputSlots()) in InferOutputShapes()
415 GetNumInputSlots(), in InferOutputShapes()
DGraph.cpp75 auto numInputSlots = it->GetNumInputSlots(); in Print()
157 for (unsigned int i=0;i<layer->GetNumInputSlots(); i++) in SerializeToDot()
621 if (slotIndex == 1 && layer->GetNumInputSlots() == 3) in ConstructErrorMessageForUnconnectedInputs()
DNetworkUtils.cpp45 convertLayers.reserve(layer.GetNumInputSlots()); in InsertConvertFp16ToFp32LayersBefore()
DSubgraphView.cpp176 unsigned int numInputSlots = layer->GetNumInputSlots();
293 unsigned int SubgraphView::GetNumInputSlots() const in GetNumInputSlots() function in armnn::SubgraphView
DLayer.hpp321 …unsigned int GetNumInputSlots() const override { return static_cast<unsigned int>(m_InputSlots.siz… in GetNumInputSlots() function in armnn::Layer
/external/armnn/src/armnn/test/
DNetworkTests.cpp19 for (unsigned int i = 0; i < layer.GetNumInputSlots(); ++i) in AreAllLayerInputSlotsConnected()
182 CHECK(srcLayer->GetNumInputSlots() == expectedSrcNumInputs); in __anonb52bfdf10202()
184 CHECK(tgtLayer->GetNumInputSlots() == 1); in __anonb52bfdf10202()
197 CHECK(srcLayer->GetNumInputSlots() == expectedSrcNumInputs); in __anonb52bfdf10302()
199 CHECK(tgtLayer->GetNumInputSlots() == 2); in __anonb52bfdf10302()
217 CHECK(srcLayer1->GetNumInputSlots() == expectedSrcNumInputs1); in __anonb52bfdf10402()
219 CHECK(srcLayer2->GetNumInputSlots() == expectedSrcNumInputs2); in __anonb52bfdf10402()
221 CHECK(tgtLayer->GetNumInputSlots() == 2); in __anonb52bfdf10402()
306 CHECK(concatLayer->GetNumInputSlots() == 2);
426 CHECK(layer->GetNumInputSlots() == 1); in ExecuteStrategy()
[all …]
/external/armnn/src/armnn/layers/
DStackLayer.cpp69 GetNumInputSlots()); in ValidateTensorShapesFromInputs()
79 for (unsigned int i = 0; i < GetNumInputSlots(); ++i) in ValidateTensorShapesFromInputs()
DConcatLayer.cpp69 const unsigned int numInputSlots = currentLayer->GetNumInputSlots(); in CreateTensors()
300 GetNumInputSlots()); in ValidateTensorShapesFromInputs()
309 for (unsigned int i = 0; i < GetNumInputSlots(); ++i) in ValidateTensorShapesFromInputs()
DLayerWithParameters.hpp62 for (unsigned int i = 0; i < GetNumInputSlots(); ++i) in GetConnectedConstantAsInputTensors()
/external/armnn/delegate/classic/src/
DMultiLayerFacade.hpp56 virtual unsigned int GetNumInputSlots() const override in GetNumInputSlots() function in armnnDelegate::MultiLayerFacade
58 return m_FirstLayer->GetNumInputSlots(); in GetNumInputSlots()
DTranspose.hpp97 …ARMNN_ASSERT(transposeLayer->GetNumInputSlots() == 1); // permutation vector given to descript… in VisitTransposeOperator()
DClassicDelegateUtils.hpp161 for (unsigned int inputIndex = 0; inputIndex < layer->GetNumInputSlots(); ++inputIndex) in Connect()
486 for (unsigned int inputIndex = 0; inputIndex < layer->GetNumInputSlots(); ++inputIndex) in ProcessInputs()
/external/armnn/delegate/opaque/src/
DOpaqueDelegateUtils.hpp183 if(numInputs != static_cast<int>(layer->GetNumInputSlots())) in Connect()
191 for (unsigned int inputIndex = 0; inputIndex < layer->GetNumInputSlots(); ++inputIndex) in Connect()
592 …for (int32_t inputIndex = 0; inputIndex < static_cast<int32_t>(layer->GetNumInputSlots()); ++input… in ProcessInputs()
/external/armnn/src/armnnTestUtils/
DGraphUtils.cpp36 return layer->GetNumInputSlots() == num; in CheckNumberOfInputSlot()
DMockBackend.cpp230 … PreCompiledDescriptor(supportedSubgraph->GetNumInputSlots(), in OptimizeSubgraphView()
/external/armnn/src/backends/backendsCommon/
DSubgraphUtils.hpp25 for (unsigned int i = 0 ; i < layer->GetNumInputSlots(); ++i) in CreateIInputsFrom()
/external/armnn/src/backends/neon/test/
DNeonTensorHandleTests.cpp153 for (unsigned int i = 0; i < layer->GetNumInputSlots(); ++i)
236 for (unsigned int i = 0; i < layer->GetNumInputSlots(); ++i)
405 for (unsigned int i = 0; i < layer->GetNumInputSlots(); ++i)
582 for (unsigned int i = 0; i < layer->GetNumInputSlots(); ++i)
/external/armnn/include/armnn/backends/
DSubgraphView.hpp124 unsigned int GetNumInputSlots() const;
/external/armnn/src/backends/tosaReference/
DTosaRefBackend.cpp149 PreCompiledDescriptor(subgraph.GetNumInputSlots(), subgraph.GetNumOutputSlots()), in OptimizeSubgraphView()
/external/armnn/src/armnnSerializer/test/
DSerializerTestUtils.cpp44 CHECK(layer->GetNumInputSlots() == m_InputTensorInfos.size()); in VerifyNameAndConnections()
/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
Darmnn_network.i261 ") GetNumInputSlots;
262 unsigned int GetNumInputSlots();
/external/armnn/src/armnnTfLiteParser/test/
DUnsupported.cpp44 CHECK(layer->GetNumInputSlots() == numInputs); in ExecuteStrategy()
/external/armnn/src/armnn/optimizations/
DFoldPadIntoLayer2d.hpp166 for (unsigned int i = 1; i < layer2d.GetNumInputSlots(); ++i) in FoldPadIntoLayer2dImpl()
DFuseBatchNorm.hpp173 if (newConv2dLayer.GetNumInputSlots() > 1) in Run()

12