code: def assign_add(ref, value):
      if hasattr(tf, "assign_add"):
          return tf.assign_add(ref, value)
      else:
          return ref.assign_add(value)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.keras import compat
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_impl
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_schedule
label_desc: Use image node_id 1 for calling a global function with example usage: assign_add(ref, value) and returns: tf, ref