Searched refs:original_body_fn (Results 1 – 1 of 1) sorted by relevance
1371 mlir::TF::WhileOp op, mlir::FlatSymbolRefAttr original_body_fn, in GetWhileBodyFunction() argument1377 std::string body_fn_name = original_body_fn.getValue().str() + "/tfrt_body_" + in GetWhileBodyFunction()1422 op.getLoc(), while_result_types, original_body_fn, block->getArguments()); in GetWhileBodyFunction()