Home
last modified time | relevance | path

Searched refs:tfLiteOutputTensor (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DQuantization.hpp38 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitDequantizeOperator() local
39 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitDequantizeOperator()
49 armnn::TensorInfo outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitDequantizeOperator()
112 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitQuantizeOperator() local
113 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitQuantizeOperator()
123 if (!IsAffineQuantization(tfLiteOutputTensor)) in VisitQuantizeOperator()
133 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitQuantizeOperator()
H A DConvolution.hpp65 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitConv2dOperator() local
66 if(!IsValid(&tfLiteOutputTensor)) in VisitConv2dOperator()
74 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitConv2dOperator()
102 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitConv2dOperator()
278 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitConv3dOperator() local
279 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitConv3dOperator()
291 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitConv3dOperator()
476 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitDepthwiseConv2dOperator() local
477 if(!IsValid(&tfLiteOutputTensor)) in VisitDepthwiseConv2dOperator()
485 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitDepthwiseConv2dOperator()
[all …]
H A DSpaceDepth.hpp32 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitSpaceToDepthOperator() local
33 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitSpaceToDepthOperator()
39 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitSpaceToDepthOperator()
100 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitDepthToSpaceOperator() local
101 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitDepthToSpaceOperator()
107 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitDepthToSpaceOperator()
H A DNormalization.hpp32 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitL2NormalizationOperator() local
33 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitL2NormalizationOperator()
39 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitL2NormalizationOperator()
100 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitLocalResponseNormalizationOperator() local
101 if (!IsValid(tfLiteContext, tfLiteOutputTensor, normalizationOperatorCode, nodeIndex)) in VisitLocalResponseNormalizationOperator()
107 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitLocalResponseNormalizationOperator()
H A DControl.hpp91 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitConcatenationOperator() local
92 if (!IsValid(tfLiteContext, tfLiteOutputTensor, tfLiteConcatOperatorCode, nodeIndex)) in VisitConcatenationOperator()
124 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitConcatenationOperator()
236 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitMeanOperator() local
237 if(!IsValid(&tfLiteOutputTensor)) in VisitMeanOperator()
245 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitMeanOperator()
256 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitMeanOperator()
H A DBatchSpace.hpp44 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitBatchToSpaceNdOperator() local
45 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitBatchToSpaceNdOperator()
53 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, t… in VisitBatchToSpaceNdOperator()
144 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitSpaceToBatchNdOperator() local
145 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitSpaceToBatchNdOperator()
153 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, t… in VisitSpaceToBatchNdOperator()
H A DRedefine.hpp37 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitCastOperator() local
38 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitCastOperator()
44 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitCastOperator()
142 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitReshapeOperator() local
143 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitReshapeOperator()
149 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitReshapeOperator()
H A DPooling.hpp39 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitPooling2dOperator() local
40 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitPooling2dOperator()
50 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitPooling2dOperator()
162 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitPooling3dOperator() local
163 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitPooling3dOperator()
173 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitPooling3dOperator()
H A DSplit.hpp68 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[i]]; in VisitSplitOperator() local
69 if (!IsValid(tfLiteContext, tfLiteOutputTensor, tfLiteSplitOperatorCode, nodeIndex)) in VisitSplitOperator()
73 outputs.push_back(GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true)); in VisitSplitOperator()
225 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[i]]; in VisitSplitVOperator() local
226 if (!IsValid(tfLiteContext, tfLiteOutputTensor, tfLiteSplitVOperatorCode, nodeIndex)) in VisitSplitVOperator()
230 outputs.push_back(GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true)); in VisitSplitVOperator()
H A DRound.hpp34 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitFloorOperator() local
35 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitFloorOperator()
44 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitFloorOperator()
H A DGatherNd.hpp40 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitGatherNdOperator() local
41 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitGatherNdOperator()
48 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitGatherNdOperator()
H A DElementwiseUnary.hpp39 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitElementwiseUnaryOperator() local
40 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitElementwiseUnaryOperator()
50 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitElementwiseUnaryOperator()
H A DShape.hpp35 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitShapeOperator() local
36 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitShapeOperator()
42 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitShapeOperator()
H A DLogicalBinary.hpp39 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitLogicalBinaryOperator() local
40 if (!IsValid(tfLiteContext, tfLiteOutputTensor, logicalOperatorCode, nodeIndex)) in VisitLogicalBinaryOperator()
47 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitLogicalBinaryOperator()
H A DFill.hpp48 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitFillOperator() local
49 if (!IsValid(tfLiteContext, tfLiteOutputTensor, tfLiteFillOperatorCode, nodeIndex)) in VisitFillOperator()
55 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitFillOperator()
H A DFullyConnected.hpp45 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitFullyConnectedOperator() local
46 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitFullyConnectedOperator()
59 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitFullyConnectedOperator()
120 armnn::TensorInfo reshapedOutputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor); in VisitFullyConnectedOperator()
H A DPrelu.hpp65 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitPreluOperator() local
66 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitPreluOperator()
73 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitPreluOperator()
H A DTranspose.hpp50 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitTransposeOperator() local
51 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitTransposeOperator()
61 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitTransposeOperator()
H A DPack.hpp57 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitPackOperator() local
58 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitPackOperator()
63 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitPackOperator()
H A DGather.hpp40 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitGatherOperator() local
41 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitGatherOperator()
51 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitGatherOperator()
H A DActivation.hpp58 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitActivationOperator() local
59 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitActivationOperator()
65 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitActivationOperator()
H A DArgMinMax.hpp33 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitArgMinMaxOperator() local
34 if (!IsValid(tfLiteContext, tfLiteOutputTensor, argMinMaxOperatorCode, nodeIndex)) in VisitArgMinMaxOperator()
40 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitArgMinMaxOperator()
H A DComparison.hpp51 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitComparisonOperator() local
52 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitComparisonOperator()
63 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitComparisonOperator()
H A DSoftmax.hpp76 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; in VisitSoftmaxOperator() local
77 if (IsDynamicTensor(tfLiteOutputTensor)) in VisitSoftmaxOperator()
87 …const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); in VisitSoftmaxOperator()
/aosp_15_r20/external/armnn/delegate/opaque/src/
H A DRedefine.hpp52tfLiteOutputTensor = TfLiteOpaqueContextGetOpaqueTensor(tfLiteContext, outputTensors[numOutputs-1]… in VisitCastOperator() local
53 if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) in VisitCastOperator()
59 …armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteOpaqueTensor(tfLiteOutputTensor, true); in VisitCastOperator()

12