Home
last modified time | relevance | path

Searched defs:axisTensorInfo (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DReduce.hpp49 const armnn::TensorInfo& axisTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAxisTensor); in VisitReduceOperator() local
H A DControl.hpp255 const armnn::TensorInfo& axisTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAxisTensor); in VisitMeanOperator() local
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp3939 armnn::TensorInfo axisTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 0); in ParseSplit() local
4046 armnn::TensorInfo axisTensorInfo = ToTensorInfo(axisTensor); in ParseSplitV() local
4206 armnn::TensorInfo axisTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 1); in ParseArgMinMax() local