Python make_vjp Examples

Example #1

File: backprop_test.py Project: yuguanching/tensorflow

    def testMakeVJP(self):
        def f(x):
            return x * x

        wrapped_fn = backprop.make_vjp(f, persistent=False)
        result, vjp = wrapped_fn(constant_op.constant(3.0))
        self.assertAllEqual(result, 9.0)
        self.assertAllEqual(vjp(2.0)[0], 12.0)

Example #2

    def testPersistentMakeVJP(self):
        def f(x):
            return x * x

        wrapped_fn = backprop.make_vjp(f, persistent=True)
        _, vjp = wrapped_fn(constant_op.constant(3.0))
        vjp_result1 = vjp(2.0)[0]
        vjp_result2 = vjp(2.0)[0]
        self.assertAllEqual(vjp_result1, vjp_result2, 12.0)

Example #3

File: backprop_test.py Project: Wajih-O/tensorflow

  def testMakeVJP(self):

    def f(x):
      return x * x

    wrapped_fn = backprop.make_vjp(f, persistent=False)
    result, vjp = wrapped_fn(constant_op.constant(3.0))
    self.assertAllEqual(result, 9.0)
    self.assertAllEqual(vjp(2.0)[0], 12.0)

Example #4

File: backprop_test.py Project: Wajih-O/tensorflow

  def testPersistentMakeVJP(self):

    def f(x):
      return x * x

    wrapped_fn = backprop.make_vjp(f, persistent=True)
    _, vjp = wrapped_fn(constant_op.constant(3.0))
    vjp_result1 = vjp(2.0)[0]
    vjp_result2 = vjp(2.0)[0]
    self.assertAllEqual(vjp_result1, vjp_result2, 12.0)

Example #5

File: tracing.py Project: zhouyonglong/tangent

def trace_grad(fn, args):
    """Trace a function, and return a VJP and the function's output."""
    result, vjp = make_vjp(fn)(*args)
    return result, vjp

Example #6

def trace_grad(fn, args):
    """Trace a function, and return a VJP and the function's output."""
    from tensorflow.python.eager.backprop import make_vjp
    result, vjp = make_vjp(fn)(*args)
    return result, vjp