| /external/llvm-project/mlir/lib/Dialect/Async/Transforms/ |
| D | AsyncParallelFor.cpp | 132 SmallVector<Value, 4> tripCounts(op.getNumLoops()); in matchAndRewrite()
133 for (size_t i = 0; i < op.getNumLoops(); ++i) { in matchAndRewrite()
152 SmallVector<Value, 4> targetNumBlocks(op.getNumLoops()); in matchAndRewrite()
153 SmallVector<Value, 4> blockSize(op.getNumLoops()); in matchAndRewrite()
154 SmallVector<Value, 4> numBlocks(op.getNumLoops()); in matchAndRewrite()
162 for (size_t i = 1; i < op.getNumLoops(); ++i) { in matchAndRewrite()
174 SmallVector<Value, 4> blockLowerBounds(op.getNumLoops()); in matchAndRewrite()
175 SmallVector<Value, 4> blockUpperBounds(op.getNumLoops()); in matchAndRewrite()
176 SmallVector<Value, 4> blockInductionVars(op.getNumLoops()); in matchAndRewrite()
189 if (loopIdx < op.getNumLoops() - 1) { in matchAndRewrite()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | ParallelLoopMapper.cpp | 132 attrs.reserve(parallelOp.getNumLoops()); in mapParallelOp()
133 for (int i = 0, e = parallelOp.getNumLoops(); i < e; ++i) { in mapParallelOp()
|
| /external/llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| D | ParallelLoopFusion.cpp | 35 if (firstPloop.getNumLoops() != secondPloop.getNumLoops()) in equalIterationSpaces()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | FusionOnTensors.cpp | 38 if (producer.getNumParallelLoops() != producer.getNumLoops()) in areTensorOpsFusable()
44 if (consumerIndexMap.getNumResults() != producer.getNumLoops()) in areTensorOpsFusable()
114 ? producer.getNumLoops() in generateFusedTensorOpRegion()
117 ? consumer.getNumLoops() in generateFusedTensorOpRegion()
122 ? std::max(producer.getNumLoops(), consumer.getNumLoops()) in generateFusedTensorOpRegion()
263 consumer.getNumLoops()); in fuseTensorOpsImpl()
|
| D | Transforms.cpp | 211 if (unfusedLoopTileSizes.size() > linalgOp.getNumLoops()) in matchAndRewrite()
212 unfusedLoopTileSizes.resize(linalgOp.getNumLoops()); in matchAndRewrite()
424 for (unsigned idx = 0, e = parallelForOp.getNumLoops(); idx < e; in substitute()
|
| D | Tiling.cpp | 186 for (unsigned i = 0; i < indexedGenericOp.getNumLoops(); ++i) { in transformIndexedGenericOpIndices()
320 auto nLoops = op.getNumLoops(); in tileLinalgOpImpl()
487 auto nLoops = op.getNumLoops(); in tileLinalgOpImpl()
|
| D | Vectorization.cpp | 105 if (genericOp.getNumLoops() != genericOp.getNumParallelLoops()) in isElementwise()
|
| D | Fusion.cpp | 142 for (unsigned i = 0, e = indexedGenericOp.getNumLoops(); i < e; ++i) { in cloneWithLoopRanges()
|
| D | Sparsification.cpp | 327 unsigned n = op.getNumLoops(); in computeIterationGraph()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToStandard/ |
| D | LinalgToStandard.cpp | 30 auto numLoops = indexedGenericOp.getNumLoops(); in extractOperandTypes()
173 auto numLoops = indexedGenericOp.getNumLoops(); in matchAndRewrite()
|
| /external/llvm-project/llvm/include/llvm/Analysis/ |
| D | LoopNestAnalysis.h | 89 size_t getNumLoops() const { return Loops.size(); } in getNumLoops() function
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_fuse_linalg.cc | 155 tile_sizes = SmallVector<int64_t, 2>(generic_op.getNumLoops(), 1); in runOnFunction()
|
| D | lhlo_legalize_to_parallel_loops.cc | 267 out_indices.reserve(outer.getNumLoops()); in CreateReduceOpInNestedParallelLoops()
544 auto rank = loop_over_src.getNumLoops(); in InsertWindowLoops()
|
| /external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/ |
| D | passes.cc | 335 unsigned num_loops = op.getNumLoops(); in runOnFunction()
|
| /external/llvm-project/mlir/lib/Conversion/SCFToStandard/ |
| D | SCFToStandard.cpp | 415 ivs.reserve(parallelOp.getNumLoops()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | kernel_creator.cc | 138 unsigned num_loops = op.getNumLoops(); in runOnFunction()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 426 auto nLoops = op.getNumLoops(); in verify()
500 auto nLoops = op.getNumLoops(); in verifyGenericOp()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgStructuredOpsInterface.td | 106 /*methodName=*/"getNumLoops",
|
| /external/llvm-project/mlir/include/mlir/Dialect/SCF/ |
| D | SCFOps.td | 369 unsigned getNumLoops() { return step().size(); }
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopUtils.cpp | 2039 for (unsigned i = 0, e = loops.getNumLoops(); i < e; ++i) { in collapseParallelLoops()
|
| /external/llvm-project/clang/lib/Serialization/ |
| D | ASTWriter.cpp | 6561 Record.push_back(C->getNumLoops()); in VisitOMPDependClause()
6569 for (unsigned I = 0, E = C->getNumLoops(); I < E; ++I) in VisitOMPDependClause()
|
| D | ASTReader.cpp | 12565 for (unsigned I = 0, E = C->getNumLoops(); I < E; ++I) in VisitOMPDependClause()
|
| /external/llvm-project/clang/include/clang/AST/ |
| D | OpenMPClause.h | 4524 unsigned getNumLoops() const { return NumLoops; } in getNumLoops() function
|
| /external/llvm-project/clang/lib/CodeGen/ |
| D | CGOpenMPRuntime.cpp | 11730 llvm::APInt Size(/*numBits=*/32, C->getNumLoops()); in emitDoacrossOrdered()
11734 for (unsigned I = 0, E = C->getNumLoops(); I < E; ++I) { in emitDoacrossOrdered()
|
| /external/llvm-project/clang/lib/Sema/ |
| D | SemaOpenMP.cpp | 7953 if (CurrentNestedLoopCount >= Pair.first->getNumLoops()) { in checkOpenMPIterationSpace()
|