Function cb

imperative/python/test/unit/autodiff/test_grad_manager.py:87–89

(x, g)

    gm.attach(w)

    def cb(x, g):
        assert x is ref()
        cb.called = True

    for i in range(3):
        with gm: