Home
last modified time | relevance | path

Searched refs:splitOutShape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DUnpack.hpp122 …armnn::TensorShape splitOutShape = armnn::TensorShape(static_cast<unsigned int>(unpackDimSizes.siz… in VisitUnpackOperator() local
128 splitterOutputs.push_back(armnn::TensorInfo(splitOutShape, in VisitUnpackOperator()
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp3878 TensorShape splitOutShape = TensorShape(static_cast<unsigned int>(unpackDimSizes.size()), in ParseUnpack() local
3885 for (unsigned int axis = 0; axis < splitOutShape.GetNumDimensions(); ++axis) in ParseUnpack()
3889 reshapeDims.push_back(splitOutShape[axis]); in ParseUnpack()
3893 TensorShape reshapeOutputShape(splitOutShape.GetNumDimensions() -1, reshapeDims.data()); in ParseUnpack()
3904 layer->GetOutputSlot(k).SetTensorInfo(armnn::TensorInfo(splitOutShape, in ParseUnpack()