Fix typo

apache · Jun 2, 2020 · 76cb069 · 76cb069
1 parent af14752
commit 76cb069
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/topi/include/topi/transform.h b/topi/include/topi/transform.h
@@ -991,7 +991,7 @@ inline Tensor tile(const Tensor& x, Array<Integer> reps, std::string name = "T_t
  * \param name The name of the operation.
  * \param tag The tag to mark the operation.
  *
- * \return A Tensor whose op member is the gather_nd operation
+ * \return A Tensor whose op member is the gather operation
  */
 inline Tensor gather(const Tensor& data, int axis, const Tensor& indices,
                      std::string name = "T_gather", std::string tag = kInjective) {

diff --git a/topi/python/topi/testing/gather_python.py b/topi/python/topi/testing/gather_python.py
@@ -15,7 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 # pylint: disable=invalid-name, line-too-long, unused-variable, too-many-locals
-"""gather_nd in python"""
+"""gather in python"""
 import numpy as np
 
 def gather_python(data, axis, indices):