Home
last modified time | relevance | path

Searched refs:qLstmLayer (Results 1 – 3 of 3) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/
DQLstmEndToEndTestImpl.cpp218 armnn::IConnectableLayer* const qLstmLayer = net->AddQLstmLayer(descriptor, params, "qLstm"); in QLstmEndToEnd() local
225 Connect(input, qLstmLayer, inputInfo, 0, 0); in QLstmEndToEnd()
226 Connect(outputStateIn, qLstmLayer, outputStateInfo, 0, 1); in QLstmEndToEnd()
227 Connect(cellStateIn, qLstmLayer, cellStateInfo, 0, 2); in QLstmEndToEnd()
229 Connect(qLstmLayer, outputStateOut, outputStateInfo, 0, 0); in QLstmEndToEnd()
230 Connect(qLstmLayer, cellStateOut, cellStateInfo, 1, 0); in QLstmEndToEnd()
231 Connect(qLstmLayer, output, outputStateInfo, 2, 0); in QLstmEndToEnd()
/external/armnn/src/armnnSerializer/test/
DLstmSerializationTests.cpp1730 …armnn::IConnectableLayer* const qLstmLayer = network->AddQLstmLayer(descriptor, params, layerName.… variable
1753 input->GetOutputSlot(0).Connect(qLstmLayer->GetInputSlot(0));
1756 outputStateIn->GetOutputSlot(0).Connect(qLstmLayer->GetInputSlot(1));
1759 cellStateIn->GetOutputSlot(0).Connect(qLstmLayer->GetInputSlot(2));
1762 qLstmLayer->GetOutputSlot(0).Connect(outputStateOut->GetInputSlot(0));
1763 qLstmLayer->GetOutputSlot(0).SetTensorInfo(outputStateInfo);
1765 qLstmLayer->GetOutputSlot(1).Connect(cellStateOut->GetInputSlot(0));
1766 qLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateInfo);
1768 qLstmLayer->GetOutputSlot(2).Connect(outputLayer->GetInputSlot(0));
1769 qLstmLayer->GetOutputSlot(2).SetTensorInfo(outputStateInfo);
[all …]
/external/armnn/src/armnn/test/
DInferOutputTests.hpp608 armnn::QLstmLayer* const qLstmLayer = graph.AddLayer<armnn::QLstmLayer>(descriptor, "qLstm"); in QLstmInferOutputShapeImpl() local
609 outputShapes = qLstmLayer->InferOutputShapes(inputShapes); in QLstmInferOutputShapeImpl()