Searched refs:splitOutShape (Results 1 – 2 of 2) sorted by relevance
122 …armnn::TensorShape splitOutShape = armnn::TensorShape(static_cast<unsigned int>(unpackDimSizes.siz… in VisitUnpackOperator() local128 splitterOutputs.push_back(armnn::TensorInfo(splitOutShape, in VisitUnpackOperator()
3878 TensorShape splitOutShape = TensorShape(static_cast<unsigned int>(unpackDimSizes.size()), in ParseUnpack() local3885 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()