Home
last modified time | relevance | path

Searched refs:m_RecurrentToForgetWeights (Results 1 – 25 of 46) sorted by relevance

12

/aosp_15_r20/external/armnn/src/armnn/layers/
H A DQuantizedLstmLayer.cpp33 …descriptor.m_RecurrentToForgetWeights = m_QuantizedLstmParameters.m_RecurrentToForgetWeights.get(); in CreateWorkload()
62 …layer->m_QuantizedLstmParameters.m_RecurrentToForgetWeights = m_QuantizedLstmParameters.m_Recurren… in Clone()
63 ? m_QuantizedLstmParameters.m_RecurrentToForgetWeights : nullptr; in Clone()
125 ARMNN_ASSERT_MSG(m_QuantizedLstmParameters.m_RecurrentToForgetWeights != nullptr, in ValidateTensorShapesFromInputs()
162 m_QuantizedLstmParameters.m_RecurrentToForgetWeights, in GetConstantTensorsByRef()
183 …TensorHandle managedRecurrentToForgetWeights(m_QuantizedLstmParameters.m_RecurrentToForgetWeights); in ExecuteStrategy()
225 if (m_QuantizedLstmParameters.m_RecurrentToForgetWeights != nullptr) in ExecuteStrategy()
H A DUnidirectionalSequenceLstmLayer.cpp30 descriptor.m_RecurrentToForgetWeights = m_BasicParameters.m_RecurrentToForgetWeights.get(); in CreateWorkload()
91 …layer->m_BasicParameters.m_RecurrentToForgetWeights = m_BasicParameters.m_RecurrentToForgetWeights in Clone()
92 m_BasicParameters.m_RecurrentToForgetWeights : nullptr; in Clone()
193 ARMNN_ASSERT_MSG(m_BasicParameters.m_RecurrentToForgetWeights != nullptr, in ValidateTensorShapesFromInputs()
283 m_BasicParameters.m_RecurrentToForgetWeights, in GetConstantTensorsByRef()
320 …gedConstTensorHandle managedRecurrentToForgetWeights(m_BasicParameters.m_RecurrentToForgetWeights); in ExecuteStrategy()
363 if (m_BasicParameters.m_RecurrentToForgetWeights != nullptr) in ExecuteStrategy()
H A DLstmLayer.cpp30 descriptor.m_RecurrentToForgetWeights = m_BasicParameters.m_RecurrentToForgetWeights.get(); in CreateWorkload()
91 …layer->m_BasicParameters.m_RecurrentToForgetWeights = m_BasicParameters.m_RecurrentToForgetWeights in Clone()
92 m_BasicParameters.m_RecurrentToForgetWeights : nullptr; in Clone()
191 ARMNN_ASSERT_MSG(m_BasicParameters.m_RecurrentToForgetWeights != nullptr, in ValidateTensorShapesFromInputs()
276 m_BasicParameters.m_RecurrentToForgetWeights, in GetConstantTensorsByRef()
313 …gedConstTensorHandle managedRecurrentToForgetWeights(m_BasicParameters.m_RecurrentToForgetWeights); in ExecuteStrategy()
356 if (m_BasicParameters.m_RecurrentToForgetWeights != nullptr) in ExecuteStrategy()
H A DQLstmLayer.cpp30 descriptor.m_RecurrentToForgetWeights = m_BasicParameters.m_RecurrentToForgetWeights.get(); in CreateWorkload()
91 …layer->m_BasicParameters.m_RecurrentToForgetWeights = m_BasicParameters.m_RecurrentToForgetWeights in Clone()
92 m_BasicParameters.m_RecurrentToForgetWeights : nullptr; in Clone()
194 ARMNN_ASSERT_MSG(m_BasicParameters.m_RecurrentToForgetWeights != nullptr, in ValidateTensorShapesFromInputs()
278 m_BasicParameters.m_RecurrentToForgetWeights, in GetConstantTensorsByRef()
313 …gedConstTensorHandle managedRecurrentToForgetWeights(m_BasicParameters.m_RecurrentToForgetWeights); in ExecuteStrategy()
356 if (m_BasicParameters.m_RecurrentToForgetWeights != nullptr) in ExecuteStrategy()
/aosp_15_r20/external/armnn/include/armnn/
H A DQuantizedLstmParams.hpp22 , m_RecurrentToForgetWeights(nullptr) in QuantizedLstmInputParams()
39 const ConstTensor* m_RecurrentToForgetWeights; member
85 return Deref(m_RecurrentToForgetWeights); in GetRecurrentToForgetWeights()
128 , m_RecurrentToForgetWeights(nullptr) in QuantizedLstmInputParamsInfo()
145 const TensorInfo* m_RecurrentToForgetWeights; member
188 return Deref(m_RecurrentToForgetWeights); in GetRecurrentToForgetWeights()
H A DLstmParams.hpp21 , m_RecurrentToForgetWeights(nullptr) in LstmInputParams()
45 const ConstTensor* m_RecurrentToForgetWeights; member
71 , m_RecurrentToForgetWeights(nullptr) in LstmInputParamsInfo()
94 const TensorInfo* m_RecurrentToForgetWeights; member
143 return Deref(m_RecurrentToForgetWeights); in GetRecurrentToForgetWeights()
/aosp_15_r20/external/armnn/include/armnn/backends/
H A DWorkloadData.hpp403 , m_RecurrentToForgetWeights(nullptr) in LstmQueueDescriptor()
427 const ConstTensorHandle* m_RecurrentToForgetWeights; member
565 , m_RecurrentToForgetWeights(nullptr) in QLstmQueueDescriptor()
589 const ConstTensorHandle* m_RecurrentToForgetWeights; member
618 , m_RecurrentToForgetWeights(nullptr) in QuantizedLstmQueueDescriptor()
634 const ConstTensorHandle* m_RecurrentToForgetWeights; member
699 , m_RecurrentToForgetWeights(nullptr) in UnidirectionalSequenceLstmQueueDescriptor()
723 const ConstTensorHandle* m_RecurrentToForgetWeights; member
/aosp_15_r20/external/armnn/src/armnn/test/
H A DConstTensorLayerVisitor.cpp93 m_InputParams.m_RecurrentToForgetWeights, in CheckInputParameters()
94 inputParams.m_RecurrentToForgetWeights); in CheckInputParameters()
649 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
731 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
830 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
934 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
1033 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
1155 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
1260 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
1358 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
[all …]
H A DShapeInferenceTests.cpp464 …layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<ScopedTensorHandle>(constTe…
552 …layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<ScopedTensorHandle>(constTe…
580 …layer->m_QuantizedLstmParameters.m_RecurrentToForgetWeights = std::make_unique<ScopedTensorHandle>…
H A DConstTensorLayerVisitor.hpp265 inputParams.m_RecurrentToForgetWeights, in CheckInputParameters()
266 layer->m_BasicParameters.m_RecurrentToForgetWeights); in CheckInputParameters()
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DLstmSerializationTests.cpp34 lstmInputParams.m_RecurrentToForgetWeights = &constants[i++]; in ConstantVector2LstmInputParams()
141 …"m_RecurrentToForgetWeights", m_InputParams.m_RecurrentToForgetWeights, params.m_RecurrentToForget… in VerifyInputParameters()
233 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
367 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
526 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
1299 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
1350 params.m_RecurrentToForgetWeights = &constants[i++]; in ConstantsVector2QuantizedLstmInputParams()
1412 … m_InputParams.m_RecurrentToForgetWeights, params.m_RecurrentToForgetWeights); in VerifyInputParameters()
1568 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
1714 params.m_RecurrentToForgetWeights = &recurrentToForgetWeights;
[all …]
/aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
H A Darmnn_lstmparam.i81 const armnn::ConstTensor* m_RecurrentToForgetWeights; member
149 const armnn::ConstTensor* m_RecurrentToForgetWeights; member
/aosp_15_r20/external/armnn/delegate/classic/src/
H A DLstm.hpp70 params.m_RecurrentToForgetWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 6); in VisitLstmOperator()
174 paramsInfo.m_RecurrentToForgetWeights = &(params.m_RecurrentToForgetWeights->GetInfo()); in VisitLstmOperator()
H A DUnidirectionalSequenceLstm.hpp72 params.m_RecurrentToForgetWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 6); in VisitUnidirectionalSequenceLstmOperator()
211 paramsInfo.m_RecurrentToForgetWeights = &(params.m_RecurrentToForgetWeights->GetInfo()); in VisitUnidirectionalSequenceLstmOperator()
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQuantizedLstmWorkload.cpp38 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonQuantizedLstmWorkload()
99 m_Data.m_RecurrentToForgetWeights); in NeonQuantizedLstmWorkload()
H A DNeonQLstmWorkload.cpp39 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonQLstmWorkload()
182 …itializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights); in NeonQLstmWorkload()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQuantizedLstmWorkload.cpp85 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClQuantizedLstmWorkload()
130 …ializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights); in ClQuantizedLstmWorkload()
H A DClQLstmWorkload.cpp41 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClQLstmWorkload()
188 …ializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights); in ClQLstmWorkload()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DUnidirectionalSequenceLstmTestImpl.cpp173 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
361 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TestImpl()
550 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerFloat32TimeMajorTestImpl()
875 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest()
1111 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest()
1295 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest()
1450 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerInt8Test()
1604 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerInt8TimeMajorTest()
1778 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest()
1981 data.m_RecurrentToForgetWeights = &recurrentToForgetWeightsTensor; in UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest()
[all …]
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils_1_3.hpp540 params.m_RecurrentToForgetWeights = recurrentToForgetWeightsPin.GetConstTensorPtr(); in ConvertQuantizedLstm()
618 paramsInfo.m_RecurrentToForgetWeights = &(params.m_RecurrentToForgetWeights->GetInfo()); in ConvertQuantizedLstm()
H A DConversionUtils_1_2.hpp1967 params.m_RecurrentToForgetWeights = recurrentToForgetWeightsPin.GetConstTensorPtr(); in ConvertQuantized16BitLstm()
1981 paramsInfo.m_RecurrentToForgetWeights = &(params.m_RecurrentToForgetWeights->GetInfo()); in ConvertQuantized16BitLstm()
2692 params.m_RecurrentToForgetWeights = recurrentToForgetWeightsPin.GetConstTensorPtr(); in ConvertLstm()
2776 paramsInfo.m_RecurrentToForgetWeights = &(params.m_RecurrentToForgetWeights->GetInfo()); in ConvertLstm()
3358 params.m_RecurrentToForgetWeights = recurrentToForgetWeightsPin.GetConstTensorPtr(); in ConvertUnidirectionalSequenceLstm()
3449 paramsInfo.m_RecurrentToForgetWeights = &(params.m_RecurrentToForgetWeights->GetInfo()); in ConvertUnidirectionalSequenceLstm()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadFactory.cpp643 …= OverrideDataType(cLayer->m_BasicParameters.m_RecurrentToForgetWeights->GetTensorInfo(), dataType… in IsLayerConfigurationSupported()
660 paramsInfo.m_RecurrentToForgetWeights = &recurrentToForgetWeights; in IsLayerConfigurationSupported()
955 paramsInfo.m_RecurrentToForgetWeights = in IsLayerConfigurationSupported()
956 &cLayer->m_BasicParameters.m_RecurrentToForgetWeights->GetTensorInfo(); in IsLayerConfigurationSupported()
1052 paramsInfo.m_RecurrentToForgetWeights = in IsLayerConfigurationSupported()
1053 &cLayer->m_QuantizedLstmParameters.m_RecurrentToForgetWeights->GetTensorInfo(); in IsLayerConfigurationSupported()
1422 …= OverrideDataType(cLayer->m_BasicParameters.m_RecurrentToForgetWeights->GetTensorInfo(), dataType… in IsLayerConfigurationSupported()
1439 paramsInfo.m_RecurrentToForgetWeights = &recurrentToForgetWeights; in IsLayerConfigurationSupported()
/aosp_15_r20/external/android-nn-driver/1.0/
H A DHalPolicy.cpp369 params.m_RecurrentToForgetWeights = recurrentToForgetWeightsPin.GetConstTensorPtr(); in ConvertLstm()
435 paramsInfo.m_RecurrentToForgetWeights = &(params.m_RecurrentToForgetWeights->GetInfo()); in ConvertLstm()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DIsLayerSupportedTestImpl.hpp324 …m_Layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<armnn::ScopedTensorHandle… in DummyLstmLayer()
374 …m_Layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<armnn::ScopedTensorHandle… in DummyUnidirectionalSequenceLstmLayer()
430 …m_Layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<armnn::ScopedTensorHandle… in DummyLayer()
503 …m_Layer->m_QuantizedLstmParameters.m_RecurrentToForgetWeights = std::make_unique<armnn::ScopedTens… in DummyLayer()
/aosp_15_r20/external/armnn/src/armnn/
H A DNetwork.cpp2339 layer->m_BasicParameters.m_RecurrentToForgetWeights = in AddLstmLayer()
2340 std::make_shared<ScopedTensorHandle>(*(params.m_RecurrentToForgetWeights)); in AddLstmLayer()
2588 layer->m_QuantizedLstmParameters.m_RecurrentToForgetWeights = in AddQuantizedLstmLayer()
2621 layer->m_BasicParameters.m_RecurrentToForgetWeights = in AddQLstmLayer()
2622 std::make_shared<ScopedTensorHandle>(*(params.m_RecurrentToForgetWeights)); in AddQLstmLayer()
2770 layer->m_BasicParameters.m_RecurrentToForgetWeights = in AddUnidirectionalSequenceLstmLayer()
2771 std::make_shared<ScopedTensorHandle>(*(params.m_RecurrentToForgetWeights)); in AddUnidirectionalSequenceLstmLayer()

12