This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 39fa759  fix docs of threefry_split and threefry_generate (#8035)
39fa759 is described below

commit 39fa759906f526f72ea02c8f4caa4828a2dfacbd
Author: Zilin Zhu <[email protected]>
AuthorDate: Thu May 13 21:22:07 2021 +0800

    fix docs of threefry_split and threefry_generate (#8035)
---
 python/tvm/topi/random/kernel.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/python/tvm/topi/random/kernel.py b/python/tvm/topi/random/kernel.py
index b6c0b3f..9366e8e 100644
--- a/python/tvm/topi/random/kernel.py
+++ b/python/tvm/topi/random/kernel.py
@@ -212,8 +212,8 @@ def threefry_generate(gen, out_shape):
     Parameters
     ----------
     gen : Tensor[10, uint64]
-        Generator state. Can be create with :py:func:`tvm.relay.threefry_key`. 
This should not be
-        reused in another function, otherwise random numbers will be repeated.
+        Generator state. Can be create with 
:py:func:`tvm.relay.random.threefry_key`. This should
+        not be reused in another function, otherwise random numbers will be 
repeated.
 
     out_shape : Sequence[int]
         Output shape of the random numbers. Product of all dimensions must be 
a multiple of 4.
@@ -352,8 +352,8 @@ def threefry_split(gen):
     Parameters
     ----------
     gen : Tensor[10, uint64]
-        Generator state. Can be create with :py:func:`tvm.relay.threefry_key`. 
This should not be
-        reused in another function, otherwise random numbers will be repeated.
+        Generator state. Can be create with 
:py:func:`tvm.relay.random.threefry_key`. This should
+        not be reused in another function, otherwise random numbers will be 
repeated.
 
     Returns
     -------

Reply via email to