| /llvm-project-15.0.7/flang/runtime/ |
| H A D | transformational.cpp | 396 SubscriptValue resultRank{shape.GetDimension(0).Extent()}; in RTNAME() local
397 if (resultRank < 0 || resultRank > static_cast<SubscriptValue>(maxRank)) { in RTNAME()
400 static_cast<std::intmax_t>(resultRank)); in RTNAME()
408 for (int j{0}; j < resultRank; ++j, ++shapeSubscript) { in RTNAME()
442 if (order->GetDimension(0).Extent() != resultRank) { in RTNAME()
446 resultRank); in RTNAME()
451 for (SubscriptValue j{0}; j < resultRank; ++j, ++orderSubscript) { in RTNAME()
454 if (k < 1 || k > resultRank || ((values >> k) & 1)) { in RTNAME()
462 for (int j{0}; j < resultRank; ++j) { in RTNAME()
469 result, source, resultRank, resultExtent, terminator, "RESHAPE"); in RTNAME()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.h | 245 int64_t resultRank = resultType.getRank(); in matchAndRewrite() local
252 bool isResultCollapsed = srcRank > resultRank; in matchAndRewrite()
304 int64_t resultRank = resultType.getRank(); in matchAndRewrite() local
310 if (srcRank > resultRank) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalgNamed.cpp | 345 int64_t resultRank = resultTy.getRank(); in matchAndRewrite() local
423 /*dimCount=*/resultRank, /*symbolCount=*/0, in matchAndRewrite()
425 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultRank)); in matchAndRewrite()
426 indexingMaps.push_back(rewriter.getMultiDimIdentityMap(resultRank)); in matchAndRewrite()
447 createDepthwiseConvCollapseMap(resultRank, reassociationMap, rewriter); in matchAndRewrite()
456 getNParallelLoopsAttrs(resultRank), in matchAndRewrite()
475 createDepthwiseConvCollapseMap(resultRank, reassociationMap, rewriter); in matchAndRewrite()
483 getNParallelLoopsAttrs(resultRank), in matchAndRewrite()
|
| /llvm-project-15.0.7/flang/lib/Evaluate/ |
| H A D | intrinsics.cpp | 1765 int resultRank{0}; in Match() local
1768 resultRank = elementalRank; in Match()
1771 resultRank = 0; in Match()
1774 resultRank = 1; in Match()
1777 resultRank = 2; in Match()
1781 resultRank = arrayArg->Rank(); in Match()
1785 resultRank = hasDimArg ? arrayArg->Rank() - 1 : 0; in Match()
1789 resultRank = hasDimArg ? arrayArg->Rank() - 1 : 1; in Match()
1793 resultRank = knownArg->Rank() + 1; in Match()
1797 resultRank = *shapeArgSize; in Match()
[all …]
|
| /llvm-project-15.0.7/flang/unittests/Evaluate/ |
| H A D | ISO-Fortran-binding.cpp | 438 constexpr CFI_rank_t resultRank{rank - 1}; in run_CFI_section_tests() local
440 CFI_CDESC_T(resultRank) resultStorage; in run_CFI_section_tests()
443 CFI_type_long_long, 0, resultRank, nullptr); in run_CFI_section_tests()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1209 Value resultRank = shapeDesc.size(rewriter, loc, 0); in convertSourceMemRefToDescriptor() local
1221 targetDesc.setRank(rewriter, loc, resultRank); in convertSourceMemRefToDescriptor()
1254 rewriter, loc, *getTypeConverter(), targetSizesBase, resultRank); in convertSourceMemRefToDescriptor()
1258 rewriter.create<LLVM::SubOp>(loc, resultRank, oneIndex); in convertSourceMemRefToDescriptor()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Tensor/IR/ |
| H A D | TensorOps.td | 328 unsigned resultRank,
334 unsigned resultRank,
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/MemRef/IR/ |
| H A D | MemRefOps.td | 1067 unsigned resultRank = getResult().getType().cast<ShapedType>().getRank();
1068 return {1, resultRank, resultRank};
|
| /llvm-project-15.0.7/flang/lib/Lower/ |
| H A D | IntrinsicCall.cpp | 3295 unsigned resultRank = in genMatmul() local
3299 mlir::Type resultArrayType = builder.getVarLenSeqTy(resultType, resultRank); in genMatmul()
3640 auto resultRank = shapeArrTy.cast<fir::SequenceType>().getShape()[0]; in genReshape() local
3642 if (resultRank == fir::SequenceType::getUnknownExtent()) in genReshape()
3658 mlir::Type type = builder.getVarLenSeqTy(resultType, resultRank); in genReshape()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 4301 unsigned resultRank = resultVectorType.getRank(); in verifyVectorShapeCast() local
4302 if (sourceRank < resultRank) { in verifyVectorShapeCast()
4305 } else if (sourceRank > resultRank) { in verifyVectorShapeCast()
|