Lines Matching refs:cellStateInTensorInfo
40 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
61 … tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
76 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
236 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
257 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
272 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
425 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
446 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
461 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
684 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
711 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
726 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
937 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
963 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
978 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
1177 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest() local
1203 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest()
1218 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest()
1352 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8Test() local
1378 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmLayerInt8Test()
1394 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmLayerInt8Test()
1506 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8TimeMajorTest() local
1531 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmLayerInt8TimeMajorTest()
1547 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmLayerInt8TimeMajorTest()
1660 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest() local
1686 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest()
1701 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest()
1840 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1866 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest()
1881 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest()
2047 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest() local
2073 tensorHandleFactory.CreateTensorHandle(cellStateInTensorInfo); in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest()
2088 AddInputToWorkload(data, info, cellStateInTensorInfo, cellStateInHandle.get()); in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest()