fix sort docstring

apache · Dec 16, 2020 · d67cfdd · d67cfdd
1 parent b20e671
commit d67cfdd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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
     ----------