Files
apache--tvm/python
Masahiro Hiramori 86794e7d91 [Relax][Frontend] Add ParameterList and ParameterDict containers (#19495)
This PR adds first-class `nn.ParameterList` and `nn.ParameterDict`
containers to the Relax frontend.

These containers provide PyTorch-like list/dict registration for raw
`nn.Parameter` objects while preserving Relax frontend semantics: values
must be explicit `nn.Parameter` instances, with no automatic
tensor-to-parameter conversion.

### Changes

- Add public `nn.ParameterList` and `nn.ParameterDict` exports.
- Support stable parameter names in traversal:
  - `params.0`, `params.1`
  - `params.foo`, `params.bar`
- Integrate the new containers with:
  - `named_parameters()`
  - `parameters()`
  - `state_dict()`
  - `load_state_dict()`
  - `to(dtype=...)`
  - `export_tvm()`
  - `nn.Mutator`
- Add focused tests for basic container behavior, type validation,
nested traversal, export parameter names, state loading, dtype
conversion, and mutator naming.

---------

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
2026-05-02 22:55:52 +08:00
..