LstmSerializationTests.cpp - OpenGrok cross reference for /external/armnn/src/armnnSerializer/test/LstmSerializationTests.cpp

Lines Matching refs:SetTensorInfo
260     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
263     outputStateIn->GetOutputSlot(0).SetTensorInfo(outputStateTensorInfo);
266     cellStateIn->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
269     lstmLayer->GetOutputSlot(0).SetTensorInfo(lstmTensorInfoScratchBuff);
272     lstmLayer->GetOutputSlot(1).SetTensorInfo(outputStateTensorInfo);
275     lstmLayer->GetOutputSlot(2).SetTensorInfo(cellStateTensorInfo);
278     lstmLayer->GetOutputSlot(3).SetTensorInfo(outputStateTensorInfo);
406     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
409     outputStateIn->GetOutputSlot(0).SetTensorInfo(outputStateTensorInfo);
412     cellStateIn->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
415     lstmLayer->GetOutputSlot(0).SetTensorInfo(lstmTensorInfoScratchBuff);
418     lstmLayer->GetOutputSlot(1).SetTensorInfo(outputStateTensorInfo);
421     lstmLayer->GetOutputSlot(2).SetTensorInfo(cellStateTensorInfo);
424     lstmLayer->GetOutputSlot(3).SetTensorInfo(outputStateTensorInfo);
571     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
574     outputStateIn->GetOutputSlot(0).SetTensorInfo(outputStateTensorInfo);
577     cellStateIn->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
580     lstmLayer->GetOutputSlot(0).SetTensorInfo(lstmTensorInfoScratchBuff);
583     lstmLayer->GetOutputSlot(1).SetTensorInfo(outputStateTensorInfo);
586     lstmLayer->GetOutputSlot(2).SetTensorInfo(cellStateTensorInfo);
589     lstmLayer->GetOutputSlot(3).SetTensorInfo(outputStateTensorInfo);
1600     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
1603     cellStateIn->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
1606     outputStateIn->GetOutputSlot(0).SetTensorInfo(outputStateTensorInfo);
1609     quantizedLstmLayer->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
1612     quantizedLstmLayer->GetOutputSlot(1).SetTensorInfo(outputStateTensorInfo);
1754     input->GetOutputSlot(0).SetTensorInfo(inputInfo);
1757     outputStateIn->GetOutputSlot(0).SetTensorInfo(cellStateInfo);
1760     cellStateIn->GetOutputSlot(0).SetTensorInfo(outputStateInfo);
1763     qLstmLayer->GetOutputSlot(0).SetTensorInfo(outputStateInfo);
1766     qLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateInfo);
1769     qLstmLayer->GetOutputSlot(2).SetTensorInfo(outputStateInfo);
1947     input->GetOutputSlot(0).SetTensorInfo(inputInfo);
1950     outputStateIn->GetOutputSlot(0).SetTensorInfo(cellStateInfo);
1953     cellStateIn->GetOutputSlot(0).SetTensorInfo(outputStateInfo);
1956     qLstmLayer->GetOutputSlot(0).SetTensorInfo(outputStateInfo);
1959     qLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateInfo);
1962     qLstmLayer->GetOutputSlot(2).SetTensorInfo(outputStateInfo);
2194     input->GetOutputSlot(0).SetTensorInfo(inputInfo);
2197     outputStateIn->GetOutputSlot(0).SetTensorInfo(cellStateInfo);
2200     cellStateIn->GetOutputSlot(0).SetTensorInfo(outputStateInfo);
2203     qLstmLayer->GetOutputSlot(0).SetTensorInfo(outputStateInfo);
2206     qLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateInfo);
2209     qLstmLayer->GetOutputSlot(2).SetTensorInfo(outputStateInfo);
2307     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
2310     outputStateIn->GetOutputSlot(0).SetTensorInfo(outputStateTensorInfo);
2313     cellStateIn->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
2315     unidirectionalSequenceLstmLayer->GetOutputSlot(0).SetTensorInfo(outputStateOutTensorInfo);
2316     unidirectionalSequenceLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateOutTensorInfo);
2318     unidirectionalSequenceLstmLayer->GetOutputSlot(2).SetTensorInfo(outputTensorInfo);
2448     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
2451     outputStateIn->GetOutputSlot(0).SetTensorInfo(outputStateTensorInfo);
2454     cellStateIn->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
2456     unidirectionalSequenceLstmLayer->GetOutputSlot(0).SetTensorInfo(outputStateOutTensorInfo);
2457     unidirectionalSequenceLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateOutTensorInfo);
2459     unidirectionalSequenceLstmLayer->GetOutputSlot(2).SetTensorInfo(outputTensorInfo);
2608     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
2611     outputStateIn->GetOutputSlot(0).SetTensorInfo(outputStateTensorInfo);
2614     cellStateIn->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
2616     unidirectionalSequenceLstmLayer->GetOutputSlot(0).SetTensorInfo(outputStateOutTensorInfo);
2617     unidirectionalSequenceLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateOutTensorInfo);
2619     unidirectionalSequenceLstmLayer->GetOutputSlot(2).SetTensorInfo(outputTensorInfo);
2717     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
2720     outputStateIn->GetOutputSlot(0).SetTensorInfo(outputStateTensorInfo);
2723     cellStateIn->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
2725     unidirectionalSequenceLstmLayer->GetOutputSlot(0).SetTensorInfo(outputStateOutTensorInfo);
2726     unidirectionalSequenceLstmLayer->GetOutputSlot(1).SetTensorInfo(cellStateOutTensorInfo);
2728     unidirectionalSequenceLstmLayer->GetOutputSlot(2).SetTensorInfo(outputTensorInfo);