From 77c98651aa80fae70ad9e56b70df22f71d2e7102 Mon Sep 17 00:00:00 2001 From: notoraptor Date: Tue, 2 Jun 2020 13:31:43 -0400 Subject: [PATCH] Fix typo --- topi/include/topi/transform.h | 2 +- topi/python/topi/testing/gather_python.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/topi/include/topi/transform.h b/topi/include/topi/transform.h index 5fb0732da32c..0537c0370670 100644 --- a/topi/include/topi/transform.h +++ b/topi/include/topi/transform.h @@ -997,7 +997,7 @@ inline Tensor tile(const Tensor& x, Array 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 index cbe50017c879..0f3573cb1679 100644 --- 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):