Home
last modified time | relevance | path

Searched refs:walked_op (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_extract_outside_compilation.cc62 ->walk([&](Operation* walked_op) { in HasOutsideCompilationNested() argument
63 if (op == walked_op) return WalkResult::advance(); in HasOutsideCompilationNested()
64 if (walked_op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in HasOutsideCompilationNested()
201 op->walk([&](Operation* walked_op) { in GetExternalOperands() argument
202 if (llvm::isa<TF::_XlaRecvAtHostV2Op, TF::_XlaSendFromHostV2Op>(walked_op)) in GetExternalOperands()
204 for (Value v : walked_op->getOperands()) { in GetExternalOperands()
Dmerge_control_flow.cc182 source_op->walk([&](Operation* walked_op) { in MoveResultsAfter() argument
184 side_effect_analysis.DirectControlSuccessors(walked_op)) { in MoveResultsAfter()