Searched refs:GenericWalk (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | test_visitor_util.cc | 47 GenericWalk(func, [&](mlir::Operation *op, const WalkStage &stage) { in runOnFunction()
52 GenericWalk(func, [&](mlir::TF::IfRegionOp op, const WalkStage &stage) { in runOnFunction()
87 auto result = GenericWalk(func, walker); in runOnFunction()
94 GenericWalk(func, [&](mlir::TF::IfRegionOp op, const WalkStage &stage) { in runOnFunction()
|
| D | resource_device_inference.cc | 173 auto walk_res = tensorflow::GenericWalk( in ComputeResourceDevicesInComputation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | visitor_util.h | 115 GenericWalk(mlir::Operation *op, FuncTy &&callback) { in GenericWalk() function
132 GenericWalk(mlir::Operation *op, FuncTy &&callback) { in GenericWalk() function
156 GenericWalk(mlir::Operation *op, FuncTy &&callback) { in GenericWalk() function
|