Searched defs:outputStateOutDimensions (Results 1 – 2 of 2) sorted by relevance
/external/android-nn-driver/test/1.3/ |
D | QLstm.cpp | 62 void QLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QLstmTestImpl() 628 hidl_vec<uint32_t> outputStateOutDimensions{batchSize, outputSize}; in QLstmWithProjection() local 788 hidl_vec<uint32_t> outputStateOutDimensions{batchSize, outputSize}; in QLstmWithNoProjection() local 949 hidl_vec<uint32_t> outputStateOutDimensions{batchSize, outputSize}; in DynamicOutputQLstmWithNoProjection() local
|
/external/android-nn-driver/test/ |
D | Lstm.hpp | 83 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl() 809 hidl_vec<uint32_t> outputStateOutDimensions{batchSize, outputSize}; in LstmNoCifgNoPeepholeNoProjection() local 999 hidl_vec<uint32_t> outputStateOutDimensions{batchSize, outputSize}; in LstmCifgPeepholeNoProjection() local 1696 hidl_vec<uint32_t> outputStateOutDimensions{batchSize, outputSize}; in LstmNoCifgPeepholeProjection() local 1909 hidl_vec<uint32_t> outputStateOutDimensions{batchSize, outputSize}; in LstmCifgPeepholeNoProjectionBatch2() local 2111 hidl_vec<uint32_t> outputStateOutDimensions{batchSize, outputSize}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local 2307 hidl_vec<uint32_t> outputStateOutDimensions{batchSize, outputSize}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
|