o s::j @sBddlmZddlmZddlZddlmZGdddej Z dS)) annotations)SequenceN)buildercsLeZdZdZ   ddfd d ZedddZdddZdddZZ S) ParameteraA module parameter that is also an ``ir.Value``. Since ``Parameter`` subclasses ``ir.Value``, it can be passed directly to ONNX ops inside ``Module.forward()`` without any conversion. Calling :meth:`_realize` qualifies the name with the current module context and registers the parameter as a graph initializer. Args: shape: Shape of the parameter tensor. dtype: Data type of the parameter. If None and data is not provided, defaults to float32. name: Name for the parameter. If None, the attribute name from the parent Module is used. data: Optional initial tensor data. If provided, the initializer will carry this as its const_value. Nshape Sequence[int]dtypeir.DataType | Nonename str | Nonedatair.TensorProtocol | NonereturnNonecs||dur!|dur|j|krtd|jd|d|dur |j}n|dur)tjj}tj|t|t||dd|_ dS)NzData type of provided data (z&) does not match the specified dtype (z).)r rtype const_valueF) r ValueErrorirDataTypeFLOATsuper__init__Shape TensorType _realized)selfrrr r  __class___/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnxscript/nn/_parameter.pyrs" zParameter.__init__cCs|jdur |jjSdS)z/Return the element data type of this parameter.N)rrrrrrr7szParameter.dtyper_builder.GraphBuildercCsH|jr|S|j}|std|j}||}|_||jj|<d|_|S)u'Qualify the name and register as a graph initializer. Uses the builder's *root* graph builder to qualify the name and register the initializer. When the builder is a sub-builder (e.g. for a Scan body), this ensures the parameter is stored in the main graph — making it visible as an implicit input to the subgraph rather than incorrectly placed inside it. Uses direct assignment to ``graph.initializers[...]`` to skip the const_value check. Idempotent: subsequent calls are no-ops. zParameter._realize() called on a Parameter without a name. Ensure the Parameter is attached to a Module attribute or otherwise initialized with a name before realization.T)rr rroot_qualify_initializer_namegraph initializers)rr self_namer"rrr_realize<s  zParameter._realizestrcCs"dt|jd|jd|jdS)NzParameter(shape=z, dtype=z, name=))listrrr r rrr__repr__Xs"zParameter.__repr__)NNN) rrrr r r r r rr)rr )rr!rr)rr() __name__ __module__ __qualname____doc__rpropertyrr'r+ __classcell__rrrrr s  r) __future__rtypingronnx_irronnxscript._internalr_builderValuerrrrrs