junrushao opened a new pull request, #15398: URL: https://github.com/apache/tvm/pull/15398
This PR introduces the core data structure for the new `nn.Module` API, including: - Tensor, a wrapper on top of relax.Expr whose struct_info is a TensorStructInfo, providing more convenient access shape and dtype information. Tensor is always symbolc and not bound to any concrete values. - Parameter, a special tensor which could be bound or not bound to concrete values. - Module, a container of nn.Parameters and sub nn.Modules. - Effect, a non-user-facing class that encloses potential side effects, for example, IO, impure external function callings, inplace mutation, etc. -- 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. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
