|
<?xml version="1.0"?> |
|
<net name="Model6" version="11"> |
|
<layers> |
|
<layer id="0" name="input" type="Parameter" version="opset1"> |
|
<data shape="?,?" element_type="i64" /> |
|
<output> |
|
<port id="0" precision="I64" names="input"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="1" name="self.weight" type="Const" version="opset1"> |
|
<data element_type="f32" shape="32002, 16" offset="0" size="2048128" /> |
|
<output> |
|
<port id="0" precision="FP32" names="self.weight"> |
|
<dim>32002</dim> |
|
<dim>16</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="2048128" 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>32002</dim> |
|
<dim>16</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="inputs_embeds"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
<dim>16</dim> |
|
</port> |
|
</output> |
|
</layer> |
|
<layer id="5" name="Result_13835" type="Result" version="opset1"> |
|
<input> |
|
<port id="0" precision="FP32"> |
|
<dim>-1</dim> |
|
<dim>-1</dim> |
|
<dim>16</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="2024.5.0-17202-a7ccc5e0efc" /> |
|
<conversion_parameters> |
|
<framework value="pytorch" /> |
|
<is_python_object value="True" /> |
|
</conversion_parameters> |
|
<optimum> |
|
<optimum_intel_version value="1.20.0.dev0+7cc52a7" /> |
|
<optimum_version value="1.23.2" /> |
|
<pytorch_version value="2.5.1" /> |
|
<transformers_version value="4.45.2" /> |
|
</optimum> |
|
</rt_info> |
|
</net> |
|
|