Home
last modified time | relevance | path

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

/external/armnn/delegate/classic/src/
DReduce.hpp49 const armnn::TensorInfo& axisTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAxisTensor); in VisitReduceOperator() local
DControl.hpp255 const armnn::TensorInfo& axisTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAxisTensor); in VisitMeanOperator() local
/external/armnn/src/armnnTfLiteParser/
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