Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_communication.cc137 llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs_to_rewrite) { in GetFunctionsToRewrite() argument
151 funcs_to_rewrite.insert({func.getName(), func_to_rewrite}); in GetFunctionsToRewrite()
178 auto it = funcs_to_rewrite.insert( in GetFunctionsToRewrite()
199 for (auto& func : funcs_to_rewrite) { in GetFunctionsToRewrite()
904 const llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs_to_rewrite) { in IsFunctionCallWithCommunication() argument
906 return funcs_to_rewrite.count(call.getCallee()); in IsFunctionCallWithCommunication()
915 const llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs_to_rewrite, in GetCommunicationControlFlowOps() argument
920 IsFunctionCallWithCommunication(op, funcs_to_rewrite)) in GetCommunicationControlFlowOps()
931 llvm::SmallDenseMap<StringRef, FuncToRewrite> funcs_to_rewrite; in runOnOperation() local
932 if (failed(GetFunctionsToRewrite(module, funcs_to_rewrite))) in runOnOperation()
[all …]