You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2021/05/13 13:22:21 UTC
[tvm] branch main updated: fix docs of threefry_split and
threefry_generate (#8035)
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 <zi...@tencent.com>
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
-------