| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | Convolution.hpp | 48 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 D | SpaceDepth.hpp | 26 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 D | Quantization.hpp | 28 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 D | Normalization.hpp | 26 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 D | Control.hpp | 74 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 D | BatchSpace.hpp | 26 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 D | Pooling.hpp | 29 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 D | Split.hpp | 39 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 D | FullyConnected.hpp | 39 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 D | Round.hpp | 28 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 D | GatherNd.hpp | 28 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 D | ElementwiseUnary.hpp | 30 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 D | Shape.hpp | 29 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 D | Fill.hpp | 36 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 D | Prelu.hpp | 53 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 D | Pack.hpp | 39 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 D | Gather.hpp | 28 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 D | Activation.hpp | 52 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 D | ArgMinMax.hpp | 27 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 D | Softmax.hpp | 67 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 D | Reduce.hpp | 27 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 D | Resize.hpp | 56 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 D | Pad.hpp | 38 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 D | Unpack.hpp | 30 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 D | Redefine.hpp | 37 … tfLiteInputTensor = 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()
|