Home
last modified time | relevance | path

Searched refs:SparsificationOptions (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Transforms/
H A DPasses.h68 struct SparsificationOptions { struct
69 SparsificationOptions(SparseParallelizationStrategy p, in SparsificationOptions() argument
74 SparsificationOptions() in SparsificationOptions() argument
75 : SparsificationOptions(SparseParallelizationStrategy::kNone, in SparsificationOptions()
88 const SparsificationOptions &options = SparsificationOptions()); argument
92 createSparsificationPass(const SparsificationOptions &options);
/llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Pipelines/
H A DPasses.h52 SparsificationOptions sparsificationOptions() const { in sparsificationOptions()
53 return SparsificationOptions(sparseParallelizationStrategy(parallelization), in sparsificationOptions()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorPasses.cpp42 SparsificationPass(const SparsificationOptions &options) { in SparsificationPass()
57 SparsificationOptions options( in runOnOperation()
204 mlir::createSparsificationPass(const SparsificationOptions &options) { in createSparsificationPass()
H A DSparsification.cpp57 CodeGen(SparsificationOptions o, unsigned numTensors, unsigned numLoops, in CodeGen()
68 SparsificationOptions options;
1712 GenericOpSparsifier(MLIRContext *context, SparsificationOptions o) in GenericOpSparsifier()
1802 SparsificationOptions options;
1810 RewritePatternSet &patterns, const SparsificationOptions &options) { in populateSparsificationPatterns()