/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()
|