Searched refs:batchNormalizationLayer (Results 1 – 2 of 2) sorted by relevance
129 …armnn::IConnectableLayer* const batchNormalizationLayer = net.AddBatchNormalizationLayer(batchNorm… variable135 CHECK(batchNormalizationLayer);137 softmaxLayer->GetOutputSlot(0).Connect(batchNormalizationLayer->GetInputSlot(0));143 batchNormalizationLayer->GetOutputSlot(0).Connect(additionLayer->GetInputSlot(0));144 batchNormalizationLayer->GetOutputSlot(0).Connect(additionLayer->GetInputSlot(1));238 CHECK(AreAllLayerInputSlotsConnected(*batchNormalizationLayer));250 checkOneOutputToOneInputConnection(softmaxLayer, batchNormalizationLayer);251 checkOneOutputToTwoInputsConnections(batchNormalizationLayer, additionLayer, 1);
166 armnn::IConnectableLayer* const batchNormalizationLayer = variable175 inputLayer->GetOutputSlot(0).Connect(batchNormalizationLayer->GetInputSlot(0));176 batchNormalizationLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0));179 batchNormalizationLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);