Searched refs:scratchBufferTensorInfo (Results 1 – 2 of 2) sorted by relevance
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | LstmTestImpl.cpp | 158 … armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 4}, ArmnnType, qScale, qOffset); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
176 auto scratchBufferTensor = MakeTensor<T,2>(scratchBufferTensorInfo, scratchBufferVector); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
195 tensorHandleFactory.CreateTensorHandle(scratchBufferTensorInfo); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
209 AddOutputToWorkload(data, info, scratchBufferTensorInfo, scratchHandle.get()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
362 … armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 4}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
380 auto scratchBufferTensor = MakeTensor<T,2>(scratchBufferTensorInfo, scratchBufferVector); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
399 tensorHandleFactory.CreateTensorHandle(scratchBufferTensorInfo); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
413 AddOutputToWorkload(data, info, scratchBufferTensorInfo, scratchHandle.get()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1086 …armnn::TensorInfo scratchBufferTensorInfo({batchSize, scratchBufferSize}, ArmnnType, qScale, qOffs… in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1204 auto scratchBufferTensor = MakeTensor<T,2>(scratchBufferTensorInfo, scratchBufferVector); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | WorkloadDataValidation.cpp | 495 armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 4}, dataType, qScale, qOffset); in BOOST_AUTO_TEST_CASE() local
513 AddOutputToWorkload(data, info, scratchBufferTensorInfo, nullptr); in BOOST_AUTO_TEST_CASE()
580 SetWorkloadOutput(data, info, 0, scratchBufferTensorInfo, nullptr); in BOOST_AUTO_TEST_CASE()
|