File size: 3,161 Bytes
efc9c0a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 |
<?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="bf16" shape="32000, 4096" offset="0" size="262144000" />
<output>
<port id="0" precision="BF16" names="self.weight">
<dim>32000</dim>
<dim>4096</dim>
</port>
</output>
</layer>
<layer id="2" name="ov_ext::embedding/Convert" type="Convert" version="opset1">
<data destination_type="f32" />
<rt_info>
<attribute name="decompression" version="0" />
</rt_info>
<input>
<port id="0" precision="BF16">
<dim>32000</dim>
<dim>4096</dim>
</port>
</input>
<output>
<port id="1" precision="FP32">
<dim>32000</dim>
<dim>4096</dim>
</port>
</output>
</layer>
<layer id="3" name="ov_ext::embedding/Convert_1" 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="4" name="ov_ext::embedding/Constant" type="Const" version="opset1">
<data element_type="i32" shape="" offset="262144000" size="4" />
<output>
<port id="0" precision="I32" />
</output>
</layer>
<layer id="5" name="ov_ext::embedding/Gather" type="Gather" version="opset8">
<data batch_dims="0" />
<input>
<port id="0" precision="FP32">
<dim>32000</dim>
<dim>4096</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>4096</dim>
</port>
</output>
</layer>
<layer id="6" name="Result_65226" type="Result" version="opset1">
<input>
<port id="0" precision="FP32">
<dim>-1</dim>
<dim>-1</dim>
<dim>4096</dim>
</port>
</input>
</layer>
</layers>
<edges>
<edge from-layer="0" from-port="0" to-layer="3" to-port="0" />
<edge from-layer="1" from-port="0" to-layer="2" to-port="0" />
<edge from-layer="2" from-port="1" to-layer="5" to-port="0" />
<edge from-layer="3" from-port="1" to-layer="5" to-port="1" />
<edge from-layer="4" from-port="0" to-layer="5" to-port="2" />
<edge from-layer="5" from-port="3" to-layer="6" to-port="0" />
</edges>
<rt_info>
<Runtime_version value="2024.5.0-17288-7975fa5da0c-refs/pull/3856/head" />
<conversion_parameters>
<framework value="pytorch" />
<is_python_object value="True" />
</conversion_parameters>
<optimum>
<optimum_intel_version value="1.22.0.dev0+35cf1d2" />
<optimum_version value="1.23.3" />
<pytorch_version value="2.3.0" />
<transformers_version value="4.46.3" />
</optimum>
</rt_info>
</net>
|