Lines Matching full:layer
129 throw ParseException(fmt::format("{0} was called with an invalid layer index. " in CheckLayers()
130 "layers:{1} layer:{2} at {3}", in CheckLayers()
289 case Layer::Layer_AbsLayer: in GetBaseLayer()
291 case Layer::Layer_ActivationLayer: in GetBaseLayer()
293 case Layer::Layer_AdditionLayer: in GetBaseLayer()
295 case Layer::Layer_ArgMinMaxLayer: in GetBaseLayer()
297 case Layer::Layer_BatchMatMulLayer: in GetBaseLayer()
299 case Layer::Layer_BatchToSpaceNdLayer: in GetBaseLayer()
301 case Layer::Layer_BatchNormalizationLayer: in GetBaseLayer()
303 case Layer::Layer_CastLayer: in GetBaseLayer()
305 case Layer::Layer_ChannelShuffleLayer: in GetBaseLayer()
307 case Layer::Layer_ComparisonLayer: in GetBaseLayer()
309 case Layer::Layer_ConcatLayer: in GetBaseLayer()
311 case Layer::Layer_ConstantLayer: in GetBaseLayer()
313 case Layer::Layer_Convolution2dLayer: in GetBaseLayer()
315 case Layer::Layer_Convolution3dLayer: in GetBaseLayer()
317 case Layer::Layer_DepthToSpaceLayer: in GetBaseLayer()
319 case Layer::Layer_DepthwiseConvolution2dLayer: in GetBaseLayer()
321 case Layer::Layer_DequantizeLayer: in GetBaseLayer()
323 case Layer::Layer_DetectionPostProcessLayer: in GetBaseLayer()
325 case Layer::Layer_DivisionLayer: in GetBaseLayer()
327 case Layer::Layer_EqualLayer: in GetBaseLayer()
329 case Layer::Layer_ElementwiseBinaryLayer: in GetBaseLayer()
331 case Layer::Layer_ElementwiseUnaryLayer: in GetBaseLayer()
333 case Layer::Layer_FullyConnectedLayer: in GetBaseLayer()
335 case Layer::Layer_FillLayer: in GetBaseLayer()
337 case Layer::Layer_FloorLayer: in GetBaseLayer()
339 case Layer::Layer_GatherLayer: in GetBaseLayer()
341 case Layer::Layer_GatherNdLayer: in GetBaseLayer()
343 case Layer::Layer_GreaterLayer: in GetBaseLayer()
345 case Layer::Layer_InputLayer: in GetBaseLayer()
347 case Layer::Layer_InstanceNormalizationLayer: in GetBaseLayer()
349 case Layer::Layer_L2NormalizationLayer: in GetBaseLayer()
351 case Layer::Layer_LogicalBinaryLayer: in GetBaseLayer()
353 case Layer::Layer_LogSoftmaxLayer: in GetBaseLayer()
355 case Layer::Layer_LstmLayer: in GetBaseLayer()
357 case Layer::Layer_MeanLayer: in GetBaseLayer()
359 case Layer::Layer_MinimumLayer: in GetBaseLayer()
361 case Layer::Layer_MaximumLayer: in GetBaseLayer()
363 case Layer::Layer_MergeLayer: in GetBaseLayer()
365 case Layer::Layer_MergerLayer: in GetBaseLayer()
367 case Layer::Layer_MultiplicationLayer: in GetBaseLayer()
369 case Layer::Layer_NormalizationLayer: in GetBaseLayer()
371 case Layer::Layer_OutputLayer: in GetBaseLayer()
373 case Layer::Layer_PadLayer: in GetBaseLayer()
375 case Layer::Layer_PermuteLayer: in GetBaseLayer()
377 case Layer::Layer_Pooling2dLayer: in GetBaseLayer()
379 case Layer::Layer_Pooling3dLayer: in GetBaseLayer()
381 case Layer::Layer_PreluLayer: in GetBaseLayer()
383 case Layer::Layer_QLstmLayer: in GetBaseLayer()
385 case Layer::Layer_QuantizeLayer: in GetBaseLayer()
387 case Layer::Layer_QuantizedLstmLayer: in GetBaseLayer()
389 case Layer::Layer_RankLayer: in GetBaseLayer()
391 case Layer::Layer_ReduceLayer: in GetBaseLayer()
393 case Layer::Layer_ReshapeLayer: in GetBaseLayer()
395 case Layer::Layer_ResizeBilinearLayer: in GetBaseLayer()
397 case Layer::Layer_ResizeLayer: in GetBaseLayer()
399 case Layer::Layer_RsqrtLayer: in GetBaseLayer()
401 case Layer::Layer_ShapeLayer: in GetBaseLayer()
403 case Layer::Layer_SliceLayer: in GetBaseLayer()
405 case Layer::Layer_SoftmaxLayer: in GetBaseLayer()
407 case Layer::Layer_SpaceToBatchNdLayer: in GetBaseLayer()
409 case Layer::Layer_SpaceToDepthLayer: in GetBaseLayer()
411 case Layer::Layer_SplitterLayer: in GetBaseLayer()
413 case Layer::Layer_StackLayer: in GetBaseLayer()
415 case Layer::Layer_StandInLayer: in GetBaseLayer()
417 case Layer::Layer_StridedSliceLayer: in GetBaseLayer()
419 case Layer::Layer_SubtractionLayer: in GetBaseLayer()
421 case Layer::Layer_SwitchLayer: in GetBaseLayer()
423 case Layer::Layer_TransposeConvolution2dLayer: in GetBaseLayer()
425 case Layer::Layer_TransposeLayer: in GetBaseLayer()
427 case Layer::Layer_UnidirectionalSequenceLstmLayer: in GetBaseLayer()
429 case Layer::Layer_NONE: in GetBaseLayer()
431 throw ParseException(fmt::format("Layer type {} not recognized", layerType)); in GetBaseLayer()
437 auto layer = GetBaseLayer(graph, index); in GetLayerName() local
438 assert(layer); in GetLayerName()
439 return layer->layerName()->str(); in GetLayerName()
446 if (layerType == Layer::Layer_InputLayer) in GetBindingLayerInfo()
450 else if ( layerType == Layer::Layer_OutputLayer ) in GetBindingLayerInfo()
794 auto layer = GetBaseLayer(graphPtr, layerIndex); in GetInputs() local
795 const auto& numInputs = layer->inputSlots()->size(); in GetInputs()
802 … (layer->inputSlots()->Get(i)->connection()->sourceLayerIndex())); in GetInputs()
811 auto layer = GetBaseLayer(graphPtr, layerIndex); in GetOutputs() local
812 const auto& numOutputs = layer->outputSlots()->size(); in GetOutputs()
818 result[i] = layer->outputSlots()->Get(i)->tensorInfo(); in GetOutputs()
827 throw ParseException(fmt::format("Layer not supported. layerIndex: {0} " in ParseUnsupportedLayer()
888 for (AnyLayer const* layer : *graph->layers()) in CreateNetworkFromGraph()
890 if (layer->layer_type() != Layer_InputLayer && in CreateNetworkFromGraph()
891 layer->layer_type() != Layer_OutputLayer) in CreateNetworkFromGraph()
894 auto& parserFunction = m_ParserFunctions[layer->layer_type()]; in CreateNetworkFromGraph()
903 // establish the connections from the layer outputs to the inputs of the subsequent layers in CreateNetworkFromGraph()
935 throw ParseException(fmt::format("No input binding found for layer:{0} / {1}", in GetNetworkInputBindingInfo()
951 throw ParseException(fmt::format("No output binding found for layer:{0} / {1}", in GetNetworkOutputBindingInfo()
960 auto layer = graph->layers()->Get(i); in GetInputLayerInVector() local
961 if (layer->layer_type() == Layer::Layer_InputLayer) in GetInputLayerInVector()
963 auto layerBindingId = layer->layer_as_InputLayer()->base()->layerBindingId(); in GetInputLayerInVector()
970 throw ParseException("Input layer with given layerBindingId not found"); in GetInputLayerInVector()
977 auto layer = graph->layers()->Get(i); in GetOutputLayerInVector() local
978 if (layer->layer_type() == Layer::Layer_OutputLayer) in GetOutputLayerInVector()
980 auto layerBindingId = layer->layer_as_OutputLayer()->base()->layerBindingId(); in GetOutputLayerInVector()
987 throw ParseException("Output layer with given layerBindingId not found"); in GetOutputLayerInVector()
994 LayerBaseRawPtr layer = GetBaseLayer(graph, i); in GetLayerIndexInVector() local
995 if (layer->index() == targetIndex) in GetLayerIndexInVector()
1000 throw ParseException("Layer with given index not found"); in GetLayerIndexInVector()
1040 …etBindingLayerInfo expect the index to be index in the vector not index property on each layer base in SetupInputLayers()
1079 …etBindingLayerInfo expect the index to be index in the vector not index property on each layer base in SetupOutputLayers()
1101 IConnectableLayer* layer) in RegisterOutputSlots() argument
1104 ARMNN_ASSERT(layer != nullptr); in RegisterOutputSlots()
1106 if (baseLayer->outputSlots()->size() != layer->GetNumOutputSlots()) in RegisterOutputSlots()
1109 " for layer index: {2} {3}", in RegisterOutputSlots()
1111 layer->GetNumOutputSlots(), in RegisterOutputSlots()
1116 for (unsigned int i = 0; i < layer->GetNumOutputSlots(); ++i) in RegisterOutputSlots()
1119 armnn::IOutputSlot* outputSlot = &(layer->GetOutputSlot(slotIndex)); in RegisterOutputSlots()
1127 armnn::IConnectableLayer* layer, in RegisterInputSlots() argument
1131 ARMNN_ASSERT(layer != nullptr); in RegisterInputSlots()
1134 if (baseLayer->inputSlots()->size() != (layer->GetNumInputSlots() - ignoreSlots.size())) in RegisterInputSlots()
1137 " for layer index:{2} {3}", in RegisterInputSlots()
1139 layer->GetNumInputSlots(), in RegisterInputSlots()
1144 for (unsigned int i = 0; i < layer->GetNumInputSlots(); ++i) in RegisterInputSlots()
1151 armnn::IInputSlot* inputSlot = &(layer->GetInputSlot(fbInputSlot->index())); in RegisterInputSlots()
1208 IConnectableLayer* layer = m_Network->AddElementwiseUnaryLayer(descriptor, layerName.c_str()); in ParseAbs() local
1210 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseAbs()
1212 RegisterInputSlots(graph, layerIndex, layer); in ParseAbs()
1213 RegisterOutputSlots(graph, layerIndex, layer); in ParseAbs()
1235 IConnectableLayer* layer = m_Network->AddActivationLayer(descriptor, in ParseActivation() local
1238 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseActivation()
1240 RegisterInputSlots(graph, layerIndex, layer); in ParseActivation()
1241 RegisterOutputSlots(graph, layerIndex, layer); in ParseActivation()
1256 IConnectableLayer* layer = m_Network->AddElementwiseBinaryLayer(descriptor, layerName.c_str()); in ParseAdd() local
1259 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseAdd()
1261 RegisterInputSlots(graph, layerIndex, layer); in ParseAdd()
1262 RegisterOutputSlots(graph, layerIndex, layer); in ParseAdd()
1282 IConnectableLayer* layer = m_Network->AddArgMinMaxLayer(descriptor, layerName.c_str()); in ParseArgMinMax() local
1285 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseArgMinMax()
1287 RegisterInputSlots(graph, layerIndex, layer); in ParseArgMinMax()
1288 RegisterOutputSlots(graph, layerIndex, layer); in ParseArgMinMax()
1313 IConnectableLayer* layer = m_Network->AddBatchMatMulLayer(descriptor, layerName.c_str()); in ParseBatchMatMul() local
1316 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseBatchMatMul()
1318 RegisterInputSlots(graph, layerIndex, layer); in ParseBatchMatMul()
1319 RegisterOutputSlots(graph, layerIndex, layer); in ParseBatchMatMul()
1355 IConnectableLayer* layer = m_Network->AddBatchToSpaceNdLayer(descriptor, layerName.c_str()); in ParseBatchToSpaceNd() local
1358 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseBatchToSpaceNd()
1360 RegisterInputSlots(graph, layerIndex, layer); in ParseBatchToSpaceNd()
1361 RegisterOutputSlots(graph, layerIndex, layer); in ParseBatchToSpaceNd()
1389 IConnectableLayer* layer = m_Network->AddBatchNormalizationLayer(descriptor, in ParseBatchNormalization() local
1395 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseBatchNormalization()
1397 RegisterInputSlots(graph, layerIndex, layer); in ParseBatchNormalization()
1398 RegisterOutputSlots(graph, layerIndex, layer); in ParseBatchNormalization()
1413 IConnectableLayer* layer = m_Network->AddCastLayer(layerName.c_str()); in ParseCast() local
1416 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseCast()
1418 RegisterInputSlots(graph, layerIndex, layer); in ParseCast()
1419 RegisterOutputSlots(graph, layerIndex, layer); in ParseCast()
1436 IConnectableLayer* layer; in ParseConstant() local
1438 // Required for when Constant Layer is used as an inputs to DepthwiseConvolution2d Layer. in ParseConstant()
1464 layer = m_Network->AddConstantLayer(weightsPermuted, layerName.c_str()); in ParseConstant()
1466 layer->GetOutputSlot(0).SetTensorInfo(weightsPermuted.GetInfo()); in ParseConstant()
1468 RegisterOutputSlots(graph, layerIndex, layer); in ParseConstant()
1474 layer = m_Network->AddConstantLayer(input, layerName.c_str()); in ParseConstant()
1478 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseConstant()
1481 RegisterOutputSlots(graph, layerIndex, layer); in ParseConstant()
1510 armnn::IConnectableLayer* layer; in ParseConvolution2d() local
1514 …// Weights and biases used to be always constant and were stored as members of the layer. This has… in ParseConvolution2d()
1518 … // If the model stores weights and biases as members of the layer we have to read them from there in ParseConvolution2d()
1522 layer = m_Network->AddConvolution2dLayer(descriptor, in ParseConvolution2d()
1527 weightsLayer->GetOutputSlot(0).Connect(layer->GetInputSlot(1u)); in ParseConvolution2d()
1535 biasLayer->GetOutputSlot(0).Connect(layer->GetInputSlot(2u)); in ParseConvolution2d()
1542 layer = m_Network->AddConvolution2dLayer(descriptor, in ParseConvolution2d()
1549 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseConvolution2d()
1551 RegisterInputSlots(graph, layerIndex, layer, ignoreSlots); in ParseConvolution2d()
1552 RegisterOutputSlots(graph, layerIndex, layer); in ParseConvolution2d()
1587 IConnectableLayer* layer = m_Network->AddConvolution3dLayer(descriptor, layerName.c_str()); in ParseConvolution3d() local
1590 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseConvolution3d()
1592 RegisterInputSlots(graph, layerIndex, layer); in ParseConvolution3d()
1593 RegisterOutputSlots(graph, layerIndex, layer); in ParseConvolution3d()
1613 IConnectableLayer* layer = m_Network->AddDepthToSpaceLayer(descriptor, layerName.c_str()); in ParseDepthToSpace() local
1616 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseDepthToSpace()
1618 RegisterInputSlots(graph, layerIndex, layer); in ParseDepthToSpace()
1619 RegisterOutputSlots(graph, layerIndex, layer); in ParseDepthToSpace()
1647 IConnectableLayer* layer; in ParseDepthwiseConvolution2d() local
1650 …// Weights and biases used to be always constant and were stored as members of the layer. This has… in ParseDepthwiseConvolution2d()
1656 … // If the model stores weights and biases as members of the layer we have to read them from there in ParseDepthwiseConvolution2d()
1661 layer = m_Network->AddDepthwiseConvolution2dLayer(descriptor, in ParseDepthwiseConvolution2d()
1671 biasLayer->GetOutputSlot(0).Connect(layer->GetInputSlot(2u)); in ParseDepthwiseConvolution2d()
1697 weightsLayer->GetOutputSlot(0).Connect(layer->GetInputSlot(1u)); in ParseDepthwiseConvolution2d()
1703 weightsLayer->GetOutputSlot(0).Connect(layer->GetInputSlot(1u)); in ParseDepthwiseConvolution2d()
1709 layer = m_Network->AddDepthwiseConvolution2dLayer(descriptor, in ParseDepthwiseConvolution2d()
1716 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseDepthwiseConvolution2d()
1718 RegisterInputSlots(graph, layerIndex, layer, ignoreSlots); in ParseDepthwiseConvolution2d()
1719 RegisterOutputSlots(graph, layerIndex, layer); in ParseDepthwiseConvolution2d()
1751 IConnectableLayer* layer = m_Network->AddDetectionPostProcessLayer(descriptor, in ParseDetectionPostProcess() local
1757 layer->GetOutputSlot(i).SetTensorInfo(ToTensorInfo(outputs[i])); in ParseDetectionPostProcess()
1760 RegisterInputSlots(graph, layerIndex, layer); in ParseDetectionPostProcess()
1761 RegisterOutputSlots(graph, layerIndex, layer); in ParseDetectionPostProcess()
1776 IConnectableLayer* layer = m_Network->AddElementwiseBinaryLayer(descriptor, layerName.c_str()); in ParseDivision() local
1779 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseDivision()
1781 RegisterInputSlots(graph, layerIndex, layer); in ParseDivision()
1782 RegisterOutputSlots(graph, layerIndex, layer); in ParseDivision()
1797 IConnectableLayer* layer = m_Network->AddComparisonLayer(descriptor, layerName.c_str()); in ParseEqual() local
1800 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseEqual()
1802 RegisterInputSlots(graph, layerIndex, layer); in ParseEqual()
1803 RegisterOutputSlots(graph, layerIndex, layer); in ParseEqual()
1819 IConnectableLayer* layer = m_Network->AddFillLayer(descriptor, layerName.c_str()); in ParseFill() local
1822 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseFill()
1824 RegisterInputSlots(graph, layerIndex, layer); in ParseFill()
1825 RegisterOutputSlots(graph, layerIndex, layer); in ParseFill()
1840 IConnectableLayer* layer = m_Network->AddComparisonLayer(descriptor, layerName.c_str()); in ParseGreater() local
1843 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseGreater()
1845 RegisterInputSlots(graph, layerIndex, layer); in ParseGreater()
1846 RegisterOutputSlots(graph, layerIndex, layer); in ParseGreater()
1871 …IConnectableLayer* layer = m_Network->AddInstanceNormalizationLayer(descriptor, layerName.c_str()); in ParseInstanceNormalization() local
1872 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseInstanceNormalization()
1874 RegisterInputSlots(graph, layerIndex, layer); in ParseInstanceNormalization()
1875 RegisterOutputSlots(graph, layerIndex, layer); in ParseInstanceNormalization()
1897 IConnectableLayer* layer = m_Network->AddL2NormalizationLayer(descriptor, layerName.c_str()); in ParseL2Normalization() local
1898 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseL2Normalization()
1900 RegisterInputSlots(graph, layerIndex, layer); in ParseL2Normalization()
1901 RegisterOutputSlots(graph, layerIndex, layer); in ParseL2Normalization()
1922 IConnectableLayer* layer = m_Network->AddLogicalBinaryLayer(descriptor, layerName.c_str()); in ParseLogicalBinary() local
1925 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseLogicalBinary()
1927 RegisterInputSlots(graph, layerIndex, layer); in ParseLogicalBinary()
1928 RegisterOutputSlots(graph, layerIndex, layer); in ParseLogicalBinary()
1946 IConnectableLayer* layer = m_Network->AddLogSoftmaxLayer(descriptor, layerName.c_str()); in ParseLogSoftmax() local
1949 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseLogSoftmax()
1951 RegisterInputSlots(graph, layerIndex, layer); in ParseLogSoftmax()
1952 RegisterOutputSlots(graph, layerIndex, layer); in ParseLogSoftmax()
1967 IConnectableLayer* layer = m_Network->AddElementwiseBinaryLayer(descriptor, layerName.c_str()); in ParseMinimum() local
1970 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseMinimum()
1972 RegisterInputSlots(graph, layerIndex, layer); in ParseMinimum()
1973 RegisterOutputSlots(graph, layerIndex, layer); in ParseMinimum()
1988 IConnectableLayer* layer = m_Network->AddElementwiseBinaryLayer(descriptor, layerName.c_str()); in ParseMaximum() local
1991 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseMaximum()
1993 RegisterInputSlots(graph, layerIndex, layer); in ParseMaximum()
1994 RegisterOutputSlots(graph, layerIndex, layer); in ParseMaximum()
2004 case Layer::Layer_ConcatLayer: in GetOriginsDescriptor()
2006 case Layer::Layer_MergerLayer: in GetOriginsDescriptor()
2009 throw armnn::Exception("unknown layer type, should be concat or merger"); in GetOriginsDescriptor()
2028 IConnectableLayer* layer = m_Network->AddChannelShuffleLayer(descriptor, layerName.c_str()); in ParseChannelShuffle() local
2031 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseChannelShuffle()
2033 RegisterInputSlots(graph, layerIndex, layer); in ParseChannelShuffle()
2034 RegisterOutputSlots(graph, layerIndex, layer); in ParseChannelShuffle()
2054 IConnectableLayer* layer = m_Network->AddComparisonLayer(descriptor, layerName.c_str()); in ParseComparison() local
2057 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseComparison()
2059 RegisterInputSlots(graph, layerIndex, layer); in ParseComparison()
2060 RegisterOutputSlots(graph, layerIndex, layer); in ParseComparison()
2081 …IConnectableLayer* layer = m_Network->AddElementwiseBinaryLayer(descriptor, layerName.c_str()); in ParseElementwiseBinary() local
2084 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseElementwiseBinary()
2086 RegisterInputSlots(graph, layerIndex, layer); in ParseElementwiseBinary()
2087 RegisterOutputSlots(graph, layerIndex, layer); in ParseElementwiseBinary()
2108 … IConnectableLayer* layer = m_Network->AddElementwiseUnaryLayer(descriptor, layerName.c_str()); in ParseElementwiseUnary() local
2111 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseElementwiseUnary()
2113 RegisterInputSlots(graph, layerIndex, layer); in ParseElementwiseUnary()
2114 RegisterOutputSlots(graph, layerIndex, layer); in ParseElementwiseUnary()
2147 IConnectableLayer* layer = m_Network->AddConcatLayer(descriptor, layerName.c_str()); in ParseConcat() local
2149 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseConcat()
2151 RegisterInputSlots(graph, layerIndex, layer); in ParseConcat()
2152 RegisterOutputSlots(graph, layerIndex, layer); in ParseConcat()
2167 IConnectableLayer* layer = m_Network->AddElementwiseBinaryLayer(descriptor, layerName.c_str()); in ParseMultiplication() local
2170 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseMultiplication()
2172 RegisterInputSlots(graph, layerIndex, layer); in ParseMultiplication()
2173 RegisterOutputSlots(graph, layerIndex, layer); in ParseMultiplication()
2189 armnn::IConnectableLayer* layer; in ParseFloor() local
2191 layer = m_Network->AddFloorLayer(layerName.c_str()); in ParseFloor()
2194 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseFloor()
2196 RegisterInputSlots(graph, layerIndex, layer); in ParseFloor()
2197 RegisterOutputSlots(graph, layerIndex, layer); in ParseFloor()
2218 armnn::IConnectableLayer* layer; in ParseFullyConnected() local
2221 …// Weights and biases used to be always constant and were stored as members of the layer. This has… in ParseFullyConnected()
2225 … // If the model stores weights and biases as members of the layer we have to read them from there in ParseFullyConnected()
2228 layer = m_Network->AddFullyConnectedLayer(fullyConnectedDescriptor, in ParseFullyConnected()
2233 weightsLayer->GetOutputSlot(0).Connect(layer->GetInputSlot(1u)); in ParseFullyConnected()
2241 biasLayer->GetOutputSlot(0).Connect(layer->GetInputSlot(2u)); in ParseFullyConnected()
2248 layer = m_Network->AddFullyConnectedLayer(fullyConnectedDescriptor, in ParseFullyConnected()
2255 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseFullyConnected()
2257 RegisterInputSlots(graph, layerIndex, layer, ignoreSlots); in ParseFullyConnected()
2258 RegisterOutputSlots(graph, layerIndex, layer); in ParseFullyConnected()
2292 IConnectableLayer* layer = m_Network->AddPadLayer(descriptor, layerName.c_str()); in ParsePad() local
2295 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParsePad()
2297 RegisterInputSlots(graph, layerIndex, layer); in ParsePad()
2298 RegisterOutputSlots(graph, layerIndex, layer); in ParsePad()
2318 IConnectableLayer* layer = m_Network->AddPermuteLayer(descriptor, layerName.c_str()); in ParsePermute() local
2319 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParsePermute()
2321 RegisterInputSlots(graph, layerIndex, layer); in ParsePermute()
2322 RegisterOutputSlots(graph, layerIndex, layer); in ParsePermute()
2533 IConnectableLayer* layer = m_Network->AddPooling2dLayer(pooling2dDescriptor, layerName.c_str()); in ParsePooling2d() local
2534 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParsePooling2d()
2536 RegisterInputSlots(graph, layerIndex, layer); in ParsePooling2d()
2537 RegisterOutputSlots(graph, layerIndex, layer); in ParsePooling2d()
2554 IConnectableLayer* layer = m_Network->AddPooling3dLayer(pooling3dDescriptor, layerName.c_str()); in ParsePooling3d() local
2555 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParsePooling3d()
2557 RegisterInputSlots(graph, layerIndex, layer); in ParsePooling3d()
2558 RegisterOutputSlots(graph, layerIndex, layer); in ParsePooling3d()
2573 IConnectableLayer* layer = m_Network->AddQuantizeLayer(layerName.c_str()); in ParseQuantize() local
2574 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseQuantize()
2576 RegisterInputSlots(graph, layerIndex, layer); in ParseQuantize()
2577 RegisterOutputSlots(graph, layerIndex, layer); in ParseQuantize()
2621 IConnectableLayer* layer = m_Network->AddRankLayer( layerName.c_str()); in ParseRank() local
2624 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseRank()
2626 RegisterInputSlots(graph, layerIndex, layer); in ParseRank()
2627 RegisterOutputSlots(graph, layerIndex, layer); in ParseRank()
2651 IConnectableLayer* layer = m_Network->AddReduceLayer(descriptor, layerName.c_str()); in ParseReduce() local
2654 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseReduce()
2656 RegisterInputSlots(graph, layerIndex, layer); in ParseReduce()
2657 RegisterOutputSlots(graph, layerIndex, layer); in ParseReduce()
2696 IConnectableLayer* layer = m_Network->AddReshapeLayer(reshapeDesc, layerName.c_str()); in ParseReshape() local
2697 layer->GetOutputSlot(0).SetTensorInfo(reshapeOutputTensorInfo); in ParseReshape()
2699 RegisterInputSlots(graph, layerIndex, layer); in ParseReshape()
2700 RegisterOutputSlots(graph, layerIndex, layer); in ParseReshape()
2724 IConnectableLayer* layer = m_Network->AddResizeLayer(descriptor, layerName.c_str()); in ParseResize() local
2727 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseResize()
2729 RegisterInputSlots(graph, layerIndex, layer); in ParseResize()
2730 RegisterOutputSlots(graph, layerIndex, layer); in ParseResize()
2757 IConnectableLayer* layer = m_Network->AddResizeLayer(descriptor, layerName.c_str()); in ParseResizeBilinear() local
2760 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseResizeBilinear()
2762 RegisterInputSlots(graph, layerIndex, layer); in ParseResizeBilinear()
2763 RegisterOutputSlots(graph, layerIndex, layer); in ParseResizeBilinear()
2777 IConnectableLayer* layer = m_Network->AddShapeLayer( layerName.c_str()); in ParseShape() local
2780 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseShape()
2782 RegisterInputSlots(graph, layerIndex, layer); in ParseShape()
2783 RegisterOutputSlots(graph, layerIndex, layer); in ParseShape()
2801 IConnectableLayer* layer = m_Network->AddSoftmaxLayer(descriptor, layerName.c_str()); in ParseSoftmax() local
2804 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseSoftmax()
2806 RegisterInputSlots(graph, layerIndex, layer); in ParseSoftmax()
2807 RegisterOutputSlots(graph, layerIndex, layer); in ParseSoftmax()
2844 IConnectableLayer* layer = m_Network->AddSpaceToBatchNdLayer(descriptor, layerName.c_str()); in ParseSpaceToBatchNd() local
2847 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseSpaceToBatchNd()
2849 RegisterInputSlots(graph, layerIndex, layer); in ParseSpaceToBatchNd()
2850 RegisterOutputSlots(graph, layerIndex, layer); in ParseSpaceToBatchNd()
2870 IConnectableLayer* layer = m_Network->AddSpaceToDepthLayer(descriptor, layerName.c_str()); in ParseSpaceToDepth() local
2873 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseSpaceToDepth()
2875 RegisterInputSlots(graph, layerIndex, layer); in ParseSpaceToDepth()
2876 RegisterOutputSlots(graph, layerIndex, layer); in ParseSpaceToDepth()
2965 …IConnectableLayer* layer = m_Network->AddNormalizationLayer(normalizationDescriptor, layerName.c_s… in ParseNormalization() local
2966 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseNormalization()
2968 RegisterInputSlots(graph, layerIndex, layer); in ParseNormalization()
2969 RegisterOutputSlots(graph, layerIndex, layer); in ParseNormalization()
2985 IConnectableLayer* layer = m_Network->AddElementwiseUnaryLayer(descriptor, layerName.c_str()); in ParseRsqrt() local
2987 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseRsqrt()
2989 RegisterInputSlots(graph, layerIndex, layer); in ParseRsqrt()
2990 RegisterOutputSlots(graph, layerIndex, layer); in ParseRsqrt()
3019 IConnectableLayer* layer = m_Network->AddSliceLayer(descriptor, layerName.c_str()); in ParseSlice() local
3022 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseSlice()
3024 RegisterInputSlots(graph, layerIndex, layer); in ParseSlice()
3025 RegisterOutputSlots(graph, layerIndex, layer); in ParseSlice()
3064 IConnectableLayer* layer = m_Network->AddStridedSliceLayer(descriptor, layerName.c_str()); in ParseStridedSlice() local
3067 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseStridedSlice()
3069 RegisterInputSlots(graph, layerIndex, layer); in ParseStridedSlice()
3070 RegisterOutputSlots(graph, layerIndex, layer); in ParseStridedSlice()
3085 IConnectableLayer* layer = m_Network->AddElementwiseBinaryLayer(descriptor, layerName.c_str()); in ParseSubtraction() local
3088 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseSubtraction()
3090 RegisterInputSlots(graph, layerIndex, layer); in ParseSubtraction()
3091 RegisterOutputSlots(graph, layerIndex, layer); in ParseSubtraction()
3108 IConnectableLayer* layer = m_Network->AddGatherLayer(descriptor, layerName.c_str()); in ParseGather() local
3111 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseGather()
3113 RegisterInputSlots(graph, layerIndex, layer); in ParseGather()
3114 RegisterOutputSlots(graph, layerIndex, layer); in ParseGather()
3128 IConnectableLayer* layer = m_Network->AddGatherNdLayer(layerName.c_str()); in ParseGatherNd() local
3131 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseGatherNd()
3133 RegisterInputSlots(graph, layerIndex, layer); in ParseGatherNd()
3134 RegisterOutputSlots(graph, layerIndex, layer); in ParseGatherNd()
3156 IConnectableLayer* layer = m_Network->AddMeanLayer(descriptor, layerName.c_str()); in ParseMean() local
3159 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseMean()
3161 RegisterInputSlots(graph, layerIndex, layer); in ParseMean()
3162 RegisterOutputSlots(graph, layerIndex, layer); in ParseMean()
3196 IConnectableLayer* layer = m_Network->AddSplitterLayer(viewsDescriptor, layerName.c_str()); in ParseSplitter() local
3202 layer->GetOutputSlot(vIdx).SetTensorInfo(outputTensorInfo); in ParseSplitter()
3205 RegisterInputSlots(graph, layerIndex, layer); in ParseSplitter()
3206 RegisterOutputSlots(graph, layerIndex, layer); in ParseSplitter()
3322 …IConnectableLayer* layer = m_Network->AddLstmLayer(lstmDescriptor, lstmInputParams, layerName.c_st… in ParseLstm() local
3325 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo1); in ParseLstm()
3328 layer->GetOutputSlot(1).SetTensorInfo(outputTensorInfo2); in ParseLstm()
3331 layer->GetOutputSlot(2).SetTensorInfo(outputTensorInfo3); in ParseLstm()
3334 layer->GetOutputSlot(3).SetTensorInfo(outputTensorInfo4); in ParseLstm()
3336 RegisterInputSlots(graph, layerIndex, layer); in ParseLstm()
3337 RegisterOutputSlots(graph, layerIndex, layer); in ParseLstm()
3451 // Optional layer norm params in ParseQLstm()
3474 …IConnectableLayer* layer = m_Network->AddQLstmLayer(qLstmDescriptor, qLstmInputParams, layerName.c… in ParseQLstm() local
3477 layer->GetOutputSlot(0).SetTensorInfo(outputStateOutInfo); in ParseQLstm()
3480 layer->GetOutputSlot(1).SetTensorInfo(cellStateOutInfo); in ParseQLstm()
3483 layer->GetOutputSlot(2).SetTensorInfo(outputInfo); in ParseQLstm()
3485 RegisterInputSlots(graph, layerIndex, layer); in ParseQLstm()
3486 RegisterOutputSlots(graph, layerIndex, layer); in ParseQLstm()
3531 IConnectableLayer* layer = m_Network->AddQuantizedLstmLayer(lstmInputParams, layerName.c_str()); in ParseQuantizedLstm() local
3534 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo1); in ParseQuantizedLstm()
3537 layer->GetOutputSlot(1).SetTensorInfo(outputTensorInfo2); in ParseQuantizedLstm()
3539 RegisterInputSlots(graph, layerIndex, layer); in ParseQuantizedLstm()
3540 RegisterOutputSlots(graph, layerIndex, layer); in ParseQuantizedLstm()
3554 IConnectableLayer* layer = m_Network->AddDequantizeLayer(layerName.c_str()); in ParseDequantize() local
3557 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseDequantize()
3559 RegisterInputSlots(graph, layerIndex, layer); in ParseDequantize()
3560 RegisterOutputSlots(graph, layerIndex, layer); in ParseDequantize()
3574 IConnectableLayer* layer = m_Network->AddMergeLayer(layerName.c_str()); in ParseMerge() local
3577 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseMerge()
3579 RegisterInputSlots(graph, layerIndex, layer); in ParseMerge()
3580 RegisterOutputSlots(graph, layerIndex, layer); in ParseMerge()
3594 IConnectableLayer* layer = m_Network->AddSwitchLayer(layerName.c_str()); in ParseSwitch() local
3597 layer->GetOutputSlot(0).SetTensorInfo(output0TensorInfo); in ParseSwitch()
3600 layer->GetOutputSlot(1).SetTensorInfo(output1TensorInfo); in ParseSwitch()
3602 RegisterInputSlots(graph, layerIndex, layer); in ParseSwitch()
3603 RegisterOutputSlots(graph, layerIndex, layer); in ParseSwitch()
3617 IConnectableLayer* layer = m_Network->AddPreluLayer(layerName.c_str()); in ParsePrelu() local
3620 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParsePrelu()
3622 RegisterInputSlots(graph, layerIndex, layer); in ParsePrelu()
3623 RegisterOutputSlots(graph, layerIndex, layer); in ParsePrelu()
3642 IConnectableLayer* layer = m_Network->AddTransposeLayer(descriptor, layerName.c_str()); in ParseTranspose() local
3643 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseTranspose()
3645 RegisterInputSlots(graph, layerIndex, layer); in ParseTranspose()
3646 RegisterOutputSlots(graph, layerIndex, layer); in ParseTranspose()
3682 IConnectableLayer* layer = m_Network->AddTransposeConvolution2dLayer(descriptor, in ParseTransposeConvolution2d() local
3688 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseTransposeConvolution2d()
3690 RegisterInputSlots(graph, layerIndex, layer); in ParseTransposeConvolution2d()
3691 RegisterOutputSlots(graph, layerIndex, layer); in ParseTransposeConvolution2d()
3733 IConnectableLayer* layer = m_Network->AddStackLayer(descriptor, layerName.c_str()); in ParseStack() local
3736 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseStack()
3738 RegisterInputSlots(graph, layerIndex, layer); in ParseStack()
3739 RegisterOutputSlots(graph, layerIndex, layer); in ParseStack()
3760 armnn::IConnectableLayer* layer = m_Network->AddStandInLayer(descriptor, layerName.c_str()); in ParseStandIn() local
3765 layer->GetOutputSlot(i).SetTensorInfo(outputInfo); in ParseStandIn()
3768 RegisterInputSlots(graph, layerIndex, layer); in ParseStandIn()
3769 RegisterOutputSlots(graph, layerIndex, layer); in ParseStandIn()
3891 IConnectableLayer* layer = m_Network->AddUnidirectionalSequenceLstmLayer(descriptor, in ParseUnidirectionalSequenceLstm() local
3896 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo0); in ParseUnidirectionalSequenceLstm()
3899 layer->GetOutputSlot(1).SetTensorInfo(outputTensorInfo1); in ParseUnidirectionalSequenceLstm()
3902 layer->GetOutputSlot(2).SetTensorInfo(outputTensorInfo2); in ParseUnidirectionalSequenceLstm()
3904 RegisterInputSlots(graph, layerIndex, layer); in ParseUnidirectionalSequenceLstm()
3905 RegisterOutputSlots(graph, layerIndex, layer); in ParseUnidirectionalSequenceLstm()