Searched defs:cellStateInfo (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/armnn/src/armnnSerializer/test/ |
| H A D | LstmSerializationTests.cpp | 1742 armnn::TensorInfo cellStateInfo({numBatches , numUnits}, variable
1935 armnn::TensorInfo cellStateInfo({numBatches , numUnits}, variable
2182 armnn::TensorInfo cellStateInfo({numBatches , numUnits}, variable
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/ |
| H A D | LstmTestImpl.cpp | 1524 armnn::TensorInfo cellStateInfo({numBatches , outputSize}, in QuantizedLstmTestImpl() local
1736 armnn::TensorInfo cellStateInfo({numBatches , numUnits}, in QLstmTestImpl() local
1976 armnn::TensorInfo cellStateInfo({numBatches , numUnits}, in QLstmTestImpl1() local
2241 armnn::TensorInfo cellStateInfo({numBatches , numUnits}, in QLstmTestImpl2() local
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 62 const TensorInfo& cellStateInfo = GetTensorInfo(inputs[2]); in Execute() local
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
| H A D | QLstmEndToEndTestImpl.cpp | 189 const armnn::TensorInfo cellStateInfo({numBatches , numUnits}, in QLstmEndToEnd() local
|
| /aosp_15_r20/external/armnn/src/armnnTestUtils/ |
| H A D | CreateWorkload.hpp | 920 armnn::TensorInfo cellStateInfo({numBatches , outputSize}, in CreateQuantizedLstmWorkloadTest() local
1084 armnn::TensorInfo cellStateInfo({numBatches , numUnits}, in CreateQLstmWorkloadTest() local
|
| /aosp_15_r20/external/armnn/src/backends/reference/test/ |
| H A D | RefCreateWorkloadTests.cpp | 1255 armnn::TensorInfo cellStateInfo({2 , 4}, armnn::DataType::QSymmS16, 3.05176e-05f, 0); in RefCreateQLstmWorkloadTest() local
|