Lines Matching refs:cellStateOutTensorInfo
43 …armnn::TensorInfo cellStateOutTensorInfo({ batchSize, 1, outputSize }, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
53 std::vector<T> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
68 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
79 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
239 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, ArmnnType, qScale, qOf… in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
249 std::vector<T> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
264 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
275 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
428 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
438 std::vector<T> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
453 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
464 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
687 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
699 std::vector<float> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
718 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
729 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
940 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
951 std::vector<float> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
970 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
981 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
1180 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest() local
1191 std::vector<float> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest()
1210 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest()
1221 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest()
1355 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerInt8Test() local
1366 std::vector<float> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmLayerInt8Test()
1385 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmLayerInt8Test()
1397 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmLayerInt8Test()
1509 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerInt8TimeMajorTest() local
1520 std::vector<float> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmLayerInt8TimeMajorTest()
1538 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmLayerInt8TimeMajorTest()
1550 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmLayerInt8TimeMajorTest()
1663 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest() local
1674 std::vector<float> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest()
1693 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest()
1704 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest()
1843 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1854 std::vector<float> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest()
1873 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest()
1884 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest()
2050 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest() local
2061 std::vector<float> actualCellStateOut(cellStateOutTensorInfo.GetNumElements()); in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest()
2080 tensorHandleFactory.CreateTensorHandle(cellStateOutTensorInfo); in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest()
2091 AddOutputToWorkload(data, info, cellStateOutTensorInfo, cellStateOutHandle.get()); in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest()