Home
last modified time | relevance | path

Searched refs:GetTensorInfoForTfLiteTensor (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DConvolution.hpp101 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitConv2dOperator()
102 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitConv2dOperator()
118 const armnn::TensorInfo& filterTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteFilterTensor); in VisitConv2dOperator()
140 biasTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteBiasTensor); in VisitConv2dOperator()
290 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitConv3dOperator()
291 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitConv3dOperator()
307 const armnn::TensorInfo& filterTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteFilterTensor); in VisitConv3dOperator()
317 biasTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteBiasTensor); in VisitConv3dOperator()
512 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitDepthwiseConv2dOperator()
513 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitDepthwiseConv2dOperator()
[all …]
H A DBatchSpace.hpp50 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitBatchToSpaceNdOperator()
51 …const armnn::TensorInfo& blockShapeTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteBlockShapeTenso… in VisitBatchToSpaceNdOperator()
52 const armnn::TensorInfo& cropsTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteCropsTensor); in VisitBatchToSpaceNdOperator()
53 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, t… in VisitBatchToSpaceNdOperator()
150 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSpaceToBatchNdOperator()
151 …const armnn::TensorInfo& blockShapeTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteBlockShapeTenso… in VisitSpaceToBatchNdOperator()
152 … const armnn::TensorInfo& padListTensorInfo = GetTensorInfoForTfLiteTensor(tfLitePadListTensor); in VisitSpaceToBatchNdOperator()
153 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, t… in VisitSpaceToBatchNdOperator()
H A DUnidirectionalSequenceLstm.hpp114 …armnn::TensorInfo outputStateInInfo = GetTensorInfoForTfLiteTensor(tfLiteTensors[tfLiteNode->input… in VisitUnidirectionalSequenceLstmOperator()
115 …armnn::TensorInfo cellStateInInfo = GetTensorInfoForTfLiteTensor(tfLiteTensors[tfLiteNode->inputs-… in VisitUnidirectionalSequenceLstmOperator()
156 auto inputIntermediateTensorInfo = GetTensorInfoForTfLiteTensor( in VisitUnidirectionalSequenceLstmOperator()
158 auto forgetIntermediateTensorInfo = GetTensorInfoForTfLiteTensor( in VisitUnidirectionalSequenceLstmOperator()
160 auto cellIntermediateTensorInfo = GetTensorInfoForTfLiteTensor( in VisitUnidirectionalSequenceLstmOperator()
162 auto outputIntermediateTensorInfo = GetTensorInfoForTfLiteTensor( in VisitUnidirectionalSequenceLstmOperator()
180 …auto hiddentensorInfo = GetTensorInfoForTfLiteTensor(tfLiteTensors[tfLiteNode->intermediates->data… in VisitUnidirectionalSequenceLstmOperator()
184 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitUnidirectionalSequenceLstmOperator()
185 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitUnidirectionalSequenceLstmOperator()
H A DSplit.hpp45 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSplitOperator()
47 ARMNN_ASSERT(GetTensorInfoForTfLiteTensor(tfLiteAxisTensor).GetNumElements() == 1); in VisitSplitOperator()
73 outputs.push_back(GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true)); in VisitSplitOperator()
183 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSplitVOperator()
184 const armnn::TensorInfo& splitsTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteSplitsTensor); in VisitSplitVOperator()
186 ARMNN_ASSERT(GetTensorInfoForTfLiteTensor(tfLiteAxisTensor).GetNumElements() == 1); in VisitSplitVOperator()
230 outputs.push_back(GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true)); in VisitSplitVOperator()
H A DFullyConnected.hpp57 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitFullyConnectedOperator()
58 const armnn::TensorInfo& weightsTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteWeightsTensor); in VisitFullyConnectedOperator()
59 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitFullyConnectedOperator()
94 biasTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteBiasTensor); in VisitFullyConnectedOperator()
101 armnn::TensorInfo reshapedTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitFullyConnectedOperator()
120 armnn::TensorInfo reshapedOutputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor); in VisitFullyConnectedOperator()
H A DControl.hpp80 armnn::TensorInfo inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitConcatenationOperator()
117 armnn::TensorInfo inputTensorInfo = GetTensorInfoForTfLiteTensor( in VisitConcatenationOperator()
124 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitConcatenationOperator()
254 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitMeanOperator()
255 const armnn::TensorInfo& axisTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAxisTensor); in VisitMeanOperator()
256 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitMeanOperator()
H A DSpaceDepth.hpp38 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSpaceToDepthOperator()
39 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitSpaceToDepthOperator()
106 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitDepthToSpaceOperator()
107 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitDepthToSpaceOperator()
H A DQuantization.hpp48 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitDequantizeOperator()
49 armnn::TensorInfo outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitDequantizeOperator()
132 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitQuantizeOperator()
133 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitQuantizeOperator()
H A DPad.hpp60 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitPadOperator()
61 const armnn::TensorInfo& paddingTensorInfo = GetTensorInfoForTfLiteTensor(tfLitepaddingTensor); in VisitPadOperator()
62 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitPadOperator()
81 armnn::TensorInfo paddingValueTensorInfo = GetTensorInfoForTfLiteTensor(tfLitepaddingValue); in VisitPadOperator()
H A DNormalization.hpp38 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitL2NormalizationOperator()
39 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitL2NormalizationOperator()
106 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitLocalResponseNormalizationOperator()
107 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitLocalResponseNormalizationOperator()
H A DGatherNd.hpp46 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitGatherNdOperator()
47 const armnn::TensorInfo& indicesTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteIndicesTensor); in VisitGatherNdOperator()
48 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitGatherNdOperator()
H A DBatchMatMul.hpp58 …const armnn::TensorInfo& armnnLHSInputTensorInfo = GetTensorInfoForTfLiteTensor(kTfLiteLHSInputTen… in VisitBatchMatMulOperator()
59 …const armnn::TensorInfo& armnnRHSInputTensorInfo = GetTensorInfoForTfLiteTensor(kTfLiteRHSInputTen… in VisitBatchMatMulOperator()
60 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(kTfLiteOutputTensor, true… in VisitBatchMatMulOperator()
H A DLogicalBinary.hpp45 armnn::TensorInfo inputTensorInfo0 = GetTensorInfoForTfLiteTensor(tfLiteInputTensor0); in VisitLogicalBinaryOperator()
46 armnn::TensorInfo inputTensorInfo1 = GetTensorInfoForTfLiteTensor(tfLiteInputTensor1); in VisitLogicalBinaryOperator()
47 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitLogicalBinaryOperator()
H A DRedefine.hpp43 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitCastOperator()
44 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitCastOperator()
148 const armnn::TensorInfo& inputTensorInfo0 = GetTensorInfoForTfLiteTensor(tfLiteInputTensor0); in VisitReshapeOperator()
149 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitReshapeOperator()
H A DPrelu.hpp71 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitPreluOperator()
72 const armnn::TensorInfo& alphaTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAlphaTensor); in VisitPreluOperator()
73 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitPreluOperator()
H A DGather.hpp49 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitGatherOperator()
50 const armnn::TensorInfo& indicesTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteIndicesTensor); in VisitGatherOperator()
51 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitGatherOperator()
H A DComparison.hpp61 armnn::TensorInfo inputTensorInfo0 = GetTensorInfoForTfLiteTensor(tfLiteInputTensor0); in VisitComparisonOperator()
62 armnn::TensorInfo inputTensorInfo1 = GetTensorInfoForTfLiteTensor(tfLiteInputTensor1); in VisitComparisonOperator()
63 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitComparisonOperator()
H A DPooling.hpp49 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitPooling2dOperator()
50 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitPooling2dOperator()
172 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitPooling3dOperator()
173 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitPooling3dOperator()
H A DLstm.hpp112 …armnn::TensorInfo outputStateInInfo = GetTensorInfoForTfLiteTensor(tfLiteTensors[tfLiteNode->input… in VisitLstmOperator()
113 …armnn::TensorInfo cellStateInInfo = GetTensorInfoForTfLiteTensor(tfLiteTensors[tfLiteNode->inputs-… in VisitLstmOperator()
151 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitLstmOperator()
152 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitLstmOperator()
H A DReduce.hpp39 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitReduceOperator()
40 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitReduceOperator()
49 const armnn::TensorInfo& axisTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAxisTensor); in VisitReduceOperator()
H A DRound.hpp40 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitFloorOperator()
44 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitFloorOperator()
H A DElementwiseUnary.hpp49 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitElementwiseUnaryOperator()
50 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitElementwiseUnaryOperator()
H A DShape.hpp41 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitShapeOperator()
42 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitShapeOperator()
H A DFill.hpp54 armnn::TensorInfo inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitFillOperator()
55 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitFillOperator()
H A DTranspose.hpp60 const armnn::TensorInfo& inputTensorInfo0 = GetTensorInfoForTfLiteTensor(tfLiteInputTensor0); in VisitTransposeOperator()
61 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitTransposeOperator()

12