Searched defs:outputStateInfo (Results 1 – 5 of 5) sorted by relevance
| /external/armnn/src/backends/reference/workloads/ |
| D | RefUnidirectionalSequenceLstmWorkload.cpp | 61 const TensorInfo& outputStateInfo = GetTensorInfo(inputs[1]); in Execute() local
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | QLstmEndToEndTestImpl.cpp | 195 const armnn::TensorInfo outputStateInfo({numBatches , outputSize}, in QLstmEndToEnd() local
|
| /external/armnn/src/armnnSerializer/test/ |
| D | LstmSerializationTests.cpp | 1747 armnn::TensorInfo outputStateInfo({numBatches , outputSize}, variable
1940 armnn::TensorInfo outputStateInfo({numBatches , outputSize}, variable
2187 armnn::TensorInfo outputStateInfo({numBatches , outputSize}, variable
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | LstmTestImpl.cpp | 1529 armnn::TensorInfo outputStateInfo({numBatches , outputSize}, in QuantizedLstmTestImpl() local
1741 armnn::TensorInfo outputStateInfo({numBatches , outputSize}, in QLstmTestImpl() local
1981 armnn::TensorInfo outputStateInfo({numBatches , outputSize}, in QLstmTestImpl1() local
2246 armnn::TensorInfo outputStateInfo({numBatches , outputSize}, in QLstmTestImpl2() local
|
| /external/armnn/src/armnnTestUtils/ |
| D | CreateWorkload.hpp | 925 armnn::TensorInfo outputStateInfo({numBatches , outputSize}, in CreateQuantizedLstmWorkloadTest() local
1089 armnn::TensorInfo outputStateInfo({numBatches , outputSize}, in CreateQLstmWorkloadTest() local
|