Searched defs:forgetLayerNormWeightsDimensions (Results 1 – 4 of 4) sorted by relevance
84 void UnidirectionalSequenceLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in UnidirectionalSequenceLstmTestImpl()497 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in UnidirectionalSequenceLstmLayerFloat32TestImpl() local702 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local942 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local1157 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local1354 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTestImpl() local
83 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()788 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in LstmNoCifgNoPeepholeNoProjection() local978 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in LstmCifgPeepholeNoProjection() local1675 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in LstmNoCifgPeepholeProjection() local1888 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in LstmCifgPeepholeNoProjectionBatch2() local2090 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local2286 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
62 void QLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QLstmTestImpl()609 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in QLstmWithProjection() local769 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in QLstmWithNoProjection() local930 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in DynamicOutputQLstmWithNoProjection() local
2168 std::vector<unsigned int> forgetLayerNormWeightsDimensions = {1, 1, 3, 3}; variable