Method grad

ot/backend.py:2937–2938

(upstream)

        @tf.custom_gradient
        def tmp(input):
            def grad(upstream):
                return grads

            return val, grad