// Replaces `xla_call_module_op` with a newly created `func::CallOp`, where the
// callee is `callee_func_op`. The existence of `kQuantizationMethodAttr` in
// `xla_call_module_op` should be guaranteed.
void ReplaceXlaCallModuleOpWithNewCallOp(TF::XlaCallModuleOp xla_call_module_op,
                                         func::FuncOp callee_func_op,
                                         PatternRewriter& rewriter) {