/aosp_15_r20/external/armnn/src/armnn/layers/ |
H A D | QuantizedLstmLayer.cpp | 33 …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 D | UnidirectionalSequenceLstmLayer.cpp | 30 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 D | LstmLayer.cpp | 30 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 D | QLstmLayer.cpp | 30 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 D | QuantizedLstmParams.hpp | 22 , 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 D | LstmParams.hpp | 21 , 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 D | WorkloadData.hpp | 403 , 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 D | ConstTensorLayerVisitor.cpp | 93 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 D | ShapeInferenceTests.cpp | 464 …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 D | ConstTensorLayerVisitor.hpp | 265 inputParams.m_RecurrentToForgetWeights, in CheckInputParameters() 266 layer->m_BasicParameters.m_RecurrentToForgetWeights); in CheckInputParameters()
|
/aosp_15_r20/external/armnn/src/armnnSerializer/test/ |
H A D | LstmSerializationTests.cpp | 34 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 D | armnn_lstmparam.i | 81 const armnn::ConstTensor* m_RecurrentToForgetWeights; member 149 const armnn::ConstTensor* m_RecurrentToForgetWeights; member
|
/aosp_15_r20/external/armnn/delegate/classic/src/ |
H A D | Lstm.hpp | 70 params.m_RecurrentToForgetWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 6); in VisitLstmOperator() 174 paramsInfo.m_RecurrentToForgetWeights = &(params.m_RecurrentToForgetWeights->GetInfo()); in VisitLstmOperator()
|
H A D | UnidirectionalSequenceLstm.hpp | 72 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 D | NeonQuantizedLstmWorkload.cpp | 38 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonQuantizedLstmWorkload() 99 m_Data.m_RecurrentToForgetWeights); in NeonQuantizedLstmWorkload()
|
H A D | NeonQLstmWorkload.cpp | 39 …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 D | ClQuantizedLstmWorkload.cpp | 85 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClQuantizedLstmWorkload() 130 …ializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights); in ClQuantizedLstmWorkload()
|
H A D | ClQLstmWorkload.cpp | 41 …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 D | UnidirectionalSequenceLstmTestImpl.cpp | 173 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 D | ConversionUtils_1_3.hpp | 540 params.m_RecurrentToForgetWeights = recurrentToForgetWeightsPin.GetConstTensorPtr(); in ConvertQuantizedLstm() 618 paramsInfo.m_RecurrentToForgetWeights = &(params.m_RecurrentToForgetWeights->GetInfo()); in ConvertQuantizedLstm()
|
H A D | ConversionUtils_1_2.hpp | 1967 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 D | WorkloadFactory.cpp | 643 …= 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 D | HalPolicy.cpp | 369 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 D | IsLayerSupportedTestImpl.hpp | 324 …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 D | Network.cpp | 2339 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()
|