[doc] Update docstring for grad replaced (#6800)

Update docstring for grad replaced from `ti.grad_replaced` to `ti.ad.grad_replaced`.
taichi-dev · Dec 3, 2022 · 61effe9 · 61effe9
1 parent 21d8177
commit 61effe9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/taichi/ad/_ad.py b/python/taichi/ad/_ad.py
@@ -295,11 +295,11 @@ def grad_replaced(func):
         >>>     for I in ti.grouped(x):
         >>>         x.grad[I] = y.grad[I] / a
         >>>
-        >>> @ti.grad_replaced
+        >>> @ti.ad.grad_replaced
         >>> def foo(a):
         >>>     multiply(a)
         >>>
-        >>> @ti.grad_for(foo)
+        >>> @ti.ad.grad_for(foo)
         >>> def foo_grad(a):
         >>>     multiply_grad(a)"""
     def decorated(*args, **kwargs):