Home
last modified time | relevance | path

Searched defs:whileOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_control_flow_to_scf.cc48 getFunction().walk([&](WhileOp whileOp) { MatchAndRewrite(whileOp); }); in runOnFunction()
53 void MatchAndRewrite(WhileOp whileOp) { in MatchAndRewrite()
Dmhlo_flatten_tuple.cc149 auto whileOp = cast<mhlo::WhileOp>(op); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc1142 if (auto whileOp = dyn_cast<mlir::TFL::WhileOp>(inst)) { in BuildOperator() local