*/
void PostOrderVisit(const Expr& node, std::function<void(const Expr&)> fvisit);
-/*
- * \brief Bind function parameters or free variables.
- *
- * Parameter binding can only happen if expr is a Function.
- * binds cannot change internal arguments of internal functions.
- *
- * \param expr The function to be binded.
- * \param binds The map of arguments to
- */
-Expr Bind(const Expr& expr, const tvm::Map<Var, Expr>& binds);
-
} // namespace relay
} // namespace tvm
#endif // TVM_RELAY_EXPR_FUNCTOR_H_
*/
TVM_DLL Expr PartialEval(const Expr& e, const Module& mod);
-/*!
- * \brief Bind the free variables to a Relay expression.
+/*
+ * \brief Bind function parameters or free variables.
*
- * \param expr The expression.
- * \param bind_map The variable to expression map that will be used to help the
- * binding.
+ * Parameter binding can only happen if expr is a Function.
+ * binds cannot change internal arguments of internal functions.
*
- * \return The updated expression.
+ * \param expr The function to be binded.
+ * \param binds The map of arguments to
*/
TVM_DLL Expr Bind(const Expr& expr, const tvm::Map<Var, Expr>& bind_map);
}
case Opcode::AllocTensor: {
auto shape = std::vector<int64_t>(instr.alloc_tensor.ndim);
- for (uint i = 0; i < instr.alloc_tensor.ndim; ++i) {
+ for (uint32_t i = 0; i < instr.alloc_tensor.ndim; ++i) {
shape[i] = instr.alloc_tensor.shape[i];
}
auto allocator = MemoryManager::Global()->GetAllocator(ctxs[0]);