Home
last modified time | relevance | path

Searched refs:distributionOptions (Results 1 – 3 of 3) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/Utils/
DUtils.cpp130 Optional<LinalgLoopDistributionOptions> distributionOptions) { in doit() argument
135 if (distributionOptions.hasValue()) in doit()
136 procInfo = distributionOptions->procInfo(builder, loc, loopRanges); in doit()
143 if (!distributionOptions.hasValue() || loopNest.loops.empty()) in doit()
148 distributionOptions->distributionMethod)) in doit()
318 Optional<LinalgLoopDistributionOptions> distributionOptions) { in doit() argument
336 if (distributionOptions) { in doit()
337 auto &options = distributionOptions.getValue(); in doit()
340 distributionMethod.assign(distributionOptions->distributionMethod.begin(), in doit()
341 distributionOptions->distributionMethod.end()); in doit()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/
DTransforms.h379 setDistributionOptions(LinalgLoopDistributionOptions distributionOptions) { in setDistributionOptions()
380 distribution = std::move(distributionOptions); in setDistributionOptions()
/external/llvm-project/mlir/test/lib/Transforms/
DTestLinalgTransforms.cpp165 LinalgLoopDistributionOptions distributionOptions; in applyPatterns() local