Searched refs:numOutputSlots (Results 1 – 12 of 12) sorted by relevance
| /external/armnn/src/armnn/ |
| D | NetworkUtils.cpp | 81 const unsigned int numOutputSlots = layer.GetNumOutputSlots(); in InsertConvertFp32ToFp16LayersAfter() local
84 convertLayers.reserve(numOutputSlots); in InsertConvertFp32ToFp16LayersAfter()
90 for (unsigned int slotIndex = 0u; slotIndex < numOutputSlots; ++slotIndex) in InsertConvertFp32ToFp16LayersAfter()
|
| D | Layer.cpp | 221 unsigned int numOutputSlots, in Layer() argument
225 : m_OutputHandlers(numOutputSlots) in Layer()
240 m_OutputSlots.reserve(numOutputSlots); in Layer()
241 for (unsigned int i = 0; i < numOutputSlots; ++i) in Layer()
248 unsigned int numOutputSlots, in Layer() argument
251 : Layer(numInputSlots, numOutputSlots, type, DataLayout::NCHW, name) in Layer()
|
| D | Layer.hpp | 221 … Layer(unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, const char* name);
222 …Layer(unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, DataLayout layout, …
460 unsigned int numOutputSlots, in BindableLayer() argument
464 : Layer(numInputSlots, numOutputSlots, type, name) in BindableLayer()
|
| D | SubgraphView.cpp | 185 unsigned int numOutputSlots = layer->GetNumOutputSlots(); local
186 m_OutputSlots.resize(numOutputSlots);
187 m_IOutputSlots.resize(numOutputSlots);
188 for (unsigned int i = 0; i < numOutputSlots; i++)
|
| D | Graph.cpp | 76 auto numOutputSlots = it->GetNumOutputSlots(); in Print() local
81 << " and " << numOutputSlots << " output slots."; in Print()
|
| /external/armnn/src/armnn/layers/ |
| D | LayerWithParameters.hpp | 31 unsigned int numOutputSlots, in LayerWithParameters() argument
35 : Layer(numInputSlots, numOutputSlots, type, name) in LayerWithParameters()
|
| D | ElementwiseBaseLayer.cpp | 17 unsigned int numOutputSlots, in ElementwiseBaseLayer() argument
20 : Layer(numInputSlots, numOutputSlots, type, name) in ElementwiseBaseLayer()
|
| D | ElementwiseBaseLayer.hpp | 37 …ElementwiseBaseLayer(unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, cons…
|
| D | SplitterLayer.cpp | 96 const unsigned int numOutputSlots = GetNumOutputSlots(); in CreateTensors() local
103 for (unsigned int it = 0; it < numOutputSlots; ++it) in CreateTensors()
|
| /external/armnn/src/armnn/test/ |
| D | GraphTests.cpp | 258 const unsigned int numOutputSlots = srcLayer->GetNumOutputSlots(); in GetEdgeList() local
259 for (unsigned int s = 0; s < numOutputSlots; ++s) in GetEdgeList()
|
| D | SubgraphViewTests.cpp | 1629 …unsigned int numOutputSlots = armnn::numeric_cast<unsigned int>(subgraphs[0]->GetIOutputSlots().si… variable
1632 CHECK((numOutputSlots == 1));
1639 PreCompiledDescriptor preCompiledDescriptor(numInputSlots, numOutputSlots);
|
| /external/armnn/include/armnn/ |
| D | Descriptors.hpp | 1336 PreCompiledDescriptor(unsigned int numInputSlots = 1u, unsigned int numOutputSlots = 1u) in PreCompiledDescriptor()
1337 : m_NumInputSlots(numInputSlots), m_NumOutputSlots(numOutputSlots) in PreCompiledDescriptor()
|