Home
last modified time | relevance | path

Searched refs:GetInputSlots (Results 1 – 15 of 15) sorted by relevance

/external/armnn/src/armnn/layers/
DSplitterLayer.cpp49 …const OutputHandler& outputHandler = GetInputSlots()[0].GetConnectedOutputSlot()->GetOutputHandler… in CreateTensors()
50 const OutputSlot* slot = GetInputSlots()[0].GetConnectedOutputSlot(); in CreateTensors()
/external/armnn/src/armnn/
DLayer.cpp257 for (auto&& inputSlot : GetInputSlots()) in CollectWorkloadInputs()
364 …LayerPriority parentPrio = std::accumulate(GetInputSlots().cbegin(), GetInputSlots().cend(), 0U, m… in GetPriority()
DLoadedNetwork.cpp60 for (auto&& input : layer.GetInputSlots()) in AddLayerStructure()
1117 …const OutputHandler& outputHandler = layer.GetInputSlots()[0].GetConnectedOutputSlot()->GetOutputH… in EnqueueOutput()
1132 (layer.GetInputSlots()[0].GetConnectedOutputSlot()->GetNumConnections() == 1)) in EnqueueOutput()
1134 …if(layer.GetInputSlots()[0].GetConnectedOutputSlot()->GetOwningLayer().GetType() != LayerType::Inp… in EnqueueOutput()
1651 const InputSlot& inputSlot = layer->GetInputSlots()[0]; in ImportOutputs()
2085 for (auto& slot : layer->GetInputSlots()) in CreateWorkingMemHandle()
2287 for (auto& inputSlot : layer->GetInputSlots()) in CreateMemoryProfileAsync()
2426 for (auto& inputSlot : layer->GetInputSlots()) in CreateMemoryProfile()
DGraph.cpp83 for (auto i : it->GetInputSlots()) in Print()
572 for (auto&& input : layer->GetInputSlots()) in InferTensorInfos()
DLayer.hpp245 const std::vector<InputSlot>& GetInputSlots() const { return m_InputSlots; } in GetInputSlots() function in armnn::Layer
DSubgraphViewSelector.cpp273 for (auto inputSlot : layer.GetInputSlots()) in ForEachLayerInput()
DSubgraphView.cpp233 const SubgraphView::InputSlots& SubgraphView::GetInputSlots() const in GetInputSlots() function in armnn::SubgraphView
DNetwork.cpp873 for (auto inputSlot : layer->GetInputSlots()) in AttemptBackendAssignment()
1501 for (auto&& inputSlot : layer.GetInputSlots()) in CalculateSlotOption()
/external/armnn/src/armnn/optimizations/
DConvertFp32NetworkToFp16.hpp32 … Layer& connectedLayer = layer.GetInputSlots()[0].GetConnectedOutputSlot()->GetOwningLayer(); in Run()
DFuseBatchNorm.hpp55 ARMNN_ASSERT_MSG(convLayer->GetInputSlots()[1].GetConnection() != nullptr, in Run()
134 ARMNN_ASSERT_MSG(convLayer->GetInputSlots()[2].GetConnection() != nullptr, in Run()
/external/armnn/src/backends/tosaCommon/
DTosaMappings.cpp102 for (auto inputSlot : layer->GetInputSlots()) in GetTosaMappingFromLayer()
/external/armnn/include/armnn/backends/
DSubgraphView.hpp93 const InputSlots& GetInputSlots() const;
/external/armnn/src/backends/backendsCommon/
DWorkloadFactory.cpp817 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfo); in IsLayerConfigurationSupported()
818 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfo); in IsLayerConfigurationSupported()
1211 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfo); in IsLayerConfigurationSupported()
1212 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfo); in IsLayerConfigurationSupported()
1242 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfoIn); in IsLayerConfigurationSupported()
1243 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfoIn); in IsLayerConfigurationSupported()
/external/armnn/src/backends/backendsCommon/test/
DOptimizeSubgraphViewTests.cpp1216 ConvertReferenceTypeToPointerType(layersInGraph.at("pooling1 layer")->GetInputSlots())), in PartiallySupportedSubgraphTestImpl()
1218 ConvertReferenceTypeToPointerType(layersInGraph.at("pooling3 layer")->GetInputSlots())) in PartiallySupportedSubgraphTestImpl()
1520 … ConvertReferenceTypeToPointerType(layersInGraph.at("conv1 layer")->GetInputSlots()[0])})[0], in PartiallyOptimizableSubgraphTestImpl2()
1522 … ConvertReferenceTypeToPointerType(layersInGraph.at("conv3 layer")->GetInputSlots()[0])})[0] in PartiallyOptimizableSubgraphTestImpl2()
/external/armnn/src/armnn/test/
DSubgraphViewTests.cpp268 CHECK(subgraph->GetInputSlots().size() == 1);
309 CHECK(subgraph->GetInputSlots() == subgraph2.GetInputSlots());
321 CHECK(subgraph->GetInputSlots() == subgraph3.GetInputSlots());
1567 for (const InputSlot& is : l->GetInputSlots())