code: def add_to_multiply(fndef):
      for node in fndef.node_def:
          if node.name.endswith("x_plus_y"):
              node.name = node.name.replace("x_plus_y", "x_times_y")
              node.op = "Mul"
          for idx, inp in enumerate(node.input):
              if inp.endswith("x_plus_y:z:0"):
                  node.input[idx] = inp.replace("x_plus_y", "x_times_y")
  
imports: import collections
  from absl.testing import parameterized
  from tensorflow.core.function.testing import test_pass
  from tensorflow.core.function.transform import transform
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.framework import test_util
  from tensorflow.python.module import module
  from tensorflow.python.ops import custom_gradient
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import while_loop
  from tensorflow.python.platform import test
  from tensorflow.python.saved_model import load
  from tensorflow.python.saved_model import save
label_desc: Use image node_id 1 for calling a global function with example usage: add_to_multiply(fndef) without return types