Searched refs:lstmTensorInfoScratchBuff (Results 1 – 3 of 3) sorted by relevance
| /external/armnn/src/armnnSerializer/test/ |
| D | LstmSerializationTests.cpp | 257 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 3 }, armnn::DataType::Float32); variable
269 lstmLayer->GetOutputSlot(0).SetTensorInfo(lstmTensorInfoScratchBuff);
286 … {lstmTensorInfoScratchBuff, outputStateTensorInfo, cellStateTensorInfo, outputStateTensorInfo},
403 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 4 }, armnn::DataType::Float32); variable
415 lstmLayer->GetOutputSlot(0).SetTensorInfo(lstmTensorInfoScratchBuff);
432 … {lstmTensorInfoScratchBuff, outputStateTensorInfo, cellStateTensorInfo, outputStateTensorInfo},
568 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 4 }, armnn::DataType::Float32); variable
580 lstmLayer->GetOutputSlot(0).SetTensorInfo(lstmTensorInfoScratchBuff);
597 … {lstmTensorInfoScratchBuff, outputStateTensorInfo, cellStateTensorInfo, outputStateTensorInfo},
1324 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 4 }, armnn::DataType::Float32,… variable
[all …]
|
| /external/armnn/src/armnn/test/ |
| D | OptimizerTests.cpp | 129 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * (layerDesc.m_CifgEnabled ? 3 :… in CreateLSTMLayerHelper() local
135 Connect(layer, scratchBuffer, lstmTensorInfoScratchBuff, 0, 0); in CreateLSTMLayerHelper()
|
| /external/armnn/src/armnnTestUtils/ |
| D | CreateWorkload.hpp | 799 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * (layerDesc.m_CifgEnabled ? 3 :… in CreateLstmWorkloadTest() local
804 Connect(layer, scratchBuffer, lstmTensorInfoScratchBuff, 0, 0); in CreateLstmWorkloadTest()
|