Searched defs:lstmTensorInfoScratchBuff (Results 1 – 3 of 3) sorted by relevance
| /external/armnn/src/armnn/test/ |
| D | OptimizerTests.cpp | 128 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * (layerDesc.m_CifgEnabled ? 3 :… in CreateLSTMLayerHelper() local
|
| D | CreateWorkload.hpp | 748 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * (layerDesc.m_CifgEnabled ? 3 :… in CreateLstmWorkloadTest() local
|
| /external/armnn/src/armnnSerializer/test/ |
| D | SerializerTests.cpp | 3155 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 3 }, armnn::DataType::Float32); in BOOST_AUTO_TEST_CASE() local
3301 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 4 }, armnn::DataType::Float32); in BOOST_AUTO_TEST_CASE() local
3466 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 4 }, armnn::DataType::Float32); in BOOST_AUTO_TEST_CASE() local
4222 …armnn::TensorInfo lstmTensorInfoScratchBuff({ batchSize, numUnits * 4 }, armnn::DataType::Float32); in BOOST_AUTO_TEST_CASE() local
|