Home
last modified time | relevance | path

Searched defs:forgetLayerNormWeightsDimensions (Results 1 – 3 of 3) sorted by relevance

/external/android-nn-driver/test/1.3/
DQLstm.cpp95 void QLstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in QLstmTestImpl()
641 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in QLstmWithProjection() local
801 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in QLstmWithNoProjection() local
962 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in DynamicOutputQLstmWithNoProjection() local
/external/android-nn-driver/test/
DLstm.hpp106 void LstmTestImpl(const hidl_vec<uint32_t>& inputDimensions, in LstmTestImpl()
808 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in LstmNoCifgNoPeepholeNoProjection() local
998 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in LstmCifgPeepholeNoProjection() local
1695 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in LstmNoCifgPeepholeProjection() local
1908 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{0}; in LstmCifgPeepholeNoProjectionBatch2() local
2110 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local
2306 hidl_vec<uint32_t> forgetLayerNormWeightsDimensions{numUnits}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
/external/armnn/src/armnn/test/
DConstTensorLayerVisitor.cpp2060 std::vector<unsigned int> forgetLayerNormWeightsDimensions = {1, 1, 3, 3}; in BOOST_AUTO_TEST_CASE() local