manupa-arm commented on a change in pull request #6917:
URL: https://github.com/apache/incubator-tvm/pull/6917#discussion_r529437902
##########
File path: include/tvm/runtime/crt/graph_runtime.h
##########
@@ -61,14 +61,20 @@ typedef struct TVMGraphRuntime TVMGraphRuntime;
* \brief Allocate a new GraphRuntime with vmalloc and initialize it.
*
* \param sym_json JSON-encoded graph.
- * \param m TVM Module that exposes the functions to call.
+ * \param module_handle TVM Module that exposes the functions to call.
* \param ctxs runtime execution context.
*/
-TVMGraphRuntime* TVMGraphRuntime_Create(const char* sym_json, const struct
TVMModule* m,
+TVMGraphRuntime* TVMGraphRuntime_Create(const char* sym_json, TVMModuleHandle
module_handle,
const TVMContext* ctxs);
int TVMGraphRuntime_GetInputIndex(TVMGraphRuntime* runtime, const char* name);
+/*!
+ * \brief get number of input tensors allocated.
+ * \return integer number of tensors available to use.
+ */
+int TVMGraphRuntime_GetNumInputs();
Review comment:
fine by me, then.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
