Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Conversion/LinalgToSPIRV/
DLinalgToSPIRV.cpp66 linalg::GenericOp genericOp) { in matchAsPerformingReduction()
112 linalg::GenericOp genericOp, ArrayRef<Value> operands, in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DGeneralization.cpp79 linalg::GenericOp genericOp = pattern->createGenericOp(rootOp, rewriter); in matchAndRewrite() local
120 linalg::GenericOp genericOp = in matchAndRewrite() local
DBufferize.cpp115 GenericOpTy genericOp, ValueRange inputs, in finalizeBufferAllocationForGenericOp()
209 if (auto genericOp = dyn_cast<linalg::GenericOp>(op)) { in matchAndRewrite() local
216 if (auto genericOp = dyn_cast<linalg::IndexedGenericOp>(op)) { in matchAndRewrite() local
DVectorization.cpp75 auto genericOp = dyn_cast<linalg::GenericOp>(op); in isContraction() local
102 auto genericOp = dyn_cast<linalg::GenericOp>(op); in isElementwise() local
/external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/
DExpandOps.cpp57 auto genericOp = in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dbuffer_reuse_pass.cc88 f.walk([&](linalg::GenericOp genericOp) { in build()