Lines Matching refs:scratchBufferTensorInfo
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()
1205 LayerTestResult<T, 2> ret0(scratchBufferTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1234 tensorHandleFactory.CreateTensorHandle(scratchBufferTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1247 AddOutputToWorkload(data, info, scratchBufferTensorInfo, scratchBufferHandle.get()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1305 … armnn::TensorInfo scratchBufferTensorInfo({batchSize, numUnits * 4}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1323 auto scratchBufferTensor = MakeTensor<float,2>(scratchBufferTensorInfo, scratchBufferVector); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1342 tensorHandleFactory.CreateTensorHandle(scratchBufferTensorInfo); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1356 AddOutputToWorkload(data, info, scratchBufferTensorInfo, scratchHandle.get()); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()