Home
last modified time | relevance | path

Searched refs:BuildArmComputeTensorInfo (Results 1 – 25 of 131) sorted by relevance

123456

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQLstmWorkload.cpp255 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input); in NeonQLstmWorkloadValidate()
256 const arm_compute::TensorInfo aclOutputStateInInfo = BuildArmComputeTensorInfo(outputStateIn); in NeonQLstmWorkloadValidate()
257 const arm_compute::TensorInfo aclCellStateInInfo = BuildArmComputeTensorInfo(cellStateIn); in NeonQLstmWorkloadValidate()
259 const arm_compute::TensorInfo aclOutputStateOutInfo = BuildArmComputeTensorInfo(outputStateOut); in NeonQLstmWorkloadValidate()
260 const arm_compute::TensorInfo aclCellStateOutInfo = BuildArmComputeTensorInfo(cellStateOut); in NeonQLstmWorkloadValidate()
261 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output); in NeonQLstmWorkloadValidate()
265 = BuildArmComputeTensorInfo(paramsInfo.GetInputToForgetWeights()); in NeonQLstmWorkloadValidate()
267 = BuildArmComputeTensorInfo(paramsInfo.GetInputToCellWeights()); in NeonQLstmWorkloadValidate()
269 = BuildArmComputeTensorInfo(paramsInfo.GetInputToOutputWeights()); in NeonQLstmWorkloadValidate()
271 = BuildArmComputeTensorInfo(paramsInfo.GetRecurrentToForgetWeights()); in NeonQLstmWorkloadValidate()
[all …]
H A DNeonLstmFloatWorkload.cpp266 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input); in NeonLstmFloatWorkloadValidate()
267 const arm_compute::TensorInfo aclOutputStateInInfo = BuildArmComputeTensorInfo(outputStateIn); in NeonLstmFloatWorkloadValidate()
268 const arm_compute::TensorInfo aclCellStateInInfo = BuildArmComputeTensorInfo(cellStateIn); in NeonLstmFloatWorkloadValidate()
269 const arm_compute::TensorInfo aclScratchBufferInfo = BuildArmComputeTensorInfo(scratchBuffer); in NeonLstmFloatWorkloadValidate()
270 const arm_compute::TensorInfo aclOutputStateOutInfo = BuildArmComputeTensorInfo(outputStateOut); in NeonLstmFloatWorkloadValidate()
271 const arm_compute::TensorInfo aclCellStateOutInfo = BuildArmComputeTensorInfo(cellStateOut); in NeonLstmFloatWorkloadValidate()
272 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output); in NeonLstmFloatWorkloadValidate()
276 = BuildArmComputeTensorInfo(paramsInfo.GetInputToForgetWeights()); in NeonLstmFloatWorkloadValidate()
278 = BuildArmComputeTensorInfo(paramsInfo.GetInputToCellWeights()); in NeonLstmFloatWorkloadValidate()
280 = BuildArmComputeTensorInfo(paramsInfo.GetInputToOutputWeights()); in NeonLstmFloatWorkloadValidate()
[all …]
H A DNeonQuantizedLstmWorkload.cpp139 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input); in NeonQuantizedLstmWorkloadValidate()
140 const arm_compute::TensorInfo aclCellStateInInfo = BuildArmComputeTensorInfo(cellStateIn); in NeonQuantizedLstmWorkloadValidate()
141 const arm_compute::TensorInfo aclOutputStateInInfo = BuildArmComputeTensorInfo(outputStateIn); in NeonQuantizedLstmWorkloadValidate()
142 const arm_compute::TensorInfo aclCellStateOutInfo = BuildArmComputeTensorInfo(cellStateOut); in NeonQuantizedLstmWorkloadValidate()
143 const arm_compute::TensorInfo aclOutputStateOutInfo = BuildArmComputeTensorInfo(outputStateOut); in NeonQuantizedLstmWorkloadValidate()
147 = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in NeonQuantizedLstmWorkloadValidate()
149 = BuildArmComputeTensorInfo(paramsInfo.GetInputToForgetWeights()); in NeonQuantizedLstmWorkloadValidate()
151 = BuildArmComputeTensorInfo(paramsInfo.GetInputToCellWeights()); in NeonQuantizedLstmWorkloadValidate()
153 = BuildArmComputeTensorInfo(paramsInfo.GetInputToOutputWeights()); in NeonQuantizedLstmWorkloadValidate()
156 … = BuildArmComputeTensorInfo(paramsInfo.GetRecurrentToInputWeights()); in NeonQuantizedLstmWorkloadValidate()
[all …]
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp541 …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()
578 = armcomputetensorutils::BuildArmComputeTensorInfo(concatTensorInfo); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
615 const arm_compute::TensorInfo aclOutputStateInInfo = BuildArmComputeTensorInfo(outputStateIn); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
616 const arm_compute::TensorInfo aclCellStateInInfo = BuildArmComputeTensorInfo(cellStateIn); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
617 const arm_compute::TensorInfo aclScratchBufferInfo = BuildArmComputeTensorInfo(scratchBuffer); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
618 const arm_compute::TensorInfo aclOutputStateOutInfo = BuildArmComputeTensorInfo(outputStateOut); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
619 const arm_compute::TensorInfo aclCellStateOutInfo = BuildArmComputeTensorInfo(cellStateOut); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
[all …]
H A DNeonUnidirectionalSequenceLstmWorkload.cpp522 …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()
598 const arm_compute::TensorInfo aclOutputStateInInfo = BuildArmComputeTensorInfo(outputStateIn); in NeonUnidirectionalSequenceLstmWorkloadValidate()
599 const arm_compute::TensorInfo aclCellStateInInfo = BuildArmComputeTensorInfo(cellStateIn); in NeonUnidirectionalSequenceLstmWorkloadValidate()
600 const arm_compute::TensorInfo aclScratchBufferInfo = BuildArmComputeTensorInfo(scratchBuffer); in NeonUnidirectionalSequenceLstmWorkloadValidate()
601 const arm_compute::TensorInfo aclOutputStateOutInfo = BuildArmComputeTensorInfo(outputStateOut); in NeonUnidirectionalSequenceLstmWorkloadValidate()
602 const arm_compute::TensorInfo aclCellStateOutInfo = BuildArmComputeTensorInfo(cellStateOut); in NeonUnidirectionalSequenceLstmWorkloadValidate()
[all …]
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 DNeonGatherNdWorkload.cpp25 const arm_compute::TensorInfo aclIndicesInfo = BuildArmComputeTensorInfo(indices_W_ND_Info); in NeonGatherNdWorkloadValidate()
30 …const arm_compute::TensorInfo aclFlattenedCoeffInfo = BuildArmComputeTensorInfo(flattenedCoeff_Inf… in NeonGatherNdWorkloadValidate()
33 const arm_compute::TensorInfo aclOutputMulInfo = BuildArmComputeTensorInfo(indices_W_ND_Info); in NeonGatherNdWorkloadValidate()
47 …const arm_compute::TensorInfo aclFlattenedIndicesInfo = BuildArmComputeTensorInfo(flattenedIndices… in NeonGatherNdWorkloadValidate()
64 const arm_compute::TensorInfo aclParamsInfo = BuildArmComputeTensorInfo(params_K_C_Info); in NeonGatherNdWorkloadValidate()
69 … const arm_compute::TensorInfo aclOutputGatherInfo = BuildArmComputeTensorInfo(outputGather_Info); in NeonGatherNdWorkloadValidate()
76 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(outputInfo); in NeonGatherNdWorkloadValidate()
H A DNeonBatchNormalizationWorkload.cpp34 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 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()
53 aclPermutedXInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permutedXInfo); in NeonBatchMatMulValidate()
65 aclPermutedYInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permutedYInfo); in NeonBatchMatMulValidate()
H A DNeonConvolution3dWorkload.cpp32 …const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLay… in NeonConvolution3dWorkloadValidate()
33 …const arm_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.m_Dat… in NeonConvolution3dWorkloadValidate()
40 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in NeonConvolution3dWorkloadValidate()
43 …const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataL… in NeonConvolution3dWorkloadValidate()
H A DNeonFullyConnectedWorkload.cpp31 const arm_compute::TensorInfo aclInput = BuildArmComputeTensorInfo(input); in NeonFullyConnectedWorkloadValidate()
32 const arm_compute::TensorInfo aclOutput = BuildArmComputeTensorInfo(output); in NeonFullyConnectedWorkloadValidate()
33 arm_compute::TensorInfo aclWeights = BuildArmComputeTensorInfo(weights); in NeonFullyConnectedWorkloadValidate()
47 aclBiases = BuildArmComputeTensorInfo(biases.value()); in NeonFullyConnectedWorkloadValidate()
H A DNeonTransposeConvolution2dWorkload.cpp31 …const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataL… in NeonTransposeConvolution2dWorkloadValidate()
32 …const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_Data… in NeonTransposeConvolution2dWorkloadValidate()
33 …const arm_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.m_Dat… in NeonTransposeConvolution2dWorkloadValidate()
42 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in NeonTransposeConvolution2dWorkloadValidate()
H A DNeonDepthwiseConvolutionWorkload.cpp36 …const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataL… in NeonDepthwiseConvolutionWorkloadValidate()
37 …const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_Data… in NeonDepthwiseConvolutionWorkloadValidate()
53 …arm_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weightsPermuted, descriptor.m_D… in NeonDepthwiseConvolutionWorkloadValidate()
67 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in NeonDepthwiseConvolutionWorkloadValidate()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQLstmWorkload.cpp259 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input); in ClQLstmWorkloadValidate()
260 const arm_compute::TensorInfo aclOutputStateInInfo = BuildArmComputeTensorInfo(outputStateIn); in ClQLstmWorkloadValidate()
261 const arm_compute::TensorInfo aclCellStateInInfo = BuildArmComputeTensorInfo(cellStateIn); in ClQLstmWorkloadValidate()
263 const arm_compute::TensorInfo aclOutputStateOutInfo = BuildArmComputeTensorInfo(outputStateOut); in ClQLstmWorkloadValidate()
264 const arm_compute::TensorInfo aclCellStateOutInfo = BuildArmComputeTensorInfo(cellStateOut); in ClQLstmWorkloadValidate()
265 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output); in ClQLstmWorkloadValidate()
269 = BuildArmComputeTensorInfo(paramsInfo.GetInputToForgetWeights()); in ClQLstmWorkloadValidate()
271 = BuildArmComputeTensorInfo(paramsInfo.GetInputToCellWeights()); in ClQLstmWorkloadValidate()
273 = BuildArmComputeTensorInfo(paramsInfo.GetInputToOutputWeights()); in ClQLstmWorkloadValidate()
275 = BuildArmComputeTensorInfo(paramsInfo.GetRecurrentToForgetWeights()); in ClQLstmWorkloadValidate()
[all …]
H A DClLstmFloatWorkload.cpp253 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input); in ClLstmFloatWorkloadValidate()
254 const arm_compute::TensorInfo aclOutputStateInInfo = BuildArmComputeTensorInfo(outputStateIn); in ClLstmFloatWorkloadValidate()
255 const arm_compute::TensorInfo aclCellStateInInfo = BuildArmComputeTensorInfo(cellStateIn); in ClLstmFloatWorkloadValidate()
256 const arm_compute::TensorInfo aclScratchBufferInfo = BuildArmComputeTensorInfo(scratchBuffer); in ClLstmFloatWorkloadValidate()
257 const arm_compute::TensorInfo aclOutputStateOutInfo = BuildArmComputeTensorInfo(outputStateOut); in ClLstmFloatWorkloadValidate()
258 const arm_compute::TensorInfo aclCellStateOutInfo = BuildArmComputeTensorInfo(cellStateOut); in ClLstmFloatWorkloadValidate()
259 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output); in ClLstmFloatWorkloadValidate()
263 = BuildArmComputeTensorInfo(paramsInfo.GetInputToForgetWeights()); in ClLstmFloatWorkloadValidate()
265 = BuildArmComputeTensorInfo(paramsInfo.GetInputToCellWeights()); in ClLstmFloatWorkloadValidate()
267 = BuildArmComputeTensorInfo(paramsInfo.GetInputToOutputWeights()); in ClLstmFloatWorkloadValidate()
[all …]
H A DClQuantizedLstmWorkload.cpp24 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input); in ClQuantizedLstmWorkloadValidate()
25 …const arm_compute::TensorInfo aclPreviousCellStateInInfo = BuildArmComputeTensorInfo(previousCellS… in ClQuantizedLstmWorkloadValidate()
26 …const arm_compute::TensorInfo aclPreviousOutputInInfo = BuildArmComputeTensorInfo(previousOutpu… in ClQuantizedLstmWorkloadValidate()
29 …const arm_compute::TensorInfo aclCellStateOutInfo = BuildArmComputeTensorInfo(cellStateOut); in ClQuantizedLstmWorkloadValidate()
30 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output); in ClQuantizedLstmWorkloadValidate()
34 = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in ClQuantizedLstmWorkloadValidate()
36 = BuildArmComputeTensorInfo(paramsInfo.GetInputToForgetWeights()); in ClQuantizedLstmWorkloadValidate()
38 = BuildArmComputeTensorInfo(paramsInfo.GetInputToCellWeights()); in ClQuantizedLstmWorkloadValidate()
40 = BuildArmComputeTensorInfo(paramsInfo.GetInputToOutputWeights()); in ClQuantizedLstmWorkloadValidate()
42 … = BuildArmComputeTensorInfo(paramsInfo.GetRecurrentToInputWeights()); in ClQuantizedLstmWorkloadValidate()
[all …]
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp541 …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()
578 … = armcomputetensorutils::BuildArmComputeTensorInfo(concatTensorInfo); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
618 const arm_compute::TensorInfo aclOutputStateInInfo = BuildArmComputeTensorInfo(outputStateIn); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
619 const arm_compute::TensorInfo aclCellStateInInfo = BuildArmComputeTensorInfo(cellStateIn); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
620 const arm_compute::TensorInfo aclScratchBufferInfo = BuildArmComputeTensorInfo(scratchBuffer); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
621 const arm_compute::TensorInfo aclOutputStateOutInfo = BuildArmComputeTensorInfo(outputStateOut); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
622 const arm_compute::TensorInfo aclCellStateOutInfo = BuildArmComputeTensorInfo(cellStateOut); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
[all …]
H A DClGatherNdWorkload.cpp27 const arm_compute::TensorInfo aclIndicesInfo = BuildArmComputeTensorInfo(indices_W_ND_Info); in ClGatherNdWorkloadValidate()
32 …const arm_compute::TensorInfo aclFlattenedCoeffInfo = BuildArmComputeTensorInfo(flattenedCoeff_Inf… in ClGatherNdWorkloadValidate()
35 const arm_compute::TensorInfo aclOutputMulInfo = BuildArmComputeTensorInfo(indices_W_ND_Info); in ClGatherNdWorkloadValidate()
49 …const arm_compute::TensorInfo aclFlattenedIndicesInfo = BuildArmComputeTensorInfo(flattenedIndices… in ClGatherNdWorkloadValidate()
66 const arm_compute::TensorInfo aclParamsInfo = BuildArmComputeTensorInfo(params_K_C_Info); in ClGatherNdWorkloadValidate()
71 … const arm_compute::TensorInfo aclOutputGatherInfo = BuildArmComputeTensorInfo(outputGather_Info); in ClGatherNdWorkloadValidate()
78 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(outputInfo); in ClGatherNdWorkloadValidate()
H A DClBatchNormalizationFloatWorkload.cpp29 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 DClConvolution3dWorkload.cpp31 …const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLay… in ClConvolution3dWorkloadValidate()
32 …const arm_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.m_Dat… in ClConvolution3dWorkloadValidate()
39 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in ClConvolution3dWorkloadValidate()
43 …const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataL… in ClConvolution3dWorkloadValidate()
H A DClFullyConnectedWorkload.cpp26 const arm_compute::TensorInfo aclInput = BuildArmComputeTensorInfo(input); in ClFullyConnectedWorkloadValidate()
27 const arm_compute::TensorInfo aclOutput = BuildArmComputeTensorInfo(output); in ClFullyConnectedWorkloadValidate()
28 arm_compute::TensorInfo aclWeights = BuildArmComputeTensorInfo(weights); in ClFullyConnectedWorkloadValidate()
42 aclBiases = BuildArmComputeTensorInfo(biases.value()); in ClFullyConnectedWorkloadValidate()
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()
63 aclPermutedXInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permutedXInfo, 3); in ClBatchMatMulValidate()
77 aclPermutedYInfo = armcomputetensorutils::BuildArmComputeTensorInfo(permutedYInfo, 3); in ClBatchMatMulValidate()
H A DClTransposeConvolution2dWorkload.cpp32 …arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in ClTransposeConvolution2dWorkloadValidate()
33 …arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout… in ClTransposeConvolution2dWorkloadValidate()
34 …arm_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.m_DataLayou… in ClTransposeConvolution2dWorkloadValidate()
43 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in ClTransposeConvolution2dWorkloadValidate()
H A DClDepthwiseConvolutionWorkload.cpp33 …const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataL… in ClDepthwiseConvolutionWorkloadValidate()
34 …const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataL… in ClDepthwiseConvolutionWorkloadValidate()
50 …arm_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weightsPermuted, descriptor.m_D… in ClDepthwiseConvolutionWorkloadValidate()
64 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in ClDepthwiseConvolutionWorkloadValidate()
/aosp_15_r20/external/armnn/src/backends/aclCommon/
H A DArmComputeTensorUtils.hpp45 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo);
50 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo, unsigned int…
55 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo,
62 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo,
68 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo,
128 tensor.allocator()->init(BuildArmComputeTensorInfo(tensorInfo)); in BuildArmComputeTensor()
135 tensor.allocator()->init(BuildArmComputeTensorInfo(tensorInfo, dataLayout)); in BuildArmComputeTensor()

123456