Searched defs:cellStateOutDimensions (Results 1 – 3 of 3) sorted by relevance
| /external/android-nn-driver/test/ |
| D | Lstm.hpp | 83 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()
387 void QuantizedLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QuantizedLstmTestImpl()
812 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmNoCifgNoPeepholeNoProjection() local
1002 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmCifgPeepholeNoProjection() local
1705 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmNoCifgPeepholeProjection() local
1913 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmCifgPeepholeNoProjectionBatch2() local
2115 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local
2311 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
2451 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, outputSize}; in QuantizedLstm() local
|
| D | UnidirectionalSequenceLstm.hpp | 526 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
731 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
971 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1186 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1383 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTestImpl() local
|
| /external/android-nn-driver/test/1.3/ |
| D | QLstm.cpp | 62 void QLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QLstmTestImpl()
631 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in QLstmWithProjection() local
791 hidl_vec<uint32_t> cellStateOutDimensions{batchSize, numUnits}; in QLstmWithNoProjection() local
952 hidl_vec<uint32_t> cellStateOutDimensions{}; in DynamicOutputQLstmWithNoProjection() local
|