Home
last modified time | relevance | path

Searched defs:forgetLayerNormWeights (Results 1 – 13 of 13) sorted by relevance

/external/armnn/delegate/test/
DUnidirectionalSequenceLstmTest.cpp91 std::vector<float> forgetLayerNormWeights; in UnidirectionalSequenceLstmTest() local
240 std::vector<float> forgetLayerNormWeights; in UnidirectionalSequenceLstmTimeMajorTest() local
427 std::vector<float> forgetLayerNormWeights; in UnidirectionalSequenceLstmNoCifgWithPeepholeWithProjectionTest() local
565 std::vector<float> forgetLayerNormWeights; in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest() local
720 std::vector<float> forgetLayerNormWeights = { 0.1f, 0.2f, 0.3f, 0.5f, 0.2f }; in UnidirectionalSequenceLstmNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
840 std::vector<float> forgetLayerNormWeights; in UnidirectionalSequenceLstmInt8Test() local
962 std::vector<float> forgetLayerNormWeights; in UnidirectionalSequenceLstmInt8TimeMajorTest() local
1082 std::vector<float> forgetLayerNormWeights; in UnidirectionalSequenceLstmInt8NoCifgWithPeepholeWithProjectionTest() local
1203 std::vector<float> forgetLayerNormWeights; in UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest() local
1328 std::vector<float> forgetLayerNormWeights = { -0.1f, 0.2f, 0.3f, 0.5f, 0.2f }; in UnidirectionalSequenceLstmInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest() local
DUnidirectionalSequenceLstmTestHelper.hpp35 std::vector<char> CreateUnidirectionalSequenceLstmTfLiteModel(tflite::TensorType tensorType, in CreateUnidirectionalSequenceLstmTfLiteModel()
578 void UnidirectionalSequenceLstmTestImpl(std::vector<armnn::BackendId>& backends, in UnidirectionalSequenceLstmTestImpl()
DLstmTestHelper.hpp25 std::vector<char> CreateLstmTfLiteModel(tflite::TensorType tensorType, in CreateLstmTfLiteModel()
548 void LstmTestImpl(std::vector<armnn::BackendId>& backends, in LstmTestImpl()
DLstmTest.cpp107 std::vector<float> forgetLayerNormWeights; in LstmTest() local
/external/armnn/src/backends/reference/workloads/
DLstm.cpp13 void LstmImpl(const LstmDescriptor& descriptor, in LstmImpl()
DRefLstmWorkload.cpp151 std::unique_ptr<Decoder<float>> forgetLayerNormWeights; in Execute() local
DRefUnidirectionalSequenceLstmWorkload.cpp182 std::unique_ptr<Decoder<float>> forgetLayerNormWeights; in Execute() local
/external/armnn/src/backends/backendsCommon/test/
DQLstmEndToEndTestImpl.cpp143 …armnn::ConstTensor forgetLayerNormWeights(layerNormWeightsInfo, forgetLayerNormWeightsVector.data(… in QLstmEndToEnd() local
/external/armnn/src/backends/backendsCommon/test/layerTests/
DLstmTestImpl.cpp1383 std::vector<float> forgetLayerNormWeights = {0.2f, 0.2f, 0.4f, 0.3f}; //{numUnits} in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
1823 std::vector<int16_t> forgetLayerNormWeights = {6553, 6553, 13107, 9830}; in QLstmTestImpl() local
2068 std::vector<int16_t> forgetLayerNormWeights = {6553, 6553, 13107, 9830}; in QLstmTestImpl1() local
2331 std::vector<int16_t> forgetLayerNormWeights = {6553, 6553, 13107, 9830}; in QLstmTestImpl2() local
DUnidirectionalSequenceLstmTestImpl.cpp1056 std::vector<float> forgetLayerNormWeights{ 0.1f, 0.2f, 0.3f, 0.5f, 0.2f }; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest() local
1926 std::vector<float> forgetLayerNormWeights = { -0.1f, 0.2f, 0.3f, 0.5f, 0.2f }; in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest() local
/external/armnn/src/armnnSerializer/
DSerializer.cpp741 flatbuffers::Offset<serializer::ConstTensor> forgetLayerNormWeights; in SerializeLstmLayer() local
1574 flatbuffers::Offset<serializer::ConstTensor> forgetLayerNormWeights; in SerializeQLstmLayer() local
1718 flatbuffers::Offset<serializer::ConstTensor> forgetLayerNormWeights; in SerializeUnidirectionalSequenceLstmLayer() local
/external/armnn/src/armnnDeserializer/
DDeserializer.cpp3303 armnn::ConstTensor forgetLayerNormWeights; in ParseLstm() local
3453 armnn::ConstTensor forgetLayerNormWeights; in ParseQLstm() local
3872 armnn::ConstTensor forgetLayerNormWeights; in ParseUnidirectionalSequenceLstm() local
/external/armnn/src/armnn/test/
DConstTensorLayerVisitor.cpp2169 ConstTensor forgetLayerNormWeights( variable