raymondlo84 commited on
Commit
8e2d2fa
1 Parent(s): 4d72995

Upload 2 files

Browse files

fp16 version -- name matched.

Files changed (1) hide show
  1. unet.xml +38 -38
unet.xml CHANGED
@@ -1,13 +1,13 @@
1
  <?xml version="1.0"?>
2
  <net name="torch_jit" version="11">
3
  <layers>
4
- <layer id="2" name="sample" type="Parameter" version="opset1">
5
  <data shape="2,4,64,64" element_type="f32" />
6
  <rt_info>
7
  <attribute name="old_api_map_element_type" version="0" value="f16" />
8
  </rt_info>
9
  <output>
10
- <port id="0" precision="FP32" names="sample">
11
  <dim>2</dim>
12
  <dim>4</dim>
13
  <dim>64</dim>
@@ -15,22 +15,22 @@
15
  </port>
16
  </output>
17
  </layer>
18
- <layer id="1" name="timestep" type="Parameter" version="opset1">
19
  <data shape="" element_type="f64" />
20
  <rt_info>
21
  <attribute name="old_api_map_element_type" version="0" value="f16" />
22
  </rt_info>
23
  <output>
24
- <port id="0" precision="FP64" names="timestep" />
25
  </output>
26
  </layer>
27
- <layer id="0" name="encoder_hidden_state" type="Parameter" version="opset1">
28
  <data shape="2,77,768" element_type="f32" />
29
  <rt_info>
30
  <attribute name="old_api_map_element_type" version="0" value="f16" />
31
  </rt_info>
32
  <output>
33
- <port id="0" precision="FP32" names="encoder_hidden_state">
34
  <dim>2</dim>
35
  <dim>77</dim>
36
  <dim>768</dim>
@@ -4054,7 +4054,7 @@
4054
  </port>
4055
  </output>
4056
  </layer>
4057
- <layer id="258" name="Constant_14076" type="Const" version="opset1">
4058
  <data element_type="i64" shape="4" offset="9322776" size="32" />
4059
  <output>
4060
  <port id="0" precision="I64">
@@ -4883,7 +4883,7 @@
4883
  </port>
4884
  </output>
4885
  </layer>
4886
- <layer id="310" name="Constant_14169" type="Const" version="opset1">
4887
  <data element_type="i64" shape="4" offset="9322776" size="32" />
4888
  <output>
4889
  <port id="0" precision="I64">
@@ -7825,7 +7825,7 @@
7825
  </port>
7826
  </output>
7827
  </layer>
7828
- <layer id="489" name="Constant_15002" type="Const" version="opset1">
7829
  <data element_type="i64" shape="4" offset="9322776" size="32" />
7830
  <output>
7831
  <port id="0" precision="I64">
@@ -8654,7 +8654,7 @@
8654
  </port>
8655
  </output>
8656
  </layer>
8657
- <layer id="541" name="Constant_15095" type="Const" version="opset1">
8658
  <data element_type="i64" shape="4" offset="9322776" size="32" />
8659
  <output>
8660
  <port id="0" precision="I64">
@@ -11828,7 +11828,7 @@
11828
  </port>
11829
  </output>
11830
  </layer>
11831
- <layer id="732" name="Constant_16024" type="Const" version="opset1">
11832
  <data element_type="i64" shape="4" offset="9322776" size="32" />
11833
  <output>
11834
  <port id="0" precision="I64">
@@ -12657,7 +12657,7 @@
12657
  </port>
12658
  </output>
12659
  </layer>
12660
- <layer id="784" name="Constant_16117" type="Const" version="opset1">
12661
  <data element_type="i64" shape="4" offset="9322776" size="32" />
12662
  <output>
12663
  <port id="0" precision="I64">
@@ -15599,7 +15599,7 @@
15599
  </port>
15600
  </output>
15601
  </layer>
15602
- <layer id="963" name="Constant_16950" type="Const" version="opset1">
15603
  <data element_type="i64" shape="4" offset="9322776" size="32" />
15604
  <output>
15605
  <port id="0" precision="I64">
@@ -16428,7 +16428,7 @@
16428
  </port>
16429
  </output>
16430
  </layer>
16431
- <layer id="1015" name="Constant_17043" type="Const" version="opset1">
16432
  <data element_type="i64" shape="4" offset="9322776" size="32" />
16433
  <output>
16434
  <port id="0" precision="I64">
@@ -19602,7 +19602,7 @@
19602
  </port>
19603
  </output>
19604
  </layer>
19605
- <layer id="1206" name="Constant_17972" type="Const" version="opset1">
19606
  <data element_type="i64" shape="4" offset="9322776" size="32" />
19607
  <output>
19608
  <port id="0" precision="I64">
@@ -20431,7 +20431,7 @@
20431
  </port>
20432
  </output>
20433
  </layer>
20434
- <layer id="1258" name="Constant_18065" type="Const" version="opset1">
20435
  <data element_type="i64" shape="4" offset="9322776" size="32" />
20436
  <output>
20437
  <port id="0" precision="I64">
@@ -23373,7 +23373,7 @@
23373
  </port>
23374
  </output>
23375
  </layer>
23376
- <layer id="1437" name="Constant_18898" type="Const" version="opset1">
23377
  <data element_type="i64" shape="4" offset="9322776" size="32" />
23378
  <output>
23379
  <port id="0" precision="I64">
@@ -24202,7 +24202,7 @@
24202
  </port>
24203
  </output>
24204
  </layer>
24205
- <layer id="1489" name="Constant_18991" type="Const" version="opset1">
24206
  <data element_type="i64" shape="4" offset="9322776" size="32" />
24207
  <output>
24208
  <port id="0" precision="I64">
@@ -29028,7 +29028,7 @@
29028
  </port>
29029
  </output>
29030
  </layer>
29031
- <layer id="1774" name="Constant_20548" type="Const" version="opset1">
29032
  <data element_type="i64" shape="4" offset="9322776" size="32" />
29033
  <output>
29034
  <port id="0" precision="I64">
@@ -29857,7 +29857,7 @@
29857
  </port>
29858
  </output>
29859
  </layer>
29860
- <layer id="1826" name="Constant_20641" type="Const" version="opset1">
29861
  <data element_type="i64" shape="4" offset="9322776" size="32" />
29862
  <output>
29863
  <port id="0" precision="I64">
@@ -37106,7 +37106,7 @@
37106
  </port>
37107
  </output>
37108
  </layer>
37109
- <layer id="2245" name="Constant_23077" type="Const" version="opset1">
37110
  <data element_type="i64" shape="4" offset="9322776" size="32" />
37111
  <output>
37112
  <port id="0" precision="I64">
@@ -37935,7 +37935,7 @@
37935
  </port>
37936
  </output>
37937
  </layer>
37938
- <layer id="2297" name="Constant_23170" type="Const" version="opset1">
37939
  <data element_type="i64" shape="4" offset="9322776" size="32" />
37940
  <output>
37941
  <port id="0" precision="I64">
@@ -41018,7 +41018,7 @@
41018
  </port>
41019
  </output>
41020
  </layer>
41021
- <layer id="2483" name="Constant_24052" type="Const" version="opset1">
41022
  <data element_type="i64" shape="4" offset="9322776" size="32" />
41023
  <output>
41024
  <port id="0" precision="I64">
@@ -41847,7 +41847,7 @@
41847
  </port>
41848
  </output>
41849
  </layer>
41850
- <layer id="2535" name="Constant_24145" type="Const" version="opset1">
41851
  <data element_type="i64" shape="4" offset="9322776" size="32" />
41852
  <output>
41853
  <port id="0" precision="I64">
@@ -44930,7 +44930,7 @@
44930
  </port>
44931
  </output>
44932
  </layer>
44933
- <layer id="2721" name="Constant_25027" type="Const" version="opset1">
44934
  <data element_type="i64" shape="4" offset="9322776" size="32" />
44935
  <output>
44936
  <port id="0" precision="I64">
@@ -45759,7 +45759,7 @@
45759
  </port>
45760
  </output>
45761
  </layer>
45762
- <layer id="2773" name="Constant_25120" type="Const" version="opset1">
45763
  <data element_type="i64" shape="4" offset="9322776" size="32" />
45764
  <output>
45765
  <port id="0" precision="I64">
@@ -49049,7 +49049,7 @@
49049
  </port>
49050
  </output>
49051
  </layer>
49052
- <layer id="2971" name="Constant_26057" type="Const" version="opset1">
49053
  <data element_type="i64" shape="4" offset="9322776" size="32" />
49054
  <output>
49055
  <port id="0" precision="I64">
@@ -49878,7 +49878,7 @@
49878
  </port>
49879
  </output>
49880
  </layer>
49881
- <layer id="3023" name="Constant_26150" type="Const" version="opset1">
49882
  <data element_type="i64" shape="4" offset="9322776" size="32" />
49883
  <output>
49884
  <port id="0" precision="I64">
@@ -52961,7 +52961,7 @@
52961
  </port>
52962
  </output>
52963
  </layer>
52964
- <layer id="3209" name="Constant_27032" type="Const" version="opset1">
52965
  <data element_type="i64" shape="4" offset="9322776" size="32" />
52966
  <output>
52967
  <port id="0" precision="I64">
@@ -53790,7 +53790,7 @@
53790
  </port>
53791
  </output>
53792
  </layer>
53793
- <layer id="3261" name="Constant_27125" type="Const" version="opset1">
53794
  <data element_type="i64" shape="4" offset="9322776" size="32" />
53795
  <output>
53796
  <port id="0" precision="I64">
@@ -56873,7 +56873,7 @@
56873
  </port>
56874
  </output>
56875
  </layer>
56876
- <layer id="3447" name="Constant_28007" type="Const" version="opset1">
56877
  <data element_type="i64" shape="4" offset="9322776" size="32" />
56878
  <output>
56879
  <port id="0" precision="I64">
@@ -57702,7 +57702,7 @@
57702
  </port>
57703
  </output>
57704
  </layer>
57705
- <layer id="3499" name="Constant_28100" type="Const" version="opset1">
57706
  <data element_type="i64" shape="4" offset="9322776" size="32" />
57707
  <output>
57708
  <port id="0" precision="I64">
@@ -60992,7 +60992,7 @@
60992
  </port>
60993
  </output>
60994
  </layer>
60995
- <layer id="3697" name="Constant_29037" type="Const" version="opset1">
60996
  <data element_type="i64" shape="4" offset="9322776" size="32" />
60997
  <output>
60998
  <port id="0" precision="I64">
@@ -61821,7 +61821,7 @@
61821
  </port>
61822
  </output>
61823
  </layer>
61824
- <layer id="3749" name="Constant_29130" type="Const" version="opset1">
61825
  <data element_type="i64" shape="4" offset="9322776" size="32" />
61826
  <output>
61827
  <port id="0" precision="I64">
@@ -64904,7 +64904,7 @@
64904
  </port>
64905
  </output>
64906
  </layer>
64907
- <layer id="3935" name="Constant_30012" type="Const" version="opset1">
64908
  <data element_type="i64" shape="4" offset="9322776" size="32" />
64909
  <output>
64910
  <port id="0" precision="I64">
@@ -65733,7 +65733,7 @@
65733
  </port>
65734
  </output>
65735
  </layer>
65736
- <layer id="3987" name="Constant_30105" type="Const" version="opset1">
65737
  <data element_type="i64" shape="4" offset="9322776" size="32" />
65738
  <output>
65739
  <port id="0" precision="I64">
@@ -68816,7 +68816,7 @@
68816
  </port>
68817
  </output>
68818
  </layer>
68819
- <layer id="4173" name="Constant_30987" type="Const" version="opset1">
68820
  <data element_type="i64" shape="4" offset="9322776" size="32" />
68821
  <output>
68822
  <port id="0" precision="I64">
@@ -69645,7 +69645,7 @@
69645
  </port>
69646
  </output>
69647
  </layer>
69648
- <layer id="4225" name="Constant_31080" type="Const" version="opset1">
69649
  <data element_type="i64" shape="4" offset="9322776" size="32" />
69650
  <output>
69651
  <port id="0" precision="I64">
 
1
  <?xml version="1.0"?>
2
  <net name="torch_jit" version="11">
3
  <layers>
4
+ <layer id="2" name="latent_model_input" type="Parameter" version="opset1">
5
  <data shape="2,4,64,64" element_type="f32" />
6
  <rt_info>
7
  <attribute name="old_api_map_element_type" version="0" value="f16" />
8
  </rt_info>
9
  <output>
10
+ <port id="0" precision="FP32" names="latent_model_input">
11
  <dim>2</dim>
12
  <dim>4</dim>
13
  <dim>64</dim>
 
15
  </port>
16
  </output>
17
  </layer>
18
+ <layer id="1" name="t" type="Parameter" version="opset1">
19
  <data shape="" element_type="f64" />
20
  <rt_info>
21
  <attribute name="old_api_map_element_type" version="0" value="f16" />
22
  </rt_info>
23
  <output>
24
+ <port id="0" precision="FP64" names="t" />
25
  </output>
26
  </layer>
27
+ <layer id="0" name="encoder_hidden_states" type="Parameter" version="opset1">
28
  <data shape="2,77,768" element_type="f32" />
29
  <rt_info>
30
  <attribute name="old_api_map_element_type" version="0" value="f16" />
31
  </rt_info>
32
  <output>
33
+ <port id="0" precision="FP32" names="encoder_hidden_states">
34
  <dim>2</dim>
35
  <dim>77</dim>
36
  <dim>768</dim>
 
4054
  </port>
4055
  </output>
4056
  </layer>
4057
+ <layer id="258" name="Constant_14078" type="Const" version="opset1">
4058
  <data element_type="i64" shape="4" offset="9322776" size="32" />
4059
  <output>
4060
  <port id="0" precision="I64">
 
4883
  </port>
4884
  </output>
4885
  </layer>
4886
+ <layer id="310" name="Constant_14171" type="Const" version="opset1">
4887
  <data element_type="i64" shape="4" offset="9322776" size="32" />
4888
  <output>
4889
  <port id="0" precision="I64">
 
7825
  </port>
7826
  </output>
7827
  </layer>
7828
+ <layer id="489" name="Constant_15004" type="Const" version="opset1">
7829
  <data element_type="i64" shape="4" offset="9322776" size="32" />
7830
  <output>
7831
  <port id="0" precision="I64">
 
8654
  </port>
8655
  </output>
8656
  </layer>
8657
+ <layer id="541" name="Constant_15097" type="Const" version="opset1">
8658
  <data element_type="i64" shape="4" offset="9322776" size="32" />
8659
  <output>
8660
  <port id="0" precision="I64">
 
11828
  </port>
11829
  </output>
11830
  </layer>
11831
+ <layer id="732" name="Constant_16026" type="Const" version="opset1">
11832
  <data element_type="i64" shape="4" offset="9322776" size="32" />
11833
  <output>
11834
  <port id="0" precision="I64">
 
12657
  </port>
12658
  </output>
12659
  </layer>
12660
+ <layer id="784" name="Constant_16119" type="Const" version="opset1">
12661
  <data element_type="i64" shape="4" offset="9322776" size="32" />
12662
  <output>
12663
  <port id="0" precision="I64">
 
15599
  </port>
15600
  </output>
15601
  </layer>
15602
+ <layer id="963" name="Constant_16952" type="Const" version="opset1">
15603
  <data element_type="i64" shape="4" offset="9322776" size="32" />
15604
  <output>
15605
  <port id="0" precision="I64">
 
16428
  </port>
16429
  </output>
16430
  </layer>
16431
+ <layer id="1015" name="Constant_17045" type="Const" version="opset1">
16432
  <data element_type="i64" shape="4" offset="9322776" size="32" />
16433
  <output>
16434
  <port id="0" precision="I64">
 
19602
  </port>
19603
  </output>
19604
  </layer>
19605
+ <layer id="1206" name="Constant_17974" type="Const" version="opset1">
19606
  <data element_type="i64" shape="4" offset="9322776" size="32" />
19607
  <output>
19608
  <port id="0" precision="I64">
 
20431
  </port>
20432
  </output>
20433
  </layer>
20434
+ <layer id="1258" name="Constant_18067" type="Const" version="opset1">
20435
  <data element_type="i64" shape="4" offset="9322776" size="32" />
20436
  <output>
20437
  <port id="0" precision="I64">
 
23373
  </port>
23374
  </output>
23375
  </layer>
23376
+ <layer id="1437" name="Constant_18900" type="Const" version="opset1">
23377
  <data element_type="i64" shape="4" offset="9322776" size="32" />
23378
  <output>
23379
  <port id="0" precision="I64">
 
24202
  </port>
24203
  </output>
24204
  </layer>
24205
+ <layer id="1489" name="Constant_18993" type="Const" version="opset1">
24206
  <data element_type="i64" shape="4" offset="9322776" size="32" />
24207
  <output>
24208
  <port id="0" precision="I64">
 
29028
  </port>
29029
  </output>
29030
  </layer>
29031
+ <layer id="1774" name="Constant_20550" type="Const" version="opset1">
29032
  <data element_type="i64" shape="4" offset="9322776" size="32" />
29033
  <output>
29034
  <port id="0" precision="I64">
 
29857
  </port>
29858
  </output>
29859
  </layer>
29860
+ <layer id="1826" name="Constant_20643" type="Const" version="opset1">
29861
  <data element_type="i64" shape="4" offset="9322776" size="32" />
29862
  <output>
29863
  <port id="0" precision="I64">
 
37106
  </port>
37107
  </output>
37108
  </layer>
37109
+ <layer id="2245" name="Constant_23079" type="Const" version="opset1">
37110
  <data element_type="i64" shape="4" offset="9322776" size="32" />
37111
  <output>
37112
  <port id="0" precision="I64">
 
37935
  </port>
37936
  </output>
37937
  </layer>
37938
+ <layer id="2297" name="Constant_23172" type="Const" version="opset1">
37939
  <data element_type="i64" shape="4" offset="9322776" size="32" />
37940
  <output>
37941
  <port id="0" precision="I64">
 
41018
  </port>
41019
  </output>
41020
  </layer>
41021
+ <layer id="2483" name="Constant_24054" type="Const" version="opset1">
41022
  <data element_type="i64" shape="4" offset="9322776" size="32" />
41023
  <output>
41024
  <port id="0" precision="I64">
 
41847
  </port>
41848
  </output>
41849
  </layer>
41850
+ <layer id="2535" name="Constant_24147" type="Const" version="opset1">
41851
  <data element_type="i64" shape="4" offset="9322776" size="32" />
41852
  <output>
41853
  <port id="0" precision="I64">
 
44930
  </port>
44931
  </output>
44932
  </layer>
44933
+ <layer id="2721" name="Constant_25029" type="Const" version="opset1">
44934
  <data element_type="i64" shape="4" offset="9322776" size="32" />
44935
  <output>
44936
  <port id="0" precision="I64">
 
45759
  </port>
45760
  </output>
45761
  </layer>
45762
+ <layer id="2773" name="Constant_25122" type="Const" version="opset1">
45763
  <data element_type="i64" shape="4" offset="9322776" size="32" />
45764
  <output>
45765
  <port id="0" precision="I64">
 
49049
  </port>
49050
  </output>
49051
  </layer>
49052
+ <layer id="2971" name="Constant_26059" type="Const" version="opset1">
49053
  <data element_type="i64" shape="4" offset="9322776" size="32" />
49054
  <output>
49055
  <port id="0" precision="I64">
 
49878
  </port>
49879
  </output>
49880
  </layer>
49881
+ <layer id="3023" name="Constant_26152" type="Const" version="opset1">
49882
  <data element_type="i64" shape="4" offset="9322776" size="32" />
49883
  <output>
49884
  <port id="0" precision="I64">
 
52961
  </port>
52962
  </output>
52963
  </layer>
52964
+ <layer id="3209" name="Constant_27034" type="Const" version="opset1">
52965
  <data element_type="i64" shape="4" offset="9322776" size="32" />
52966
  <output>
52967
  <port id="0" precision="I64">
 
53790
  </port>
53791
  </output>
53792
  </layer>
53793
+ <layer id="3261" name="Constant_27127" type="Const" version="opset1">
53794
  <data element_type="i64" shape="4" offset="9322776" size="32" />
53795
  <output>
53796
  <port id="0" precision="I64">
 
56873
  </port>
56874
  </output>
56875
  </layer>
56876
+ <layer id="3447" name="Constant_28009" type="Const" version="opset1">
56877
  <data element_type="i64" shape="4" offset="9322776" size="32" />
56878
  <output>
56879
  <port id="0" precision="I64">
 
57702
  </port>
57703
  </output>
57704
  </layer>
57705
+ <layer id="3499" name="Constant_28102" type="Const" version="opset1">
57706
  <data element_type="i64" shape="4" offset="9322776" size="32" />
57707
  <output>
57708
  <port id="0" precision="I64">
 
60992
  </port>
60993
  </output>
60994
  </layer>
60995
+ <layer id="3697" name="Constant_29039" type="Const" version="opset1">
60996
  <data element_type="i64" shape="4" offset="9322776" size="32" />
60997
  <output>
60998
  <port id="0" precision="I64">
 
61821
  </port>
61822
  </output>
61823
  </layer>
61824
+ <layer id="3749" name="Constant_29132" type="Const" version="opset1">
61825
  <data element_type="i64" shape="4" offset="9322776" size="32" />
61826
  <output>
61827
  <port id="0" precision="I64">
 
64904
  </port>
64905
  </output>
64906
  </layer>
64907
+ <layer id="3935" name="Constant_30014" type="Const" version="opset1">
64908
  <data element_type="i64" shape="4" offset="9322776" size="32" />
64909
  <output>
64910
  <port id="0" precision="I64">
 
65733
  </port>
65734
  </output>
65735
  </layer>
65736
+ <layer id="3987" name="Constant_30107" type="Const" version="opset1">
65737
  <data element_type="i64" shape="4" offset="9322776" size="32" />
65738
  <output>
65739
  <port id="0" precision="I64">
 
68816
  </port>
68817
  </output>
68818
  </layer>
68819
+ <layer id="4173" name="Constant_30989" type="Const" version="opset1">
68820
  <data element_type="i64" shape="4" offset="9322776" size="32" />
68821
  <output>
68822
  <port id="0" precision="I64">
 
69645
  </port>
69646
  </output>
69647
  </layer>
69648
+ <layer id="4225" name="Constant_31082" type="Const" version="opset1">
69649
  <data element_type="i64" shape="4" offset="9322776" size="32" />
69650
  <output>
69651
  <port id="0" precision="I64">