Home
last modified time | relevance | path

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

/external/armnn/src/backends/backendsCommon/test/layerTests/
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 …]
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
/external/armnn/delegate/test/
DLstmTest.cpp26 std::vector<int32_t> cellStateInTensorInfo {batchSize , numUnits}; in LstmTest() local
DUnidirectionalSequenceLstmTest.cpp170 std::vector<int32_t> cellStateInTensorInfo = {batchSize, numUnits}; in UnidirectionalSequenceLstmTimeMajorTest() local
318 std::vector<int32_t> cellStateInTensorInfo = {batchSize, numUnits}; in UnidirectionalSequenceLstmNoCifgWithPeepholeWithProjectionTest() local
/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp131 armnn::TensorInfo cellStateInTensorInfo({batchSize , outputSize}, in CreateQuantizedLstmNetwork() local
DWorkloadDataValidation.cpp507 armnn::TensorInfo cellStateInTensorInfo({batchSize , numUnits}, dataType, qScale, qOffset); variable