Home
last modified time | relevance | path

Searched refs:tfLiteInputTensor (Results 1 – 25 of 30) sorted by relevance

12

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DConvolution.hpp48 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitConv2dOperator() local
57 if (IsDynamicTensor(tfLiteInputTensor)) in VisitConv2dOperator()
101 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitConv2dOperator()
144 biasTensorInfo = armnn::TensorInfo(armnn::TensorShape({1}), GetDataType(tfLiteInputTensor)); in VisitConv2dOperator()
272 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitConv3dOperator() local
273 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitConv3dOperator()
290 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitConv3dOperator()
321 biasTensorInfo = armnn::TensorInfo(armnn::TensorShape({1}), GetDataType(tfLiteInputTensor)); in VisitConv3dOperator()
459 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitDepthwiseConv2dOperator() local
460 if(!IsValid(&tfLiteInputTensor)) in VisitDepthwiseConv2dOperator()
[all …]
H A DSpaceDepth.hpp26 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitSpaceToDepthOperator() local
27 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitSpaceToDepthOperator()
38 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSpaceToDepthOperator()
94 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitDepthToSpaceOperator() local
95 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitDepthToSpaceOperator()
106 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitDepthToSpaceOperator()
H A DQuantization.hpp28 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitDequantizeOperator() local
29 if (IsDynamicTensor(tfLiteInputTensor)) in VisitDequantizeOperator()
48 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitDequantizeOperator()
102 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitQuantizeOperator() local
103 if (IsDynamicTensor(tfLiteInputTensor)) in VisitQuantizeOperator()
132 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitQuantizeOperator()
H A DNormalization.hpp26 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitL2NormalizationOperator() local
27 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitL2NormalizationOperator()
38 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitL2NormalizationOperator()
94 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitLocalResponseNormalizationOperator() local
95 if (!IsValid(tfLiteContext, tfLiteInputTensor, normalizationOperatorCode, nodeIndex)) in VisitLocalResponseNormalizationOperator()
106 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitLocalResponseNormalizationOperator()
H A DControl.hpp74 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[i]]; in VisitConcatenationOperator() local
75 if (!IsValid(tfLiteContext, tfLiteInputTensor, tfLiteConcatOperatorCode, nodeIndex)) in VisitConcatenationOperator()
80 armnn::TensorInfo inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitConcatenationOperator()
200 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitMeanOperator() local
201 if(!IsValid(&tfLiteInputTensor)) in VisitMeanOperator()
209 if (IsDynamicTensor(tfLiteInputTensor)) in VisitMeanOperator()
254 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitMeanOperator()
H A DBatchSpace.hpp26 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitBatchToSpaceNdOperator() local
27 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitBatchToSpaceNdOperator()
50 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitBatchToSpaceNdOperator()
126 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitSpaceToBatchNdOperator() local
127 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitSpaceToBatchNdOperator()
150 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSpaceToBatchNdOperator()
H A DPooling.hpp29 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitPooling2dOperator() local
30 if (IsDynamicTensor(tfLiteInputTensor)) in VisitPooling2dOperator()
49 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitPooling2dOperator()
152 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitPooling3dOperator() local
153 if (IsDynamicTensor(tfLiteInputTensor)) in VisitPooling3dOperator()
172 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitPooling3dOperator()
H A DSplit.hpp39 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[1]]; in VisitSplitOperator() local
40 if (!IsValid(tfLiteContext, tfLiteInputTensor, tfLiteSplitOperatorCode, nodeIndex)) in VisitSplitOperator()
45 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSplitOperator()
165 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitSplitVOperator() local
166 if (!IsValid(tfLiteContext, tfLiteInputTensor, tfLiteSplitVOperatorCode, nodeIndex)) in VisitSplitVOperator()
183 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSplitVOperator()
H A DFullyConnected.hpp39 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitFullyConnectedOperator() local
40 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitFullyConnectedOperator()
57 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitFullyConnectedOperator()
98 biasTensorInfo = armnn::TensorInfo(armnn::TensorShape({1}), GetDataType(tfLiteInputTensor)); in VisitFullyConnectedOperator()
101 armnn::TensorInfo reshapedTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitFullyConnectedOperator()
H A DRound.hpp28 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitFloorOperator() local
29 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitFloorOperator()
40 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitFloorOperator()
H A DGatherNd.hpp28 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitGatherNdOperator() local
29 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitGatherNdOperator()
46 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitGatherNdOperator()
H A DElementwiseUnary.hpp30 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitElementwiseUnaryOperator() local
31 if (IsDynamicTensor(tfLiteInputTensor)) in VisitElementwiseUnaryOperator()
49 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitElementwiseUnaryOperator()
H A DShape.hpp29 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitShapeOperator() local
30 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitShapeOperator()
41 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitShapeOperator()
H A DFill.hpp36 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitFillOperator() local
37 if (!IsValid(tfLiteContext, tfLiteInputTensor, tfLiteFillOperatorCode, nodeIndex)) in VisitFillOperator()
54 armnn::TensorInfo inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitFillOperator()
H A DPrelu.hpp53 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitPreluOperator() local
54 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitPreluOperator()
71 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitPreluOperator()
H A DPack.hpp39 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[i]]; in VisitPackOperator() local
40 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitPackOperator()
45 armnn::TensorInfo inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitPackOperator()
H A DGather.hpp28 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitGatherOperator() local
29 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitGatherOperator()
49 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitGatherOperator()
H A DActivation.hpp52 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitActivationOperator() local
53 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitActivationOperator()
64 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitActivationOperator()
H A DArgMinMax.hpp27 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitArgMinMaxOperator() local
28 if (!IsValid(tfLiteContext, tfLiteInputTensor, argMinMaxOperatorCode, nodeIndex)) in VisitArgMinMaxOperator()
39 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitArgMinMaxOperator()
H A DSoftmax.hpp67 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitSoftmaxOperator() local
68 if (IsDynamicTensor(tfLiteInputTensor)) in VisitSoftmaxOperator()
86 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSoftmaxOperator()
H A DReduce.hpp27 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitReduceOperator() local
28 if (!IsValid(tfLiteContext, tfLiteInputTensor, reduceOperatorCode, nodeIndex)) in VisitReduceOperator()
39 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitReduceOperator()
H A DResize.hpp56 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitResizeOperator() local
57 if (IsDynamicTensor(tfLiteInputTensor)) in VisitResizeOperator()
89 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitResizeOperator()
H A DPad.hpp38 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitPadOperator() local
41 if (IsDynamicTensor(tfLiteInputTensor)) in VisitPadOperator()
60 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitPadOperator()
H A DUnpack.hpp30 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitUnpackOperator() local
32 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitUnpackOperator()
42 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitUnpackOperator()
/aosp_15_r20/external/armnn/delegate/opaque/src/
H A DRedefine.hpp37tfLiteInputTensor = TfLiteOpaqueContextGetOpaqueTensor(tfLiteContext, inputTensors[numInputs-1]); in VisitCastOperator() local
38 if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) in VisitCastOperator()
58 … const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteOpaqueTensor(tfLiteInputTensor); in VisitCastOperator()

12