Searched defs:cellStateInDimensions (Results 1 – 2 of 2) sorted by relevance
| /external/android-nn-driver/test/ |
| D | Lstm.hpp | 106 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()
784 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmNoCifgNoPeepholeNoProjection() local
974 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmCifgPeepholeNoProjection() local
1671 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmNoCifgPeepholeProjection() local
1884 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmCifgPeepholeNoProjectionBatch2() local
2086 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local
2282 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
|
| /external/android-nn-driver/test/1.3/ |
| D | QLstm.cpp | 634 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in QLstmWithProjection() local
794 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in QLstmWithNoProjection() local
955 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in DynamicOutputQLstmWithNoProjection() local
|