Home
last modified time | relevance | path

Searched defs:nLoops (Results 1 – 6 of 6) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DTiling.cpp320 auto nLoops = op.getNumLoops(); in tileLinalgOpImpl() local
487 auto nLoops = op.getNumLoops(); in tileLinalgOpImpl() local
DLoops.cpp469 unsigned nLoops = allIvs.size(); in emitScalarImplementation() local
DFusion.cpp223 for (unsigned i = 0, nLoops = loopRanges.size(); i < nLoops; ++i) { in fuse() local
/external/llvm-project/mlir/lib/Dialect/Linalg/Utils/
DUtils.cpp218 unsigned nLoops = iteratorTypes.size(); local
/external/llvm-project/mlir/lib/Dialect/Linalg/IR/
DLinalgOps.cpp224 unsigned nLoops = iteratorTypes.size(); in build() local
426 auto nLoops = op.getNumLoops(); in verify() local
500 auto nLoops = op.getNumLoops(); in verifyGenericOp() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc56 unsigned nLoops, unsigned nReduction) { in GetParallelAndReductionIterators()