Searched defs:lstmTensorInfoScratchBuff (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/armnn/src/armnnSerializer/test/ |
| H A D | LstmSerializationTests.cpp | 257 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 3 }, armnn::DataType::Float32); variable
403 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 4 }, armnn::DataType::Float32); variable
568 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 4 }, armnn::DataType::Float32); variable
1324 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 4 }, armnn::DataType::Float32,… variable
|
| /aosp_15_r20/external/armnn/src/armnn/test/ |
| H A D | OptimizerTests.cpp | 129 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * (layerDesc.m_CifgEnabled ? 3 :… in CreateLSTMLayerHelper() local
|
| /aosp_15_r20/external/armnn/src/armnnTestUtils/ |
| H A D | CreateWorkload.hpp | 799 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * (layerDesc.m_CifgEnabled ? 3 :… in CreateLstmWorkloadTest() local
|