Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp145 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl() local
159 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
173 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
335 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmLayerFloat32TestImpl() local
348 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in UnidirectionalSequenceLstmLayerFloat32TestImpl()
361 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl()
524 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl() local
537 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
550 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
838 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo6x5); in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest() local
[all …]
H A DLstmTestImpl.cpp275 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl() local
289 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in LstmNoCifgNoPeepholeNoProjectionTestImpl()
303 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in LstmNoCifgNoPeepholeNoProjectionTestImpl()
934 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo20x16); in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl() local
953 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
971 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in LstmLayerNoCifgWithPeepholeWithProjectionTestImpl()
1119 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfoOutput); in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl() local
1134 …AllocateAndCopyDataToITensorHandle(&recurrentToForgetWeightsTensor, recurrentToForgetWeights.data(… in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1149 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in LstmLayerWithCifgWithPeepholeNoProjectionTestImpl()
1394 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo4x3); in LstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTestImpl() local
[all …]
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp31 std::unique_ptr<Decoder<float>>& recurrentToForgetWeightsTensor, in LstmImpl() argument
119 MatrixBatchVectorMultiplyAccumulate(*recurrentToForgetWeightsTensor, in LstmImpl()
H A DRefLstmWorkload.cpp128 std::unique_ptr<Decoder<float>> recurrentToForgetWeightsTensor = MakeDecoder<float>( in Execute() local
226 recurrentToForgetWeightsTensor, in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp159 std::unique_ptr<Decoder<float>> recurrentToForgetWeightsTensor = MakeDecoder<float>( in Execute() local
259 recurrentToForgetWeightsTensor, in Execute()
H A DLstm.hpp35 std::unique_ptr<Decoder<float>>& recurrentToForgetWeightsTensor,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQLstmEndToEndTestImpl.cpp123 armnn::ConstTensor recurrentToForgetWeightsTensor(recurrentWeightsInfo, in QLstmEndToEnd() local
153 params.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in QLstmEndToEnd()
H A DQuantizedLstmEndToEndTestImpl.cpp80 armnn::ConstTensor recurrentToForgetWeightsTensor(recurrentWeightsInfo, in CreateQuantizedLstmNetwork() local
109 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in CreateQuantizedLstmNetwork()
H A DWorkloadDataValidation.cpp537 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo4x3);
560 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor;