Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DVisitors.h75 class WalkStage {
77 explicit WalkStage(Operation *op);
244 function_ref<void(Operation *, const WalkStage &stage)> callback);
253 function_ref<WalkResult(Operation *, const WalkStage &stage)> callback);
263 std::declval<ArgT>(), std::declval<const WalkStage &>()))>
278 std::declval<ArgT>(), std::declval<const WalkStage &>()))>
283 auto wrapperFn = [&](Operation *op, const WalkStage &stage) { in walk()
288 op, function_ref<RetT(Operation *, const WalkStage &)>(wrapperFn)); in walk()
303 std::declval<ArgT>(), std::declval<const WalkStage &>()))>
308 auto wrapperFn = [&](Operation *op, const WalkStage &stage) { in walk()
[all …]
/llvm-project-15.0.7/mlir/lib/IR/
H A DVisitors.cpp14 WalkStage::WalkStage(Operation *op) in WalkStage() function in WalkStage
74 function_ref<void(Operation *, const WalkStage &)> callback) { in walk()
75 WalkStage stage(op); in walk()
187 function_ref<WalkResult(Operation *, const WalkStage &)> callback) { in walk()
188 WalkStage stage(op); in walk()
/llvm-project-15.0.7/mlir/test/lib/IR/
H A DTestVisitorsGeneric.cpp14 static std::string getStageDescription(const WalkStage &stage) { in getStageDescription()
34 outerOp->walk([&](Operation *op, const WalkStage &stage) { in runOnOperation()
40 outerOp->walk([&](test::TwoRegionOp op, const WalkStage &stage) { in runOnOperation()
65 auto walker = [&](Operation *op, const WalkStage &stage) { in runOnOperation()
107 result = outerOp->walk([&](test::TwoRegionOp op, const WalkStage &stage) { in runOnOperation()