Home
last modified time | relevance | path

Searched refs:armcomputetensorutils (Results 1 – 25 of 142) sorted by relevance

123456

/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()
58 armnn::armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_Tensor); in Allocate()
99 return armcomputetensorutils::GetStrides(m_Tensor.info()->strides_in_bytes()); in GetStrides()
104 return armcomputetensorutils::GetShape(m_Tensor.info()->tensor_shape()); in GetShape()
148 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
153 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
159 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
163 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
168 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
[all …]
H A DClImportTensorHandle.hpp36 armnn::armcomputetensorutils::BuildArmComputeTensor(m_Tensor, tensorInfo); in ClImportTensorHandle()
44 armnn::armcomputetensorutils::BuildArmComputeTensor(m_Tensor, tensorInfo, dataLayout); in ClImportTensorHandle()
74 return armcomputetensorutils::GetStrides(m_Tensor.info()->strides_in_bytes()); in GetStrides()
79 return armcomputetensorutils::GetShape(m_Tensor.info()->tensor_shape()); in GetShape()
258 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
263 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
268 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
272 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
277 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
281 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
[all …]
/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()
59 armnn::armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_Tensor); in Allocate()
94 return armcomputetensorutils::GetStrides(m_Tensor.info()->strides_in_bytes()); in GetStrides()
99 return armcomputetensorutils::GetShape(m_Tensor.info()->tensor_shape()); in GetShape()
191 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
196 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
201 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
205 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
209 armcomputetensorutils::CopyArmComputeITensorData(this->GetTensor(), in CopyOutTo()
[all …]
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClBatchMatMulWorkload.cpp49 …const auto aclInputXInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inputX, descriptor.m_D… in ClBatchMatMulValidate()
50 …const auto aclInputYInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inputY, descriptor.m_D… in ClBatchMatMulValidate()
51 …const auto aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(output, descriptor.m_D… in ClBatchMatMulValidate()
61 …const auto aclPermutationXVector = armcomputetensorutils::BuildArmComputePermutationVector(permuta… in ClBatchMatMulValidate()
63 aclPermutedXInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permutedXInfo, 3); in ClBatchMatMulValidate()
75 …const auto aclPermutationYVector = armcomputetensorutils::BuildArmComputePermutationVector(permuta… in ClBatchMatMulValidate()
77 aclPermutedYInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permutedYInfo, 3); in ClBatchMatMulValidate()
142 …inputX.info()->set_data_layout(armcomputetensorutils::ConvertDataLayout(m_Data.m_Parameters.m_Data… in ClBatchMatMulWorkload()
143 arm_compute::TensorShape inputXTensorInfo = armcomputetensorutils::BuildArmComputeTensorShape( in ClBatchMatMulWorkload()
146 …inputY.info()->set_data_layout(armcomputetensorutils::ConvertDataLayout(m_Data.m_Parameters.m_Data… in ClBatchMatMulWorkload()
[all …]
H A DClBatchNormalizationFloatWorkload.cpp17 using namespace armcomputetensorutils;
29 armcomputetensorutils::BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
31 armcomputetensorutils::BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
33 armcomputetensorutils::BuildArmComputeTensorInfo(mean, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
35 armcomputetensorutils::BuildArmComputeTensorInfo(var, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
37 armcomputetensorutils::BuildArmComputeTensorInfo(beta, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
39 armcomputetensorutils::BuildArmComputeTensorInfo(gamma, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
H A DClTransposeWorkload.cpp21 …const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inpu… in ClTransposeWorkloadValidate()
22 …const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(out… in ClTransposeWorkloadValidate()
26 … armcomputetensorutils::BuildArmComputeTransposeVector(mappings)); in ClTransposeWorkloadValidate()
51 … armcomputetensorutils::BuildArmComputeTransposeVector(mappings)); in ClTransposeWorkload()
H A DClPermuteWorkload.cpp21 …const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inpu… in ClPermuteWorkloadValidate()
22 …const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(out… in ClPermuteWorkloadValidate()
26 … armcomputetensorutils::BuildArmComputePermutationVector(mappings)); in ClPermuteWorkloadValidate()
40 using armcomputetensorutils::BuildArmComputePermutationVector; in ClPermuteWorkload()
H A DClGatherNdWorkload.cpp12 using namespace armnn::armcomputetensorutils;
125 armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_FlattenedIndices); in ClGatherNdWorkload()
144 armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_FlattenedCoeff); in ClGatherNdWorkload()
153 armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_OutputMul); in ClGatherNdWorkload()
187 armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_OutputGather); in ClGatherNdWorkload()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp29 using namespace armcomputetensorutils;
75 armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_PermuteFirstOut); in ClUnidirectionalSequenceLstmFloatWorkload()
97 armcomputetensorutils::InitialiseArmComputeTensorEmpty(splitter_out); in ClUnidirectionalSequenceLstmFloatWorkload()
98 armcomputetensorutils::InitialiseArmComputeTensorEmpty(concat_in); in ClUnidirectionalSequenceLstmFloatWorkload()
349 armcomputetensorutils::InitialiseArmComputeTensorEmpty(*m_ScratchBuffer); in ClUnidirectionalSequenceLstmFloatWorkload()
436 armcomputetensorutils::InitialiseArmComputeTensorEmpty(concat_out); in ClUnidirectionalSequenceLstmFloatWorkload()
541 …const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inp… in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
542 …const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(out… in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
548 …arm_compute::TensorInfo aclPermuteOutInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permu… in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
576 … = armcomputetensorutils::BuildArmComputeTensorInfo(splitterTensorInfo); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
[all …]
H A DClPreluWorkload.cpp20 … const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input); in ClPreluWorkloadValidate()
21 … const arm_compute::TensorInfo aclAlpha = armcomputetensorutils::BuildArmComputeTensorInfo(alpha); in ClPreluWorkloadValidate()
22 …const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output); in ClPreluWorkloadValidate()
H A DClDivisionWorkload.cpp23 …const arm_compute::TensorInfo aclInput1 = armcomputetensorutils::BuildArmComputeTensorInfo(input0); in ClDivisionWorkloadValidate()
24 …const arm_compute::TensorInfo aclInput2 = armcomputetensorutils::BuildArmComputeTensorInfo(input1); in ClDivisionWorkloadValidate()
25 …const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output); in ClDivisionWorkloadValidate()
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonBatchMatMulWorkload.cpp37 …const auto aclInputXInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inputX, descriptor.m_D… in NeonBatchMatMulValidate()
38 …const auto aclInputYInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inputY, descriptor.m_D… in NeonBatchMatMulValidate()
39 const auto aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(output); in NeonBatchMatMulValidate()
51 …const auto aclPermutationXVector = armcomputetensorutils::BuildArmComputePermutationVector(permuta… in NeonBatchMatMulValidate()
53 aclPermutedXInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permutedXInfo); in NeonBatchMatMulValidate()
63 …const auto aclPermutationYVector = armcomputetensorutils::BuildArmComputePermutationVector(permuta… in NeonBatchMatMulValidate()
65 aclPermutedYInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permutedYInfo); in NeonBatchMatMulValidate()
140 …const auto aclPermutationXVector = armcomputetensorutils::BuildArmComputePermutationVector(permuta… in NeonBatchMatMulWorkload()
154 …const auto aclPermutationYVector = armcomputetensorutils::BuildArmComputePermutationVector(permuta… in NeonBatchMatMulWorkload()
H A DNeonDetectionPostProcessWorkload.cpp44 armcomputetensorutils::BuildArmComputeTensorInfo(boxEncodings); in NeonDetectionPostProcessValidate()
47 armcomputetensorutils::BuildArmComputeTensorInfo(scores); in NeonDetectionPostProcessValidate()
50 armcomputetensorutils::BuildArmComputeTensorInfo(anchors); in NeonDetectionPostProcessValidate()
53 armcomputetensorutils::BuildArmComputeTensorInfo(detectionBoxes); in NeonDetectionPostProcessValidate()
56 armcomputetensorutils::BuildArmComputeTensorInfo(detectionClasses); in NeonDetectionPostProcessValidate()
59 armcomputetensorutils::BuildArmComputeTensorInfo(detectionScores); in NeonDetectionPostProcessValidate()
62 armcomputetensorutils::BuildArmComputeTensorInfo(numDetections); in NeonDetectionPostProcessValidate()
H A DNeonBatchNormalizationWorkload.cpp21 using namespace armcomputetensorutils;
34 armcomputetensorutils::BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
36 armcomputetensorutils::BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
38 armcomputetensorutils::BuildArmComputeTensorInfo(mean, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
40 armcomputetensorutils::BuildArmComputeTensorInfo(var, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
42 armcomputetensorutils::BuildArmComputeTensorInfo(beta, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
44 armcomputetensorutils::BuildArmComputeTensorInfo(gamma, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
H A DNeonTransposeWorkload.cpp19 …const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inpu… in NeonTransposeWorkloadValidate()
20 …const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(out… in NeonTransposeWorkloadValidate()
24 … armcomputetensorutils::BuildArmComputeTransposeVector(mappings)); in NeonTransposeWorkloadValidate()
45 armcomputetensorutils::BuildArmComputeTransposeVector(mappings)); in NeonTransposeWorkload()
H A DNeonPermuteWorkload.cpp19 …const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inpu… in NeonPermuteWorkloadValidate()
20 …const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(out… in NeonPermuteWorkloadValidate()
24 … armcomputetensorutils::BuildArmComputePermutationVector(mappings)); in NeonPermuteWorkloadValidate()
37 using armcomputetensorutils::BuildArmComputePermutationVector; in NeonPermuteWorkload()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp29 using namespace armcomputetensorutils;
73 armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_PermuteFirstOut); in NeonUnidirectionalSequenceLstmFloatWorkload()
95 armcomputetensorutils::InitialiseArmComputeTensorEmpty(splitter_out); in NeonUnidirectionalSequenceLstmFloatWorkload()
96 armcomputetensorutils::InitialiseArmComputeTensorEmpty(concat_in); in NeonUnidirectionalSequenceLstmFloatWorkload()
352 armcomputetensorutils::InitialiseArmComputeTensorEmpty(*m_ScratchBuffer); in NeonUnidirectionalSequenceLstmFloatWorkload()
438 armcomputetensorutils::InitialiseArmComputeTensorEmpty(concat_out); in NeonUnidirectionalSequenceLstmFloatWorkload()
541 …const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inp… in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
542 …const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(ou… in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
548 …arm_compute::TensorInfo aclPermuteOutInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permu… in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
576 = armcomputetensorutils::BuildArmComputeTensorInfo(splitterTensorInfo); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
[all …]
H A DNeonUnidirectionalSequenceLstmWorkload.cpp30 using namespace armcomputetensorutils;
75 armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_PermuteFirstOut); in NeonUnidirectionalSequenceLstmWorkload()
97 armcomputetensorutils::InitialiseArmComputeTensorEmpty(splitter_out); in NeonUnidirectionalSequenceLstmWorkload()
98 armcomputetensorutils::InitialiseArmComputeTensorEmpty(concat_in); in NeonUnidirectionalSequenceLstmWorkload()
419 armcomputetensorutils::InitialiseArmComputeTensorEmpty(concat_out); in NeonUnidirectionalSequenceLstmWorkload()
522 …const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inp… in NeonUnidirectionalSequenceLstmWorkloadValidate()
523 …const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(ou… in NeonUnidirectionalSequenceLstmWorkloadValidate()
529 …arm_compute::TensorInfo aclPermuteOutInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permu… in NeonUnidirectionalSequenceLstmWorkloadValidate()
557 = armcomputetensorutils::BuildArmComputeTensorInfo(splitterTensorInfo); in NeonUnidirectionalSequenceLstmWorkloadValidate()
559 = armcomputetensorutils::BuildArmComputeTensorInfo(concatTensorInfo); in NeonUnidirectionalSequenceLstmWorkloadValidate()
H A DNeonQuantizeWorkload.cpp16 using namespace armcomputetensorutils;
20 …const arm_compute::TensorInfo neonInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(in… in NeonQuantizeWorkloadValidate()
21 …const arm_compute::TensorInfo neonOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(ou… in NeonQuantizeWorkloadValidate()
H A DNeonMinimumWorkload.cpp19 …const arm_compute::TensorInfo aclInput0 = armcomputetensorutils::BuildArmComputeTensorInfo(input0); in NeonMinimumWorkloadValidate()
20 …const arm_compute::TensorInfo aclInput1 = armcomputetensorutils::BuildArmComputeTensorInfo(input1); in NeonMinimumWorkloadValidate()
21 …const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output); in NeonMinimumWorkloadValidate()
H A DNeonMaximumWorkload.cpp18 …const arm_compute::TensorInfo aclInput0 = armcomputetensorutils::BuildArmComputeTensorInfo(input0); in NeonMaximumWorkloadValidate()
19 …const arm_compute::TensorInfo aclInput1 = armcomputetensorutils::BuildArmComputeTensorInfo(input1); in NeonMaximumWorkloadValidate()
20 …const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output); in NeonMaximumWorkloadValidate()
H A DNeonPreluWorkload.cpp21 … const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input); in NeonPreluWorkloadValidate()
22 … const arm_compute::TensorInfo aclAlpha = armcomputetensorutils::BuildArmComputeTensorInfo(alpha); in NeonPreluWorkloadValidate()
23 …const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output); in NeonPreluWorkloadValidate()
H A DNeonDivisionWorkload.cpp23 …const arm_compute::TensorInfo aclInput0 = armcomputetensorutils::BuildArmComputeTensorInfo(input0); in NeonDivisionWorkloadValidate()
24 …const arm_compute::TensorInfo aclInput1 = armcomputetensorutils::BuildArmComputeTensorInfo(input1); in NeonDivisionWorkloadValidate()
25 …const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output); in NeonDivisionWorkloadValidate()
H A DNeonMeanWorkload.cpp16 using namespace armcomputetensorutils;
22 …const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(inp… in NeonMeanWorkloadValidate()
23 …const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(out… in NeonMeanWorkloadValidate()
H A DNeonSubtractionWorkload.cpp27 …const arm_compute::TensorInfo aclInput0 = armcomputetensorutils::BuildArmComputeTensorInfo(input0); in NeonSubtractionWorkloadValidate()
28 …const arm_compute::TensorInfo aclInput1 = armcomputetensorutils::BuildArmComputeTensorInfo(input1); in NeonSubtractionWorkloadValidate()
29 …const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output); in NeonSubtractionWorkloadValidate()

123456