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):