Searched refs:outputShapeVector (Results 1 – 1 of 1) sorted by relevance
57 std::vector<unsigned int> outputShapeVector; in CalculateReducedOutputTensoInfo() local79 outputShapeVector.push_back(inputTensorInfo.GetShape()[i]); in CalculateReducedOutputTensoInfo()86 outputShapeVector.push_back(1); in CalculateReducedOutputTensoInfo()93 … armnn::TensorShape outputTensorShape(inputTensorInfo.GetNumDimensions(), &outputShapeVector[0]); in CalculateReducedOutputTensoInfo()109 std::vector<unsigned int> outputShapeVector; in CalculateStridedSliceOutputTensorInfo() local126 outputShapeVector.push_back(static_cast<unsigned int>(newSize)); in CalculateStridedSliceOutputTensorInfo()129 armnn::TensorShape outputTensorShape(inputTensorInfo.GetNumDimensions(), &outputShapeVector[0]); in CalculateStridedSliceOutputTensorInfo()