Searched refs:tfLiteInputTensor (Results 1 – 9 of 9) sorted by relevance
28 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitDequantizeOperator() local29 if (IsDynamicTensor(tfLiteInputTensor)) in VisitDequantizeOperator()49 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitDequantizeOperator()89 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitQuantizeOperator() local90 if (IsDynamicTensor(tfLiteInputTensor)) in VisitQuantizeOperator()119 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitQuantizeOperator()
74 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[i]]; in VisitConcatenationOperator() local75 if(!IsValid(&tfLiteInputTensor)) in VisitConcatenationOperator()83 if (IsDynamicTensor(tfLiteInputTensor)) in VisitConcatenationOperator()92 armnn::TensorInfo inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitConcatenationOperator()193 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitMeanOperator() local194 if(!IsValid(&tfLiteInputTensor)) in VisitMeanOperator()202 if (IsDynamicTensor(tfLiteInputTensor)) in VisitMeanOperator()247 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitMeanOperator()
47 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitConv2dOperator() local56 if (IsDynamicTensor(tfLiteInputTensor)) in VisitConv2dOperator()100 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitConv2dOperator()129 biasTensorInfo = armnn::TensorInfo(armnn::TensorShape({1}), GetDataType(tfLiteInputTensor)); in VisitConv2dOperator()236 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitDepthwiseConv2dOperator() local237 if(!IsValid(&tfLiteInputTensor)) in VisitDepthwiseConv2dOperator()245 if (IsDynamicTensor(tfLiteInputTensor)) in VisitDepthwiseConv2dOperator()289 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitDepthwiseConv2dOperator()340 biasTensorInfo = armnn::TensorInfo(armnn::TensorShape({1}), GetDataType(tfLiteInputTensor)); in VisitDepthwiseConv2dOperator()461 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[2]]; in VisitTransposeConv2dOperator() local[all …]
37 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitFullyConnectedOperator() local46 if (IsDynamicTensor(tfLiteInputTensor)) in VisitFullyConnectedOperator()90 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitFullyConnectedOperator()129 biasTensorInfo = armnn::TensorInfo(armnn::TensorShape({1}), GetDataType(tfLiteInputTensor)); in VisitFullyConnectedOperator()132 armnn::TensorInfo reshapedTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitFullyConnectedOperator()
30 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitElementwiseUnaryOperator() local31 if (IsDynamicTensor(tfLiteInputTensor)) in VisitElementwiseUnaryOperator()49 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitElementwiseUnaryOperator()
51 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitActivationOperator() local52 if (IsDynamicTensor(tfLiteInputTensor)) in VisitActivationOperator()70 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitActivationOperator()
65 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitSoftmaxOperator() local66 if (IsDynamicTensor(tfLiteInputTensor)) in VisitSoftmaxOperator()84 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSoftmaxOperator()
28 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitPoolingOperator() local29 if (IsDynamicTensor(tfLiteInputTensor)) in VisitPoolingOperator()48 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitPoolingOperator()
55 const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitResizeOperator() local56 if (IsDynamicTensor(tfLiteInputTensor)) in VisitResizeOperator()88 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitResizeOperator()