Vui Seng Chua
commited on
Commit
•
f9eb3e0
1
Parent(s):
28bc099
Hack input output node name of openvino_model.xml
Browse files- openvino_model.xml +4 -4
openvino_model.xml
CHANGED
|
@@ -1,13 +1,13 @@
|
|
| 1 |
<?xml version="1.0"?>
|
| 2 |
<net name="torch_jit" version="11">
|
| 3 |
<layers>
|
| 4 |
-
<layer id="0" name="
|
| 5 |
<data shape="?,?" element_type="i64" />
|
| 6 |
<rt_info>
|
| 7 |
<attribute name="old_api_map_element_type" version="0" value="i32" />
|
| 8 |
</rt_info>
|
| 9 |
<output>
|
| 10 |
-
<port id="0" precision="I64" names="
|
| 11 |
<dim>-1</dim>
|
| 12 |
<dim>-1</dim>
|
| 13 |
<rt_info>
|
|
@@ -24801,7 +24801,7 @@
|
|
| 24801 |
</port>
|
| 24802 |
</output>
|
| 24803 |
</layer>
|
| 24804 |
-
<layer id="1752" name="
|
| 24805 |
<data transpose_a="false" transpose_b="true" />
|
| 24806 |
<input>
|
| 24807 |
<port id="0" precision="FP32">
|
|
@@ -24815,7 +24815,7 @@
|
|
| 24815 |
</port>
|
| 24816 |
</input>
|
| 24817 |
<output>
|
| 24818 |
-
<port id="2" precision="FP32" names="
|
| 24819 |
<dim>-1</dim>
|
| 24820 |
<dim>-1</dim>
|
| 24821 |
<dim>50257</dim>
|
|
|
|
| 1 |
<?xml version="1.0"?>
|
| 2 |
<net name="torch_jit" version="11">
|
| 3 |
<layers>
|
| 4 |
+
<layer id="0" name="input_ids" type="Parameter" version="opset1">
|
| 5 |
<data shape="?,?" element_type="i64" />
|
| 6 |
<rt_info>
|
| 7 |
<attribute name="old_api_map_element_type" version="0" value="i32" />
|
| 8 |
</rt_info>
|
| 9 |
<output>
|
| 10 |
+
<port id="0" precision="I64" names="input_ids">
|
| 11 |
<dim>-1</dim>
|
| 12 |
<dim>-1</dim>
|
| 13 |
<rt_info>
|
|
|
|
| 24801 |
</port>
|
| 24802 |
</output>
|
| 24803 |
</layer>
|
| 24804 |
+
<layer id="1752" name="logits" type="MatMul" version="opset1">
|
| 24805 |
<data transpose_a="false" transpose_b="true" />
|
| 24806 |
<input>
|
| 24807 |
<port id="0" precision="FP32">
|
|
|
|
| 24815 |
</port>
|
| 24816 |
</input>
|
| 24817 |
<output>
|
| 24818 |
+
<port id="2" precision="FP32" names="logits">
|
| 24819 |
<dim>-1</dim>
|
| 24820 |
<dim>-1</dim>
|
| 24821 |
<dim>50257</dim>
|