util.py 文件源码

def ready_for_reuse(name):
  def decorator(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
      temp_func = tf.make_template(name, func)
      return temp_func(*args, **kwargs)

    return wrapper

  return decorator