o s::j @sFddlmZddlmZddlmZddlmZGdddej Z dS)) annotations)Any)builder) _module_listcsJeZdZdZdfdd Zdd d ZdddZdddZdddZZ S) SequentialaA sequential container that calls children in order, mirroring ``torch.nn.Sequential``. Children are registered with string keys ``"0"``, ``"1"``, etc., just like ``ModuleList``. The ``forward`` method passes the output of each child as the input to the next. Example:: class SiLU(Module): def forward(self, op, x): return op.Mul(x, op.Sigmoid(x)) # Produces parameter names: "mod.0.weight", "mod.0.bias" # SiLU at index 0 has no parameters. mod = Sequential(SiLU(), Linear(4, 4)) # Calling mod(op, x) is equivalent to: # x = silu(op, x) # x = linear(op, x) modules_module_list.ModulereturnNonecst|dS)N)super__init__)selfr __class__`/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnxscript/nn/_sequential.pyr "szSequential.__init__namestrcCs0t|d||jD] \}}||q dS)aySet this container's name. Children keep simple ``"0"``, ``"1"`` names. Unlike ``ModuleList._set_name`` which fully qualifies children (used when ModuleList is iterated externally), Sequential is called via ``__call__`` which already pushes its own name onto the builder stack. Children must keep simple keys to avoid double-prefixing. _nameN)object __setattr___modulesitems _set_name)r rkeychildrrrr%s zSequential._set_namermodulecCs4|jdur t|d|||j|<t|||dS)a!Register a child module under the given string key. Unlike ``ModuleList._register_child`` which qualifies the child name with the parent name, Sequential keeps children with simple index names because ``__call__`` already pushes the Sequential's own name. Nr)rrrr)r rrrrr_register_child1s  zSequential._register_childop_builder.OpBuilderinputrcCs,t|dkr td|D]}|||}q |S)zRun each child module sequentially, passing output to the next. Mirrors ``torch.nn.Sequential.forward``: each child receives exactly one positional argument (the output of the previous child). rz4Cannot call forward on an empty Sequential container)len RuntimeError)r rr rrrrforward=s  zSequential.forwardcCsTdg}|jD]\}}t|dd}|d|d|q|dd|S)Nz Sequential( z z (z): ))rrreprreplaceappendjoin)r linesrrmod_reprrrr__repr__Is   zSequential.__repr__)rrr r )rrr r )rrrrr r )rrr rr r)r r) __name__ __module__ __qualname____doc__r rrr#r, __classcell__rrrrr s   rN) __future__rtypingronnxscript._internalr_builder onnxscript.nnr ModuleListrrrrrs