| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonQLstmWorkload.cpp | 178 InitializeArmComputeTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); in NeonQLstmWorkload()
179 InitializeArmComputeTensorData(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights); in NeonQLstmWorkload()
180 InitializeArmComputeTensorData(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights); in NeonQLstmWorkload()
182 …InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeight… in NeonQLstmWorkload()
183 …InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights); in NeonQLstmWorkload()
184 …InitializeArmComputeTensorData(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeight… in NeonQLstmWorkload()
186 InitializeArmComputeTensorData(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias); in NeonQLstmWorkload()
187 InitializeArmComputeTensorData(*m_CellBiasTensor, m_Data.m_CellBias); in NeonQLstmWorkload()
188 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in NeonQLstmWorkload()
193 … InitializeArmComputeTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); in NeonQLstmWorkload()
[all …]
|
| H A D | NeonLstmFloatWorkload.cpp | 177 InitializeArmComputeTensorData(*m_InputToForgetWeightsTensor, in NeonLstmFloatWorkload()
179 InitializeArmComputeTensorData(*m_InputToCellWeightsTensor, in NeonLstmFloatWorkload()
181 InitializeArmComputeTensorData(*m_InputToOutputWeightsTensor, in NeonLstmFloatWorkload()
183 InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, in NeonLstmFloatWorkload()
185 InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, in NeonLstmFloatWorkload()
187 InitializeArmComputeTensorData(*m_RecurrentToOutputWeightsTensor, in NeonLstmFloatWorkload()
189 InitializeArmComputeTensorData(*m_ForgetGateBiasTensor, in NeonLstmFloatWorkload()
191 InitializeArmComputeTensorData(*m_CellBiasTensor, in NeonLstmFloatWorkload()
193 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, in NeonLstmFloatWorkload()
198 InitializeArmComputeTensorData(*m_InputToInputWeightsTensor, in NeonLstmFloatWorkload()
[all …]
|
| H A D | NeonQuantizedLstmWorkload.cpp | 83 InitializeArmComputeTensorData(*m_InputToInputWeightsTensor, in NeonQuantizedLstmWorkload()
86 InitializeArmComputeTensorData(*m_InputToForgetWeightsTensor, in NeonQuantizedLstmWorkload()
89 InitializeArmComputeTensorData(*m_InputToCellWeightsTensor, in NeonQuantizedLstmWorkload()
92 InitializeArmComputeTensorData(*m_InputToOutputWeightsTensor, in NeonQuantizedLstmWorkload()
95 InitializeArmComputeTensorData(*m_RecurrentToInputWeightsTensor, in NeonQuantizedLstmWorkload()
98 InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, in NeonQuantizedLstmWorkload()
101 InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, in NeonQuantizedLstmWorkload()
104 InitializeArmComputeTensorData(*m_RecurrentToOutputWeightsTensor, in NeonQuantizedLstmWorkload()
107 InitializeArmComputeTensorData(*m_InputGateBiasTensor, in NeonQuantizedLstmWorkload()
110 InitializeArmComputeTensorData(*m_ForgetGateBiasTensor, in NeonQuantizedLstmWorkload()
[all …]
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 354 InitializeArmComputeTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
355 InitializeArmComputeTensorData(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
356 InitializeArmComputeTensorData(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
357 …InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeight… in NeonUnidirectionalSequenceLstmFloatWorkload()
358 … InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
359 …InitializeArmComputeTensorData(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeight… in NeonUnidirectionalSequenceLstmFloatWorkload()
360 InitializeArmComputeTensorData(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias); in NeonUnidirectionalSequenceLstmFloatWorkload()
361 InitializeArmComputeTensorData(*m_CellBiasTensor, m_Data.m_CellBias); in NeonUnidirectionalSequenceLstmFloatWorkload()
362 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in NeonUnidirectionalSequenceLstmFloatWorkload()
366 InitializeArmComputeTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
[all …]
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 335 InitializeArmComputeTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); in NeonUnidirectionalSequenceLstmWorkload()
336 InitializeArmComputeTensorData(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights); in NeonUnidirectionalSequenceLstmWorkload()
337 InitializeArmComputeTensorData(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights); in NeonUnidirectionalSequenceLstmWorkload()
338 …InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeight… in NeonUnidirectionalSequenceLstmWorkload()
339 … InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights); in NeonUnidirectionalSequenceLstmWorkload()
340 …InitializeArmComputeTensorData(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeight… in NeonUnidirectionalSequenceLstmWorkload()
341 InitializeArmComputeTensorData(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias); in NeonUnidirectionalSequenceLstmWorkload()
342 InitializeArmComputeTensorData(*m_CellBiasTensor, m_Data.m_CellBias); in NeonUnidirectionalSequenceLstmWorkload()
343 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in NeonUnidirectionalSequenceLstmWorkload()
347 InitializeArmComputeTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); in NeonUnidirectionalSequenceLstmWorkload()
[all …]
|
| H A D | NeonBatchNormalizationWorkload.cpp | 103 InitializeArmComputeTensorData(*m_Mean, m_Data.m_Mean); in NeonBatchNormalizationWorkload()
104 InitializeArmComputeTensorData(*m_Variance, m_Data.m_Variance); in NeonBatchNormalizationWorkload()
105 InitializeArmComputeTensorData(*m_Gamma, m_Data.m_Gamma); in NeonBatchNormalizationWorkload()
106 InitializeArmComputeTensorData(*m_Beta, m_Data.m_Beta); in NeonBatchNormalizationWorkload()
|
| H A D | NeonFullyConnectedWorkload.cpp | 118 InitializeArmComputeTensorData(*m_WeightsTensor, m_WeightsTensorInfo, m_Data.m_Inputs[1]); in Execute()
122 InitializeArmComputeTensorData(*m_BiasesTensor, m_BiasesTensorInfo, m_Data.m_Inputs[2]); in Execute()
|
| H A D | NeonTransposeConvolution2dWorkload.cpp | 102 InitializeArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight); in NeonTransposeConvolution2dWorkload()
106 InitializeArmComputeTensorData(*m_BiasTensor, m_Data.m_Bias); in NeonTransposeConvolution2dWorkload()
|
| H A D | NeonWorkloadUtils.hpp | 60 inline void InitializeArmComputeTensorData(arm_compute::Tensor& tensor, in InitializeArmComputeTensorData() function
96 inline void InitializeArmComputeTensorData(arm_compute::Tensor& tensor, in InitializeArmComputeTensorData() function
|
| H A D | NeonConvolution2dWorkload.cpp | 164 InitializeArmComputeTensorData(*m_KernelTensor, m_KernelTensorInfo, m_Data.m_Inputs[1]); in Execute()
168 InitializeArmComputeTensorData(*m_BiasTensor, m_BiasTensorInfo, m_Data.m_Inputs[2]); in Execute()
|
| H A D | NeonDetectionPostProcessWorkload.cpp | 108 InitializeArmComputeTensorData(*m_Anchors, m_Data.m_Anchors); in NeonDetectionPostProcessWorkload()
|