Home
last modified time | relevance | path

Searched defs:cellStateOutTensorInfo (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp43 …armnn::TensorInfo cellStateOutTensorInfo({ batchSize, 1, outputSize }, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
239 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, ArmnnType, qScale, qOf… in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
428 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
687 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
940 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1180 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest() local
1355 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerInt8Test() local
1509 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerInt8TimeMajorTest() local
1663 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest() local
1843 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest() local
[all …]
H A DLstmTestImpl.cpp168 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
364 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1047 armnn::TensorInfo cellStateOutTensorInfo({batchSize, cellSize}, ArmnnType, qScale, qOffset); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1271 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DLstmSerializationTests.cpp2303 armnn::TensorInfo cellStateOutTensorInfo({ batchSize, outputSize }, armnn::DataType::Float32); variable
2444 armnn::TensorInfo cellStateOutTensorInfo({ batchSize, outputSize }, armnn::DataType::Float32); variable
2604 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… variable
2713 …armnn::TensorInfo cellStateOutTensorInfo({batchSize, timeSize, outputSize}, armnn::DataType::Float… variable
/aosp_15_r20/external/armnn/delegate/classic/src/
H A DLstm.hpp167 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, dataType, qScale, qOffset); in VisitLstmOperator() local
H A DUnidirectionalSequenceLstm.hpp200 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, in VisitUnidirectionalSequenceLstmOperator() local
/aosp_15_r20/external/armnn/delegate/test/
H A DLstmTest.cpp30 std::vector<int32_t> cellStateOutTensorInfo {batchSize, numUnits}; in LstmTest() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQuantizedLstmEndToEndTestImpl.cpp141 armnn::TensorInfo cellStateOutTensorInfo({batchSize, outputSize}, in CreateQuantizedLstmNetwork() local
H A DWorkloadDataValidation.cpp511 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, dataType, qScale, qOffset); variable
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp3727 armnn::TensorInfo cellStateOutTensorInfo({batchSize, numUnits}, in ParseUnidirectionalSequenceLSTM() local