Searched defs:cellStateOutDimensions (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()
408 void QuantizedLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QuantizedLstmTestImpl()
832 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmNoCifgNoPeepholeNoProjection() local
1022 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmCifgPeepholeNoProjection() local
1725 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmNoCifgPeepholeProjection() local
1933 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmCifgPeepholeNoProjectionBatch2() local
2135 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local
2331 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
2471 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, outputSize}; in QuantizedLstm() local
|
| /external/android-nn-driver/test/1.3/ |
| D | QLstm.cpp | 95 void QLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QLstmTestImpl()
663 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in QLstmWithProjection() local
823 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in QLstmWithNoProjection() local
984 hidl_vec<uint32_t> cellStateOutDimensions{}; in DynamicOutputQLstmWithNoProjection() local
|