Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmerge_control_flow.cc297 auto& if_ops = entry.second; in OptimizeIfRegions() local
298 for (auto it = if_ops.begin(); it != if_ops.end(); ++it) { in OptimizeIfRegions()
300 for (auto it2 = std::next(it); it2 != if_ops.end(); ++it2) { in OptimizeIfRegions()
316 if_ops.erase(it2--); in OptimizeIfRegions()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_control_flow.cc211 llvm::SmallVector<IfOp, 4> if_ops; in runOnFunction() local
212 func.walk([&](IfOp op) { if_ops.push_back(op); }); in runOnFunction()
214 for (auto& op : if_ops) { in runOnFunction()