Home
last modified time | relevance | path

Searched defs:cellStateInTensorInfo (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp40 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
236 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
425 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, ArmnnType, qScale, qOffset); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
684 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
937 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1177 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest() local
1352 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8Test() local
1506 armnn::TensorInfo cellStateInTensorInfo({batchSize, numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8TimeMajorTest() local
1660 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest() local
1840 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, armnn::DataType::Float32); in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest() local
[all …]
H A DLstmTestImpl.cpp164 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
359 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1042 armnn::TensorInfo cellStateInTensorInfo({batchSize, cellSize}, ArmnnType, qScale, qOffset); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1266 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, ArmnnType, qScale, qOffset); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
/aosp_15_r20/external/armnn/delegate/test/
H A DLstmTest.cpp26 std::vector<int32_t> cellStateInTensorInfo {batchSize , numUnits}; in LstmTest() local
H A DUnidirectionalSequenceLstmTest.cpp170 std::vector<int32_t> cellStateInTensorInfo = {batchSize, numUnits}; in UnidirectionalSequenceLstmTimeMajorTest() local
318 std::vector<int32_t> cellStateInTensorInfo = {batchSize, numUnits}; in UnidirectionalSequenceLstmNoCifgWithPeepholeWithProjectionTest() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQuantizedLstmEndToEndTestImpl.cpp131 armnn::TensorInfo cellStateInTensorInfo({batchSize , outputSize}, in CreateQuantizedLstmNetwork() local
H A DWorkloadDataValidation.cpp507 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, dataType, qScale, qOffset); variable