Searched refs:tfLiteAxisTensor (Results 1 – 4 of 4) sorted by relevance
| /external/armnn/delegate/classic/src/ |
| D | ArgMinMax.hpp | 43 const TfLiteTensor& tfLiteAxisTensor = tfLiteTensors[tfLiteNode->inputs->data[1]]; in VisitArgMinMaxOperator() local
44 if (!IsValid(tfLiteContext, tfLiteAxisTensor, argMinMaxOperatorCode, nodeIndex)) in VisitArgMinMaxOperator()
51 switch (tfLiteAxisTensor.type) in VisitArgMinMaxOperator()
55 desc.m_Axis = tflite::GetTensorData<int>(&tfLiteAxisTensor)[0]; in VisitArgMinMaxOperator()
|
| D | Split.hpp | 33 const TfLiteTensor& tfLiteAxisTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitSplitOperator() local
34 if (!IsValid(tfLiteContext, tfLiteAxisTensor, tfLiteSplitOperatorCode, nodeIndex)) in VisitSplitOperator()
47 ARMNN_ASSERT(GetTensorInfoForTfLiteTensor(tfLiteAxisTensor).GetNumElements() == 1); in VisitSplitOperator()
48 auto* axisTensorDataPtr = tflite::GetTensorData<int32_t>(&tfLiteAxisTensor); in VisitSplitOperator()
177 const TfLiteTensor& tfLiteAxisTensor = tfLiteTensors[tfLiteNode->inputs->data[2]]; in VisitSplitVOperator() local
178 if (!IsValid(tfLiteContext, tfLiteAxisTensor, tfLiteSplitVOperatorCode, nodeIndex)) in VisitSplitVOperator()
186 ARMNN_ASSERT(GetTensorInfoForTfLiteTensor(tfLiteAxisTensor).GetNumElements() == 1); in VisitSplitVOperator()
188 auto* axisTensorDataPtr = tflite::GetTensorData<int32_t>(&tfLiteAxisTensor); in VisitSplitVOperator()
|
| D | Reduce.hpp | 43 const TfLiteTensor& tfLiteAxisTensor = tfLiteTensors[tfLiteNode->inputs->data[1]]; in VisitReduceOperator() local
44 if (!IsValid(tfLiteContext, tfLiteAxisTensor, reduceOperatorCode, nodeIndex)) in VisitReduceOperator()
49 const armnn::TensorInfo& axisTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAxisTensor); in VisitReduceOperator()
50 auto* axisTensorData = tflite::GetTensorData<int32_t>(&tfLiteAxisTensor); in VisitReduceOperator()
|
| D | Control.hpp | 218 const TfLiteTensor& tfLiteAxisTensor = tfLiteTensors[tfLiteNode->inputs->data[1]]; in VisitMeanOperator() local
219 if(!IsValid(&tfLiteAxisTensor)) in VisitMeanOperator()
227 if (IsDynamicTensor(tfLiteAxisTensor)) in VisitMeanOperator()
242 tfLiteAxisTensor, nodeIndex); in VisitMeanOperator()
255 const armnn::TensorInfo& axisTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAxisTensor); in VisitMeanOperator()
258 auto* axisTensorData = tflite::GetTensorData<int32_t>(&tfLiteAxisTensor); in VisitMeanOperator()
|