Searched refs:func_to_rewrite (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 143 FuncToRewrite func_to_rewrite{/*original=*/func, /*control_flow_ops=*/{}, in GetFunctionsToRewrite() local
147 if (failed(FindCommunicationOps(func, func_to_rewrite.control_flow_ops, in GetFunctionsToRewrite()
148 func_to_rewrite.control_flow_blocks, in GetFunctionsToRewrite()
153 funcs_to_rewrite.insert({func.getName(), func_to_rewrite}); in GetFunctionsToRewrite()
170 FuncToRewrite func_to_rewrite{/*original=*/caller_parent_func, in GetFunctionsToRewrite() local
175 use.getUser(), func_to_rewrite.control_flow_ops, in GetFunctionsToRewrite()
176 func_to_rewrite.control_flow_blocks))) in GetFunctionsToRewrite()
180 {caller_parent_func.getName(), func_to_rewrite}); in GetFunctionsToRewrite()
185 func_to_rewrite.control_flow_ops.begin(), in GetFunctionsToRewrite()
186 func_to_rewrite.control_flow_ops.end()); in GetFunctionsToRewrite()
[all …]
|