Function cb1

imperative/python/test/unit/xla/test_xla_training.py:312–315

(_, grad)

        side_effect_cnt = 0

        def cb1(_, grad):
            nonlocal side_effect_cnt
            side_effect_cnt += 1
            return grad / 2

        def cb2(_, grad):
            return F.clip(grad, -2e-2, 2e-2)