Function _save_to

imperative/python/test/unit/quantization/test_fake_quant.py:89–93

(self, name="grad")

def _save_to(self, name="grad"):
    def callback(grad):
        setattr(self, name, grad)

    return callback


class Round(Function):