Home
last modified time | relevance | path

Searched refs:axisTensorData (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DSplit.hpp49 std::vector<int32_t> axisTensorData(axisTensorDataPtr, axisTensorDataPtr + 1); in VisitSplitOperator() local
50 int32_t axis = axisTensorData[0]; in VisitSplitOperator()
189 std::vector<int32_t> axisTensorData(axisTensorDataPtr, axisTensorDataPtr + 1); in VisitSplitVOperator() local
190 int32_t axis = axisTensorData[0]; in VisitSplitVOperator()
200 …const unsigned int splitDim = ComputeWrappedIndex(axisTensorData[0], inputTensorInfo.GetNumDimensi… in VisitSplitVOperator()
H A DReduce.hpp50 auto* axisTensorData = tflite::GetTensorData<int32_t>(&tfLiteAxisTensor); in VisitReduceOperator() local
54 if (axisTensorData != nullptr) in VisitReduceOperator()
58 axis.emplace_back(axisTensorData[i]); in VisitReduceOperator()
H A DControl.hpp258 auto* axisTensorData = tflite::GetTensorData<int32_t>(&tfLiteAxisTensor); in VisitMeanOperator() local
264 axis.emplace_back(axisTensorData[i]); in VisitMeanOperator()