| <?xml version="1.0"?> |
| <net name="Model505" version="11"> |
| <layers> |
| <layer id="0" name="token_ids" type="Parameter" version="opset1"> |
| <data shape="?,?" element_type="i64" /> |
| <output> |
| <port id="0" precision="I64" names="token_ids"> |
| <dim>-1</dim> |
| <dim>-1</dim> |
| </port> |
| </output> |
| </layer> |
| <layer id="1" name="self.weight" type="Const" version="opset1"> |
| <data element_type="f32" shape="3072, 2048" offset="0" size="25165824" /> |
| <output> |
| <port id="0" precision="FP32" names="self.weight"> |
| <dim>3072</dim> |
| <dim>2048</dim> |
| </port> |
| </output> |
| </layer> |
| <layer id="2" name="aten::embedding/Convert" type="Convert" version="opset1"> |
| <data destination_type="i32" /> |
| <input> |
| <port id="0" precision="I64"> |
| <dim>-1</dim> |
| <dim>-1</dim> |
| </port> |
| </input> |
| <output> |
| <port id="1" precision="I32"> |
| <dim>-1</dim> |
| <dim>-1</dim> |
| </port> |
| </output> |
| </layer> |
| <layer id="3" name="aten::embedding/Constant" type="Const" version="opset1"> |
| <data element_type="i32" shape="" offset="25165824" size="4" /> |
| <output> |
| <port id="0" precision="I32" /> |
| </output> |
| </layer> |
| <layer id="4" name="aten::embedding/Gather" type="Gather" version="opset8"> |
| <data batch_dims="0" /> |
| <input> |
| <port id="0" precision="FP32"> |
| <dim>3072</dim> |
| <dim>2048</dim> |
| </port> |
| <port id="1" precision="I32"> |
| <dim>-1</dim> |
| <dim>-1</dim> |
| </port> |
| <port id="2" precision="I32" /> |
| </input> |
| <output> |
| <port id="3" precision="FP32" names="embeddings"> |
| <dim>-1</dim> |
| <dim>-1</dim> |
| <dim>2048</dim> |
| </port> |
| </output> |
| </layer> |
| <layer id="5" name="Result_1600870" type="Result" version="opset1" output_names="embeddings"> |
| <input> |
| <port id="0" precision="FP32"> |
| <dim>-1</dim> |
| <dim>-1</dim> |
| <dim>2048</dim> |
| </port> |
| </input> |
| </layer> |
| </layers> |
| <edges> |
| <edge from-layer="0" from-port="0" to-layer="2" to-port="0" /> |
| <edge from-layer="1" from-port="0" to-layer="4" to-port="0" /> |
| <edge from-layer="2" from-port="1" to-layer="4" to-port="1" /> |
| <edge from-layer="3" from-port="0" to-layer="4" to-port="2" /> |
| <edge from-layer="4" from-port="3" to-layer="5" to-port="0" /> |
| </edges> |
| <rt_info> |
| <Runtime_version value="2026.0.0-20965-c6d6a13a886-releases/2026/0" /> |
| <conversion_parameters> |
| <framework value="pytorch" /> |
| <is_python_object value="True" /> |
| </conversion_parameters> |
| </rt_info> |
| </net> |
|
|