Searched defs:cellStateInDimensions (Results 1 – 5 of 5) sorted by relevance
| /external/android-nn-driver/test/ |
| D | UnidirectionalSequenceLstm.hpp | 84 void UnidirectionalSequenceLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in UnidirectionalSequenceLstmTestImpl()
470 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
675 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
915 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1130 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1327 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTestImpl() local
|
| D | Lstm.hpp | 83 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()
764 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmNoCifgNoPeepholeNoProjection() local
954 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmCifgPeepholeNoProjection() local
1651 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmNoCifgPeepholeProjection() local
1864 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmCifgPeepholeNoProjectionBatch2() local
2066 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local
2262 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
|
| /external/android-nn-driver/test/1.3/ |
| D | QLstm.cpp | 602 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in QLstmWithProjection() local
762 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in QLstmWithNoProjection() local
923 hidl_vec<uint32_t> cellStateInDimensions{batchSize, numUnits}; in DynamicOutputQLstmWithNoProjection() local
|
| /external/armnn/delegate/test/ |
| D | UnidirectionalSequenceLstmTestHelper.hpp | 100 std::vector<int32_t> cellStateInDimensions{batchSize, numUnits}; in CreateUnidirectionalSequenceLstmTfLiteModel() local
|
| D | LstmTestHelper.hpp | 81 std::vector<int32_t> cellStateInDimensions{batchSize, numUnits}; in CreateLstmTfLiteModel() local
|