Update docs

apache · Apr 10, 2019 · 78a26b5 · 78a26b5
1 parent ddbd4cb
commit 78a26b5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git 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
@@ -42,6 +42,5 @@ TVM_REGISTER_API("codegen._Build")
 
 TVM_REGISTER_API("module._PackImportsToC")
 .set_body_typed(PackImportsToC);
-
 }  // namespace codegen
 }  // namespace tvm