Searched defs:cellToForgetWeightsDimensions (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/android-nn-driver/test/ |
| H A D | UnidirectionalSequenceLstm.hpp | 104 const hidl_vec<uint32_t>& cellToForgetWeightsDimensions, in UnidirectionalSequenceLstmTestImpl()
441 hidl_vec<uint32_t> cellToForgetWeightsDimensions{0}; in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
646 hidl_vec<uint32_t> cellToForgetWeightsDimensions{0}; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
875 hidl_vec<uint32_t> cellToForgetWeightsDimensions{numUnits}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
1097 hidl_vec<uint32_t> cellToForgetWeightsDimensions{numUnits}; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1298 hidl_vec<uint32_t> cellToForgetWeightsDimensions{numUnits}; in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTestImpl() local
|
| H A D | Lstm.hpp | 103 const hidl_vec<uint32_t>& cellToForgetWeightsDimensions, in LstmTestImpl()
735 hidl_vec<uint32_t> cellToForgetWeightsDimensions{0}; in LstmNoCifgNoPeepholeNoProjection() local
925 hidl_vec<uint32_t> cellToForgetWeightsDimensions{4}; in LstmCifgPeepholeNoProjection() local
1520 hidl_vec<uint32_t> cellToForgetWeightsDimensions{numUnits}; in LstmNoCifgPeepholeProjection() local
1835 hidl_vec<uint32_t> cellToForgetWeightsDimensions{numUnits}; in LstmCifgPeepholeNoProjectionBatch2() local
2035 hidl_vec<uint32_t> cellToForgetWeightsDimensions{numUnits}; in LstmNoCifgPeepholeProjectionNoClippingLayerNorm() local
2231 hidl_vec<uint32_t> cellToForgetWeightsDimensions{numUnits}; in LstmCifgPeepholeProjectionNoClippingLayerNorm() local
|
| /aosp_15_r20/external/android-nn-driver/test/1.3/ |
| H A D | QLstm.cpp | 82 const hidl_vec<uint32_t>& cellToForgetWeightsDimensions, in QLstmTestImpl()
575 hidl_vec<uint32_t> cellToForgetWeightsDimensions{0}; in QLstmWithProjection() local
735 hidl_vec<uint32_t> cellToForgetWeightsDimensions{0}; in QLstmWithNoProjection() local
896 hidl_vec<uint32_t> cellToForgetWeightsDimensions{0}; in DynamicOutputQLstmWithNoProjection() local
|
| /aosp_15_r20/external/armnn/src/armnn/test/ |
| H A D | ConstTensorLayerVisitor.cpp | 1018 std::vector<unsigned int> cellToForgetWeightsDimensions = {1, 1, 3, 3}; variable
1121 std::vector<unsigned int> cellToForgetWeightsDimensions = {1, 1, 3, 3}; variable
1245 std::vector<unsigned int> cellToForgetWeightsDimensions = {1, 1, 3, 3}; variable
1834 std::vector<unsigned int> cellToForgetWeightsDimensions = {1, 1, 3, 3}; variable
1941 std::vector<unsigned int> cellToForgetWeightsDimensions = {1, 1, 3, 3}; variable
|