Home
last modified time | relevance | path

Searched refs:BuildArmComputeTensor (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonLstmFloatWorkload.cpp33 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in NeonLstmFloatWorkload()
36 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
39 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in NeonLstmFloatWorkload()
42 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonLstmFloatWorkload()
45 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonLstmFloatWorkload()
48 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in NeonLstmFloatWorkload()
51 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in NeonLstmFloatWorkload()
54 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in NeonLstmFloatWorkload()
57 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonLstmFloatWorkload()
63 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
[all …]
H A DNeonQLstmWorkload.cpp30 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in NeonQLstmWorkload()
33 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonQLstmWorkload()
36 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in NeonQLstmWorkload()
39 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonQLstmWorkload()
42 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonQLstmWorkload()
45 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in NeonQLstmWorkload()
48 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in NeonQLstmWorkload()
51 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in NeonQLstmWorkload()
54 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonQLstmWorkload()
64 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in NeonQLstmWorkload()
[all …]
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp72 BuildArmComputeTensor(m_PermuteFirstOut, permuteOutInfo); in NeonUnidirectionalSequenceLstmFloatWorkload()
92 BuildArmComputeTensor(splitter_out, splitterTensorInfo); in NeonUnidirectionalSequenceLstmFloatWorkload()
93 BuildArmComputeTensor(concat_in, concatTensorInfo); in NeonUnidirectionalSequenceLstmFloatWorkload()
152 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in NeonUnidirectionalSequenceLstmFloatWorkload()
155 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmFloatWorkload()
158 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in NeonUnidirectionalSequenceLstmFloatWorkload()
161 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonUnidirectionalSequenceLstmFloatWorkload()
164 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonUnidirectionalSequenceLstmFloatWorkload()
167 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in NeonUnidirectionalSequenceLstmFloatWorkload()
170 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in NeonUnidirectionalSequenceLstmFloatWorkload()
[all …]
H A DNeonQuantizedLstmWorkload.cpp23 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
26 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in NeonQuantizedLstmWorkload()
29 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
32 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in NeonQuantizedLstmWorkload()
35 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in NeonQuantizedLstmWorkload()
38 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonQuantizedLstmWorkload()
41 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonQuantizedLstmWorkload()
44 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in NeonQuantizedLstmWorkload()
47 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in NeonQuantizedLstmWorkload()
50 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in NeonQuantizedLstmWorkload()
[all …]
H A DNeonUnidirectionalSequenceLstmWorkload.cpp74 BuildArmComputeTensor(m_PermuteFirstOut, permuteOutInfo); in NeonUnidirectionalSequenceLstmWorkload()
94 BuildArmComputeTensor(splitter_out, splitterTensorInfo); in NeonUnidirectionalSequenceLstmWorkload()
95 BuildArmComputeTensor(concat_in, concatTensorInfo); in NeonUnidirectionalSequenceLstmWorkload()
165 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in NeonUnidirectionalSequenceLstmWorkload()
168 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmWorkload()
171 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in NeonUnidirectionalSequenceLstmWorkload()
174 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonUnidirectionalSequenceLstmWorkload()
177 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonUnidirectionalSequenceLstmWorkload()
180 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in NeonUnidirectionalSequenceLstmWorkload()
183 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in NeonUnidirectionalSequenceLstmWorkload()
[all …]
H A DNeonBatchNormalizationWorkload.cpp79 BuildArmComputeTensor(*m_Mean, m_Data.m_Mean->GetTensorInfo()); in NeonBatchNormalizationWorkload()
82 BuildArmComputeTensor(*m_Variance, m_Data.m_Variance->GetTensorInfo()); in NeonBatchNormalizationWorkload()
85 BuildArmComputeTensor(*m_Gamma, m_Data.m_Gamma->GetTensorInfo()); in NeonBatchNormalizationWorkload()
88 BuildArmComputeTensor(*m_Beta, m_Data.m_Beta->GetTensorInfo()); in NeonBatchNormalizationWorkload()
H A DNeonGatherNdWorkload.cpp121 BuildArmComputeTensor(m_FlattenedIndices, flattenedIndices_Info); in NeonGatherNdWorkload()
140 BuildArmComputeTensor(m_FlattenedCoeff, flattenedCoeff_Info); in NeonGatherNdWorkload()
149 BuildArmComputeTensor(m_OutputMul, outputMul_Info); in NeonGatherNdWorkload()
181 BuildArmComputeTensor(m_OutputGather, outputGather_Info); in NeonGatherNdWorkload()
H A DNeonFullyConnectedWorkload.cpp74 BuildArmComputeTensor(*m_WeightsTensor, m_WeightsTensorInfo); in NeonFullyConnectedWorkload()
81 BuildArmComputeTensor(*m_BiasesTensor, m_BiasesTensorInfo); in NeonFullyConnectedWorkload()
H A DNeonTransposeConvolution2dWorkload.cpp70 …BuildArmComputeTensor(*m_KernelTensor, m_Data.m_Weight->GetTensorInfo(), m_Data.m_Parameters.m_Dat… in NeonTransposeConvolution2dWorkload()
75 …BuildArmComputeTensor(*m_BiasTensor, m_Data.m_Bias->GetTensorInfo(), m_Data.m_Parameters.m_DataLay… in NeonTransposeConvolution2dWorkload()
H A DNeonConvolution2dWorkload.cpp93 …BuildArmComputeTensor(*m_KernelTensor, info.m_InputTensorInfos[1], m_Data.m_Parameters.m_DataLayou… in NeonConvolution2dWorkload()
97 …BuildArmComputeTensor(*m_BiasTensor, info.m_InputTensorInfos[2], m_Data.m_Parameters.m_DataLayout); in NeonConvolution2dWorkload()
H A DNeonBatchMatMulWorkload.cpp143 BuildArmComputeTensor(m_PermutedTensorX, permutedXInfo); in NeonBatchMatMulWorkload()
157 BuildArmComputeTensor(m_PermutedTensorY, permutedYInfo); in NeonBatchMatMulWorkload()
H A DNeonDetectionPostProcessWorkload.cpp87 BuildArmComputeTensor(*m_Anchors, descriptor.m_Anchors->GetTensorInfo()); in NeonDetectionPostProcessWorkload()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClLstmFloatWorkload.cpp38 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in ClLstmFloatWorkload()
41 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClLstmFloatWorkload()
44 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in ClLstmFloatWorkload()
47 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClLstmFloatWorkload()
50 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClLstmFloatWorkload()
53 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in ClLstmFloatWorkload()
56 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in ClLstmFloatWorkload()
59 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in ClLstmFloatWorkload()
62 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClLstmFloatWorkload()
68 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in ClLstmFloatWorkload()
[all …]
H A DClQLstmWorkload.cpp32 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in ClQLstmWorkload()
35 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClQLstmWorkload()
38 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in ClQLstmWorkload()
41 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClQLstmWorkload()
44 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClQLstmWorkload()
47 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in ClQLstmWorkload()
50 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in ClQLstmWorkload()
53 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in ClQLstmWorkload()
56 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClQLstmWorkload()
66 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in ClQLstmWorkload()
[all …]
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp74 BuildArmComputeTensor(m_PermuteFirstOut, permuteOutInfo); in ClUnidirectionalSequenceLstmFloatWorkload()
94 BuildArmComputeTensor(splitter_out, splitterTensorInfo); in ClUnidirectionalSequenceLstmFloatWorkload()
95 BuildArmComputeTensor(concat_in, concatTensorInfo); in ClUnidirectionalSequenceLstmFloatWorkload()
154 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in ClUnidirectionalSequenceLstmFloatWorkload()
157 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClUnidirectionalSequenceLstmFloatWorkload()
160 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in ClUnidirectionalSequenceLstmFloatWorkload()
163 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClUnidirectionalSequenceLstmFloatWorkload()
166 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClUnidirectionalSequenceLstmFloatWorkload()
169 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in ClUnidirectionalSequenceLstmFloatWorkload()
172 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in ClUnidirectionalSequenceLstmFloatWorkload()
[all …]
H A DClQuantizedLstmWorkload.cpp70 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
73 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in ClQuantizedLstmWorkload()
76 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
79 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in ClQuantizedLstmWorkload()
82 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in ClQuantizedLstmWorkload()
85 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClQuantizedLstmWorkload()
88 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClQuantizedLstmWorkload()
91 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in ClQuantizedLstmWorkload()
94 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in ClQuantizedLstmWorkload()
97 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in ClQuantizedLstmWorkload()
[all …]
H A DClBatchNormalizationFloatWorkload.cpp67 BuildArmComputeTensor(*m_Mean, m_Data.m_Mean->GetTensorInfo()); in ClBatchNormalizationFloatWorkload()
70 BuildArmComputeTensor(*m_Variance, m_Data.m_Variance->GetTensorInfo()); in ClBatchNormalizationFloatWorkload()
73 BuildArmComputeTensor(*m_Gamma, m_Data.m_Gamma->GetTensorInfo()); in ClBatchNormalizationFloatWorkload()
76 BuildArmComputeTensor(*m_Beta, m_Data.m_Beta->GetTensorInfo()); in ClBatchNormalizationFloatWorkload()
H A DClGatherNdWorkload.cpp124 BuildArmComputeTensor(m_FlattenedIndices, flattenedIndices_Info); in ClGatherNdWorkload()
143 BuildArmComputeTensor(m_FlattenedCoeff, flattenedCoeff_Info); in ClGatherNdWorkload()
152 BuildArmComputeTensor(m_OutputMul, outputMul_Info); in ClGatherNdWorkload()
186 BuildArmComputeTensor(m_OutputGather, outputGather_Info); in ClGatherNdWorkload()
H A DClTransposeConvolution2dWorkload.cpp84 BuildArmComputeTensor(*m_WeightsTensor, weightInfo, m_Data.m_Parameters.m_DataLayout); in ClTransposeConvolution2dWorkload()
89 …BuildArmComputeTensor(*m_BiasesTensor, m_Data.m_Bias->GetTensorInfo(), m_Data.m_Parameters.m_DataL… in ClTransposeConvolution2dWorkload()
H A DClBatchMatMulWorkload.cpp162 armcomputetensorutils::BuildArmComputeTensor(m_PermutedTensorX, permutedXInfo); in ClBatchMatMulWorkload()
181 armcomputetensorutils::BuildArmComputeTensor(m_PermutedTensorY, permutedYInfo); in ClBatchMatMulWorkload()
/aosp_15_r20/external/armnn/src/backends/aclCommon/
H A DArmComputeTensorUtils.hpp126 void BuildArmComputeTensor(Tensor& tensor, const armnn::TensorInfo& tensorInfo) in BuildArmComputeTensor() function
133 void BuildArmComputeTensor(Tensor& tensor, const armnn::TensorInfo& tensorInfo, DataLayout dataLayo… in BuildArmComputeTensor() function
/aosp_15_r20/external/armnn/src/backends/cl/
H A DClTensorHandle.hpp34 armnn::armcomputetensorutils::BuildArmComputeTensor(m_Tensor, tensorInfo); in ClTensorHandle()
44 armnn::armcomputetensorutils::BuildArmComputeTensor(m_Tensor, tensorInfo, dataLayout); in ClTensorHandle()
H A DClImportTensorHandle.hpp36 armnn::armcomputetensorutils::BuildArmComputeTensor(m_Tensor, tensorInfo); in ClImportTensorHandle()
44 armnn::armcomputetensorutils::BuildArmComputeTensor(m_Tensor, tensorInfo, dataLayout); in ClImportTensorHandle()
/aosp_15_r20/external/armnn/src/backends/neon/
H A DNeonTensorHandle.hpp35 armnn::armcomputetensorutils::BuildArmComputeTensor(m_Tensor, tensorInfo); in NeonTensorHandle()
48 armnn::armcomputetensorutils::BuildArmComputeTensor(m_Tensor, tensorInfo, dataLayout); in NeonTensorHandle()