From 78a26b5664d30d2f39e90f2ae63a3b8496f7fa02 Mon Sep 17 00:00:00 2001
From: James Gilles <jhgilles@mit.edu>
Date: Wed, 10 Apr 2019 00:18:23 -0400
Subject: [PATCH] Update docs

---
 include/tvm/runtime/registry.h | 15 ++++++++++++++-
 src/api/api_codegen.cc         |  1 -
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/include/tvm/runtime/registry.h b/include/tvm/runtime/registry.h
index 717324e4e617..40e1a520cb67 100644
--- a/include/tvm/runtime/registry.h
+++ b/include/tvm/runtime/registry.h
@@ -102,7 +102,8 @@ class Registry {
    * \endcode
    *
    * \param f The function to forward to.
-   * \tparam FType the signature of the function.
+   * \tparam R the return type of the function (inferred).
+   * \tparam Args the argument types of the function (inferred).
    */
   template<typename R, typename ...Args>
   Registry& set_body_typed(R (*f)(Args...)) {
@@ -125,6 +126,9 @@ class Registry {
    * \endcode
    *
    * \param f the method pointer to forward to.
+   * \tparam T the type containing the method (inferred).
+   * \tparam R the return type of the function (inferred).
+   * \tparam Args the argument types of the function (inferred).
    */
   template<typename T, typename R, typename ...Args>
   Registry& set_body_method(R (T::*f)(Args...)) {
@@ -150,6 +154,9 @@ class Registry {
    * \endcode
    *
    * \param f the method pointer to forward to.
+   * \tparam T the type containing the method (inferred).
+   * \tparam R the return type of the function (inferred).
+   * \tparam Args the argument types of the function (inferred).
    */
   template<typename T, typename R, typename ...Args>
   Registry& set_body_method(R (T::*f)(Args...) const) {
@@ -185,6 +192,9 @@ class Registry {
    *
    * \param f the method pointer to forward to.
    * \tparam TNodeRef the node reference type to call the method on
+   * \tparam TNode the node type containing the method (inferred).
+   * \tparam R the return type of the function (inferred).
+   * \tparam Args the argument types of the function (inferred).
    */
   template<typename TNodeRef, typename TNode, typename R, typename ...Args,
     typename = typename std::enable_if<std::is_base_of<NodeRef, TNodeRef>::value>::type>
@@ -222,6 +232,9 @@ class Registry {
    *
    * \param f the method pointer to forward to.
    * \tparam TNodeRef the node reference type to call the method on
+   * \tparam TNode the node type containing the method (inferred).
+   * \tparam R the return type of the function (inferred).
+   * \tparam Args the argument types of the function (inferred).
    */
   template<typename TNodeRef, typename TNode, typename R, typename ...Args,
     typename = typename std::enable_if<std::is_base_of<NodeRef, TNodeRef>::value>::type>
diff --git a/src/api/api_codegen.cc b/src/api/api_codegen.cc
index 3f81fdc4aca2..73e26719cf15 100644
--- a/src/api/api_codegen.cc
+++ b/src/api/api_codegen.cc
@@ -42,6 +42,5 @@ TVM_REGISTER_API("codegen._Build")
 
 TVM_REGISTER_API("module._PackImportsToC")
 .set_body_typed(PackImportsToC);
-
 }  // namespace codegen
 }  // namespace tvm