Lines Matching refs:outputStateInTensorInfo
41 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
63 … tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
75 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
237 armnn::TensorInfo outputStateInTensorInfo({batchSize, outputSize}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
259 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
271 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
426 armnn::TensorInfo outputStateInTensorInfo({batchSize, outputSize}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
448 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
460 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
685 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
713 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
725 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
938 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
965 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
977 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
1178 armnn::TensorInfo outputStateInTensorInfo({batchSize, outputSize}, armnn::DataType::Float32); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest() local
1205 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest()
1217 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest()
1353 armnn::TensorInfo outputStateInTensorInfo({batchSize, outputSize}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8Test() local
1380 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmLayerInt8Test()
1393 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmLayerInt8Test()
1507 armnn::TensorInfo outputStateInTensorInfo({batchSize, outputSize}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8TimeMajorTest() local
1533 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmLayerInt8TimeMajorTest()
1546 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmLayerInt8TimeMajorTest()
1661 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest() local
1688 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest()
1700 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest()
1841 armnn::TensorInfo outputStateInTensorInfo({batchSize , outputSize}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1868 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest()
1880 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest()
2048 armnn::TensorInfo outputStateInTensorInfo({batchSize, outputSize}, armnn::DataType::Float32); in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest() local
2075 tensorHandleFactory.CreateTensorHandle(outputStateInTensorInfo); in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest()
2087 AddInputToWorkload(data, info, outputStateInTensorInfo, outputStateInHandle.get()); in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest()