Searched defs:outputDims (Results 1 – 11 of 11) sorted by relevance
21 const armnn::TensorShape& outputDims = outputInfo.GetShape(); in Stack() local
136 … TfLiteIntArray* outputDims = m_TfLiteInterpreter->tensor(tfLiteDelegateOutputId)->dims; in Execute() local
60 std::vector<ExprHandle> outputDims; in computeSum() local
92 std::vector<unsigned int> outputDims(targetShape.begin(), targetShape.end()); in CreateOutputTensorShape() local
432 std::vector<unsigned int> outputDims = armnnUtils::SqueezeDims(outputShape); in ValidateAndCopyShape() local
709 std::vector<unsigned int> outputDims(dimensions->begin(), dimensions->begin() + size); in ToTensorInfo() local2583 std::vector<unsigned int> outputDims(targetDimsIn.begin(), targetDimsIn.end()); in OutputShapeOfReshape() local2672 std::vector<uint32_t> outputDims(targetDims->begin(), targetDims->begin() + targetDims->size()); in ParseReshape() local
271 std::vector<uint32_t> outputDims(numDims, 0u); in ValidateBroadcastTensorShapesMatch() local889 std::size_t outputDims = workloadInfo.m_OutputTensorInfos[0].GetNumDimensions(); in Validate() local
3728 std::vector<uint32_t> outputDims; in ConvertSqueeze() local3953 std::vector<uint32_t> outputDims(perm.begin(), perm.begin() + rank); in ConvertTranspose() local
5681 std::vector<uint32_t> outputDims; in ConvertSqueeze() local5893 std::vector<uint32_t> outputDims(perm.begin(), perm.begin() + rank); in ConvertTranspose() local
2200 std::vector<uint32_t> outputDims; in OutputShapeOfSqueeze() local2948 std::vector<unsigned int> outputDims(targetDimsIn.begin(), targetDimsIn.end()); in OutputShapeOfReshape() local
2364 auto outputDims = resultType.getShape(); in matchAndRewrite() local