Function grad_fn

tensorflow/contrib/layers/python/layers/rev_block_lib.py:628–629

(*output_grads, **kwargs)

    # created variables, the grad_fn should accept the "variables" kwarg.
    if original_vars:
      def grad_fn(*output_grads, **kwargs):
        return _grad_fn(output_grads, kwargs["variables"])
    else:
      def grad_fn(*output_grads):
        return _grad_fn(output_grads)