Searched refs:FuncToRewrite (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 128 struct FuncToRewrite { struct
139 llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs_to_rewrite) { in GetFunctionsToRewrite()
143 FuncToRewrite func_to_rewrite{/*original=*/func, /*control_flow_ops=*/{}, in GetFunctionsToRewrite()
170 FuncToRewrite func_to_rewrite{/*original=*/caller_parent_func, in GetFunctionsToRewrite()
736 const llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs, in RewriteFunction()
825 const llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs_to_rewrite) { in IsFunctionCallWithCommunication()
836 const llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs_to_rewrite, in GetCommunicationControlFlowOps()
852 llvm::SmallDenseMap<StringRef, FuncToRewrite> funcs_to_rewrite; in runOnOperation()
|