Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_communication.cc139 llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs_to_rewrite) { in GetFunctionsToRewrite() argument
153 funcs_to_rewrite.insert({func.getName(), func_to_rewrite}); in GetFunctionsToRewrite()
179 auto it = funcs_to_rewrite.insert( in GetFunctionsToRewrite()
200 for (auto& func : funcs_to_rewrite) { in GetFunctionsToRewrite()
825 const llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs_to_rewrite) { in IsFunctionCallWithCommunication() argument
827 return funcs_to_rewrite.count(call.callee()); in IsFunctionCallWithCommunication()
836 const llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs_to_rewrite, in GetCommunicationControlFlowOps() argument
841 IsFunctionCallWithCommunication(op, funcs_to_rewrite)) in GetCommunicationControlFlowOps()
852 llvm::SmallDenseMap<StringRef, FuncToRewrite> funcs_to_rewrite; in runOnOperation() local
853 if (failed(GetFunctionsToRewrite(module, funcs_to_rewrite))) in runOnOperation()
[all …]