Home
last modified time | relevance | path

Searched refs:forgetGateBiasTensor (Results 1 – 9 of 9) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/layerTests/
DUnidirectionalSequenceLstmTestImpl.cpp150 armnn::ScopedTensorHandle forgetGateBiasTensor(tensorInfo4); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
164 AllocateAndCopyDataToITensorHandle(&forgetGateBiasTensor, forgetGateBias.data()); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
177 data.m_ForgetGateBias = &forgetGateBiasTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
339 armnn::ScopedTensorHandle forgetGateBiasTensor(tensorInfo4); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
352 AllocateAndCopyDataToITensorHandle(&forgetGateBiasTensor, forgetGateBias.data()); in UnidirectionalSequenceLstmLayerFloat32TestImpl()
365 data.m_ForgetGateBias = &forgetGateBiasTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl()
528 armnn::ScopedTensorHandle forgetGateBiasTensor(tensorInfo4); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
541 AllocateAndCopyDataToITensorHandle(&forgetGateBiasTensor, forgetGateBias.data()); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
554 data.m_ForgetGateBias = &forgetGateBiasTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
844 armnn::ScopedTensorHandle forgetGateBiasTensor(tensorInfo6); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
[all …]
DLstmTestImpl.cpp280 armnn::ScopedTensorHandle forgetGateBiasTensor(tensorInfo4); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
294 AllocateAndCopyDataToITensorHandle(&forgetGateBiasTensor, forgetGateBias.data()); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
307 data.m_ForgetGateBias = &forgetGateBiasTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
940 armnn::ScopedTensorHandle forgetGateBiasTensor(tensorInfo20); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
958 AllocateAndCopyDataToITensorHandle(&forgetGateBiasTensor, forgetGateBias.data()); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
976 data.m_ForgetGateBias = &forgetGateBiasTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1115 armnn::ScopedTensorHandle forgetGateBiasTensor(tensorInfoNumUnits); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1130 AllocateAndCopyDataToITensorHandle(&forgetGateBiasTensor, forgetGateBias.data()); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1145 data.m_ForgetGateBias = &forgetGateBiasTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1400 armnn::ScopedTensorHandle forgetGateBiasTensor(tensorInfo4); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
[all …]
/external/armnn/src/backends/reference/workloads/
DLstm.cpp38 std::unique_ptr<Decoder<float>>& forgetGateBiasTensor, in LstmImpl()
81 VectorBatchVectorAssign(*forgetGateBiasTensor, in LstmImpl()
160 VectorBatchVectorAdd(*forgetGateBiasTensor, in LstmImpl()
DLstm.hpp42 std::unique_ptr<Decoder<float>>& forgetGateBiasTensor,
DRefLstmWorkload.cpp136 std::unique_ptr<Decoder<float>> forgetGateBiasTensor = MakeDecoder<float>( in Execute() local
233 forgetGateBiasTensor, in Execute()
DRefUnidirectionalSequenceLstmWorkload.cpp167 std::unique_ptr<Decoder<float>> forgetGateBiasTensor = MakeDecoder<float>( in Execute() local
266 forgetGateBiasTensor, in Execute()
/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp96 armnn::ConstTensor forgetGateBiasTensor(biasInfo, forgetGateBiasTensorVector.data()); in CreateQuantizedLstmNetwork() local
113 data.m_ForgetGateBias = &forgetGateBiasTensor; in CreateQuantizedLstmNetwork()
DQLstmEndToEndTestImpl.cpp134 armnn::ConstTensor forgetGateBiasTensor(biasInfo, forgetGateBiasTensorVector.data()); in QLstmEndToEnd() local
157 params.m_ForgetGateBias = &forgetGateBiasTensor; in QLstmEndToEnd()
DWorkloadDataValidation.cpp543 armnn::ScopedTensorHandle forgetGateBiasTensor(tensorInfo4);
565 data.m_ForgetGateBias = &forgetGateBiasTensor;