Lines Matching refs:cellStateOutTensorInfo
168 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
197 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
209 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
364 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
393 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
405 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1047 armnn::TensorInfo cellStateOutTensorInfo({batchSize, cellSize}, ArmnnType, qScale, qOffset); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1174 LayerTestResult<T, 2> ret2(cellStateOutTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1184 std::vector<T> actualCellStateOutput(cellStateOutTensorInfo.GetNumElements()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1200 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1211 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1271 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1300 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()
1312 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl()