autograd.rst

oneflow.autograd
================================================
Functions and classes for autograd.
---------------------------------------------------
.. currentmodule:: oneflow.autograd
.. autoclass:: oneflow.autograd.Function
    :members: apply,
    :special-members: __call__,

.. automodule:: oneflow.autograd
    :members: grad,
      backward,