Searched defs:outputStateInDimensions (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/android-nn-driver/test/ |
| H A D | UnidirectionalSequenceLstm.hpp | 84 void UnidirectionalSequenceLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in UnidirectionalSequenceLstmTestImpl()
467 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
672 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
912 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1127 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1324 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTestImpl() local
|
| H A D | Lstm.hpp | 83 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()
761 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in LstmNoCifgNoPeepholeNoProjection() local
951 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in LstmCifgPeepholeNoProjection() local
1648 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in LstmNoCifgPeepholeProjection() local
1861 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in LstmCifgPeepholeNoProjectionBatch2() local
2063 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local
2259 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
|
| /aosp_15_r20/external/android-nn-driver/test/1.3/ |
| H A D | QLstm.cpp | 599 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in QLstmWithProjection() local
759 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in QLstmWithNoProjection() local
920 hidl_vec<uint32_t> outputStateInDimensions{batchSize, outputSize}; in DynamicOutputQLstmWithNoProjection() local
|
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | LstmTestHelper.hpp | 80 std::vector<int32_t> outputStateInDimensions{batchSize, outputSize}; in CreateLstmTfLiteModel() local
|
| H A D | UnidirectionalSequenceLstmTestHelper.hpp | 99 std::vector<int32_t> outputStateInDimensions{batchSize, outputSize}; in CreateUnidirectionalSequenceLstmTfLiteModel() local
|