From d67cfdd931491c333422b7368c117702cb5774d2 Mon Sep 17 00:00:00 2001 From: mbrookhart Date: Wed, 16 Dec 2020 12:12:58 -0700 Subject: [PATCH] fix sort docstring --- python/tvm/topi/cuda/sort.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/topi/cuda/sort.py b/python/tvm/topi/cuda/sort.py index 47684b8f778e..b8873e0b546b 100644 --- a/python/tvm/topi/cuda/sort.py +++ b/python/tvm/topi/cuda/sort.py @@ -317,8 +317,8 @@ def argsort_nms_thrust(data, valid_count, axis=-1, is_ascend=1, dtype="float32") def sort(data, axis=-1, is_ascend=1): - """Performs sorting along the given axis and returns an array of indicies - having same shape as an input array that index data in sorted order. + """Performs sorting along the given axis and returns an array of + sorted values with teh same shape as the input data. Parameters ----------