tensorflow/compiler/mlir/xla/transforms/legalize_tf_communication.cc
// Checks if a function call is pointing to a function with communication ops.
Checks if a function call is pointing to a function with communication ops.
Checks if a function call is pointing to a function with communication ops .
bool IsFunctionCallWithCommunication(
    Operation* op,
    const llvm::SmallDenseMap<StringRef, FuncToRewrite>& funcs_to_rewrite) {
  if (auto call = dyn_cast<mlir::func::CallOp>(op))
    return funcs_to_rewrite.count(call.getCallee());
  return false;
}
bool IsFunctionCallWithCommunication ( Operation * op , const llvm :: SmallDenseMap < StringRef , FuncToRewrite > & funcs_to_rewrite ) { if ( auto call = dyn_cast < mlir :: func :: CallOp > ( op ) ) return funcs_to_rewrite . count ( call . getCallee ( ) ) ; return false ; }
Checks if a function call is pointing to a function with communication ops.
Checks if a function call is pointing to a function with communication ops .