alexwengg commited on
Commit
afd1b1e
·
verified ·
1 Parent(s): 162f51c

Upload 62 files

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. ANE/ANE-zh/KokoroAlbert.mlmodelc/analytics/coremldata.bin +3 -0
  2. ANE/ANE-zh/KokoroAlbert.mlmodelc/coremldata.bin +3 -0
  3. ANE/ANE-zh/KokoroAlbert.mlmodelc/metadata.json +90 -0
  4. ANE/ANE-zh/KokoroAlbert.mlmodelc/model.mil +0 -0
  5. ANE/ANE-zh/KokoroAlbert.mlmodelc/weights/weight.bin +3 -0
  6. ANE/ANE-zh/KokoroAlbert.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  7. ANE/ANE-zh/KokoroAlbert.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  8. ANE/ANE-zh/KokoroAlbert.mlpackage/Manifest.json +18 -0
  9. ANE/ANE-zh/KokoroAlignment.mlmodelc/analytics/coremldata.bin +3 -0
  10. ANE/ANE-zh/KokoroAlignment.mlmodelc/coremldata.bin +3 -0
  11. ANE/ANE-zh/KokoroAlignment.mlmodelc/metadata.json +107 -0
  12. ANE/ANE-zh/KokoroAlignment.mlmodelc/model.mil +54 -0
  13. ANE/ANE-zh/KokoroAlignment.mlmodelc/weights/weight.bin +3 -0
  14. ANE/ANE-zh/KokoroAlignment.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  15. ANE/ANE-zh/KokoroAlignment.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  16. ANE/ANE-zh/KokoroAlignment.mlpackage/Manifest.json +18 -0
  17. ANE/ANE-zh/KokoroNoise.mlmodelc/analytics/coremldata.bin +3 -0
  18. ANE/ANE-zh/KokoroNoise.mlmodelc/coremldata.bin +3 -0
  19. ANE/ANE-zh/KokoroNoise.mlmodelc/metadata.json +111 -0
  20. ANE/ANE-zh/KokoroNoise.mlmodelc/model.mil +535 -0
  21. ANE/ANE-zh/KokoroNoise.mlmodelc/weights/weight.bin +3 -0
  22. ANE/ANE-zh/KokoroNoise.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  23. ANE/ANE-zh/KokoroNoise.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  24. ANE/ANE-zh/KokoroNoise.mlpackage/Manifest.json +18 -0
  25. ANE/ANE-zh/KokoroPostAlbert.mlmodelc/analytics/coremldata.bin +3 -0
  26. ANE/ANE-zh/KokoroPostAlbert.mlmodelc/coremldata.bin +3 -0
  27. ANE/ANE-zh/KokoroPostAlbert.mlmodelc/metadata.json +148 -0
  28. ANE/ANE-zh/KokoroPostAlbert.mlmodelc/model.mil +277 -0
  29. ANE/ANE-zh/KokoroPostAlbert.mlmodelc/weights/weight.bin +3 -0
  30. ANE/ANE-zh/KokoroPostAlbert.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  31. ANE/ANE-zh/KokoroPostAlbert.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  32. ANE/ANE-zh/KokoroPostAlbert.mlpackage/Manifest.json +18 -0
  33. ANE/ANE-zh/KokoroProsody.mlmodelc/analytics/coremldata.bin +3 -0
  34. ANE/ANE-zh/KokoroProsody.mlmodelc/coremldata.bin +3 -0
  35. ANE/ANE-zh/KokoroProsody.mlmodelc/metadata.json +98 -0
  36. ANE/ANE-zh/KokoroProsody.mlmodelc/model.mil +394 -0
  37. ANE/ANE-zh/KokoroProsody.mlmodelc/weights/weight.bin +3 -0
  38. ANE/ANE-zh/KokoroProsody.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  39. ANE/ANE-zh/KokoroProsody.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  40. ANE/ANE-zh/KokoroProsody.mlpackage/Manifest.json +18 -0
  41. ANE/ANE-zh/KokoroTail.mlmodelc/analytics/coremldata.bin +3 -0
  42. ANE/ANE-zh/KokoroTail.mlmodelc/coremldata.bin +3 -0
  43. ANE/ANE-zh/KokoroTail.mlmodelc/metadata.json +70 -0
  44. ANE/ANE-zh/KokoroTail.mlmodelc/model.mil +47 -0
  45. ANE/ANE-zh/KokoroTail.mlmodelc/weights/weight.bin +3 -0
  46. ANE/ANE-zh/KokoroTail.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  47. ANE/ANE-zh/KokoroTail.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  48. ANE/ANE-zh/KokoroTail.mlpackage/Manifest.json +18 -0
  49. ANE/ANE-zh/KokoroVocoder.mlmodelc/analytics/coremldata.bin +3 -0
  50. ANE/ANE-zh/KokoroVocoder.mlmodelc/coremldata.bin +3 -0
ANE/ANE-zh/KokoroAlbert.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3b66851b7a61e9b2f2fb17a5050a46f5d98acaec46b3822b65c3e509c15c94e
3
+ size 243
ANE/ANE-zh/KokoroAlbert.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bae8190320b079bc995d69b19c2247c2e66faf5ebefe0cfb99bddf4efa98a86c
3
+ size 433
ANE/ANE-zh/KokoroAlbert.mlmodelc/metadata.json ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "metadataOutputVersion" : "3.0",
4
+ "storagePrecision" : "Mixed (Float16, Palettized (8 bits))",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float16",
10
+ "formattedType" : "MultiArray (Float16)",
11
+ "shortDescription" : "",
12
+ "shape" : "[]",
13
+ "name" : "bert_dur",
14
+ "type" : "MultiArray"
15
+ }
16
+ ],
17
+ "modelParameters" : [
18
+
19
+ ],
20
+ "specificationVersion" : 8,
21
+ "mlProgramOperationTypeHistogram" : {
22
+ "Ios17.mul" : 13,
23
+ "Ios17.linear" : 73,
24
+ "Ios17.transpose" : 48,
25
+ "Ios17.sub" : 2,
26
+ "Ios16.constexprLutToDense" : 9,
27
+ "Range1d" : 1,
28
+ "Ios17.matmul" : 24,
29
+ "Shape" : 1,
30
+ "Ios17.gather" : 4,
31
+ "Ios17.add" : 38,
32
+ "Ios16.softmax" : 12,
33
+ "Ios17.expandDims" : 3,
34
+ "Ios16.gelu" : 12,
35
+ "Ios17.cast" : 6,
36
+ "Ios17.layerNorm" : 25,
37
+ "Ios17.reshape" : 48
38
+ },
39
+ "computePrecision" : "Mixed (Float16, Int16, Int32, UInt16)",
40
+ "isUpdatable" : "0",
41
+ "stateSchema" : [
42
+
43
+ ],
44
+ "availability" : {
45
+ "macOS" : "14.0",
46
+ "tvOS" : "17.0",
47
+ "visionOS" : "1.0",
48
+ "watchOS" : "10.0",
49
+ "iOS" : "17.0",
50
+ "macCatalyst" : "17.0"
51
+ },
52
+ "modelType" : {
53
+ "name" : "MLModelType_mlProgram"
54
+ },
55
+ "userDefinedMetadata" : {
56
+ "com.github.apple.coremltools.conversion_date" : "2026-05-03",
57
+ "com.github.apple.coremltools.source" : "torch==2.11.0",
58
+ "com.github.apple.coremltools.version" : "9.0",
59
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
60
+ },
61
+ "inputSchema" : [
62
+ {
63
+ "dataType" : "Int32",
64
+ "hasShapeFlexibility" : "1",
65
+ "isOptional" : "0",
66
+ "shapeFlexibility" : "1 × 2...512",
67
+ "shapeRange" : "[[1, 1], [2, 512]]",
68
+ "formattedType" : "MultiArray (Int32 1 × 37)",
69
+ "type" : "MultiArray",
70
+ "shape" : "[1, 37]",
71
+ "name" : "input_ids",
72
+ "shortDescription" : ""
73
+ },
74
+ {
75
+ "dataType" : "Int32",
76
+ "hasShapeFlexibility" : "1",
77
+ "isOptional" : "0",
78
+ "shapeFlexibility" : "1 × 2...512",
79
+ "shapeRange" : "[[1, 1], [2, 512]]",
80
+ "formattedType" : "MultiArray (Int32 1 × 37)",
81
+ "type" : "MultiArray",
82
+ "shape" : "[1, 37]",
83
+ "name" : "attention_mask",
84
+ "shortDescription" : ""
85
+ }
86
+ ],
87
+ "generatedClassName" : "KokoroAlbert",
88
+ "method" : "predict"
89
+ }
90
+ ]
ANE/ANE-zh/KokoroAlbert.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
ANE/ANE-zh/KokoroAlbert.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c3d8458f357f2e60380de284139af566c6c22e3719bb985524616d0a4f57e5b
3
+ size 5718848
ANE/ANE-zh/KokoroAlbert.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7871b55a006dd65f8531edbbd48acefe4f554bad1182b852630641c3dfde2d79
3
+ size 92156
ANE/ANE-zh/KokoroAlbert.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c3d8458f357f2e60380de284139af566c6c22e3719bb985524616d0a4f57e5b
3
+ size 5718848
ANE/ANE-zh/KokoroAlbert.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "9FF918C7-3A11-4CA4-8998-080F81BAF843": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Weights",
7
+ "name": "weights",
8
+ "path": "com.apple.CoreML/weights"
9
+ },
10
+ "EFC4F885-8922-4367-A207-C478D0734CCA": {
11
+ "author": "com.apple.CoreML",
12
+ "description": "CoreML Model Specification",
13
+ "name": "model.mlmodel",
14
+ "path": "com.apple.CoreML/model.mlmodel"
15
+ }
16
+ },
17
+ "rootModelIdentifier": "EFC4F885-8922-4367-A207-C478D0734CCA"
18
+ }
ANE/ANE-zh/KokoroAlignment.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c84c8820e2e6f1b27229db6ce4ec2cc2817c546aa31d2d2f677d35be5445443
3
+ size 243
ANE/ANE-zh/KokoroAlignment.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bede89325746a1b540c3350a473ceeaea39a5cea7a641c340e9013655865936
3
+ size 484
ANE/ANE-zh/KokoroAlignment.mlmodelc/metadata.json ADDED
@@ -0,0 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "metadataOutputVersion" : "3.0",
4
+ "storagePrecision" : "Float16",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float16",
10
+ "formattedType" : "MultiArray (Float16)",
11
+ "shortDescription" : "",
12
+ "shape" : "[]",
13
+ "name" : "en",
14
+ "type" : "MultiArray"
15
+ },
16
+ {
17
+ "hasShapeFlexibility" : "0",
18
+ "isOptional" : "0",
19
+ "dataType" : "Float16",
20
+ "formattedType" : "MultiArray (Float16)",
21
+ "shortDescription" : "",
22
+ "shape" : "[]",
23
+ "name" : "asr",
24
+ "type" : "MultiArray"
25
+ }
26
+ ],
27
+ "modelParameters" : [
28
+
29
+ ],
30
+ "specificationVersion" : 8,
31
+ "mlProgramOperationTypeHistogram" : {
32
+ "Ios17.squeeze" : 1,
33
+ "Ios16.cumsum" : 1,
34
+ "Ios17.logicalAnd" : 1,
35
+ "Ios17.cast" : 4,
36
+ "Ios17.less" : 1,
37
+ "Ios17.matmul" : 2,
38
+ "Ios17.sliceByIndex" : 3,
39
+ "Ios17.concat" : 2,
40
+ "Ios17.greaterEqual" : 1,
41
+ "Ios17.sub" : 1,
42
+ "Ios17.expandDims" : 2
43
+ },
44
+ "computePrecision" : "Mixed (Float16, Int16, Int32)",
45
+ "isUpdatable" : "0",
46
+ "stateSchema" : [
47
+
48
+ ],
49
+ "availability" : {
50
+ "macOS" : "14.0",
51
+ "tvOS" : "17.0",
52
+ "visionOS" : "1.0",
53
+ "watchOS" : "10.0",
54
+ "iOS" : "17.0",
55
+ "macCatalyst" : "17.0"
56
+ },
57
+ "modelType" : {
58
+ "name" : "MLModelType_mlProgram"
59
+ },
60
+ "userDefinedMetadata" : {
61
+ "com.github.apple.coremltools.conversion_date" : "2026-05-03",
62
+ "com.github.apple.coremltools.source" : "torch==2.11.0",
63
+ "com.github.apple.coremltools.version" : "9.0",
64
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
65
+ },
66
+ "inputSchema" : [
67
+ {
68
+ "dataType" : "Int32",
69
+ "hasShapeFlexibility" : "1",
70
+ "isOptional" : "0",
71
+ "shapeFlexibility" : "1 × 2...512",
72
+ "shapeRange" : "[[1, 1], [2, 512]]",
73
+ "formattedType" : "MultiArray (Int32 1 × 37)",
74
+ "type" : "MultiArray",
75
+ "shape" : "[1, 37]",
76
+ "name" : "pred_dur",
77
+ "shortDescription" : ""
78
+ },
79
+ {
80
+ "dataType" : "Float16",
81
+ "hasShapeFlexibility" : "1",
82
+ "isOptional" : "0",
83
+ "shapeFlexibility" : "1 × 2...512 × 640",
84
+ "shapeRange" : "[[1, 1], [2, 512], [640, 640]]",
85
+ "formattedType" : "MultiArray (Float16 1 × 37 × 640)",
86
+ "type" : "MultiArray",
87
+ "shape" : "[1, 37, 640]",
88
+ "name" : "d",
89
+ "shortDescription" : ""
90
+ },
91
+ {
92
+ "dataType" : "Float16",
93
+ "hasShapeFlexibility" : "1",
94
+ "isOptional" : "0",
95
+ "shapeFlexibility" : "1 × 512 × 2...512",
96
+ "shapeRange" : "[[1, 1], [512, 512], [2, 512]]",
97
+ "formattedType" : "MultiArray (Float16 1 × 512 × 37)",
98
+ "type" : "MultiArray",
99
+ "shape" : "[1, 512, 37]",
100
+ "name" : "t_en",
101
+ "shortDescription" : ""
102
+ }
103
+ ],
104
+ "generatedClassName" : "KokoroAlignment",
105
+ "method" : "predict"
106
+ }
107
+ ]
ANE/ANE-zh/KokoroAlignment.mlmodelc/model.mil ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ program(1.0)
2
+ [buildInfo = dict<tensor<string, []>, tensor<string, []>>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}})]
3
+ {
4
+ func main<ios17>(tensor<fp16, [1, ?, 640]> d, tensor<int32, [1, ?]> pred_dur, tensor<fp16, [1, 512, ?]> t_en) [FlexibleShapeInformation = tuple<tuple<tensor<string, []>, dict<tensor<string, []>, tensor<int32, [?]>>>, tuple<tensor<string, []>, dict<tensor<string, []>, list<tensor<int32, [2]>, ?>>>>((("DefaultShapes", {{"d", [1, 37, 640]}, {"pred_dur", [1, 37]}, {"t_en", [1, 512, 37]}}), ("RangeDims", {{"d", [[1, 1], [2, 512], [640, 640]]}, {"pred_dur", [[1, 1], [2, 512]]}, {"t_en", [[1, 1], [512, 512], [2, 512]]}})))] {
5
+ tensor<int32, []> var_19 = const()[name = tensor<string, []>("op_19"), val = tensor<int32, []>(-1)];
6
+ tensor<bool, []> cum_dur_exclusive_0 = const()[name = tensor<string, []>("cum_dur_exclusive_0"), val = tensor<bool, []>(false)];
7
+ tensor<bool, []> cum_dur_reverse_0 = const()[name = tensor<string, []>("cum_dur_reverse_0"), val = tensor<bool, []>(false)];
8
+ tensor<string, []> dur_to_fp16_dtype_0 = const()[name = tensor<string, []>("dur_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
9
+ tensor<fp16, [1, ?]> pred_dur_to_fp16 = cast(dtype = dur_to_fp16_dtype_0, x = pred_dur)[name = tensor<string, []>("cast_3")];
10
+ tensor<fp16, [1, ?]> cum_dur_cast_fp16 = cumsum(axis = var_19, exclusive = cum_dur_exclusive_0, reverse = cum_dur_reverse_0, x = pred_dur_to_fp16)[name = tensor<string, []>("cum_dur_cast_fp16")];
11
+ tensor<fp16, [1, ?]> starts_cast_fp16 = sub(x = cum_dur_cast_fp16, y = pred_dur_to_fp16)[name = tensor<string, []>("starts_cast_fp16")];
12
+ tensor<int32, [1]> var_40_axes_0 = const()[name = tensor<string, []>("op_40_axes_0"), val = tensor<int32, [1]>([-1])];
13
+ tensor<fp16, [1, ?, 1]> var_40_cast_fp16 = expand_dims(axes = var_40_axes_0, x = starts_cast_fp16)[name = tensor<string, []>("op_40_cast_fp16")];
14
+ tensor<fp16, [1, 1, 2000]> frames_to_fp16 = const()[name = tensor<string, []>("frames_to_fp16"), val = tensor<fp16, [1, 1, 2000]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64)))];
15
+ tensor<bool, [1, ?, 2000]> var_41_cast_fp16 = greater_equal(x = frames_to_fp16, y = var_40_cast_fp16)[name = tensor<string, []>("op_41_cast_fp16")];
16
+ tensor<int32, [1]> var_43_axes_0 = const()[name = tensor<string, []>("op_43_axes_0"), val = tensor<int32, [1]>([-1])];
17
+ tensor<fp16, [1, ?, 1]> var_43_cast_fp16 = expand_dims(axes = var_43_axes_0, x = cum_dur_cast_fp16)[name = tensor<string, []>("op_43_cast_fp16")];
18
+ tensor<bool, [1, ?, 2000]> var_44_cast_fp16 = less(x = frames_to_fp16, y = var_43_cast_fp16)[name = tensor<string, []>("op_44_cast_fp16")];
19
+ tensor<bool, [1, ?, 2000]> var_45 = logical_and(x = var_41_cast_fp16, y = var_44_cast_fp16)[name = tensor<string, []>("op_45")];
20
+ tensor<bool, []> en_transpose_x_1 = const()[name = tensor<string, []>("en_transpose_x_1"), val = tensor<bool, []>(true)];
21
+ tensor<bool, []> en_transpose_y_1 = const()[name = tensor<string, []>("en_transpose_y_1"), val = tensor<bool, []>(false)];
22
+ tensor<string, []> alignment_to_fp16_dtype_0 = const()[name = tensor<string, []>("alignment_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
23
+ tensor<fp16, [1, ?, 2000]> var_45_to_fp16 = cast(dtype = alignment_to_fp16_dtype_0, x = var_45)[name = tensor<string, []>("cast_2")];
24
+ tensor<fp16, [1, 640, 2000]> en_cast_fp16 = matmul(transpose_x = en_transpose_x_1, transpose_y = en_transpose_y_1, x = d, y = var_45_to_fp16)[name = tensor<string, []>("en_cast_fp16")];
25
+ tensor<bool, []> asr_transpose_x_0 = const()[name = tensor<string, []>("asr_transpose_x_0"), val = tensor<bool, []>(false)];
26
+ tensor<bool, []> asr_transpose_y_0 = const()[name = tensor<string, []>("asr_transpose_y_0"), val = tensor<bool, []>(false)];
27
+ tensor<fp16, [1, 512, 2000]> asr_cast_fp16 = matmul(transpose_x = asr_transpose_x_0, transpose_y = asr_transpose_y_0, x = t_en, y = var_45_to_fp16)[name = tensor<string, []>("asr_cast_fp16")];
28
+ tensor<int32, [2]> var_65_begin_0 = const()[name = tensor<string, []>("op_65_begin_0"), val = tensor<int32, [2]>([0, -1])];
29
+ tensor<int32, [2]> var_65_end_0 = const()[name = tensor<string, []>("op_65_end_0"), val = tensor<int32, [2]>([1, 0])];
30
+ tensor<bool, [2]> var_65_end_mask_0 = const()[name = tensor<string, []>("op_65_end_mask_0"), val = tensor<bool, [2]>([true, true])];
31
+ tensor<fp16, [1, 1]> var_65_cast_fp16 = slice_by_index(begin = var_65_begin_0, end = var_65_end_0, end_mask = var_65_end_mask_0, x = cum_dur_cast_fp16)[name = tensor<string, []>("op_65_cast_fp16")];
32
+ tensor<string, []> var_70_to_int16_dtype_0 = const()[name = tensor<string, []>("op_70_to_int16_dtype_0"), val = tensor<string, []>("int16")];
33
+ tensor<int16, [1, 1]> var_65_cast_fp16_to_int16 = cast(dtype = var_70_to_int16_dtype_0, x = var_65_cast_fp16)[name = tensor<string, []>("cast_1")];
34
+ tensor<int16, []> T_a_cast_int16 = squeeze(x = var_65_cast_fp16_to_int16)[name = tensor<string, []>("T_a_cast_int16")];
35
+ tensor<string, []> T_a_cast_int16_to_int32_dtype_0 = const()[name = tensor<string, []>("T_a_cast_int16_to_int32_dtype_0"), val = tensor<string, []>("int32")];
36
+ tensor<int32, []> concat_0_values0_0 = const()[name = tensor<string, []>("concat_0_values0_0"), val = tensor<int32, []>(1)];
37
+ tensor<int32, []> concat_0_values1_0 = const()[name = tensor<string, []>("concat_0_values1_0"), val = tensor<int32, []>(640)];
38
+ tensor<int32, []> concat_0_axis_0 = const()[name = tensor<string, []>("concat_0_axis_0"), val = tensor<int32, []>(0)];
39
+ tensor<bool, []> concat_0_interleave_0 = const()[name = tensor<string, []>("concat_0_interleave_0"), val = tensor<bool, []>(false)];
40
+ tensor<int32, []> T_a_cast_int16_to_int32 = cast(dtype = T_a_cast_int16_to_int32_dtype_0, x = T_a_cast_int16)[name = tensor<string, []>("cast_0")];
41
+ tensor<int32, [3]> concat_0 = concat(axis = concat_0_axis_0, interleave = concat_0_interleave_0, values = (concat_0_values0_0, concat_0_values1_0, T_a_cast_int16_to_int32))[name = tensor<string, []>("concat_0")];
42
+ tensor<int32, [3]> var_87_begin_0 = const()[name = tensor<string, []>("op_87_begin_0"), val = tensor<int32, [3]>([0, 0, 0])];
43
+ tensor<bool, [3]> var_87_end_mask_0 = const()[name = tensor<string, []>("op_87_end_mask_0"), val = tensor<bool, [3]>([true, true, false])];
44
+ tensor<fp16, [1, 640, ?]> en = slice_by_index(begin = var_87_begin_0, end = concat_0, end_mask = var_87_end_mask_0, x = en_cast_fp16)[name = tensor<string, []>("op_87_cast_fp16")];
45
+ tensor<int32, []> concat_1_values0_0 = const()[name = tensor<string, []>("concat_1_values0_0"), val = tensor<int32, []>(1)];
46
+ tensor<int32, []> concat_1_values1_0 = const()[name = tensor<string, []>("concat_1_values1_0"), val = tensor<int32, []>(512)];
47
+ tensor<int32, []> concat_1_axis_0 = const()[name = tensor<string, []>("concat_1_axis_0"), val = tensor<int32, []>(0)];
48
+ tensor<bool, []> concat_1_interleave_0 = const()[name = tensor<string, []>("concat_1_interleave_0"), val = tensor<bool, []>(false)];
49
+ tensor<int32, [3]> concat_1 = concat(axis = concat_1_axis_0, interleave = concat_1_interleave_0, values = (concat_1_values0_0, concat_1_values1_0, T_a_cast_int16_to_int32))[name = tensor<string, []>("concat_1")];
50
+ tensor<int32, [3]> var_101_begin_0 = const()[name = tensor<string, []>("op_101_begin_0"), val = tensor<int32, [3]>([0, 0, 0])];
51
+ tensor<bool, [3]> var_101_end_mask_0 = const()[name = tensor<string, []>("op_101_end_mask_0"), val = tensor<bool, [3]>([true, true, false])];
52
+ tensor<fp16, [1, 512, ?]> asr = slice_by_index(begin = var_101_begin_0, end = concat_1, end_mask = var_101_end_mask_0, x = asr_cast_fp16)[name = tensor<string, []>("op_101_cast_fp16")];
53
+ } -> (en, asr);
54
+ }
ANE/ANE-zh/KokoroAlignment.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e7d69128b59d615fc3d3cf85637a687235fc086b1eb136359adb11a61615f6b
3
+ size 4128
ANE/ANE-zh/KokoroAlignment.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e5b90ecdad726428342766d68fecde858ea7218484141cde85af1672aa98279
3
+ size 6971
ANE/ANE-zh/KokoroAlignment.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e7d69128b59d615fc3d3cf85637a687235fc086b1eb136359adb11a61615f6b
3
+ size 4128
ANE/ANE-zh/KokoroAlignment.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "538DCB08-7D66-45AF-AB6B-C363411A8FC9": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Specification",
7
+ "name": "model.mlmodel",
8
+ "path": "com.apple.CoreML/model.mlmodel"
9
+ },
10
+ "910C3F65-B5FE-43B5-80BA-3B9677FBA054": {
11
+ "author": "com.apple.CoreML",
12
+ "description": "CoreML Model Weights",
13
+ "name": "weights",
14
+ "path": "com.apple.CoreML/weights"
15
+ }
16
+ },
17
+ "rootModelIdentifier": "538DCB08-7D66-45AF-AB6B-C363411A8FC9"
18
+ }
ANE/ANE-zh/KokoroNoise.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7536e2bb87ceabc3b06e8e7c44cd658e050f7c4fab89b45e3c758ff5c8f1d59
3
+ size 243
ANE/ANE-zh/KokoroNoise.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7030a1d8a73eb6cda44c7ba9afabb6856c098b62bd728b5af2875900a5e5f155
3
+ size 440
ANE/ANE-zh/KokoroNoise.mlmodelc/metadata.json ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "metadataOutputVersion" : "3.0",
4
+ "storagePrecision" : "Mixed (Float32, Palettized (8 bits))",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float32",
10
+ "formattedType" : "MultiArray (Float32)",
11
+ "shortDescription" : "",
12
+ "shape" : "[]",
13
+ "name" : "x_source_0",
14
+ "type" : "MultiArray"
15
+ },
16
+ {
17
+ "hasShapeFlexibility" : "0",
18
+ "isOptional" : "0",
19
+ "dataType" : "Float32",
20
+ "formattedType" : "MultiArray (Float32)",
21
+ "shortDescription" : "",
22
+ "shape" : "[]",
23
+ "name" : "x_source_1",
24
+ "type" : "MultiArray"
25
+ }
26
+ ],
27
+ "modelParameters" : [
28
+
29
+ ],
30
+ "specificationVersion" : 8,
31
+ "mlProgramOperationTypeHistogram" : {
32
+ "Ios16.cumsum" : 1,
33
+ "Ios17.concat" : 1,
34
+ "UpsampleNearestNeighbor" : 1,
35
+ "Ios17.equal" : 1,
36
+ "Ios17.logicalAnd" : 5,
37
+ "Ios17.reshape" : 12,
38
+ "Ios17.instanceNorm" : 12,
39
+ "Ios17.transpose" : 5,
40
+ "Ios17.sin" : 13,
41
+ "Split" : 12,
42
+ "Ios17.expandDims" : 4,
43
+ "Ios16.avgPool" : 1,
44
+ "Ios17.add" : 50,
45
+ "Ios17.squeeze" : 3,
46
+ "Pad" : 1,
47
+ "Ios16.upsampleBilinear" : 1,
48
+ "Ios17.sqrt" : 1,
49
+ "Ios17.sub" : 4,
50
+ "Ios16.constexprLutToDense" : 26,
51
+ "Ios17.conv" : 16,
52
+ "Ios17.tanh" : 1,
53
+ "Ios17.linear" : 13,
54
+ "Ios17.pow" : 14,
55
+ "Ios17.cast" : 6,
56
+ "Ios17.less" : 3,
57
+ "Ios17.realDiv" : 1,
58
+ "Ios17.atan" : 1,
59
+ "Ios17.greater" : 3,
60
+ "Ios17.mul" : 53
61
+ },
62
+ "computePrecision" : "Mixed (Float32, Int32)",
63
+ "isUpdatable" : "0",
64
+ "stateSchema" : [
65
+
66
+ ],
67
+ "availability" : {
68
+ "macOS" : "14.0",
69
+ "tvOS" : "17.0",
70
+ "visionOS" : "1.0",
71
+ "watchOS" : "10.0",
72
+ "iOS" : "17.0",
73
+ "macCatalyst" : "17.0"
74
+ },
75
+ "modelType" : {
76
+ "name" : "MLModelType_mlProgram"
77
+ },
78
+ "userDefinedMetadata" : {
79
+ "com.github.apple.coremltools.conversion_date" : "2026-05-03",
80
+ "com.github.apple.coremltools.source" : "torch==2.11.0",
81
+ "com.github.apple.coremltools.version" : "9.0",
82
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
83
+ },
84
+ "inputSchema" : [
85
+ {
86
+ "dataType" : "Float32",
87
+ "hasShapeFlexibility" : "1",
88
+ "isOptional" : "0",
89
+ "shapeFlexibility" : "1 × 2...4000",
90
+ "shapeRange" : "[[1, 1], [2, 4000]]",
91
+ "formattedType" : "MultiArray (Float32 1 × 266)",
92
+ "type" : "MultiArray",
93
+ "shape" : "[1, 266]",
94
+ "name" : "F0_curve",
95
+ "shortDescription" : ""
96
+ },
97
+ {
98
+ "hasShapeFlexibility" : "0",
99
+ "isOptional" : "0",
100
+ "dataType" : "Float32",
101
+ "formattedType" : "MultiArray (Float32 1 × 128)",
102
+ "shortDescription" : "",
103
+ "shape" : "[1, 128]",
104
+ "name" : "style_timbre",
105
+ "type" : "MultiArray"
106
+ }
107
+ ],
108
+ "generatedClassName" : "KokoroNoise",
109
+ "method" : "predict"
110
+ }
111
+ ]
ANE/ANE-zh/KokoroNoise.mlmodelc/model.mil ADDED
@@ -0,0 +1,535 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ program(1.0)
2
+ [buildInfo = dict<tensor<string, []>, tensor<string, []>>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}})]
3
+ {
4
+ func main<ios17>(tensor<fp32, [1, ?]> F0_curve, tensor<fp32, [1, 128]> style_timbre) [FlexibleShapeInformation = tuple<tuple<tensor<string, []>, dict<tensor<string, []>, tensor<int32, [?]>>>, tuple<tensor<string, []>, dict<tensor<string, []>, list<tensor<int32, [2]>, ?>>>>((("DefaultShapes", {{"F0_curve", [1, 266]}}), ("RangeDims", {{"F0_curve", [[1, 1], [2, 4000]]}})))] {
5
+ tensor<fp32, [1]> m_source_l_linear_bias = const()[name = tensor<string, []>("m_source_l_linear_bias"), val = tensor<fp32, [1]>([-0x1.e28358p-6])];
6
+ tensor<fp32, [1, 9]> m_source_l_linear_weight = const()[name = tensor<string, []>("m_source_l_linear_weight"), val = tensor<fp32, [1, 9]>([[-0x1.4dfed8p-4, -0x1.7b4864p-3, -0x1.7608cep-3, -0x1.6d4e54p-3, -0x1.946f4ap-4, 0x1.527ebcp-4, 0x1.66277ap-4, -0x1.900fdap-2, -0x1.1871f2p-1]])];
7
+ tensor<fp32, [11, 1, 20]> stft_conv_real_weight = const()[name = tensor<string, []>("stft_conv_real_weight"), val = tensor<fp32, [11, 1, 20]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64)))];
8
+ tensor<fp32, [11, 1, 20]> stft_conv_imag_weight = const()[name = tensor<string, []>("stft_conv_imag_weight"), val = tensor<fp32, [11, 1, 20]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1024)))];
9
+ tensor<fp32, [256]> noise_convs_0_bias = const()[name = tensor<string, []>("noise_convs_0_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1984)))];
10
+ tensor<fp32, [256, 22, 12]> noise_convs_0_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [67584]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3072))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(70720))), name = tensor<string, []>("noise_convs_0_weight_palettized"), shape = tensor<uint32, [3]>([256, 22, 12])];
11
+ tensor<fp32, [1, 256, 1]> noise_res_0_alpha2_2 = const()[name = tensor<string, []>("noise_res_0_alpha2_2"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(71808)))];
12
+ tensor<fp32, [1, 256, 1]> noise_res_0_alpha1_2 = const()[name = tensor<string, []>("noise_res_0_alpha1_2"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(72896)))];
13
+ tensor<fp32, [1, 256, 1]> noise_res_0_alpha2_1 = const()[name = tensor<string, []>("noise_res_0_alpha2_1"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(73984)))];
14
+ tensor<fp32, [1, 256, 1]> noise_res_0_alpha1_1 = const()[name = tensor<string, []>("noise_res_0_alpha1_1"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(75072)))];
15
+ tensor<fp32, [1, 256, 1]> noise_res_0_alpha2_0 = const()[name = tensor<string, []>("noise_res_0_alpha2_0"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(76160)))];
16
+ tensor<fp32, [1, 256, 1]> noise_res_0_alpha1_0 = const()[name = tensor<string, []>("noise_res_0_alpha1_0"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(77248)))];
17
+ tensor<fp32, [512]> noise_res_0_adain1_0_fc_bias = const()[name = tensor<string, []>("noise_res_0_adain1_0_fc_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(78336)))];
18
+ tensor<fp32, [512, 128]> noise_res_0_adain1_0_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(80448))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(146048))), name = tensor<string, []>("noise_res_0_adain1_0_fc_weight_palettized"), shape = tensor<uint32, [2]>([512, 128])];
19
+ tensor<fp32, [256]> noise_res_0_adain1_0_norm_bias = const()[name = tensor<string, []>("noise_res_0_adain1_0_norm_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(147136)))];
20
+ tensor<fp32, [256]> noise_res_0_adain1_0_norm_weight = const()[name = tensor<string, []>("noise_res_0_adain1_0_norm_weight"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(148224)))];
21
+ tensor<fp32, [256]> noise_res_0_convs1_0_bias = const()[name = tensor<string, []>("noise_res_0_convs1_0_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(149312)))];
22
+ tensor<fp32, [512]> noise_res_0_adain2_0_fc_bias = const()[name = tensor<string, []>("noise_res_0_adain2_0_fc_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(150400)))];
23
+ tensor<fp32, [512, 128]> noise_res_0_adain2_0_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(152512))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(218112))), name = tensor<string, []>("noise_res_0_adain2_0_fc_weight_palettized"), shape = tensor<uint32, [2]>([512, 128])];
24
+ tensor<fp32, [256]> noise_res_0_convs2_0_bias = const()[name = tensor<string, []>("noise_res_0_convs2_0_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(219200)))];
25
+ tensor<fp32, [512]> noise_res_0_adain1_1_fc_bias = const()[name = tensor<string, []>("noise_res_0_adain1_1_fc_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(220288)))];
26
+ tensor<fp32, [512, 128]> noise_res_0_adain1_1_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(222400))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(288000))), name = tensor<string, []>("noise_res_0_adain1_1_fc_weight_palettized"), shape = tensor<uint32, [2]>([512, 128])];
27
+ tensor<fp32, [256]> noise_res_0_convs1_1_bias = const()[name = tensor<string, []>("noise_res_0_convs1_1_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(289088)))];
28
+ tensor<fp32, [512]> noise_res_0_adain2_1_fc_bias = const()[name = tensor<string, []>("noise_res_0_adain2_1_fc_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(290176)))];
29
+ tensor<fp32, [512, 128]> noise_res_0_adain2_1_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(292288))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(357888))), name = tensor<string, []>("noise_res_0_adain2_1_fc_weight_palettized"), shape = tensor<uint32, [2]>([512, 128])];
30
+ tensor<fp32, [256]> noise_res_0_convs2_1_bias = const()[name = tensor<string, []>("noise_res_0_convs2_1_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(358976)))];
31
+ tensor<fp32, [512]> noise_res_0_adain1_2_fc_bias = const()[name = tensor<string, []>("noise_res_0_adain1_2_fc_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(360064)))];
32
+ tensor<fp32, [512, 128]> noise_res_0_adain1_2_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(362176))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(427776))), name = tensor<string, []>("noise_res_0_adain1_2_fc_weight_palettized"), shape = tensor<uint32, [2]>([512, 128])];
33
+ tensor<fp32, [256]> noise_res_0_convs1_2_bias = const()[name = tensor<string, []>("noise_res_0_convs1_2_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(428864)))];
34
+ tensor<fp32, [512]> noise_res_0_adain2_2_fc_bias = const()[name = tensor<string, []>("noise_res_0_adain2_2_fc_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(429952)))];
35
+ tensor<fp32, [512, 128]> noise_res_0_adain2_2_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(432064))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(497664))), name = tensor<string, []>("noise_res_0_adain2_2_fc_weight_palettized"), shape = tensor<uint32, [2]>([512, 128])];
36
+ tensor<fp32, [256]> noise_res_0_convs2_2_bias = const()[name = tensor<string, []>("noise_res_0_convs2_2_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(498752)))];
37
+ tensor<fp32, [128]> noise_convs_1_bias = const()[name = tensor<string, []>("noise_convs_1_bias"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(499840)))];
38
+ tensor<fp32, [128, 22, 1]> noise_convs_1_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [2816]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(500416))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(503296))), name = tensor<string, []>("noise_convs_1_weight_palettized"), shape = tensor<uint32, [3]>([128, 22, 1])];
39
+ tensor<fp32, [1, 128, 1]> noise_res_1_alpha2_2 = const()[name = tensor<string, []>("noise_res_1_alpha2_2"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(504384)))];
40
+ tensor<fp32, [1, 128, 1]> noise_res_1_alpha1_2 = const()[name = tensor<string, []>("noise_res_1_alpha1_2"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(504960)))];
41
+ tensor<fp32, [1, 128, 1]> noise_res_1_alpha2_1 = const()[name = tensor<string, []>("noise_res_1_alpha2_1"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(505536)))];
42
+ tensor<fp32, [1, 128, 1]> noise_res_1_alpha1_1 = const()[name = tensor<string, []>("noise_res_1_alpha1_1"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(506112)))];
43
+ tensor<fp32, [1, 128, 1]> noise_res_1_alpha2_0 = const()[name = tensor<string, []>("noise_res_1_alpha2_0"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(506688)))];
44
+ tensor<fp32, [1, 128, 1]> noise_res_1_alpha1_0 = const()[name = tensor<string, []>("noise_res_1_alpha1_0"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(507264)))];
45
+ tensor<fp32, [256]> noise_res_1_adain1_0_fc_bias = const()[name = tensor<string, []>("noise_res_1_adain1_0_fc_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(507840)))];
46
+ tensor<fp32, [256, 128]> noise_res_1_adain1_0_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [32768]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(508928))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(541760))), name = tensor<string, []>("noise_res_1_adain1_0_fc_weight_palettized"), shape = tensor<uint32, [2]>([256, 128])];
47
+ tensor<fp32, [128]> noise_res_1_adain1_0_norm_bias = const()[name = tensor<string, []>("noise_res_1_adain1_0_norm_bias"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(542848)))];
48
+ tensor<fp32, [128]> noise_res_1_adain1_0_norm_weight = const()[name = tensor<string, []>("noise_res_1_adain1_0_norm_weight"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(543424)))];
49
+ tensor<fp32, [128]> noise_res_1_convs1_0_bias = const()[name = tensor<string, []>("noise_res_1_convs1_0_bias"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(544000)))];
50
+ tensor<fp32, [256]> noise_res_1_adain2_0_fc_bias = const()[name = tensor<string, []>("noise_res_1_adain2_0_fc_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(544576)))];
51
+ tensor<fp32, [256, 128]> noise_res_1_adain2_0_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [32768]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(545664))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(578496))), name = tensor<string, []>("noise_res_1_adain2_0_fc_weight_palettized"), shape = tensor<uint32, [2]>([256, 128])];
52
+ tensor<fp32, [128]> noise_res_1_convs2_0_bias = const()[name = tensor<string, []>("noise_res_1_convs2_0_bias"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(579584)))];
53
+ tensor<fp32, [256]> noise_res_1_adain1_1_fc_bias = const()[name = tensor<string, []>("noise_res_1_adain1_1_fc_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(580160)))];
54
+ tensor<fp32, [256, 128]> noise_res_1_adain1_1_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [32768]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(581248))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(614080))), name = tensor<string, []>("noise_res_1_adain1_1_fc_weight_palettized"), shape = tensor<uint32, [2]>([256, 128])];
55
+ tensor<fp32, [128]> noise_res_1_convs1_1_bias = const()[name = tensor<string, []>("noise_res_1_convs1_1_bias"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(615168)))];
56
+ tensor<fp32, [256]> noise_res_1_adain2_1_fc_bias = const()[name = tensor<string, []>("noise_res_1_adain2_1_fc_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(615744)))];
57
+ tensor<fp32, [256, 128]> noise_res_1_adain2_1_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [32768]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(616832))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(649664))), name = tensor<string, []>("noise_res_1_adain2_1_fc_weight_palettized"), shape = tensor<uint32, [2]>([256, 128])];
58
+ tensor<fp32, [128]> noise_res_1_convs2_1_bias = const()[name = tensor<string, []>("noise_res_1_convs2_1_bias"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(650752)))];
59
+ tensor<fp32, [256]> noise_res_1_adain1_2_fc_bias = const()[name = tensor<string, []>("noise_res_1_adain1_2_fc_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(651328)))];
60
+ tensor<fp32, [256, 128]> noise_res_1_adain1_2_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [32768]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(652416))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(685248))), name = tensor<string, []>("noise_res_1_adain1_2_fc_weight_palettized"), shape = tensor<uint32, [2]>([256, 128])];
61
+ tensor<fp32, [128]> noise_res_1_convs1_2_bias = const()[name = tensor<string, []>("noise_res_1_convs1_2_bias"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(686336)))];
62
+ tensor<fp32, [256]> noise_res_1_adain2_2_fc_bias = const()[name = tensor<string, []>("noise_res_1_adain2_2_fc_bias"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(686912)))];
63
+ tensor<fp32, [256, 128]> noise_res_1_adain2_2_fc_weight_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [32768]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(688000))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(720832))), name = tensor<string, []>("noise_res_1_adain2_2_fc_weight_palettized"), shape = tensor<uint32, [2]>([256, 128])];
64
+ tensor<fp32, [128]> noise_res_1_convs2_2_bias = const()[name = tensor<string, []>("noise_res_1_convs2_2_bias"), val = tensor<fp32, [128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(721920)))];
65
+ tensor<int32, [1]> input_1_axes_0 = const()[name = tensor<string, []>("input_1_axes_0"), val = tensor<int32, [1]>([1])];
66
+ tensor<fp32, [1, 1, ?]> input_1 = expand_dims(axes = input_1_axes_0, x = F0_curve)[name = tensor<string, []>("input_1")];
67
+ tensor<int32, [1]> expand_dims_0_axes_0 = const()[name = tensor<string, []>("expand_dims_0_axes_0"), val = tensor<int32, [1]>([3])];
68
+ tensor<fp32, [1, 1, ?, 1]> expand_dims_0 = expand_dims(axes = expand_dims_0_axes_0, x = input_1)[name = tensor<string, []>("expand_dims_0")];
69
+ tensor<int32, []> upsample_nearest_neighbor_0_scale_factor_height_0 = const()[name = tensor<string, []>("upsample_nearest_neighbor_0_scale_factor_height_0"), val = tensor<int32, []>(300)];
70
+ tensor<int32, []> upsample_nearest_neighbor_0_scale_factor_width_0 = const()[name = tensor<string, []>("upsample_nearest_neighbor_0_scale_factor_width_0"), val = tensor<int32, []>(1)];
71
+ tensor<fp32, [1, 1, ?, 1]> upsample_nearest_neighbor_0 = upsample_nearest_neighbor(scale_factor_height = upsample_nearest_neighbor_0_scale_factor_height_0, scale_factor_width = upsample_nearest_neighbor_0_scale_factor_width_0, x = expand_dims_0)[name = tensor<string, []>("upsample_nearest_neighbor_0")];
72
+ tensor<int32, [1]> var_26_axes_0 = const()[name = tensor<string, []>("op_26_axes_0"), val = tensor<int32, [1]>([3])];
73
+ tensor<fp32, [1, 1, ?]> var_26 = squeeze(axes = var_26_axes_0, x = upsample_nearest_neighbor_0)[name = tensor<string, []>("op_26")];
74
+ tensor<int32, []> var_30 = const()[name = tensor<string, []>("op_30"), val = tensor<int32, []>(1)];
75
+ tensor<fp32, [1, 9, 1]> const_26 = const()[name = tensor<string, []>("const_26"), val = tensor<fp32, [1, 9, 1]>([[[0x1p+0], [0x1p+1], [0x1.8p+1], [0x1p+2], [0x1.4p+2], [0x1.8p+2], [0x1.cp+2], [0x1p+3], [0x1.2p+3]]])];
76
+ tensor<fp32, [1, 9, ?]> fn = mul(x = var_26, y = const_26)[name = tensor<string, []>("fn")];
77
+ tensor<fp32, []> _inversed_rad_values_y_0 = const()[name = tensor<string, []>("_inversed_rad_values_y_0"), val = tensor<fp32, []>(0x1.5d867cp-15)];
78
+ tensor<fp32, [1, 9, ?]> _inversed_rad_values = mul(x = fn, y = _inversed_rad_values_y_0)[name = tensor<string, []>("_inversed_rad_values")];
79
+ tensor<int32, [1]> var_50 = const()[name = tensor<string, []>("op_50"), val = tensor<int32, [1]>([300])];
80
+ tensor<int32, [1]> var_51 = const()[name = tensor<string, []>("op_51"), val = tensor<int32, [1]>([300])];
81
+ tensor<string, []> rv_down_pad_type_0 = const()[name = tensor<string, []>("rv_down_pad_type_0"), val = tensor<string, []>("custom")];
82
+ tensor<int32, [2]> rv_down_pad_0 = const()[name = tensor<string, []>("rv_down_pad_0"), val = tensor<int32, [2]>([0, 0])];
83
+ tensor<bool, []> rv_down_exclude_padding_from_average_0 = const()[name = tensor<string, []>("rv_down_exclude_padding_from_average_0"), val = tensor<bool, []>(false)];
84
+ tensor<bool, []> rv_down_ceil_mode_0 = const()[name = tensor<string, []>("rv_down_ceil_mode_0"), val = tensor<bool, []>(false)];
85
+ tensor<fp32, [1, 9, ?]> rv_down = avg_pool(ceil_mode = rv_down_ceil_mode_0, exclude_padding_from_average = rv_down_exclude_padding_from_average_0, kernel_sizes = var_50, pad = rv_down_pad_0, pad_type = rv_down_pad_type_0, strides = var_51, x = _inversed_rad_values)[name = tensor<string, []>("rv_down")];
86
+ tensor<int32, [3]> rad_values_down_perm_0 = const()[name = tensor<string, []>("rad_values_down_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
87
+ tensor<bool, []> var_55_exclusive_0 = const()[name = tensor<string, []>("op_55_exclusive_0"), val = tensor<bool, []>(false)];
88
+ tensor<bool, []> var_55_reverse_0 = const()[name = tensor<string, []>("op_55_reverse_0"), val = tensor<bool, []>(false)];
89
+ tensor<fp32, [1, ?, 9]> rad_values_down = transpose(perm = rad_values_down_perm_0, x = rv_down)[name = tensor<string, []>("transpose_4")];
90
+ tensor<fp32, [1, ?, 9]> var_55 = cumsum(axis = var_30, exclusive = var_55_exclusive_0, reverse = var_55_reverse_0, x = rad_values_down)[name = tensor<string, []>("op_55")];
91
+ tensor<fp32, []> var_56 = const()[name = tensor<string, []>("op_56"), val = tensor<fp32, []>(0x1.921fb6p+2)];
92
+ tensor<fp32, [1, ?, 9]> phase_1 = mul(x = var_55, y = var_56)[name = tensor<string, []>("phase_1")];
93
+ tensor<int32, [3]> var_58_perm_0 = const()[name = tensor<string, []>("op_58_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
94
+ tensor<fp32, []> var_59_promoted = const()[name = tensor<string, []>("op_59_promoted"), val = tensor<fp32, []>(0x1.2cp+8)];
95
+ tensor<fp32, [1, 9, ?]> var_58 = transpose(perm = var_58_perm_0, x = phase_1)[name = tensor<string, []>("transpose_3")];
96
+ tensor<fp32, [1, 9, ?]> input_3 = mul(x = var_58, y = var_59_promoted)[name = tensor<string, []>("input_3")];
97
+ tensor<int32, [1]> expand_dims_1_axes_0 = const()[name = tensor<string, []>("expand_dims_1_axes_0"), val = tensor<int32, [1]>([3])];
98
+ tensor<fp32, [1, 9, ?, 1]> expand_dims_1 = expand_dims(axes = expand_dims_1_axes_0, x = input_3)[name = tensor<string, []>("expand_dims_1")];
99
+ tensor<int32, []> upsample_bilinear_0_scale_factor_height_0 = const()[name = tensor<string, []>("upsample_bilinear_0_scale_factor_height_0"), val = tensor<int32, []>(300)];
100
+ tensor<bool, []> upsample_bilinear_0_align_corners_0 = const()[name = tensor<string, []>("upsample_bilinear_0_align_corners_0"), val = tensor<bool, []>(false)];
101
+ tensor<int32, []> upsample_bilinear_0_scale_factor_width_0 = const()[name = tensor<string, []>("upsample_bilinear_0_scale_factor_width_0"), val = tensor<int32, []>(1)];
102
+ tensor<fp32, [1, 9, ?, 1]> upsample_bilinear_0 = upsample_bilinear(align_corners = upsample_bilinear_0_align_corners_0, scale_factor_height = upsample_bilinear_0_scale_factor_height_0, scale_factor_width = upsample_bilinear_0_scale_factor_width_0, x = expand_dims_1)[name = tensor<string, []>("upsample_bilinear_0")];
103
+ tensor<int32, [1]> ph_up_axes_0 = const()[name = tensor<string, []>("ph_up_axes_0"), val = tensor<int32, [1]>([3])];
104
+ tensor<fp32, [1, 9, ?]> ph_up = squeeze(axes = ph_up_axes_0, x = upsample_bilinear_0)[name = tensor<string, []>("ph_up")];
105
+ tensor<int32, [3]> phase_perm_0 = const()[name = tensor<string, []>("phase_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
106
+ tensor<fp32, [1, ?, 9]> phase = transpose(perm = phase_perm_0, x = ph_up)[name = tensor<string, []>("transpose_2")];
107
+ tensor<fp32, [1, ?, 9]> var_64 = sin(x = phase)[name = tensor<string, []>("op_64")];
108
+ tensor<fp32, []> var_65 = const()[name = tensor<string, []>("op_65"), val = tensor<fp32, []>(0x1.99999ap-4)];
109
+ tensor<fp32, [1, ?, 9]> sines = mul(x = var_64, y = var_65)[name = tensor<string, []>("sines")];
110
+ tensor<fp32, []> var_31_promoted = const()[name = tensor<string, []>("op_31_promoted"), val = tensor<fp32, []>(0x1.4p+3)];
111
+ tensor<int32, [3]> transpose_0_perm_0 = const()[name = tensor<string, []>("transpose_0_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
112
+ tensor<fp32, [1, ?, 1]> transpose_0 = transpose(perm = transpose_0_perm_0, x = var_26)[name = tensor<string, []>("transpose_1")];
113
+ tensor<bool, [1, ?, 1]> var_67 = greater(x = transpose_0, y = var_31_promoted)[name = tensor<string, []>("op_67")];
114
+ tensor<string, []> uv_dtype_0 = const()[name = tensor<string, []>("uv_dtype_0"), val = tensor<string, []>("fp32")];
115
+ tensor<fp32, []> var_69 = const()[name = tensor<string, []>("op_69"), val = tensor<fp32, []>(0x1.89374cp-9)];
116
+ tensor<fp32, [1, ?, 1]> uv = cast(dtype = uv_dtype_0, x = var_67)[name = tensor<string, []>("cast_5")];
117
+ tensor<fp32, [1, ?, 1]> var_70 = mul(x = uv, y = var_69)[name = tensor<string, []>("op_70")];
118
+ tensor<fp32, []> var_30_promoted = const()[name = tensor<string, []>("op_30_promoted"), val = tensor<fp32, []>(0x1p+0)];
119
+ tensor<fp32, [1, ?, 1]> var_71 = sub(x = var_30_promoted, y = uv)[name = tensor<string, []>("op_71")];
120
+ tensor<fp32, []> var_72 = const()[name = tensor<string, []>("op_72"), val = tensor<fp32, []>(0x1.99999ap-4)];
121
+ tensor<fp32, [1, ?, 1]> var_73 = mul(x = var_71, y = var_72)[name = tensor<string, []>("op_73")];
122
+ tensor<fp32, []> _inversed_75_y_0 = const()[name = tensor<string, []>("_inversed_75_y_0"), val = tensor<fp32, []>(0x1.555556p-2)];
123
+ tensor<fp32, [1, ?, 1]> _inversed_75 = mul(x = var_73, y = _inversed_75_y_0)[name = tensor<string, []>("_inversed_75")];
124
+ tensor<fp32, [1, ?, 1]> noise_amp = add(x = var_70, y = _inversed_75)[name = tensor<string, []>("noise_amp")];
125
+ tensor<fp32, []> var_77 = const()[name = tensor<string, []>("op_77"), val = tensor<fp32, []>(0x1.47ae14p-7)];
126
+ tensor<fp32, [1, ?, 1]> noise = mul(x = noise_amp, y = var_77)[name = tensor<string, []>("noise")];
127
+ tensor<fp32, [1, ?, 9]> var_79 = mul(x = sines, y = uv)[name = tensor<string, []>("op_79")];
128
+ tensor<fp32, [1, ?, 9]> input_5 = add(x = var_79, y = noise)[name = tensor<string, []>("input_5")];
129
+ tensor<fp32, [1, ?, 1]> input_7 = linear(bias = m_source_l_linear_bias, weight = m_source_l_linear_weight, x = input_5)[name = tensor<string, []>("linear_0")];
130
+ tensor<fp32, [1, ?, 1]> har_source = tanh(x = input_7)[name = tensor<string, []>("har_source")];
131
+ tensor<int32, [3]> var_90_perm_0 = const()[name = tensor<string, []>("op_90_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
132
+ tensor<int32, [1]> input_9_axes_0 = const()[name = tensor<string, []>("input_9_axes_0"), val = tensor<int32, [1]>([1])];
133
+ tensor<fp32, [1, 1, ?]> var_90 = transpose(perm = var_90_perm_0, x = har_source)[name = tensor<string, []>("transpose_0")];
134
+ tensor<fp32, [1, ?]> input_9 = squeeze(axes = input_9_axes_0, x = var_90)[name = tensor<string, []>("input_9")];
135
+ tensor<fp32, []> const_1 = const()[name = tensor<string, []>("const_1"), val = tensor<fp32, []>(0x0p+0)];
136
+ tensor<int32, [4]> waveform_pad_0 = const()[name = tensor<string, []>("waveform_pad_0"), val = tensor<int32, [4]>([0, 0, 10, 10])];
137
+ tensor<string, []> waveform_mode_0 = const()[name = tensor<string, []>("waveform_mode_0"), val = tensor<string, []>("replicate")];
138
+ tensor<fp32, [1, ?]> waveform = pad(constant_val = const_1, mode = waveform_mode_0, pad = waveform_pad_0, x = input_9)[name = tensor<string, []>("waveform")];
139
+ tensor<int32, [1]> input_11_axes_0 = const()[name = tensor<string, []>("input_11_axes_0"), val = tensor<int32, [1]>([1])];
140
+ tensor<fp32, [1, 1, ?]> input_11 = expand_dims(axes = input_11_axes_0, x = waveform)[name = tensor<string, []>("input_11")];
141
+ tensor<string, []> real_out_pad_type_0 = const()[name = tensor<string, []>("real_out_pad_type_0"), val = tensor<string, []>("valid")];
142
+ tensor<int32, [1]> real_out_strides_0 = const()[name = tensor<string, []>("real_out_strides_0"), val = tensor<int32, [1]>([5])];
143
+ tensor<int32, [2]> real_out_pad_0 = const()[name = tensor<string, []>("real_out_pad_0"), val = tensor<int32, [2]>([0, 0])];
144
+ tensor<int32, [1]> real_out_dilations_0 = const()[name = tensor<string, []>("real_out_dilations_0"), val = tensor<int32, [1]>([1])];
145
+ tensor<int32, []> real_out_groups_0 = const()[name = tensor<string, []>("real_out_groups_0"), val = tensor<int32, []>(1)];
146
+ tensor<fp32, [1, 11, ?]> real_out = conv(dilations = real_out_dilations_0, groups = real_out_groups_0, pad = real_out_pad_0, pad_type = real_out_pad_type_0, strides = real_out_strides_0, weight = stft_conv_real_weight, x = input_11)[name = tensor<string, []>("real_out")];
147
+ tensor<string, []> imag_out_pad_type_0 = const()[name = tensor<string, []>("imag_out_pad_type_0"), val = tensor<string, []>("valid")];
148
+ tensor<int32, [1]> imag_out_strides_0 = const()[name = tensor<string, []>("imag_out_strides_0"), val = tensor<int32, [1]>([5])];
149
+ tensor<int32, [2]> imag_out_pad_0 = const()[name = tensor<string, []>("imag_out_pad_0"), val = tensor<int32, [2]>([0, 0])];
150
+ tensor<int32, [1]> imag_out_dilations_0 = const()[name = tensor<string, []>("imag_out_dilations_0"), val = tensor<int32, [1]>([1])];
151
+ tensor<int32, []> imag_out_groups_0 = const()[name = tensor<string, []>("imag_out_groups_0"), val = tensor<int32, []>(1)];
152
+ tensor<fp32, [1, 11, ?]> imag_out = conv(dilations = imag_out_dilations_0, groups = imag_out_groups_0, pad = imag_out_pad_0, pad_type = imag_out_pad_type_0, strides = imag_out_strides_0, weight = stft_conv_imag_weight, x = input_11)[name = tensor<string, []>("imag_out")];
153
+ tensor<fp32, []> var_125_promoted = const()[name = tensor<string, []>("op_125_promoted"), val = tensor<fp32, []>(0x1p+1)];
154
+ tensor<fp32, [1, 11, ?]> var_126 = pow(x = real_out, y = var_125_promoted)[name = tensor<string, []>("op_126")];
155
+ tensor<fp32, []> var_127_promoted = const()[name = tensor<string, []>("op_127_promoted"), val = tensor<fp32, []>(0x1p+1)];
156
+ tensor<fp32, [1, 11, ?]> var_128 = pow(x = imag_out, y = var_127_promoted)[name = tensor<string, []>("op_128")];
157
+ tensor<fp32, [1, 11, ?]> var_130 = add(x = var_126, y = var_128)[name = tensor<string, []>("op_130")];
158
+ tensor<fp32, []> var_132 = const()[name = tensor<string, []>("op_132"), val = tensor<fp32, []>(0x1.6849b8p-47)];
159
+ tensor<fp32, [1, 11, ?]> var_133 = add(x = var_130, y = var_132)[name = tensor<string, []>("op_133")];
160
+ tensor<fp32, [1, 11, ?]> har_spec = sqrt(x = var_133)[name = tensor<string, []>("har_spec")];
161
+ tensor<fp32, []> less_0_y_0 = const()[name = tensor<string, []>("less_0_y_0"), val = tensor<fp32, []>(0x0p+0)];
162
+ tensor<bool, [1, 11, ?]> less_0 = less(x = imag_out, y = less_0_y_0)[name = tensor<string, []>("less_0")];
163
+ tensor<fp32, []> greater_0_y_0 = const()[name = tensor<string, []>("greater_0_y_0"), val = tensor<fp32, []>(0x0p+0)];
164
+ tensor<bool, [1, 11, ?]> greater_0 = greater(x = imag_out, y = greater_0_y_0)[name = tensor<string, []>("greater_0")];
165
+ tensor<fp32, []> less_1_y_0 = const()[name = tensor<string, []>("less_1_y_0"), val = tensor<fp32, []>(0x0p+0)];
166
+ tensor<bool, [1, 11, ?]> less_1 = less(x = real_out, y = less_1_y_0)[name = tensor<string, []>("less_1")];
167
+ tensor<fp32, []> equal_0_y_0 = const()[name = tensor<string, []>("equal_0_y_0"), val = tensor<fp32, []>(0x0p+0)];
168
+ tensor<bool, [1, 11, ?]> equal_0 = equal(x = real_out, y = equal_0_y_0)[name = tensor<string, []>("equal_0")];
169
+ tensor<bool, [1, 11, ?]> logical_and_0 = logical_and(x = greater_0, y = less_1)[name = tensor<string, []>("logical_and_0")];
170
+ tensor<bool, [1, 11, ?]> logical_and_1 = logical_and(x = less_0, y = less_1)[name = tensor<string, []>("logical_and_1")];
171
+ tensor<bool, [1, 11, ?]> logical_and_2 = logical_and(x = greater_0, y = equal_0)[name = tensor<string, []>("logical_and_2")];
172
+ tensor<bool, [1, 11, ?]> logical_and_3 = logical_and(x = less_0, y = equal_0)[name = tensor<string, []>("logical_and_3")];
173
+ tensor<string, []> cast_5_dtype_0 = const()[name = tensor<string, []>("cast_5_dtype_0"), val = tensor<string, []>("fp32")];
174
+ tensor<string, []> cast_6_dtype_0 = const()[name = tensor<string, []>("cast_6_dtype_0"), val = tensor<string, []>("fp32")];
175
+ tensor<string, []> cast_7_dtype_0 = const()[name = tensor<string, []>("cast_7_dtype_0"), val = tensor<string, []>("fp32")];
176
+ tensor<string, []> cast_8_dtype_0 = const()[name = tensor<string, []>("cast_8_dtype_0"), val = tensor<string, []>("fp32")];
177
+ tensor<fp32, []> mul_0_y_0 = const()[name = tensor<string, []>("mul_0_y_0"), val = tensor<fp32, []>(0x1.921fb6p+1)];
178
+ tensor<fp32, [1, 11, ?]> cast_5 = cast(dtype = cast_5_dtype_0, x = logical_and_0)[name = tensor<string, []>("cast_4")];
179
+ tensor<fp32, [1, 11, ?]> mul_0 = mul(x = cast_5, y = mul_0_y_0)[name = tensor<string, []>("mul_0")];
180
+ tensor<fp32, []> mul_1_y_0 = const()[name = tensor<string, []>("mul_1_y_0"), val = tensor<fp32, []>(0x1.921fb6p+1)];
181
+ tensor<fp32, [1, 11, ?]> cast_6 = cast(dtype = cast_6_dtype_0, x = logical_and_1)[name = tensor<string, []>("cast_3")];
182
+ tensor<fp32, [1, 11, ?]> mul_1 = mul(x = cast_6, y = mul_1_y_0)[name = tensor<string, []>("mul_1")];
183
+ tensor<fp32, []> sub_0_x_0 = const()[name = tensor<string, []>("sub_0_x_0"), val = tensor<fp32, []>(0x1p+0)];
184
+ tensor<fp32, [1, 11, ?]> cast_7 = cast(dtype = cast_7_dtype_0, x = logical_and_2)[name = tensor<string, []>("cast_2")];
185
+ tensor<fp32, [1, 11, ?]> sub_0 = sub(x = sub_0_x_0, y = cast_7)[name = tensor<string, []>("sub_0")];
186
+ tensor<fp32, []> mul_2_y_0 = const()[name = tensor<string, []>("mul_2_y_0"), val = tensor<fp32, []>(0x1.921fb6p+0)];
187
+ tensor<fp32, [1, 11, ?]> mul_2 = mul(x = cast_7, y = mul_2_y_0)[name = tensor<string, []>("mul_2")];
188
+ tensor<fp32, []> sub_1_x_0 = const()[name = tensor<string, []>("sub_1_x_0"), val = tensor<fp32, []>(0x1p+0)];
189
+ tensor<fp32, [1, 11, ?]> cast_8 = cast(dtype = cast_8_dtype_0, x = logical_and_3)[name = tensor<string, []>("cast_1")];
190
+ tensor<fp32, [1, 11, ?]> sub_1 = sub(x = sub_1_x_0, y = cast_8)[name = tensor<string, []>("sub_1")];
191
+ tensor<fp32, []> mul_3_y_0 = const()[name = tensor<string, []>("mul_3_y_0"), val = tensor<fp32, []>(-0x1.921fb6p+0)];
192
+ tensor<fp32, [1, 11, ?]> mul_3 = mul(x = cast_8, y = mul_3_y_0)[name = tensor<string, []>("mul_3")];
193
+ tensor<fp32, []> greater_1_y_0 = const()[name = tensor<string, []>("greater_1_y_0"), val = tensor<fp32, []>(-0x1.5798eep-27)];
194
+ tensor<bool, [1, 11, ?]> greater_1 = greater(x = real_out, y = greater_1_y_0)[name = tensor<string, []>("greater_1")];
195
+ tensor<fp32, []> less_2_y_0 = const()[name = tensor<string, []>("less_2_y_0"), val = tensor<fp32, []>(0x1.5798eep-27)];
196
+ tensor<bool, [1, 11, ?]> less_2 = less(x = real_out, y = less_2_y_0)[name = tensor<string, []>("less_2")];
197
+ tensor<bool, [1, 11, ?]> logical_and_4 = logical_and(x = greater_1, y = less_2)[name = tensor<string, []>("logical_and_4")];
198
+ tensor<string, []> cast_9_dtype_0 = const()[name = tensor<string, []>("cast_9_dtype_0"), val = tensor<string, []>("fp32")];
199
+ tensor<fp32, []> mul_4_y_0 = const()[name = tensor<string, []>("mul_4_y_0"), val = tensor<fp32, []>(0x1.5798eep-26)];
200
+ tensor<fp32, [1, 11, ?]> cast_9 = cast(dtype = cast_9_dtype_0, x = logical_and_4)[name = tensor<string, []>("cast_0")];
201
+ tensor<fp32, [1, 11, ?]> mul_4 = mul(x = cast_9, y = mul_4_y_0)[name = tensor<string, []>("mul_4")];
202
+ tensor<fp32, [1, 11, ?]> add_0 = add(x = real_out, y = mul_4)[name = tensor<string, []>("add_0")];
203
+ tensor<fp32, [1, 11, ?]> real_div_0 = real_div(x = imag_out, y = add_0)[name = tensor<string, []>("real_div_0")];
204
+ tensor<fp32, [1, 11, ?]> atan_0 = atan(x = real_div_0)[name = tensor<string, []>("atan_0")];
205
+ tensor<fp32, [1, 11, ?]> add_1 = add(x = atan_0, y = mul_0)[name = tensor<string, []>("add_1")];
206
+ tensor<fp32, [1, 11, ?]> sub_2 = sub(x = add_1, y = mul_1)[name = tensor<string, []>("sub_2")];
207
+ tensor<fp32, [1, 11, ?]> mul_5 = mul(x = sub_2, y = sub_0)[name = tensor<string, []>("mul_5")];
208
+ tensor<fp32, [1, 11, ?]> add_2 = add(x = mul_5, y = mul_2)[name = tensor<string, []>("add_2")];
209
+ tensor<fp32, [1, 11, ?]> mul_6 = mul(x = add_2, y = sub_1)[name = tensor<string, []>("mul_6")];
210
+ tensor<fp32, [1, 11, ?]> har_phase = add(x = mul_6, y = mul_3)[name = tensor<string, []>("har_phase")];
211
+ tensor<int32, []> var_137 = const()[name = tensor<string, []>("op_137"), val = tensor<int32, []>(1)];
212
+ tensor<bool, []> input_13_interleave_0 = const()[name = tensor<string, []>("input_13_interleave_0"), val = tensor<bool, []>(false)];
213
+ tensor<fp32, [1, 22, ?]> input_13 = concat(axis = var_137, interleave = input_13_interleave_0, values = (har_spec, har_phase))[name = tensor<string, []>("input_13")];
214
+ tensor<string, []> input_15_pad_type_0 = const()[name = tensor<string, []>("input_15_pad_type_0"), val = tensor<string, []>("custom")];
215
+ tensor<int32, [2]> input_15_pad_0 = const()[name = tensor<string, []>("input_15_pad_0"), val = tensor<int32, [2]>([3, 3])];
216
+ tensor<int32, [1]> input_15_strides_0 = const()[name = tensor<string, []>("input_15_strides_0"), val = tensor<int32, [1]>([6])];
217
+ tensor<int32, [1]> input_15_dilations_0 = const()[name = tensor<string, []>("input_15_dilations_0"), val = tensor<int32, [1]>([1])];
218
+ tensor<int32, []> input_15_groups_0 = const()[name = tensor<string, []>("input_15_groups_0"), val = tensor<int32, []>(1)];
219
+ tensor<fp32, [1, 256, ?]> input_15 = conv(bias = noise_convs_0_bias, dilations = input_15_dilations_0, groups = input_15_groups_0, pad = input_15_pad_0, pad_type = input_15_pad_type_0, strides = input_15_strides_0, weight = noise_convs_0_weight_palettized, x = input_13)[name = tensor<string, []>("input_15")];
220
+ tensor<fp32, []> var_160 = const()[name = tensor<string, []>("op_160"), val = tensor<fp32, []>(0x1.4f8b58p-17)];
221
+ tensor<fp32, [1, 512]> h_1 = linear(bias = noise_res_0_adain1_0_fc_bias, weight = noise_res_0_adain1_0_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_1")];
222
+ tensor<int32, [3]> var_243 = const()[name = tensor<string, []>("op_243"), val = tensor<int32, [3]>([1, 512, 1])];
223
+ tensor<fp32, [1, 512, 1]> h_3 = reshape(shape = var_243, x = h_1)[name = tensor<string, []>("h_3")];
224
+ tensor<int32, [2]> var_245_split_sizes_0 = const()[name = tensor<string, []>("op_245_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
225
+ tensor<int32, []> var_245_axis_0 = const()[name = tensor<string, []>("op_245_axis_0"), val = tensor<int32, []>(1)];
226
+ tensor<fp32, [1, 256, 1]> var_245_0, tensor<fp32, [1, 256, 1]> var_245_1 = split(axis = var_245_axis_0, split_sizes = var_245_split_sizes_0, x = h_3)[name = tensor<string, []>("op_245")];
227
+ tensor<fp32, []> var_247_promoted = const()[name = tensor<string, []>("op_247_promoted"), val = tensor<fp32, []>(0x1p+0)];
228
+ tensor<fp32, [1, 256, 1]> var_248 = add(x = var_245_0, y = var_247_promoted)[name = tensor<string, []>("op_248")];
229
+ tensor<fp32, [1, 256, ?]> var_251 = instance_norm(beta = noise_res_0_adain1_0_norm_bias, epsilon = var_160, gamma = noise_res_0_adain1_0_norm_weight, x = input_15)[name = tensor<string, []>("op_251")];
230
+ tensor<fp32, [1, 256, ?]> var_252 = mul(x = var_248, y = var_251)[name = tensor<string, []>("op_252")];
231
+ tensor<fp32, [1, 256, ?]> xt_1 = add(x = var_252, y = var_245_1)[name = tensor<string, []>("xt_1")];
232
+ tensor<fp32, [1, 256, 1]> var_254 = const()[name = tensor<string, []>("op_254"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(722496)))];
233
+ tensor<fp32, [1, 256, ?]> var_257 = mul(x = noise_res_0_alpha1_0, y = xt_1)[name = tensor<string, []>("op_257")];
234
+ tensor<fp32, [1, 256, ?]> var_258 = sin(x = var_257)[name = tensor<string, []>("op_258")];
235
+ tensor<fp32, []> var_159_promoted = const()[name = tensor<string, []>("op_159_promoted"), val = tensor<fp32, []>(0x1p+1)];
236
+ tensor<fp32, [1, 256, ?]> var_259 = pow(x = var_258, y = var_159_promoted)[name = tensor<string, []>("op_259")];
237
+ tensor<fp32, [1, 256, ?]> var_260 = mul(x = var_254, y = var_259)[name = tensor<string, []>("op_260")];
238
+ tensor<fp32, [1, 256, ?]> input_17 = add(x = xt_1, y = var_260)[name = tensor<string, []>("input_17")];
239
+ tensor<fp32, [256, 256, 7]> weight_9_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [458752]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(723584))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1182400))), name = tensor<string, []>("weight_9_palettized"), shape = tensor<uint32, [3]>([256, 256, 7])];
240
+ tensor<string, []> input_19_pad_type_0 = const()[name = tensor<string, []>("input_19_pad_type_0"), val = tensor<string, []>("custom")];
241
+ tensor<int32, [2]> input_19_pad_0 = const()[name = tensor<string, []>("input_19_pad_0"), val = tensor<int32, [2]>([3, 3])];
242
+ tensor<int32, [1]> input_19_strides_0 = const()[name = tensor<string, []>("input_19_strides_0"), val = tensor<int32, [1]>([1])];
243
+ tensor<int32, [1]> input_19_dilations_0 = const()[name = tensor<string, []>("input_19_dilations_0"), val = tensor<int32, [1]>([1])];
244
+ tensor<int32, []> input_19_groups_0 = const()[name = tensor<string, []>("input_19_groups_0"), val = tensor<int32, []>(1)];
245
+ tensor<fp32, [1, 256, ?]> input_19 = conv(bias = noise_res_0_convs1_0_bias, dilations = input_19_dilations_0, groups = input_19_groups_0, pad = input_19_pad_0, pad_type = input_19_pad_type_0, strides = input_19_strides_0, weight = weight_9_palettized, x = input_17)[name = tensor<string, []>("input_19")];
246
+ tensor<fp32, [1, 512]> h_5 = linear(bias = noise_res_0_adain2_0_fc_bias, weight = noise_res_0_adain2_0_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_2")];
247
+ tensor<int32, [3]> var_276 = const()[name = tensor<string, []>("op_276"), val = tensor<int32, [3]>([1, 512, 1])];
248
+ tensor<fp32, [1, 512, 1]> h_7 = reshape(shape = var_276, x = h_5)[name = tensor<string, []>("h_7")];
249
+ tensor<int32, [2]> var_278_split_sizes_0 = const()[name = tensor<string, []>("op_278_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
250
+ tensor<int32, []> var_278_axis_0 = const()[name = tensor<string, []>("op_278_axis_0"), val = tensor<int32, []>(1)];
251
+ tensor<fp32, [1, 256, 1]> var_278_0, tensor<fp32, [1, 256, 1]> var_278_1 = split(axis = var_278_axis_0, split_sizes = var_278_split_sizes_0, x = h_7)[name = tensor<string, []>("op_278")];
252
+ tensor<fp32, []> var_280_promoted = const()[name = tensor<string, []>("op_280_promoted"), val = tensor<fp32, []>(0x1p+0)];
253
+ tensor<fp32, [1, 256, 1]> var_281 = add(x = var_278_0, y = var_280_promoted)[name = tensor<string, []>("op_281")];
254
+ tensor<fp32, [1, 256, ?]> var_284 = instance_norm(beta = noise_res_0_adain1_0_norm_bias, epsilon = var_160, gamma = noise_res_0_adain1_0_norm_weight, x = input_19)[name = tensor<string, []>("op_284")];
255
+ tensor<fp32, [1, 256, ?]> var_285 = mul(x = var_281, y = var_284)[name = tensor<string, []>("op_285")];
256
+ tensor<fp32, [1, 256, ?]> xt_3 = add(x = var_285, y = var_278_1)[name = tensor<string, []>("xt_3")];
257
+ tensor<fp32, [1, 256, 1]> var_287 = const()[name = tensor<string, []>("op_287"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1183488)))];
258
+ tensor<fp32, [1, 256, ?]> var_290 = mul(x = noise_res_0_alpha2_0, y = xt_3)[name = tensor<string, []>("op_290")];
259
+ tensor<fp32, [1, 256, ?]> var_291 = sin(x = var_290)[name = tensor<string, []>("op_291")];
260
+ tensor<fp32, []> var_159_promoted_1 = const()[name = tensor<string, []>("op_159_promoted_1"), val = tensor<fp32, []>(0x1p+1)];
261
+ tensor<fp32, [1, 256, ?]> var_292 = pow(x = var_291, y = var_159_promoted_1)[name = tensor<string, []>("op_292")];
262
+ tensor<fp32, [1, 256, ?]> var_293 = mul(x = var_287, y = var_292)[name = tensor<string, []>("op_293")];
263
+ tensor<fp32, [1, 256, ?]> input_21 = add(x = xt_3, y = var_293)[name = tensor<string, []>("input_21")];
264
+ tensor<fp32, [256, 256, 7]> weight_13_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [458752]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1184576))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1643392))), name = tensor<string, []>("weight_13_palettized"), shape = tensor<uint32, [3]>([256, 256, 7])];
265
+ tensor<string, []> xt_5_pad_type_0 = const()[name = tensor<string, []>("xt_5_pad_type_0"), val = tensor<string, []>("custom")];
266
+ tensor<int32, [2]> xt_5_pad_0 = const()[name = tensor<string, []>("xt_5_pad_0"), val = tensor<int32, [2]>([3, 3])];
267
+ tensor<int32, [1]> xt_5_strides_0 = const()[name = tensor<string, []>("xt_5_strides_0"), val = tensor<int32, [1]>([1])];
268
+ tensor<int32, [1]> xt_5_dilations_0 = const()[name = tensor<string, []>("xt_5_dilations_0"), val = tensor<int32, [1]>([1])];
269
+ tensor<int32, []> xt_5_groups_0 = const()[name = tensor<string, []>("xt_5_groups_0"), val = tensor<int32, []>(1)];
270
+ tensor<fp32, [1, 256, ?]> xt_5 = conv(bias = noise_res_0_convs2_0_bias, dilations = xt_5_dilations_0, groups = xt_5_groups_0, pad = xt_5_pad_0, pad_type = xt_5_pad_type_0, strides = xt_5_strides_0, weight = weight_13_palettized, x = input_21)[name = tensor<string, []>("xt_5")];
271
+ tensor<fp32, [1, 256, ?]> input_23 = add(x = xt_5, y = input_15)[name = tensor<string, []>("input_23")];
272
+ tensor<fp32, [1, 512]> h_9 = linear(bias = noise_res_0_adain1_1_fc_bias, weight = noise_res_0_adain1_1_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_3")];
273
+ tensor<int32, [3]> var_310 = const()[name = tensor<string, []>("op_310"), val = tensor<int32, [3]>([1, 512, 1])];
274
+ tensor<fp32, [1, 512, 1]> h_11 = reshape(shape = var_310, x = h_9)[name = tensor<string, []>("h_11")];
275
+ tensor<int32, [2]> var_312_split_sizes_0 = const()[name = tensor<string, []>("op_312_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
276
+ tensor<int32, []> var_312_axis_0 = const()[name = tensor<string, []>("op_312_axis_0"), val = tensor<int32, []>(1)];
277
+ tensor<fp32, [1, 256, 1]> var_312_0, tensor<fp32, [1, 256, 1]> var_312_1 = split(axis = var_312_axis_0, split_sizes = var_312_split_sizes_0, x = h_11)[name = tensor<string, []>("op_312")];
278
+ tensor<fp32, []> var_314_promoted = const()[name = tensor<string, []>("op_314_promoted"), val = tensor<fp32, []>(0x1p+0)];
279
+ tensor<fp32, [1, 256, 1]> var_315 = add(x = var_312_0, y = var_314_promoted)[name = tensor<string, []>("op_315")];
280
+ tensor<fp32, [1, 256, ?]> var_318 = instance_norm(beta = noise_res_0_adain1_0_norm_bias, epsilon = var_160, gamma = noise_res_0_adain1_0_norm_weight, x = input_23)[name = tensor<string, []>("op_318")];
281
+ tensor<fp32, [1, 256, ?]> var_319 = mul(x = var_315, y = var_318)[name = tensor<string, []>("op_319")];
282
+ tensor<fp32, [1, 256, ?]> xt_7 = add(x = var_319, y = var_312_1)[name = tensor<string, []>("xt_7")];
283
+ tensor<fp32, [1, 256, 1]> var_321 = const()[name = tensor<string, []>("op_321"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1644480)))];
284
+ tensor<fp32, [1, 256, ?]> var_324 = mul(x = noise_res_0_alpha1_1, y = xt_7)[name = tensor<string, []>("op_324")];
285
+ tensor<fp32, [1, 256, ?]> var_325 = sin(x = var_324)[name = tensor<string, []>("op_325")];
286
+ tensor<fp32, []> var_159_promoted_2 = const()[name = tensor<string, []>("op_159_promoted_2"), val = tensor<fp32, []>(0x1p+1)];
287
+ tensor<fp32, [1, 256, ?]> var_326 = pow(x = var_325, y = var_159_promoted_2)[name = tensor<string, []>("op_326")];
288
+ tensor<fp32, [1, 256, ?]> var_327 = mul(x = var_321, y = var_326)[name = tensor<string, []>("op_327")];
289
+ tensor<fp32, [1, 256, ?]> input_25 = add(x = xt_7, y = var_327)[name = tensor<string, []>("input_25")];
290
+ tensor<fp32, [256, 256, 7]> weight_17_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [458752]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1645568))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2104384))), name = tensor<string, []>("weight_17_palettized"), shape = tensor<uint32, [3]>([256, 256, 7])];
291
+ tensor<string, []> input_27_pad_type_0 = const()[name = tensor<string, []>("input_27_pad_type_0"), val = tensor<string, []>("custom")];
292
+ tensor<int32, [2]> input_27_pad_0 = const()[name = tensor<string, []>("input_27_pad_0"), val = tensor<int32, [2]>([9, 9])];
293
+ tensor<int32, [1]> input_27_dilations_0 = const()[name = tensor<string, []>("input_27_dilations_0"), val = tensor<int32, [1]>([3])];
294
+ tensor<int32, [1]> input_27_strides_0 = const()[name = tensor<string, []>("input_27_strides_0"), val = tensor<int32, [1]>([1])];
295
+ tensor<int32, []> input_27_groups_0 = const()[name = tensor<string, []>("input_27_groups_0"), val = tensor<int32, []>(1)];
296
+ tensor<fp32, [1, 256, ?]> input_27 = conv(bias = noise_res_0_convs1_1_bias, dilations = input_27_dilations_0, groups = input_27_groups_0, pad = input_27_pad_0, pad_type = input_27_pad_type_0, strides = input_27_strides_0, weight = weight_17_palettized, x = input_25)[name = tensor<string, []>("input_27")];
297
+ tensor<fp32, [1, 512]> h_13 = linear(bias = noise_res_0_adain2_1_fc_bias, weight = noise_res_0_adain2_1_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_4")];
298
+ tensor<int32, [3]> var_343 = const()[name = tensor<string, []>("op_343"), val = tensor<int32, [3]>([1, 512, 1])];
299
+ tensor<fp32, [1, 512, 1]> h_15 = reshape(shape = var_343, x = h_13)[name = tensor<string, []>("h_15")];
300
+ tensor<int32, [2]> var_345_split_sizes_0 = const()[name = tensor<string, []>("op_345_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
301
+ tensor<int32, []> var_345_axis_0 = const()[name = tensor<string, []>("op_345_axis_0"), val = tensor<int32, []>(1)];
302
+ tensor<fp32, [1, 256, 1]> var_345_0, tensor<fp32, [1, 256, 1]> var_345_1 = split(axis = var_345_axis_0, split_sizes = var_345_split_sizes_0, x = h_15)[name = tensor<string, []>("op_345")];
303
+ tensor<fp32, []> var_347_promoted = const()[name = tensor<string, []>("op_347_promoted"), val = tensor<fp32, []>(0x1p+0)];
304
+ tensor<fp32, [1, 256, 1]> var_348 = add(x = var_345_0, y = var_347_promoted)[name = tensor<string, []>("op_348")];
305
+ tensor<fp32, [1, 256, ?]> var_351 = instance_norm(beta = noise_res_0_adain1_0_norm_bias, epsilon = var_160, gamma = noise_res_0_adain1_0_norm_weight, x = input_27)[name = tensor<string, []>("op_351")];
306
+ tensor<fp32, [1, 256, ?]> var_352 = mul(x = var_348, y = var_351)[name = tensor<string, []>("op_352")];
307
+ tensor<fp32, [1, 256, ?]> xt_9 = add(x = var_352, y = var_345_1)[name = tensor<string, []>("xt_9")];
308
+ tensor<fp32, [1, 256, 1]> var_354 = const()[name = tensor<string, []>("op_354"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2105472)))];
309
+ tensor<fp32, [1, 256, ?]> var_357 = mul(x = noise_res_0_alpha2_1, y = xt_9)[name = tensor<string, []>("op_357")];
310
+ tensor<fp32, [1, 256, ?]> var_358 = sin(x = var_357)[name = tensor<string, []>("op_358")];
311
+ tensor<fp32, []> var_159_promoted_3 = const()[name = tensor<string, []>("op_159_promoted_3"), val = tensor<fp32, []>(0x1p+1)];
312
+ tensor<fp32, [1, 256, ?]> var_359 = pow(x = var_358, y = var_159_promoted_3)[name = tensor<string, []>("op_359")];
313
+ tensor<fp32, [1, 256, ?]> var_360 = mul(x = var_354, y = var_359)[name = tensor<string, []>("op_360")];
314
+ tensor<fp32, [1, 256, ?]> input_29 = add(x = xt_9, y = var_360)[name = tensor<string, []>("input_29")];
315
+ tensor<fp32, [256, 256, 7]> weight_21_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [458752]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2106560))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2565376))), name = tensor<string, []>("weight_21_palettized"), shape = tensor<uint32, [3]>([256, 256, 7])];
316
+ tensor<string, []> xt_11_pad_type_0 = const()[name = tensor<string, []>("xt_11_pad_type_0"), val = tensor<string, []>("custom")];
317
+ tensor<int32, [2]> xt_11_pad_0 = const()[name = tensor<string, []>("xt_11_pad_0"), val = tensor<int32, [2]>([3, 3])];
318
+ tensor<int32, [1]> xt_11_strides_0 = const()[name = tensor<string, []>("xt_11_strides_0"), val = tensor<int32, [1]>([1])];
319
+ tensor<int32, [1]> xt_11_dilations_0 = const()[name = tensor<string, []>("xt_11_dilations_0"), val = tensor<int32, [1]>([1])];
320
+ tensor<int32, []> xt_11_groups_0 = const()[name = tensor<string, []>("xt_11_groups_0"), val = tensor<int32, []>(1)];
321
+ tensor<fp32, [1, 256, ?]> xt_11 = conv(bias = noise_res_0_convs2_1_bias, dilations = xt_11_dilations_0, groups = xt_11_groups_0, pad = xt_11_pad_0, pad_type = xt_11_pad_type_0, strides = xt_11_strides_0, weight = weight_21_palettized, x = input_29)[name = tensor<string, []>("xt_11")];
322
+ tensor<fp32, [1, 256, ?]> input_31 = add(x = xt_11, y = input_23)[name = tensor<string, []>("input_31")];
323
+ tensor<fp32, [1, 512]> h_17 = linear(bias = noise_res_0_adain1_2_fc_bias, weight = noise_res_0_adain1_2_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_5")];
324
+ tensor<int32, [3]> var_377 = const()[name = tensor<string, []>("op_377"), val = tensor<int32, [3]>([1, 512, 1])];
325
+ tensor<fp32, [1, 512, 1]> h_19 = reshape(shape = var_377, x = h_17)[name = tensor<string, []>("h_19")];
326
+ tensor<int32, [2]> var_379_split_sizes_0 = const()[name = tensor<string, []>("op_379_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
327
+ tensor<int32, []> var_379_axis_0 = const()[name = tensor<string, []>("op_379_axis_0"), val = tensor<int32, []>(1)];
328
+ tensor<fp32, [1, 256, 1]> var_379_0, tensor<fp32, [1, 256, 1]> var_379_1 = split(axis = var_379_axis_0, split_sizes = var_379_split_sizes_0, x = h_19)[name = tensor<string, []>("op_379")];
329
+ tensor<fp32, []> var_381_promoted = const()[name = tensor<string, []>("op_381_promoted"), val = tensor<fp32, []>(0x1p+0)];
330
+ tensor<fp32, [1, 256, 1]> var_382 = add(x = var_379_0, y = var_381_promoted)[name = tensor<string, []>("op_382")];
331
+ tensor<fp32, [1, 256, ?]> var_385 = instance_norm(beta = noise_res_0_adain1_0_norm_bias, epsilon = var_160, gamma = noise_res_0_adain1_0_norm_weight, x = input_31)[name = tensor<string, []>("op_385")];
332
+ tensor<fp32, [1, 256, ?]> var_386 = mul(x = var_382, y = var_385)[name = tensor<string, []>("op_386")];
333
+ tensor<fp32, [1, 256, ?]> xt_13 = add(x = var_386, y = var_379_1)[name = tensor<string, []>("xt_13")];
334
+ tensor<fp32, [1, 256, 1]> var_388 = const()[name = tensor<string, []>("op_388"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2566464)))];
335
+ tensor<fp32, [1, 256, ?]> var_391 = mul(x = noise_res_0_alpha1_2, y = xt_13)[name = tensor<string, []>("op_391")];
336
+ tensor<fp32, [1, 256, ?]> var_392 = sin(x = var_391)[name = tensor<string, []>("op_392")];
337
+ tensor<fp32, []> var_159_promoted_4 = const()[name = tensor<string, []>("op_159_promoted_4"), val = tensor<fp32, []>(0x1p+1)];
338
+ tensor<fp32, [1, 256, ?]> var_393 = pow(x = var_392, y = var_159_promoted_4)[name = tensor<string, []>("op_393")];
339
+ tensor<fp32, [1, 256, ?]> var_394 = mul(x = var_388, y = var_393)[name = tensor<string, []>("op_394")];
340
+ tensor<fp32, [1, 256, ?]> input_33 = add(x = xt_13, y = var_394)[name = tensor<string, []>("input_33")];
341
+ tensor<fp32, [256, 256, 7]> weight_25_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [458752]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2567552))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3026368))), name = tensor<string, []>("weight_25_palettized"), shape = tensor<uint32, [3]>([256, 256, 7])];
342
+ tensor<string, []> input_35_pad_type_0 = const()[name = tensor<string, []>("input_35_pad_type_0"), val = tensor<string, []>("custom")];
343
+ tensor<int32, [2]> input_35_pad_0 = const()[name = tensor<string, []>("input_35_pad_0"), val = tensor<int32, [2]>([15, 15])];
344
+ tensor<int32, [1]> input_35_dilations_0 = const()[name = tensor<string, []>("input_35_dilations_0"), val = tensor<int32, [1]>([5])];
345
+ tensor<int32, [1]> input_35_strides_0 = const()[name = tensor<string, []>("input_35_strides_0"), val = tensor<int32, [1]>([1])];
346
+ tensor<int32, []> input_35_groups_0 = const()[name = tensor<string, []>("input_35_groups_0"), val = tensor<int32, []>(1)];
347
+ tensor<fp32, [1, 256, ?]> input_35 = conv(bias = noise_res_0_convs1_2_bias, dilations = input_35_dilations_0, groups = input_35_groups_0, pad = input_35_pad_0, pad_type = input_35_pad_type_0, strides = input_35_strides_0, weight = weight_25_palettized, x = input_33)[name = tensor<string, []>("input_35")];
348
+ tensor<fp32, [1, 512]> h_21 = linear(bias = noise_res_0_adain2_2_fc_bias, weight = noise_res_0_adain2_2_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_6")];
349
+ tensor<int32, [3]> var_410 = const()[name = tensor<string, []>("op_410"), val = tensor<int32, [3]>([1, 512, 1])];
350
+ tensor<fp32, [1, 512, 1]> h_23 = reshape(shape = var_410, x = h_21)[name = tensor<string, []>("h_23")];
351
+ tensor<int32, [2]> var_412_split_sizes_0 = const()[name = tensor<string, []>("op_412_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
352
+ tensor<int32, []> var_412_axis_0 = const()[name = tensor<string, []>("op_412_axis_0"), val = tensor<int32, []>(1)];
353
+ tensor<fp32, [1, 256, 1]> var_412_0, tensor<fp32, [1, 256, 1]> var_412_1 = split(axis = var_412_axis_0, split_sizes = var_412_split_sizes_0, x = h_23)[name = tensor<string, []>("op_412")];
354
+ tensor<fp32, []> var_414_promoted = const()[name = tensor<string, []>("op_414_promoted"), val = tensor<fp32, []>(0x1p+0)];
355
+ tensor<fp32, [1, 256, 1]> var_415 = add(x = var_412_0, y = var_414_promoted)[name = tensor<string, []>("op_415")];
356
+ tensor<fp32, [1, 256, ?]> var_418 = instance_norm(beta = noise_res_0_adain1_0_norm_bias, epsilon = var_160, gamma = noise_res_0_adain1_0_norm_weight, x = input_35)[name = tensor<string, []>("op_418")];
357
+ tensor<fp32, [1, 256, ?]> var_419 = mul(x = var_415, y = var_418)[name = tensor<string, []>("op_419")];
358
+ tensor<fp32, [1, 256, ?]> xt_15 = add(x = var_419, y = var_412_1)[name = tensor<string, []>("xt_15")];
359
+ tensor<fp32, [1, 256, 1]> var_421 = const()[name = tensor<string, []>("op_421"), val = tensor<fp32, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3027456)))];
360
+ tensor<fp32, [1, 256, ?]> var_424 = mul(x = noise_res_0_alpha2_2, y = xt_15)[name = tensor<string, []>("op_424")];
361
+ tensor<fp32, [1, 256, ?]> var_425 = sin(x = var_424)[name = tensor<string, []>("op_425")];
362
+ tensor<fp32, []> var_159_promoted_5 = const()[name = tensor<string, []>("op_159_promoted_5"), val = tensor<fp32, []>(0x1p+1)];
363
+ tensor<fp32, [1, 256, ?]> var_426 = pow(x = var_425, y = var_159_promoted_5)[name = tensor<string, []>("op_426")];
364
+ tensor<fp32, [1, 256, ?]> var_427 = mul(x = var_421, y = var_426)[name = tensor<string, []>("op_427")];
365
+ tensor<fp32, [1, 256, ?]> input_37 = add(x = xt_15, y = var_427)[name = tensor<string, []>("input_37")];
366
+ tensor<fp32, [256, 256, 7]> weight_29_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [458752]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3028544))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3487360))), name = tensor<string, []>("weight_29_palettized"), shape = tensor<uint32, [3]>([256, 256, 7])];
367
+ tensor<string, []> xt_17_pad_type_0 = const()[name = tensor<string, []>("xt_17_pad_type_0"), val = tensor<string, []>("custom")];
368
+ tensor<int32, [2]> xt_17_pad_0 = const()[name = tensor<string, []>("xt_17_pad_0"), val = tensor<int32, [2]>([3, 3])];
369
+ tensor<int32, [1]> xt_17_strides_0 = const()[name = tensor<string, []>("xt_17_strides_0"), val = tensor<int32, [1]>([1])];
370
+ tensor<int32, [1]> xt_17_dilations_0 = const()[name = tensor<string, []>("xt_17_dilations_0"), val = tensor<int32, [1]>([1])];
371
+ tensor<int32, []> xt_17_groups_0 = const()[name = tensor<string, []>("xt_17_groups_0"), val = tensor<int32, []>(1)];
372
+ tensor<fp32, [1, 256, ?]> xt_17 = conv(bias = noise_res_0_convs2_2_bias, dilations = xt_17_dilations_0, groups = xt_17_groups_0, pad = xt_17_pad_0, pad_type = xt_17_pad_type_0, strides = xt_17_strides_0, weight = weight_29_palettized, x = input_37)[name = tensor<string, []>("xt_17")];
373
+ tensor<fp32, [1, 256, ?]> x_source_0 = add(x = xt_17, y = input_31)[name = tensor<string, []>("op_436")];
374
+ tensor<string, []> input_39_pad_type_0 = const()[name = tensor<string, []>("input_39_pad_type_0"), val = tensor<string, []>("valid")];
375
+ tensor<int32, [1]> input_39_strides_0 = const()[name = tensor<string, []>("input_39_strides_0"), val = tensor<int32, [1]>([1])];
376
+ tensor<int32, [2]> input_39_pad_0 = const()[name = tensor<string, []>("input_39_pad_0"), val = tensor<int32, [2]>([0, 0])];
377
+ tensor<int32, [1]> input_39_dilations_0 = const()[name = tensor<string, []>("input_39_dilations_0"), val = tensor<int32, [1]>([1])];
378
+ tensor<int32, []> input_39_groups_0 = const()[name = tensor<string, []>("input_39_groups_0"), val = tensor<int32, []>(1)];
379
+ tensor<fp32, [1, 128, ?]> input_39 = conv(bias = noise_convs_1_bias, dilations = input_39_dilations_0, groups = input_39_groups_0, pad = input_39_pad_0, pad_type = input_39_pad_type_0, strides = input_39_strides_0, weight = noise_convs_1_weight_palettized, x = input_13)[name = tensor<string, []>("input_39")];
380
+ tensor<fp32, []> var_456 = const()[name = tensor<string, []>("op_456"), val = tensor<fp32, []>(0x1.4f8b58p-17)];
381
+ tensor<fp32, [1, 256]> h_25 = linear(bias = noise_res_1_adain1_0_fc_bias, weight = noise_res_1_adain1_0_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_7")];
382
+ tensor<int32, [3]> var_539 = const()[name = tensor<string, []>("op_539"), val = tensor<int32, [3]>([1, 256, 1])];
383
+ tensor<fp32, [1, 256, 1]> h_27 = reshape(shape = var_539, x = h_25)[name = tensor<string, []>("h_27")];
384
+ tensor<int32, [2]> var_541_split_sizes_0 = const()[name = tensor<string, []>("op_541_split_sizes_0"), val = tensor<int32, [2]>([128, 128])];
385
+ tensor<int32, []> var_541_axis_0 = const()[name = tensor<string, []>("op_541_axis_0"), val = tensor<int32, []>(1)];
386
+ tensor<fp32, [1, 128, 1]> var_541_0, tensor<fp32, [1, 128, 1]> var_541_1 = split(axis = var_541_axis_0, split_sizes = var_541_split_sizes_0, x = h_27)[name = tensor<string, []>("op_541")];
387
+ tensor<fp32, []> var_543_promoted = const()[name = tensor<string, []>("op_543_promoted"), val = tensor<fp32, []>(0x1p+0)];
388
+ tensor<fp32, [1, 128, 1]> var_544 = add(x = var_541_0, y = var_543_promoted)[name = tensor<string, []>("op_544")];
389
+ tensor<fp32, [1, 128, ?]> var_547 = instance_norm(beta = noise_res_1_adain1_0_norm_bias, epsilon = var_456, gamma = noise_res_1_adain1_0_norm_weight, x = input_39)[name = tensor<string, []>("op_547")];
390
+ tensor<fp32, [1, 128, ?]> var_548 = mul(x = var_544, y = var_547)[name = tensor<string, []>("op_548")];
391
+ tensor<fp32, [1, 128, ?]> xt_19 = add(x = var_548, y = var_541_1)[name = tensor<string, []>("xt_19")];
392
+ tensor<fp32, [1, 128, 1]> var_550 = const()[name = tensor<string, []>("op_550"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3488448)))];
393
+ tensor<fp32, [1, 128, ?]> var_553 = mul(x = noise_res_1_alpha1_0, y = xt_19)[name = tensor<string, []>("op_553")];
394
+ tensor<fp32, [1, 128, ?]> var_554 = sin(x = var_553)[name = tensor<string, []>("op_554")];
395
+ tensor<fp32, []> var_455_promoted = const()[name = tensor<string, []>("op_455_promoted"), val = tensor<fp32, []>(0x1p+1)];
396
+ tensor<fp32, [1, 128, ?]> var_555 = pow(x = var_554, y = var_455_promoted)[name = tensor<string, []>("op_555")];
397
+ tensor<fp32, [1, 128, ?]> var_556 = mul(x = var_550, y = var_555)[name = tensor<string, []>("op_556")];
398
+ tensor<fp32, [1, 128, ?]> input_41 = add(x = xt_19, y = var_556)[name = tensor<string, []>("input_41")];
399
+ tensor<fp32, [128, 128, 11]> weight_35_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [180224]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3489024))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3669312))), name = tensor<string, []>("weight_35_palettized"), shape = tensor<uint32, [3]>([128, 128, 11])];
400
+ tensor<string, []> input_43_pad_type_0 = const()[name = tensor<string, []>("input_43_pad_type_0"), val = tensor<string, []>("custom")];
401
+ tensor<int32, [2]> input_43_pad_0 = const()[name = tensor<string, []>("input_43_pad_0"), val = tensor<int32, [2]>([5, 5])];
402
+ tensor<int32, [1]> input_43_strides_0 = const()[name = tensor<string, []>("input_43_strides_0"), val = tensor<int32, [1]>([1])];
403
+ tensor<int32, [1]> input_43_dilations_0 = const()[name = tensor<string, []>("input_43_dilations_0"), val = tensor<int32, [1]>([1])];
404
+ tensor<int32, []> input_43_groups_0 = const()[name = tensor<string, []>("input_43_groups_0"), val = tensor<int32, []>(1)];
405
+ tensor<fp32, [1, 128, ?]> input_43 = conv(bias = noise_res_1_convs1_0_bias, dilations = input_43_dilations_0, groups = input_43_groups_0, pad = input_43_pad_0, pad_type = input_43_pad_type_0, strides = input_43_strides_0, weight = weight_35_palettized, x = input_41)[name = tensor<string, []>("input_43")];
406
+ tensor<fp32, [1, 256]> h_29 = linear(bias = noise_res_1_adain2_0_fc_bias, weight = noise_res_1_adain2_0_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_8")];
407
+ tensor<int32, [3]> var_572 = const()[name = tensor<string, []>("op_572"), val = tensor<int32, [3]>([1, 256, 1])];
408
+ tensor<fp32, [1, 256, 1]> h_31 = reshape(shape = var_572, x = h_29)[name = tensor<string, []>("h_31")];
409
+ tensor<int32, [2]> var_574_split_sizes_0 = const()[name = tensor<string, []>("op_574_split_sizes_0"), val = tensor<int32, [2]>([128, 128])];
410
+ tensor<int32, []> var_574_axis_0 = const()[name = tensor<string, []>("op_574_axis_0"), val = tensor<int32, []>(1)];
411
+ tensor<fp32, [1, 128, 1]> var_574_0, tensor<fp32, [1, 128, 1]> var_574_1 = split(axis = var_574_axis_0, split_sizes = var_574_split_sizes_0, x = h_31)[name = tensor<string, []>("op_574")];
412
+ tensor<fp32, []> var_576_promoted = const()[name = tensor<string, []>("op_576_promoted"), val = tensor<fp32, []>(0x1p+0)];
413
+ tensor<fp32, [1, 128, 1]> var_577 = add(x = var_574_0, y = var_576_promoted)[name = tensor<string, []>("op_577")];
414
+ tensor<fp32, [1, 128, ?]> var_580 = instance_norm(beta = noise_res_1_adain1_0_norm_bias, epsilon = var_456, gamma = noise_res_1_adain1_0_norm_weight, x = input_43)[name = tensor<string, []>("op_580")];
415
+ tensor<fp32, [1, 128, ?]> var_581 = mul(x = var_577, y = var_580)[name = tensor<string, []>("op_581")];
416
+ tensor<fp32, [1, 128, ?]> xt_21 = add(x = var_581, y = var_574_1)[name = tensor<string, []>("xt_21")];
417
+ tensor<fp32, [1, 128, 1]> var_583 = const()[name = tensor<string, []>("op_583"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3670400)))];
418
+ tensor<fp32, [1, 128, ?]> var_586 = mul(x = noise_res_1_alpha2_0, y = xt_21)[name = tensor<string, []>("op_586")];
419
+ tensor<fp32, [1, 128, ?]> var_587 = sin(x = var_586)[name = tensor<string, []>("op_587")];
420
+ tensor<fp32, []> var_455_promoted_1 = const()[name = tensor<string, []>("op_455_promoted_1"), val = tensor<fp32, []>(0x1p+1)];
421
+ tensor<fp32, [1, 128, ?]> var_588 = pow(x = var_587, y = var_455_promoted_1)[name = tensor<string, []>("op_588")];
422
+ tensor<fp32, [1, 128, ?]> var_589 = mul(x = var_583, y = var_588)[name = tensor<string, []>("op_589")];
423
+ tensor<fp32, [1, 128, ?]> input_45 = add(x = xt_21, y = var_589)[name = tensor<string, []>("input_45")];
424
+ tensor<fp32, [128, 128, 11]> weight_39_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [180224]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3670976))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3851264))), name = tensor<string, []>("weight_39_palettized"), shape = tensor<uint32, [3]>([128, 128, 11])];
425
+ tensor<string, []> xt_23_pad_type_0 = const()[name = tensor<string, []>("xt_23_pad_type_0"), val = tensor<string, []>("custom")];
426
+ tensor<int32, [2]> xt_23_pad_0 = const()[name = tensor<string, []>("xt_23_pad_0"), val = tensor<int32, [2]>([5, 5])];
427
+ tensor<int32, [1]> xt_23_strides_0 = const()[name = tensor<string, []>("xt_23_strides_0"), val = tensor<int32, [1]>([1])];
428
+ tensor<int32, [1]> xt_23_dilations_0 = const()[name = tensor<string, []>("xt_23_dilations_0"), val = tensor<int32, [1]>([1])];
429
+ tensor<int32, []> xt_23_groups_0 = const()[name = tensor<string, []>("xt_23_groups_0"), val = tensor<int32, []>(1)];
430
+ tensor<fp32, [1, 128, ?]> xt_23 = conv(bias = noise_res_1_convs2_0_bias, dilations = xt_23_dilations_0, groups = xt_23_groups_0, pad = xt_23_pad_0, pad_type = xt_23_pad_type_0, strides = xt_23_strides_0, weight = weight_39_palettized, x = input_45)[name = tensor<string, []>("xt_23")];
431
+ tensor<fp32, [1, 128, ?]> input_47 = add(x = xt_23, y = input_39)[name = tensor<string, []>("input_47")];
432
+ tensor<fp32, [1, 256]> h_33 = linear(bias = noise_res_1_adain1_1_fc_bias, weight = noise_res_1_adain1_1_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_9")];
433
+ tensor<int32, [3]> var_606 = const()[name = tensor<string, []>("op_606"), val = tensor<int32, [3]>([1, 256, 1])];
434
+ tensor<fp32, [1, 256, 1]> h_35 = reshape(shape = var_606, x = h_33)[name = tensor<string, []>("h_35")];
435
+ tensor<int32, [2]> var_608_split_sizes_0 = const()[name = tensor<string, []>("op_608_split_sizes_0"), val = tensor<int32, [2]>([128, 128])];
436
+ tensor<int32, []> var_608_axis_0 = const()[name = tensor<string, []>("op_608_axis_0"), val = tensor<int32, []>(1)];
437
+ tensor<fp32, [1, 128, 1]> var_608_0, tensor<fp32, [1, 128, 1]> var_608_1 = split(axis = var_608_axis_0, split_sizes = var_608_split_sizes_0, x = h_35)[name = tensor<string, []>("op_608")];
438
+ tensor<fp32, []> var_610_promoted = const()[name = tensor<string, []>("op_610_promoted"), val = tensor<fp32, []>(0x1p+0)];
439
+ tensor<fp32, [1, 128, 1]> var_611 = add(x = var_608_0, y = var_610_promoted)[name = tensor<string, []>("op_611")];
440
+ tensor<fp32, [1, 128, ?]> var_614 = instance_norm(beta = noise_res_1_adain1_0_norm_bias, epsilon = var_456, gamma = noise_res_1_adain1_0_norm_weight, x = input_47)[name = tensor<string, []>("op_614")];
441
+ tensor<fp32, [1, 128, ?]> var_615 = mul(x = var_611, y = var_614)[name = tensor<string, []>("op_615")];
442
+ tensor<fp32, [1, 128, ?]> xt_25 = add(x = var_615, y = var_608_1)[name = tensor<string, []>("xt_25")];
443
+ tensor<fp32, [1, 128, 1]> var_617 = const()[name = tensor<string, []>("op_617"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3852352)))];
444
+ tensor<fp32, [1, 128, ?]> var_620 = mul(x = noise_res_1_alpha1_1, y = xt_25)[name = tensor<string, []>("op_620")];
445
+ tensor<fp32, [1, 128, ?]> var_621 = sin(x = var_620)[name = tensor<string, []>("op_621")];
446
+ tensor<fp32, []> var_455_promoted_2 = const()[name = tensor<string, []>("op_455_promoted_2"), val = tensor<fp32, []>(0x1p+1)];
447
+ tensor<fp32, [1, 128, ?]> var_622 = pow(x = var_621, y = var_455_promoted_2)[name = tensor<string, []>("op_622")];
448
+ tensor<fp32, [1, 128, ?]> var_623 = mul(x = var_617, y = var_622)[name = tensor<string, []>("op_623")];
449
+ tensor<fp32, [1, 128, ?]> input_49 = add(x = xt_25, y = var_623)[name = tensor<string, []>("input_49")];
450
+ tensor<fp32, [128, 128, 11]> weight_43_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [180224]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3852928))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4033216))), name = tensor<string, []>("weight_43_palettized"), shape = tensor<uint32, [3]>([128, 128, 11])];
451
+ tensor<string, []> input_51_pad_type_0 = const()[name = tensor<string, []>("input_51_pad_type_0"), val = tensor<string, []>("custom")];
452
+ tensor<int32, [2]> input_51_pad_0 = const()[name = tensor<string, []>("input_51_pad_0"), val = tensor<int32, [2]>([15, 15])];
453
+ tensor<int32, [1]> input_51_dilations_0 = const()[name = tensor<string, []>("input_51_dilations_0"), val = tensor<int32, [1]>([3])];
454
+ tensor<int32, [1]> input_51_strides_0 = const()[name = tensor<string, []>("input_51_strides_0"), val = tensor<int32, [1]>([1])];
455
+ tensor<int32, []> input_51_groups_0 = const()[name = tensor<string, []>("input_51_groups_0"), val = tensor<int32, []>(1)];
456
+ tensor<fp32, [1, 128, ?]> input_51 = conv(bias = noise_res_1_convs1_1_bias, dilations = input_51_dilations_0, groups = input_51_groups_0, pad = input_51_pad_0, pad_type = input_51_pad_type_0, strides = input_51_strides_0, weight = weight_43_palettized, x = input_49)[name = tensor<string, []>("input_51")];
457
+ tensor<fp32, [1, 256]> h_37 = linear(bias = noise_res_1_adain2_1_fc_bias, weight = noise_res_1_adain2_1_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_10")];
458
+ tensor<int32, [3]> var_639 = const()[name = tensor<string, []>("op_639"), val = tensor<int32, [3]>([1, 256, 1])];
459
+ tensor<fp32, [1, 256, 1]> h_39 = reshape(shape = var_639, x = h_37)[name = tensor<string, []>("h_39")];
460
+ tensor<int32, [2]> var_641_split_sizes_0 = const()[name = tensor<string, []>("op_641_split_sizes_0"), val = tensor<int32, [2]>([128, 128])];
461
+ tensor<int32, []> var_641_axis_0 = const()[name = tensor<string, []>("op_641_axis_0"), val = tensor<int32, []>(1)];
462
+ tensor<fp32, [1, 128, 1]> var_641_0, tensor<fp32, [1, 128, 1]> var_641_1 = split(axis = var_641_axis_0, split_sizes = var_641_split_sizes_0, x = h_39)[name = tensor<string, []>("op_641")];
463
+ tensor<fp32, []> var_643_promoted = const()[name = tensor<string, []>("op_643_promoted"), val = tensor<fp32, []>(0x1p+0)];
464
+ tensor<fp32, [1, 128, 1]> var_644 = add(x = var_641_0, y = var_643_promoted)[name = tensor<string, []>("op_644")];
465
+ tensor<fp32, [1, 128, ?]> var_647 = instance_norm(beta = noise_res_1_adain1_0_norm_bias, epsilon = var_456, gamma = noise_res_1_adain1_0_norm_weight, x = input_51)[name = tensor<string, []>("op_647")];
466
+ tensor<fp32, [1, 128, ?]> var_648 = mul(x = var_644, y = var_647)[name = tensor<string, []>("op_648")];
467
+ tensor<fp32, [1, 128, ?]> xt_27 = add(x = var_648, y = var_641_1)[name = tensor<string, []>("xt_27")];
468
+ tensor<fp32, [1, 128, 1]> var_650 = const()[name = tensor<string, []>("op_650"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4034304)))];
469
+ tensor<fp32, [1, 128, ?]> var_653 = mul(x = noise_res_1_alpha2_1, y = xt_27)[name = tensor<string, []>("op_653")];
470
+ tensor<fp32, [1, 128, ?]> var_654 = sin(x = var_653)[name = tensor<string, []>("op_654")];
471
+ tensor<fp32, []> var_455_promoted_3 = const()[name = tensor<string, []>("op_455_promoted_3"), val = tensor<fp32, []>(0x1p+1)];
472
+ tensor<fp32, [1, 128, ?]> var_655 = pow(x = var_654, y = var_455_promoted_3)[name = tensor<string, []>("op_655")];
473
+ tensor<fp32, [1, 128, ?]> var_656 = mul(x = var_650, y = var_655)[name = tensor<string, []>("op_656")];
474
+ tensor<fp32, [1, 128, ?]> input_53 = add(x = xt_27, y = var_656)[name = tensor<string, []>("input_53")];
475
+ tensor<fp32, [128, 128, 11]> weight_47_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [180224]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4034880))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4215168))), name = tensor<string, []>("weight_47_palettized"), shape = tensor<uint32, [3]>([128, 128, 11])];
476
+ tensor<string, []> xt_29_pad_type_0 = const()[name = tensor<string, []>("xt_29_pad_type_0"), val = tensor<string, []>("custom")];
477
+ tensor<int32, [2]> xt_29_pad_0 = const()[name = tensor<string, []>("xt_29_pad_0"), val = tensor<int32, [2]>([5, 5])];
478
+ tensor<int32, [1]> xt_29_strides_0 = const()[name = tensor<string, []>("xt_29_strides_0"), val = tensor<int32, [1]>([1])];
479
+ tensor<int32, [1]> xt_29_dilations_0 = const()[name = tensor<string, []>("xt_29_dilations_0"), val = tensor<int32, [1]>([1])];
480
+ tensor<int32, []> xt_29_groups_0 = const()[name = tensor<string, []>("xt_29_groups_0"), val = tensor<int32, []>(1)];
481
+ tensor<fp32, [1, 128, ?]> xt_29 = conv(bias = noise_res_1_convs2_1_bias, dilations = xt_29_dilations_0, groups = xt_29_groups_0, pad = xt_29_pad_0, pad_type = xt_29_pad_type_0, strides = xt_29_strides_0, weight = weight_47_palettized, x = input_53)[name = tensor<string, []>("xt_29")];
482
+ tensor<fp32, [1, 128, ?]> input_55 = add(x = xt_29, y = input_47)[name = tensor<string, []>("input_55")];
483
+ tensor<fp32, [1, 256]> h_41 = linear(bias = noise_res_1_adain1_2_fc_bias, weight = noise_res_1_adain1_2_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_11")];
484
+ tensor<int32, [3]> var_673 = const()[name = tensor<string, []>("op_673"), val = tensor<int32, [3]>([1, 256, 1])];
485
+ tensor<fp32, [1, 256, 1]> h_43 = reshape(shape = var_673, x = h_41)[name = tensor<string, []>("h_43")];
486
+ tensor<int32, [2]> var_675_split_sizes_0 = const()[name = tensor<string, []>("op_675_split_sizes_0"), val = tensor<int32, [2]>([128, 128])];
487
+ tensor<int32, []> var_675_axis_0 = const()[name = tensor<string, []>("op_675_axis_0"), val = tensor<int32, []>(1)];
488
+ tensor<fp32, [1, 128, 1]> var_675_0, tensor<fp32, [1, 128, 1]> var_675_1 = split(axis = var_675_axis_0, split_sizes = var_675_split_sizes_0, x = h_43)[name = tensor<string, []>("op_675")];
489
+ tensor<fp32, []> var_677_promoted = const()[name = tensor<string, []>("op_677_promoted"), val = tensor<fp32, []>(0x1p+0)];
490
+ tensor<fp32, [1, 128, 1]> var_678 = add(x = var_675_0, y = var_677_promoted)[name = tensor<string, []>("op_678")];
491
+ tensor<fp32, [1, 128, ?]> var_681 = instance_norm(beta = noise_res_1_adain1_0_norm_bias, epsilon = var_456, gamma = noise_res_1_adain1_0_norm_weight, x = input_55)[name = tensor<string, []>("op_681")];
492
+ tensor<fp32, [1, 128, ?]> var_682 = mul(x = var_678, y = var_681)[name = tensor<string, []>("op_682")];
493
+ tensor<fp32, [1, 128, ?]> xt_31 = add(x = var_682, y = var_675_1)[name = tensor<string, []>("xt_31")];
494
+ tensor<fp32, [1, 128, 1]> var_684 = const()[name = tensor<string, []>("op_684"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4216256)))];
495
+ tensor<fp32, [1, 128, ?]> var_687 = mul(x = noise_res_1_alpha1_2, y = xt_31)[name = tensor<string, []>("op_687")];
496
+ tensor<fp32, [1, 128, ?]> var_688 = sin(x = var_687)[name = tensor<string, []>("op_688")];
497
+ tensor<fp32, []> var_455_promoted_4 = const()[name = tensor<string, []>("op_455_promoted_4"), val = tensor<fp32, []>(0x1p+1)];
498
+ tensor<fp32, [1, 128, ?]> var_689 = pow(x = var_688, y = var_455_promoted_4)[name = tensor<string, []>("op_689")];
499
+ tensor<fp32, [1, 128, ?]> var_690 = mul(x = var_684, y = var_689)[name = tensor<string, []>("op_690")];
500
+ tensor<fp32, [1, 128, ?]> input_57 = add(x = xt_31, y = var_690)[name = tensor<string, []>("input_57")];
501
+ tensor<fp32, [128, 128, 11]> weight_51_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [180224]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4216832))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4397120))), name = tensor<string, []>("weight_51_palettized"), shape = tensor<uint32, [3]>([128, 128, 11])];
502
+ tensor<string, []> input_59_pad_type_0 = const()[name = tensor<string, []>("input_59_pad_type_0"), val = tensor<string, []>("custom")];
503
+ tensor<int32, [2]> input_59_pad_0 = const()[name = tensor<string, []>("input_59_pad_0"), val = tensor<int32, [2]>([25, 25])];
504
+ tensor<int32, [1]> input_59_dilations_0 = const()[name = tensor<string, []>("input_59_dilations_0"), val = tensor<int32, [1]>([5])];
505
+ tensor<int32, [1]> input_59_strides_0 = const()[name = tensor<string, []>("input_59_strides_0"), val = tensor<int32, [1]>([1])];
506
+ tensor<int32, []> input_59_groups_0 = const()[name = tensor<string, []>("input_59_groups_0"), val = tensor<int32, []>(1)];
507
+ tensor<fp32, [1, 128, ?]> input_59 = conv(bias = noise_res_1_convs1_2_bias, dilations = input_59_dilations_0, groups = input_59_groups_0, pad = input_59_pad_0, pad_type = input_59_pad_type_0, strides = input_59_strides_0, weight = weight_51_palettized, x = input_57)[name = tensor<string, []>("input_59")];
508
+ tensor<fp32, [1, 256]> h_45 = linear(bias = noise_res_1_adain2_2_fc_bias, weight = noise_res_1_adain2_2_fc_weight_palettized, x = style_timbre)[name = tensor<string, []>("linear_12")];
509
+ tensor<int32, [3]> var_706 = const()[name = tensor<string, []>("op_706"), val = tensor<int32, [3]>([1, 256, 1])];
510
+ tensor<fp32, [1, 256, 1]> h = reshape(shape = var_706, x = h_45)[name = tensor<string, []>("h")];
511
+ tensor<int32, [2]> var_708_split_sizes_0 = const()[name = tensor<string, []>("op_708_split_sizes_0"), val = tensor<int32, [2]>([128, 128])];
512
+ tensor<int32, []> var_708_axis_0 = const()[name = tensor<string, []>("op_708_axis_0"), val = tensor<int32, []>(1)];
513
+ tensor<fp32, [1, 128, 1]> var_708_0, tensor<fp32, [1, 128, 1]> var_708_1 = split(axis = var_708_axis_0, split_sizes = var_708_split_sizes_0, x = h)[name = tensor<string, []>("op_708")];
514
+ tensor<fp32, []> var_710_promoted = const()[name = tensor<string, []>("op_710_promoted"), val = tensor<fp32, []>(0x1p+0)];
515
+ tensor<fp32, [1, 128, 1]> var_711 = add(x = var_708_0, y = var_710_promoted)[name = tensor<string, []>("op_711")];
516
+ tensor<fp32, [1, 128, ?]> var_714 = instance_norm(beta = noise_res_1_adain1_0_norm_bias, epsilon = var_456, gamma = noise_res_1_adain1_0_norm_weight, x = input_59)[name = tensor<string, []>("op_714")];
517
+ tensor<fp32, [1, 128, ?]> var_715 = mul(x = var_711, y = var_714)[name = tensor<string, []>("op_715")];
518
+ tensor<fp32, [1, 128, ?]> xt_33 = add(x = var_715, y = var_708_1)[name = tensor<string, []>("xt_33")];
519
+ tensor<fp32, [1, 128, 1]> var_717 = const()[name = tensor<string, []>("op_717"), val = tensor<fp32, [1, 128, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4398208)))];
520
+ tensor<fp32, [1, 128, ?]> var_720 = mul(x = noise_res_1_alpha2_2, y = xt_33)[name = tensor<string, []>("op_720")];
521
+ tensor<fp32, [1, 128, ?]> var_721 = sin(x = var_720)[name = tensor<string, []>("op_721")];
522
+ tensor<fp32, []> var_455_promoted_5 = const()[name = tensor<string, []>("op_455_promoted_5"), val = tensor<fp32, []>(0x1p+1)];
523
+ tensor<fp32, [1, 128, ?]> var_722 = pow(x = var_721, y = var_455_promoted_5)[name = tensor<string, []>("op_722")];
524
+ tensor<fp32, [1, 128, ?]> var_723 = mul(x = var_717, y = var_722)[name = tensor<string, []>("op_723")];
525
+ tensor<fp32, [1, 128, ?]> input = add(x = xt_33, y = var_723)[name = tensor<string, []>("input")];
526
+ tensor<fp32, [128, 128, 11]> weight_55_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [180224]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4398784))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4579072))), name = tensor<string, []>("weight_55_palettized"), shape = tensor<uint32, [3]>([128, 128, 11])];
527
+ tensor<string, []> xt_pad_type_0 = const()[name = tensor<string, []>("xt_pad_type_0"), val = tensor<string, []>("custom")];
528
+ tensor<int32, [2]> xt_pad_0 = const()[name = tensor<string, []>("xt_pad_0"), val = tensor<int32, [2]>([5, 5])];
529
+ tensor<int32, [1]> xt_strides_0 = const()[name = tensor<string, []>("xt_strides_0"), val = tensor<int32, [1]>([1])];
530
+ tensor<int32, [1]> xt_dilations_0 = const()[name = tensor<string, []>("xt_dilations_0"), val = tensor<int32, [1]>([1])];
531
+ tensor<int32, []> xt_groups_0 = const()[name = tensor<string, []>("xt_groups_0"), val = tensor<int32, []>(1)];
532
+ tensor<fp32, [1, 128, ?]> xt = conv(bias = noise_res_1_convs2_2_bias, dilations = xt_dilations_0, groups = xt_groups_0, pad = xt_pad_0, pad_type = xt_pad_type_0, strides = xt_strides_0, weight = weight_55_palettized, x = input)[name = tensor<string, []>("xt")];
533
+ tensor<fp32, [1, 128, ?]> x_source_1 = add(x = xt, y = input_55)[name = tensor<string, []>("op_732")];
534
+ } -> (x_source_0, x_source_1);
535
+ }
ANE/ANE-zh/KokoroNoise.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1a1edc7ac86b9899cb908d6e4e65f3def26ee7720d8730444bee29385596ab5
3
+ size 4580160
ANE/ANE-zh/KokoroNoise.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8674162d48113407da93aeee21d7c2433738494bd3662094ba0249b8d0445e9b
3
+ size 74960
ANE/ANE-zh/KokoroNoise.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1a1edc7ac86b9899cb908d6e4e65f3def26ee7720d8730444bee29385596ab5
3
+ size 4580160
ANE/ANE-zh/KokoroNoise.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "510CA943-DA71-4F61-A81C-5534318B48C9": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Weights",
7
+ "name": "weights",
8
+ "path": "com.apple.CoreML/weights"
9
+ },
10
+ "E69B7771-1977-4192-A042-A8C3807DC461": {
11
+ "author": "com.apple.CoreML",
12
+ "description": "CoreML Model Specification",
13
+ "name": "model.mlmodel",
14
+ "path": "com.apple.CoreML/model.mlmodel"
15
+ }
16
+ },
17
+ "rootModelIdentifier": "E69B7771-1977-4192-A042-A8C3807DC461"
18
+ }
ANE/ANE-zh/KokoroPostAlbert.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ac50dad158a0dc349a240675d84ad920ae3202f282fc70fb2b833d78eb6829f
3
+ size 243
ANE/ANE-zh/KokoroPostAlbert.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9246412170c6118149ccf412efafbb76db5bd47946776a1897b046ca2576f95
3
+ size 556
ANE/ANE-zh/KokoroPostAlbert.mlmodelc/metadata.json ADDED
@@ -0,0 +1,148 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "metadataOutputVersion" : "3.0",
4
+ "storagePrecision" : "Mixed (Float16, Palettized (8 bits))",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float16",
10
+ "formattedType" : "MultiArray (Float16)",
11
+ "shortDescription" : "",
12
+ "shape" : "[]",
13
+ "name" : "duration",
14
+ "type" : "MultiArray"
15
+ },
16
+ {
17
+ "hasShapeFlexibility" : "0",
18
+ "isOptional" : "0",
19
+ "dataType" : "Float16",
20
+ "formattedType" : "MultiArray (Float16)",
21
+ "shortDescription" : "",
22
+ "shape" : "[]",
23
+ "name" : "d",
24
+ "type" : "MultiArray"
25
+ },
26
+ {
27
+ "hasShapeFlexibility" : "0",
28
+ "isOptional" : "0",
29
+ "dataType" : "Float16",
30
+ "formattedType" : "MultiArray (Float16)",
31
+ "shortDescription" : "",
32
+ "shape" : "[]",
33
+ "name" : "t_en",
34
+ "type" : "MultiArray"
35
+ }
36
+ ],
37
+ "modelParameters" : [
38
+
39
+ ],
40
+ "specificationVersion" : 8,
41
+ "mlProgramOperationTypeHistogram" : {
42
+ "Ios17.equal" : 2,
43
+ "Ios17.reshape" : 3,
44
+ "Ios17.transpose" : 30,
45
+ "Split" : 3,
46
+ "Select" : 10,
47
+ "Ios17.expandDims" : 3,
48
+ "Ios17.add" : 6,
49
+ "Tile" : 1,
50
+ "Ios16.sigmoid" : 1,
51
+ "Ios16.reduceSum" : 1,
52
+ "Shape" : 1,
53
+ "Ios17.gather" : 2,
54
+ "Ios17.layerNorm" : 6,
55
+ "Ios17.lstm" : 5,
56
+ "Ios17.cast" : 3,
57
+ "Ios16.constexprLutToDense" : 29,
58
+ "Ios17.conv" : 3,
59
+ "Ios17.realDiv" : 2,
60
+ "Ios17.linear" : 5,
61
+ "Ios17.concat" : 5,
62
+ "Ios17.leakyRelu" : 3,
63
+ "Ios17.mul" : 3
64
+ },
65
+ "computePrecision" : "Mixed (Float16, Float32, Int16, Int32, UInt16)",
66
+ "isUpdatable" : "0",
67
+ "stateSchema" : [
68
+
69
+ ],
70
+ "availability" : {
71
+ "macOS" : "14.0",
72
+ "tvOS" : "17.0",
73
+ "visionOS" : "1.0",
74
+ "watchOS" : "10.0",
75
+ "iOS" : "17.0",
76
+ "macCatalyst" : "17.0"
77
+ },
78
+ "modelType" : {
79
+ "name" : "MLModelType_mlProgram"
80
+ },
81
+ "userDefinedMetadata" : {
82
+ "com.github.apple.coremltools.conversion_date" : "2026-05-03",
83
+ "com.github.apple.coremltools.source" : "torch==2.11.0",
84
+ "com.github.apple.coremltools.version" : "9.0",
85
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
86
+ },
87
+ "inputSchema" : [
88
+ {
89
+ "dataType" : "Float16",
90
+ "hasShapeFlexibility" : "1",
91
+ "isOptional" : "0",
92
+ "shapeFlexibility" : "1 × 2...512 × 768",
93
+ "shapeRange" : "[[1, 1], [2, 512], [768, 768]]",
94
+ "formattedType" : "MultiArray (Float16 1 × 37 × 768)",
95
+ "type" : "MultiArray",
96
+ "shape" : "[1, 37, 768]",
97
+ "name" : "bert_dur",
98
+ "shortDescription" : ""
99
+ },
100
+ {
101
+ "dataType" : "Int32",
102
+ "hasShapeFlexibility" : "1",
103
+ "isOptional" : "0",
104
+ "shapeFlexibility" : "1 × 2...512",
105
+ "shapeRange" : "[[1, 1], [2, 512]]",
106
+ "formattedType" : "MultiArray (Int32 1 × 37)",
107
+ "type" : "MultiArray",
108
+ "shape" : "[1, 37]",
109
+ "name" : "input_ids",
110
+ "shortDescription" : ""
111
+ },
112
+ {
113
+ "hasShapeFlexibility" : "0",
114
+ "isOptional" : "0",
115
+ "dataType" : "Float16",
116
+ "formattedType" : "MultiArray (Float16 1 × 128)",
117
+ "shortDescription" : "",
118
+ "shape" : "[1, 128]",
119
+ "name" : "style_s",
120
+ "type" : "MultiArray"
121
+ },
122
+ {
123
+ "hasShapeFlexibility" : "0",
124
+ "isOptional" : "0",
125
+ "dataType" : "Float16",
126
+ "formattedType" : "MultiArray (Float16 1)",
127
+ "shortDescription" : "",
128
+ "shape" : "[1]",
129
+ "name" : "speed",
130
+ "type" : "MultiArray"
131
+ },
132
+ {
133
+ "dataType" : "Int32",
134
+ "hasShapeFlexibility" : "1",
135
+ "isOptional" : "0",
136
+ "shapeFlexibility" : "1 × 2...512",
137
+ "shapeRange" : "[[1, 1], [2, 512]]",
138
+ "formattedType" : "MultiArray (Int32 1 × 37)",
139
+ "type" : "MultiArray",
140
+ "shape" : "[1, 37]",
141
+ "name" : "attention_mask",
142
+ "shortDescription" : ""
143
+ }
144
+ ],
145
+ "generatedClassName" : "KokoroPostAlbert",
146
+ "method" : "predict"
147
+ }
148
+ ]
ANE/ANE-zh/KokoroPostAlbert.mlmodelc/model.mil ADDED
@@ -0,0 +1,277 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ program(1.0)
2
+ [buildInfo = dict<tensor<string, []>, tensor<string, []>>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}})]
3
+ {
4
+ func main<ios17>(tensor<int32, [1, ?]> attention_mask, tensor<fp16, [1, ?, 768]> bert_dur, tensor<int32, [1, ?]> input_ids, tensor<fp16, [1]> speed, tensor<fp16, [1, 128]> style_s) [FlexibleShapeInformation = tuple<tuple<tensor<string, []>, dict<tensor<string, []>, tensor<int32, [?]>>>, tuple<tensor<string, []>, dict<tensor<string, []>, list<tensor<int32, [2]>, ?>>>>((("DefaultShapes", {{"attention_mask", [1, 37]}, {"bert_dur", [1, 37, 768]}, {"input_ids", [1, 37]}}), ("RangeDims", {{"attention_mask", [[1, 1], [2, 512]]}, {"bert_dur", [[1, 1], [2, 512], [768, 768]]}, {"input_ids", [[1, 1], [2, 512]]}})))] {
5
+ tensor<int32, []> var_11 = const()[name = tensor<string, []>("op_11"), val = tensor<int32, []>(0)];
6
+ tensor<bool, [1, ?]> m = equal(x = attention_mask, y = var_11)[name = tensor<string, []>("m")];
7
+ tensor<fp16, [512, 768]> bert_encoder_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [393216]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(393344))), name = tensor<string, []>("bert_encoder_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 768])];
8
+ tensor<fp16, [512]> bert_encoder_bias_to_fp16 = const()[name = tensor<string, []>("bert_encoder_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(393920)))];
9
+ tensor<fp16, [1, ?, 512]> linear_0_cast_fp16 = linear(bias = bert_encoder_bias_to_fp16, weight = bert_encoder_weight_to_fp16_palettized, x = bert_dur)[name = tensor<string, []>("linear_0_cast_fp16")];
10
+ tensor<int32, [3]> transpose_6_perm_0 = const()[name = tensor<string, []>("transpose_6_perm_0"), val = tensor<int32, [3]>([-2, 0, -1])];
11
+ tensor<int32, []> var_30 = const()[name = tensor<string, []>("op_30"), val = tensor<int32, []>(-1)];
12
+ tensor<int32, []> var_31 = const()[name = tensor<string, []>("op_31"), val = tensor<int32, []>(1)];
13
+ tensor<fp16, [?, 1, 512]> transpose_6_cast_fp16 = transpose(perm = transpose_6_perm_0, x = linear_0_cast_fp16)[name = tensor<string, []>("transpose_47")];
14
+ tensor<int32, [3]> var_48_shape_cast_fp16 = shape(x = transpose_6_cast_fp16)[name = tensor<string, []>("op_48_shape_cast_fp16")];
15
+ tensor<int32, []> gather_0_axis_0 = const()[name = tensor<string, []>("gather_0_axis_0"), val = tensor<int32, []>(0)];
16
+ tensor<int32, []> gather_0_batch_dims_0 = const()[name = tensor<string, []>("gather_0_batch_dims_0"), val = tensor<int32, []>(0)];
17
+ tensor<bool, []> gather_0_validate_indices_0 = const()[name = tensor<string, []>("gather_0_validate_indices_0"), val = tensor<bool, []>(false)];
18
+ tensor<string, []> var_48_shape_cast_fp16_to_int16_dtype_0 = const()[name = tensor<string, []>("op_48_shape_cast_fp16_to_int16_dtype_0"), val = tensor<string, []>("int16")];
19
+ tensor<uint16, []> gather_0_indices_0_to_uint16 = const()[name = tensor<string, []>("gather_0_indices_0_to_uint16"), val = tensor<uint16, []>(0)];
20
+ tensor<int16, [3]> var_48_shape_cast_fp16_to_int16 = cast(dtype = var_48_shape_cast_fp16_to_int16_dtype_0, x = var_48_shape_cast_fp16)[name = tensor<string, []>("cast_2")];
21
+ tensor<int16, []> gather_0_cast_uint16 = gather(axis = gather_0_axis_0, batch_dims = gather_0_batch_dims_0, indices = gather_0_indices_0_to_uint16, validate_indices = gather_0_validate_indices_0, x = var_48_shape_cast_fp16_to_int16)[name = tensor<string, []>("gather_0_cast_uint16")];
22
+ tensor<string, []> gather_0_cast_uint16_to_int32_dtype_0 = const()[name = tensor<string, []>("gather_0_cast_uint16_to_int32_dtype_0"), val = tensor<string, []>("int32")];
23
+ tensor<int32, [1]> var_49_axes_0 = const()[name = tensor<string, []>("op_49_axes_0"), val = tensor<int32, [1]>([0])];
24
+ tensor<fp16, [1, 1, 128]> var_49_cast_fp16 = expand_dims(axes = var_49_axes_0, x = style_s)[name = tensor<string, []>("op_49_cast_fp16")];
25
+ tensor<int32, []> concat_0_axis_0 = const()[name = tensor<string, []>("concat_0_axis_0"), val = tensor<int32, []>(0)];
26
+ tensor<bool, []> concat_0_interleave_0 = const()[name = tensor<string, []>("concat_0_interleave_0"), val = tensor<bool, []>(false)];
27
+ tensor<int32, []> gather_0_cast_uint16_to_int32 = cast(dtype = gather_0_cast_uint16_to_int32_dtype_0, x = gather_0_cast_uint16)[name = tensor<string, []>("cast_1")];
28
+ tensor<int32, [3]> concat_0 = concat(axis = concat_0_axis_0, interleave = concat_0_interleave_0, values = (gather_0_cast_uint16_to_int32, var_30, var_30))[name = tensor<string, []>("concat_0")];
29
+ tensor<int32, [3]> shape_0 = const()[name = tensor<string, []>("shape_0"), val = tensor<int32, [3]>([1, 1, 128])];
30
+ tensor<int32, []> equal_0_y_0 = const()[name = tensor<string, []>("equal_0_y_0"), val = tensor<int32, []>(-1)];
31
+ tensor<bool, [3]> equal_0 = equal(x = concat_0, y = equal_0_y_0)[name = tensor<string, []>("equal_0")];
32
+ tensor<int32, [3]> select_0 = select(a = shape_0, b = concat_0, cond = equal_0)[name = tensor<string, []>("select_0")];
33
+ tensor<int32, [3]> real_div_0 = real_div(x = select_0, y = shape_0)[name = tensor<string, []>("real_div_0")];
34
+ tensor<fp16, [?, ?, ?]> s_cast_fp16 = tile(reps = real_div_0, x = var_49_cast_fp16)[name = tensor<string, []>("s_cast_fp16")];
35
+ tensor<bool, []> x_5_interleave_0 = const()[name = tensor<string, []>("x_5_interleave_0"), val = tensor<bool, []>(false)];
36
+ tensor<fp16, [?, 1, ?]> x_5_cast_fp16 = concat(axis = var_30, interleave = x_5_interleave_0, values = (transpose_6_cast_fp16, s_cast_fp16))[name = tensor<string, []>("x_5_cast_fp16")];
37
+ tensor<int32, [1]> var_54_axes_0 = const()[name = tensor<string, []>("op_54_axes_0"), val = tensor<int32, [1]>([-1])];
38
+ tensor<bool, [1, ?, 1]> var_54 = expand_dims(axes = var_54_axes_0, x = m)[name = tensor<string, []>("op_54")];
39
+ tensor<int32, [3]> var_55_perm_0 = const()[name = tensor<string, []>("op_55_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
40
+ tensor<fp16, []> var_28_to_fp16 = const()[name = tensor<string, []>("op_28_to_fp16"), val = tensor<fp16, []>(0x0p+0)];
41
+ tensor<bool, [?, 1, 1]> var_55 = transpose(perm = var_55_perm_0, x = var_54)[name = tensor<string, []>("transpose_46")];
42
+ tensor<fp16, [?, 1, ?]> x_7_cast_fp16 = select(a = var_28_to_fp16, b = x_5_cast_fp16, cond = var_55)[name = tensor<string, []>("x_7_cast_fp16")];
43
+ tensor<string, []> input_3_batch_first_direction_0 = const()[name = tensor<string, []>("input_3_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
44
+ tensor<bool, []> input_3_batch_first_output_sequence_0 = const()[name = tensor<string, []>("input_3_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
45
+ tensor<string, []> input_3_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("input_3_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
46
+ tensor<string, []> input_3_batch_first_cell_activation_0 = const()[name = tensor<string, []>("input_3_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
47
+ tensor<string, []> input_3_batch_first_activation_0 = const()[name = tensor<string, []>("input_3_batch_first_activation_0"), val = tensor<string, []>("tanh")];
48
+ tensor<fp16, [1, 512]> input_3_batch_first_lstm_h0_reshaped_to_fp16 = const()[name = tensor<string, []>("input_3_batch_first_lstm_h0_reshaped_to_fp16"), val = tensor<fp16, [1, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(395008)))];
49
+ tensor<fp16, [1024, 640]> concat_5_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(396096))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1051520))), name = tensor<string, []>("concat_5_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
50
+ tensor<fp16, [1024, 256]> concat_6_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1052096))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1314304))), name = tensor<string, []>("concat_6_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
51
+ tensor<fp16, [1024]> add_0_to_fp16 = const()[name = tensor<string, []>("add_0_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1314880)))];
52
+ tensor<fp16, [1024, 640]> concat_7_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1316992))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1972416))), name = tensor<string, []>("concat_7_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
53
+ tensor<fp16, [1024, 256]> concat_8_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1972992))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2235200))), name = tensor<string, []>("concat_8_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
54
+ tensor<fp16, [1024]> add_1_to_fp16 = const()[name = tensor<string, []>("add_1_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2235776)))];
55
+ tensor<fp16, [?, 1, 512]> input_3_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> input_3_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> input_3_batch_first_cast_fp16_2 = lstm(activation = input_3_batch_first_activation_0, bias = add_0_to_fp16, bias_back = add_1_to_fp16, cell_activation = input_3_batch_first_cell_activation_0, direction = input_3_batch_first_direction_0, initial_c = input_3_batch_first_lstm_h0_reshaped_to_fp16, initial_h = input_3_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = input_3_batch_first_output_sequence_0, recurrent_activation = input_3_batch_first_recurrent_activation_0, weight_hh = concat_6_to_fp16_palettized, weight_hh_back = concat_8_to_fp16_palettized, weight_ih = concat_5_to_fp16_palettized, weight_ih_back = concat_7_to_fp16_palettized, x = x_7_cast_fp16)[name = tensor<string, []>("input_3_batch_first_cast_fp16")];
56
+ tensor<int32, [3]> transpose_17_perm_0 = const()[name = tensor<string, []>("transpose_17_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
57
+ tensor<fp16, [1024, 128]> dur_encoder_norms_0_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2237888))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2369024))), name = tensor<string, []>("dur_encoder_norms_0_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
58
+ tensor<fp16, [1024]> dur_encoder_norms_0_fc_bias_to_fp16 = const()[name = tensor<string, []>("dur_encoder_norms_0_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2369600)))];
59
+ tensor<fp16, [1, 1024]> linear_1_cast_fp16 = linear(bias = dur_encoder_norms_0_fc_bias_to_fp16, weight = dur_encoder_norms_0_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_1_cast_fp16")];
60
+ tensor<int32, [3]> var_89 = const()[name = tensor<string, []>("op_89"), val = tensor<int32, [3]>([1, 1024, 1])];
61
+ tensor<fp16, [1, 1024, 1]> h_3_cast_fp16 = reshape(shape = var_89, x = linear_1_cast_fp16)[name = tensor<string, []>("h_3_cast_fp16")];
62
+ tensor<int32, [2]> var_91_split_sizes_0 = const()[name = tensor<string, []>("op_91_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
63
+ tensor<int32, []> var_91_axis_0 = const()[name = tensor<string, []>("op_91_axis_0"), val = tensor<int32, []>(1)];
64
+ tensor<fp16, [1, 512, 1]> var_91_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_91_cast_fp16_1 = split(axis = var_91_axis_0, split_sizes = var_91_split_sizes_0, x = h_3_cast_fp16)[name = tensor<string, []>("op_91_cast_fp16")];
65
+ tensor<int32, [3]> gamma_3_perm_0 = const()[name = tensor<string, []>("gamma_3_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
66
+ tensor<int32, [3]> beta_3_perm_0 = const()[name = tensor<string, []>("beta_3_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
67
+ tensor<int32, [1]> x_21_axes_0 = const()[name = tensor<string, []>("x_21_axes_0"), val = tensor<int32, [1]>([-1])];
68
+ tensor<fp16, []> var_20_to_fp16 = const()[name = tensor<string, []>("op_20_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
69
+ tensor<fp16, [1, ?, 512]> transpose_17_cast_fp16 = transpose(perm = transpose_17_perm_0, x = input_3_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_45")];
70
+ tensor<fp16, [1, ?, 512]> x_21_cast_fp16 = layer_norm(axes = x_21_axes_0, epsilon = var_20_to_fp16, x = transpose_17_cast_fp16)[name = tensor<string, []>("x_21_cast_fp16")];
71
+ tensor<fp16, []> var_97_promoted_to_fp16 = const()[name = tensor<string, []>("op_97_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
72
+ tensor<fp16, [1, 1, 512]> gamma_3_cast_fp16 = transpose(perm = gamma_3_perm_0, x = var_91_cast_fp16_0)[name = tensor<string, []>("transpose_44")];
73
+ tensor<fp16, [1, 1, 512]> var_98_cast_fp16 = add(x = gamma_3_cast_fp16, y = var_97_promoted_to_fp16)[name = tensor<string, []>("op_98_cast_fp16")];
74
+ tensor<fp16, [1, ?, 512]> var_99_cast_fp16 = mul(x = var_98_cast_fp16, y = x_21_cast_fp16)[name = tensor<string, []>("op_99_cast_fp16")];
75
+ tensor<fp16, [1, 1, 512]> beta_3_cast_fp16 = transpose(perm = beta_3_perm_0, x = var_91_cast_fp16_1)[name = tensor<string, []>("transpose_43")];
76
+ tensor<fp16, [1, ?, 512]> x_23_cast_fp16 = add(x = var_99_cast_fp16, y = beta_3_cast_fp16)[name = tensor<string, []>("x_23_cast_fp16")];
77
+ tensor<bool, []> x_27_interleave_0 = const()[name = tensor<string, []>("x_27_interleave_0"), val = tensor<bool, []>(false)];
78
+ tensor<int32, [3]> transpose_12_perm_0 = const()[name = tensor<string, []>("transpose_12_perm_0"), val = tensor<int32, [3]>([0, -1, -2])];
79
+ tensor<int32, [3]> transpose_13_perm_0 = const()[name = tensor<string, []>("transpose_13_perm_0"), val = tensor<int32, [3]>([1, 2, 0])];
80
+ tensor<fp16, [?, ?, ?]> transpose_13 = transpose(perm = transpose_13_perm_0, x = s_cast_fp16)[name = tensor<string, []>("transpose_41")];
81
+ tensor<fp16, [1, 512, ?]> transpose_12 = transpose(perm = transpose_12_perm_0, x = x_23_cast_fp16)[name = tensor<string, []>("transpose_42")];
82
+ tensor<fp16, [1, ?, ?]> x_27_cast_fp16 = concat(axis = var_31, interleave = x_27_interleave_0, values = (transpose_12, transpose_13))[name = tensor<string, []>("x_27_cast_fp16")];
83
+ tensor<int32, [3]> var_109_perm_0 = const()[name = tensor<string, []>("op_109_perm_0"), val = tensor<int32, [3]>([0, -1, -2])];
84
+ tensor<bool, [1, 1, ?]> var_109 = transpose(perm = var_109_perm_0, x = var_54)[name = tensor<string, []>("transpose_40")];
85
+ tensor<fp16, [1, ?, ?]> x_29_cast_fp16 = select(a = var_28_to_fp16, b = x_27_cast_fp16, cond = var_109)[name = tensor<string, []>("x_29_cast_fp16")];
86
+ tensor<int32, [3]> transpose_10_perm_0 = const()[name = tensor<string, []>("transpose_10_perm_0"), val = tensor<int32, [3]>([-1, 0, -2])];
87
+ tensor<string, []> input_9_batch_first_direction_0 = const()[name = tensor<string, []>("input_9_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
88
+ tensor<bool, []> input_9_batch_first_output_sequence_0 = const()[name = tensor<string, []>("input_9_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
89
+ tensor<string, []> input_9_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("input_9_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
90
+ tensor<string, []> input_9_batch_first_cell_activation_0 = const()[name = tensor<string, []>("input_9_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
91
+ tensor<string, []> input_9_batch_first_activation_0 = const()[name = tensor<string, []>("input_9_batch_first_activation_0"), val = tensor<string, []>("tanh")];
92
+ tensor<fp16, [1024, 640]> concat_15_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2371712))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3027136))), name = tensor<string, []>("concat_15_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
93
+ tensor<fp16, [1024, 256]> concat_16_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3027712))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3289920))), name = tensor<string, []>("concat_16_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
94
+ tensor<fp16, [1024]> add_2_to_fp16 = const()[name = tensor<string, []>("add_2_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3290496)))];
95
+ tensor<fp16, [1024, 640]> concat_17_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3292608))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3948032))), name = tensor<string, []>("concat_17_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
96
+ tensor<fp16, [1024, 256]> concat_18_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3948608))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4210816))), name = tensor<string, []>("concat_18_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
97
+ tensor<fp16, [1024]> add_3_to_fp16 = const()[name = tensor<string, []>("add_3_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4211392)))];
98
+ tensor<fp16, [?, 1, ?]> transpose_10_cast_fp16 = transpose(perm = transpose_10_perm_0, x = x_29_cast_fp16)[name = tensor<string, []>("transpose_39")];
99
+ tensor<fp16, [?, 1, 512]> input_9_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> input_9_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> input_9_batch_first_cast_fp16_2 = lstm(activation = input_9_batch_first_activation_0, bias = add_2_to_fp16, bias_back = add_3_to_fp16, cell_activation = input_9_batch_first_cell_activation_0, direction = input_9_batch_first_direction_0, initial_c = input_3_batch_first_lstm_h0_reshaped_to_fp16, initial_h = input_3_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = input_9_batch_first_output_sequence_0, recurrent_activation = input_9_batch_first_recurrent_activation_0, weight_hh = concat_16_to_fp16_palettized, weight_hh_back = concat_18_to_fp16_palettized, weight_ih = concat_15_to_fp16_palettized, weight_ih_back = concat_17_to_fp16_palettized, x = transpose_10_cast_fp16)[name = tensor<string, []>("input_9_batch_first_cast_fp16")];
100
+ tensor<int32, [3]> transpose_18_perm_0 = const()[name = tensor<string, []>("transpose_18_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
101
+ tensor<fp16, [1024, 128]> dur_encoder_norms_1_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4213504))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4344640))), name = tensor<string, []>("dur_encoder_norms_1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
102
+ tensor<fp16, [1024]> dur_encoder_norms_1_fc_bias_to_fp16 = const()[name = tensor<string, []>("dur_encoder_norms_1_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4345216)))];
103
+ tensor<fp16, [1, 1024]> linear_2_cast_fp16 = linear(bias = dur_encoder_norms_1_fc_bias_to_fp16, weight = dur_encoder_norms_1_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_2_cast_fp16")];
104
+ tensor<int32, [3]> var_141 = const()[name = tensor<string, []>("op_141"), val = tensor<int32, [3]>([1, 1024, 1])];
105
+ tensor<fp16, [1, 1024, 1]> h_7_cast_fp16 = reshape(shape = var_141, x = linear_2_cast_fp16)[name = tensor<string, []>("h_7_cast_fp16")];
106
+ tensor<int32, [2]> var_143_split_sizes_0 = const()[name = tensor<string, []>("op_143_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
107
+ tensor<int32, []> var_143_axis_0 = const()[name = tensor<string, []>("op_143_axis_0"), val = tensor<int32, []>(1)];
108
+ tensor<fp16, [1, 512, 1]> var_143_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_143_cast_fp16_1 = split(axis = var_143_axis_0, split_sizes = var_143_split_sizes_0, x = h_7_cast_fp16)[name = tensor<string, []>("op_143_cast_fp16")];
109
+ tensor<int32, [3]> gamma_7_perm_0 = const()[name = tensor<string, []>("gamma_7_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
110
+ tensor<int32, [3]> beta_7_perm_0 = const()[name = tensor<string, []>("beta_7_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
111
+ tensor<int32, [1]> x_39_axes_0 = const()[name = tensor<string, []>("x_39_axes_0"), val = tensor<int32, [1]>([-1])];
112
+ tensor<fp16, [1, ?, 512]> transpose_18_cast_fp16 = transpose(perm = transpose_18_perm_0, x = input_9_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_38")];
113
+ tensor<fp16, [1, ?, 512]> x_39_cast_fp16 = layer_norm(axes = x_39_axes_0, epsilon = var_20_to_fp16, x = transpose_18_cast_fp16)[name = tensor<string, []>("x_39_cast_fp16")];
114
+ tensor<fp16, []> var_149_promoted_to_fp16 = const()[name = tensor<string, []>("op_149_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
115
+ tensor<fp16, [1, 1, 512]> gamma_7_cast_fp16 = transpose(perm = gamma_7_perm_0, x = var_143_cast_fp16_0)[name = tensor<string, []>("transpose_37")];
116
+ tensor<fp16, [1, 1, 512]> var_150_cast_fp16 = add(x = gamma_7_cast_fp16, y = var_149_promoted_to_fp16)[name = tensor<string, []>("op_150_cast_fp16")];
117
+ tensor<fp16, [1, ?, 512]> var_151_cast_fp16 = mul(x = var_150_cast_fp16, y = x_39_cast_fp16)[name = tensor<string, []>("op_151_cast_fp16")];
118
+ tensor<fp16, [1, 1, 512]> beta_7_cast_fp16 = transpose(perm = beta_7_perm_0, x = var_143_cast_fp16_1)[name = tensor<string, []>("transpose_36")];
119
+ tensor<fp16, [1, ?, 512]> x_41_cast_fp16 = add(x = var_151_cast_fp16, y = beta_7_cast_fp16)[name = tensor<string, []>("x_41_cast_fp16")];
120
+ tensor<bool, []> x_45_interleave_0 = const()[name = tensor<string, []>("x_45_interleave_0"), val = tensor<bool, []>(false)];
121
+ tensor<int32, [3]> transpose_16_perm_0 = const()[name = tensor<string, []>("transpose_16_perm_0"), val = tensor<int32, [3]>([0, -1, -2])];
122
+ tensor<fp16, [1, 512, ?]> transpose_16 = transpose(perm = transpose_16_perm_0, x = x_41_cast_fp16)[name = tensor<string, []>("transpose_35")];
123
+ tensor<fp16, [1, ?, ?]> x_45_cast_fp16 = concat(axis = var_31, interleave = x_45_interleave_0, values = (transpose_16, transpose_13))[name = tensor<string, []>("x_45_cast_fp16")];
124
+ tensor<fp16, [1, ?, ?]> x_47_cast_fp16 = select(a = var_28_to_fp16, b = x_45_cast_fp16, cond = var_109)[name = tensor<string, []>("x_47_cast_fp16")];
125
+ tensor<int32, [3]> transpose_12_perm_0_1 = const()[name = tensor<string, []>("transpose_12_perm_0_1"), val = tensor<int32, [3]>([-1, 0, -2])];
126
+ tensor<string, []> input_15_batch_first_direction_0 = const()[name = tensor<string, []>("input_15_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
127
+ tensor<bool, []> input_15_batch_first_output_sequence_0 = const()[name = tensor<string, []>("input_15_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
128
+ tensor<string, []> input_15_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("input_15_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
129
+ tensor<string, []> input_15_batch_first_cell_activation_0 = const()[name = tensor<string, []>("input_15_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
130
+ tensor<string, []> input_15_batch_first_activation_0 = const()[name = tensor<string, []>("input_15_batch_first_activation_0"), val = tensor<string, []>("tanh")];
131
+ tensor<fp16, [1024, 640]> concat_25_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4347328))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5002752))), name = tensor<string, []>("concat_25_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
132
+ tensor<fp16, [1024, 256]> concat_26_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5003328))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5265536))), name = tensor<string, []>("concat_26_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
133
+ tensor<fp16, [1024]> add_4_to_fp16 = const()[name = tensor<string, []>("add_4_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5266112)))];
134
+ tensor<fp16, [1024, 640]> concat_27_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5268224))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5923648))), name = tensor<string, []>("concat_27_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
135
+ tensor<fp16, [1024, 256]> concat_28_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5924224))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6186432))), name = tensor<string, []>("concat_28_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
136
+ tensor<fp16, [1024]> add_5_to_fp16 = const()[name = tensor<string, []>("add_5_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6187008)))];
137
+ tensor<fp16, [?, 1, ?]> transpose_12_cast_fp16 = transpose(perm = transpose_12_perm_0_1, x = x_47_cast_fp16)[name = tensor<string, []>("transpose_34")];
138
+ tensor<fp16, [?, 1, 512]> input_15_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> input_15_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> input_15_batch_first_cast_fp16_2 = lstm(activation = input_15_batch_first_activation_0, bias = add_4_to_fp16, bias_back = add_5_to_fp16, cell_activation = input_15_batch_first_cell_activation_0, direction = input_15_batch_first_direction_0, initial_c = input_3_batch_first_lstm_h0_reshaped_to_fp16, initial_h = input_3_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = input_15_batch_first_output_sequence_0, recurrent_activation = input_15_batch_first_recurrent_activation_0, weight_hh = concat_26_to_fp16_palettized, weight_hh_back = concat_28_to_fp16_palettized, weight_ih = concat_25_to_fp16_palettized, weight_ih_back = concat_27_to_fp16_palettized, x = transpose_12_cast_fp16)[name = tensor<string, []>("input_15_batch_first_cast_fp16")];
139
+ tensor<int32, [3]> transpose_19_perm_0 = const()[name = tensor<string, []>("transpose_19_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
140
+ tensor<fp16, [1024, 128]> dur_encoder_norms_2_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6189120))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6320256))), name = tensor<string, []>("dur_encoder_norms_2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
141
+ tensor<fp16, [1024]> dur_encoder_norms_2_fc_bias_to_fp16 = const()[name = tensor<string, []>("dur_encoder_norms_2_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6320832)))];
142
+ tensor<fp16, [1, 1024]> linear_3_cast_fp16 = linear(bias = dur_encoder_norms_2_fc_bias_to_fp16, weight = dur_encoder_norms_2_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_3_cast_fp16")];
143
+ tensor<int32, [3]> var_193 = const()[name = tensor<string, []>("op_193"), val = tensor<int32, [3]>([1, 1024, 1])];
144
+ tensor<fp16, [1, 1024, 1]> h_cast_fp16 = reshape(shape = var_193, x = linear_3_cast_fp16)[name = tensor<string, []>("h_cast_fp16")];
145
+ tensor<int32, [2]> var_195_split_sizes_0 = const()[name = tensor<string, []>("op_195_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
146
+ tensor<int32, []> var_195_axis_0 = const()[name = tensor<string, []>("op_195_axis_0"), val = tensor<int32, []>(1)];
147
+ tensor<fp16, [1, 512, 1]> var_195_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_195_cast_fp16_1 = split(axis = var_195_axis_0, split_sizes = var_195_split_sizes_0, x = h_cast_fp16)[name = tensor<string, []>("op_195_cast_fp16")];
148
+ tensor<int32, [3]> gamma_11_perm_0 = const()[name = tensor<string, []>("gamma_11_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
149
+ tensor<int32, [3]> beta_11_perm_0 = const()[name = tensor<string, []>("beta_11_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
150
+ tensor<int32, [1]> x_57_axes_0 = const()[name = tensor<string, []>("x_57_axes_0"), val = tensor<int32, [1]>([-1])];
151
+ tensor<fp16, [1, ?, 512]> transpose_19_cast_fp16 = transpose(perm = transpose_19_perm_0, x = input_15_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_33")];
152
+ tensor<fp16, [1, ?, 512]> x_57_cast_fp16 = layer_norm(axes = x_57_axes_0, epsilon = var_20_to_fp16, x = transpose_19_cast_fp16)[name = tensor<string, []>("x_57_cast_fp16")];
153
+ tensor<fp16, []> var_201_promoted_to_fp16 = const()[name = tensor<string, []>("op_201_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
154
+ tensor<fp16, [1, 1, 512]> gamma_11_cast_fp16 = transpose(perm = gamma_11_perm_0, x = var_195_cast_fp16_0)[name = tensor<string, []>("transpose_32")];
155
+ tensor<fp16, [1, 1, 512]> var_202_cast_fp16 = add(x = gamma_11_cast_fp16, y = var_201_promoted_to_fp16)[name = tensor<string, []>("op_202_cast_fp16")];
156
+ tensor<fp16, [1, ?, 512]> var_203_cast_fp16 = mul(x = var_202_cast_fp16, y = x_57_cast_fp16)[name = tensor<string, []>("op_203_cast_fp16")];
157
+ tensor<fp16, [1, 1, 512]> beta_11_cast_fp16 = transpose(perm = beta_11_perm_0, x = var_195_cast_fp16_1)[name = tensor<string, []>("transpose_31")];
158
+ tensor<fp16, [1, ?, 512]> x_59_cast_fp16 = add(x = var_203_cast_fp16, y = beta_11_cast_fp16)[name = tensor<string, []>("x_59_cast_fp16")];
159
+ tensor<bool, []> x_63_interleave_0 = const()[name = tensor<string, []>("x_63_interleave_0"), val = tensor<bool, []>(false)];
160
+ tensor<int32, [3]> transpose_17_perm_0_1 = const()[name = tensor<string, []>("transpose_17_perm_0_1"), val = tensor<int32, [3]>([0, -1, -2])];
161
+ tensor<fp16, [1, 512, ?]> transpose_17 = transpose(perm = transpose_17_perm_0_1, x = x_59_cast_fp16)[name = tensor<string, []>("transpose_30")];
162
+ tensor<fp16, [1, ?, ?]> x_63_cast_fp16 = concat(axis = var_31, interleave = x_63_interleave_0, values = (transpose_17, transpose_13))[name = tensor<string, []>("x_63_cast_fp16")];
163
+ tensor<fp16, [1, ?, ?]> x_65_cast_fp16 = select(a = var_28_to_fp16, b = x_63_cast_fp16, cond = var_109)[name = tensor<string, []>("x_65_cast_fp16")];
164
+ tensor<int32, [3]> input_19_perm_0 = const()[name = tensor<string, []>("input_19_perm_0"), val = tensor<int32, [3]>([0, -1, -2])];
165
+ tensor<int32, [3]> input_19_batch_first_transpose_perm_0 = const()[name = tensor<string, []>("input_19_batch_first_transpose_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
166
+ tensor<string, []> input_21_batch_first_direction_0 = const()[name = tensor<string, []>("input_21_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
167
+ tensor<bool, []> input_21_batch_first_output_sequence_0 = const()[name = tensor<string, []>("input_21_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
168
+ tensor<string, []> input_21_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("input_21_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
169
+ tensor<string, []> input_21_batch_first_cell_activation_0 = const()[name = tensor<string, []>("input_21_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
170
+ tensor<string, []> input_21_batch_first_activation_0 = const()[name = tensor<string, []>("input_21_batch_first_activation_0"), val = tensor<string, []>("tanh")];
171
+ tensor<fp16, [1024, 640]> concat_35_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6322944))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6978368))), name = tensor<string, []>("concat_35_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
172
+ tensor<fp16, [1024, 256]> concat_36_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6978944))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7241152))), name = tensor<string, []>("concat_36_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
173
+ tensor<fp16, [1024]> add_6_to_fp16 = const()[name = tensor<string, []>("add_6_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7241728)))];
174
+ tensor<fp16, [1024, 640]> concat_37_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7243840))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7899264))), name = tensor<string, []>("concat_37_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
175
+ tensor<fp16, [1024, 256]> concat_38_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7899840))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8162048))), name = tensor<string, []>("concat_38_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
176
+ tensor<fp16, [1024]> add_7_to_fp16 = const()[name = tensor<string, []>("add_7_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8162624)))];
177
+ tensor<fp16, [1, ?, ?]> d = transpose(perm = input_19_perm_0, x = x_65_cast_fp16)[name = tensor<string, []>("transpose_29")];
178
+ tensor<fp16, [?, 1, ?]> input_19_batch_first_transpose_cast_fp16 = transpose(perm = input_19_batch_first_transpose_perm_0, x = d)[name = tensor<string, []>("transpose_28")];
179
+ tensor<fp16, [?, 1, 512]> input_21_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> input_21_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> input_21_batch_first_cast_fp16_2 = lstm(activation = input_21_batch_first_activation_0, bias = add_6_to_fp16, bias_back = add_7_to_fp16, cell_activation = input_21_batch_first_cell_activation_0, direction = input_21_batch_first_direction_0, initial_c = input_3_batch_first_lstm_h0_reshaped_to_fp16, initial_h = input_3_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = input_21_batch_first_output_sequence_0, recurrent_activation = input_21_batch_first_recurrent_activation_0, weight_hh = concat_36_to_fp16_palettized, weight_hh_back = concat_38_to_fp16_palettized, weight_ih = concat_35_to_fp16_palettized, weight_ih_back = concat_37_to_fp16_palettized, x = input_19_batch_first_transpose_cast_fp16)[name = tensor<string, []>("input_21_batch_first_cast_fp16")];
180
+ tensor<int32, [3]> input_21_perm_0 = const()[name = tensor<string, []>("input_21_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
181
+ tensor<fp16, [50, 512]> duration_proj_linear_layer_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [25600]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8164736))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8190400))), name = tensor<string, []>("duration_proj_linear_layer_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([50, 512])];
182
+ tensor<fp16, [50]> duration_proj_linear_layer_bias_to_fp16 = const()[name = tensor<string, []>("duration_proj_linear_layer_bias_to_fp16"), val = tensor<fp16, [50]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8190976)))];
183
+ tensor<fp16, [1, ?, 512]> input_21_cast_fp16 = transpose(perm = input_21_perm_0, x = input_21_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_27")];
184
+ tensor<fp16, [1, ?, 50]> linear_4_cast_fp16 = linear(bias = duration_proj_linear_layer_bias_to_fp16, weight = duration_proj_linear_layer_weight_to_fp16_palettized, x = input_21_cast_fp16)[name = tensor<string, []>("linear_4_cast_fp16")];
185
+ tensor<fp16, [1, ?, 50]> var_248_cast_fp16 = sigmoid(x = linear_4_cast_fp16)[name = tensor<string, []>("op_248_cast_fp16")];
186
+ tensor<int32, [1]> var_253_axes_0 = const()[name = tensor<string, []>("op_253_axes_0"), val = tensor<int32, [1]>([-1])];
187
+ tensor<bool, []> var_253_keep_dims_0 = const()[name = tensor<string, []>("op_253_keep_dims_0"), val = tensor<bool, []>(false)];
188
+ tensor<fp16, [1, ?]> var_253_cast_fp16 = reduce_sum(axes = var_253_axes_0, keep_dims = var_253_keep_dims_0, x = var_248_cast_fp16)[name = tensor<string, []>("op_253_cast_fp16")];
189
+ tensor<fp16, [1, ?]> duration = real_div(x = var_253_cast_fp16, y = speed)[name = tensor<string, []>("op_254_cast_fp16")];
190
+ tensor<fp32, []> var_263 = const()[name = tensor<string, []>("op_263"), val = tensor<fp32, []>(0x1.99999ap-3)];
191
+ tensor<int32, []> x_67_axis_0 = const()[name = tensor<string, []>("x_67_axis_0"), val = tensor<int32, []>(0)];
192
+ tensor<int32, []> x_67_batch_dims_0 = const()[name = tensor<string, []>("x_67_batch_dims_0"), val = tensor<int32, []>(0)];
193
+ tensor<bool, []> x_67_validate_indices_0 = const()[name = tensor<string, []>("x_67_validate_indices_0"), val = tensor<bool, []>(false)];
194
+ tensor<fp16, [178, 512]> text_encoder_embedding_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [91136]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8191168))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8282368))), name = tensor<string, []>("text_encoder_embedding_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([178, 512])];
195
+ tensor<string, []> input_ids_to_uint16_dtype_0 = const()[name = tensor<string, []>("input_ids_to_uint16_dtype_0"), val = tensor<string, []>("uint16")];
196
+ tensor<uint16, [1, ?]> input_ids_to_uint16 = cast(dtype = input_ids_to_uint16_dtype_0, x = input_ids)[name = tensor<string, []>("cast_0")];
197
+ tensor<fp16, [1, ?, 512]> x_67_cast_fp16_cast_uint16 = gather(axis = x_67_axis_0, batch_dims = x_67_batch_dims_0, indices = input_ids_to_uint16, validate_indices = x_67_validate_indices_0, x = text_encoder_embedding_weight_to_fp16_palettized)[name = tensor<string, []>("x_67_cast_fp16_cast_uint16")];
198
+ tensor<int32, [3]> x_69_perm_0 = const()[name = tensor<string, []>("x_69_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
199
+ tensor<int32, [1]> m_unsq_axes_0 = const()[name = tensor<string, []>("m_unsq_axes_0"), val = tensor<int32, [1]>([1])];
200
+ tensor<bool, [1, 1, ?]> m_unsq = expand_dims(axes = m_unsq_axes_0, x = m)[name = tensor<string, []>("m_unsq")];
201
+ tensor<fp16, []> var_265_to_fp16 = const()[name = tensor<string, []>("op_265_to_fp16"), val = tensor<fp16, []>(0x0p+0)];
202
+ tensor<fp16, [1, 512, ?]> x_69_cast_fp16 = transpose(perm = x_69_perm_0, x = x_67_cast_fp16_cast_uint16)[name = tensor<string, []>("transpose_26")];
203
+ tensor<fp16, [1, 512, ?]> input_23_cast_fp16 = select(a = var_265_to_fp16, b = x_69_cast_fp16, cond = m_unsq)[name = tensor<string, []>("input_23_cast_fp16")];
204
+ tensor<string, []> x_71_pad_type_0 = const()[name = tensor<string, []>("x_71_pad_type_0"), val = tensor<string, []>("custom")];
205
+ tensor<int32, [2]> x_71_pad_0 = const()[name = tensor<string, []>("x_71_pad_0"), val = tensor<int32, [2]>([2, 2])];
206
+ tensor<int32, [1]> x_71_strides_0 = const()[name = tensor<string, []>("x_71_strides_0"), val = tensor<int32, [1]>([1])];
207
+ tensor<int32, [1]> x_71_dilations_0 = const()[name = tensor<string, []>("x_71_dilations_0"), val = tensor<int32, [1]>([1])];
208
+ tensor<int32, []> x_71_groups_0 = const()[name = tensor<string, []>("x_71_groups_0"), val = tensor<int32, []>(1)];
209
+ tensor<fp16, [512, 512, 5]> weight_3_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [1310720]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8282944))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9593728))), name = tensor<string, []>("weight_3_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 5])];
210
+ tensor<fp16, [512]> text_encoder_cnn_0_0_bias_to_fp16 = const()[name = tensor<string, []>("text_encoder_cnn_0_0_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9594304)))];
211
+ tensor<fp16, [1, 512, ?]> x_71_cast_fp16 = conv(bias = text_encoder_cnn_0_0_bias_to_fp16, dilations = x_71_dilations_0, groups = x_71_groups_0, pad = x_71_pad_0, pad_type = x_71_pad_type_0, strides = x_71_strides_0, weight = weight_3_to_fp16_palettized, x = input_23_cast_fp16)[name = tensor<string, []>("x_71_cast_fp16")];
212
+ tensor<int32, [3]> input_25_perm_0 = const()[name = tensor<string, []>("input_25_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
213
+ tensor<int32, [1]> x_73_axes_0 = const()[name = tensor<string, []>("x_73_axes_0"), val = tensor<int32, [1]>([-1])];
214
+ tensor<fp16, [512]> text_encoder_cnn_0_1_gamma_to_fp16 = const()[name = tensor<string, []>("text_encoder_cnn_0_1_gamma_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9595392)))];
215
+ tensor<fp16, [512]> text_encoder_cnn_0_1_beta_to_fp16 = const()[name = tensor<string, []>("text_encoder_cnn_0_1_beta_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9596480)))];
216
+ tensor<fp16, []> var_261_to_fp16 = const()[name = tensor<string, []>("op_261_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
217
+ tensor<fp16, [1, ?, 512]> input_25_cast_fp16 = transpose(perm = input_25_perm_0, x = x_71_cast_fp16)[name = tensor<string, []>("transpose_25")];
218
+ tensor<fp16, [1, ?, 512]> x_73_cast_fp16 = layer_norm(axes = x_73_axes_0, beta = text_encoder_cnn_0_1_beta_to_fp16, epsilon = var_261_to_fp16, gamma = text_encoder_cnn_0_1_gamma_to_fp16, x = input_25_cast_fp16)[name = tensor<string, []>("x_73_cast_fp16")];
219
+ tensor<int32, [3]> input_27_perm_0 = const()[name = tensor<string, []>("input_27_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
220
+ tensor<fp16, [1, 512, ?]> input_27_cast_fp16 = transpose(perm = input_27_perm_0, x = x_73_cast_fp16)[name = tensor<string, []>("transpose_24")];
221
+ tensor<fp16, [1, 512, ?]> x_75_cast_fp16 = leaky_relu(alpha = var_263, x = input_27_cast_fp16)[name = tensor<string, []>("x_75_cast_fp16")];
222
+ tensor<fp16, [1, 512, ?]> input_29_cast_fp16 = select(a = var_265_to_fp16, b = x_75_cast_fp16, cond = m_unsq)[name = tensor<string, []>("input_29_cast_fp16")];
223
+ tensor<string, []> x_77_pad_type_0 = const()[name = tensor<string, []>("x_77_pad_type_0"), val = tensor<string, []>("custom")];
224
+ tensor<int32, [2]> x_77_pad_0 = const()[name = tensor<string, []>("x_77_pad_0"), val = tensor<int32, [2]>([2, 2])];
225
+ tensor<int32, [1]> x_77_strides_0 = const()[name = tensor<string, []>("x_77_strides_0"), val = tensor<int32, [1]>([1])];
226
+ tensor<int32, [1]> x_77_dilations_0 = const()[name = tensor<string, []>("x_77_dilations_0"), val = tensor<int32, [1]>([1])];
227
+ tensor<int32, []> x_77_groups_0 = const()[name = tensor<string, []>("x_77_groups_0"), val = tensor<int32, []>(1)];
228
+ tensor<fp16, [512, 512, 5]> weight_7_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [1310720]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9597568))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10908352))), name = tensor<string, []>("weight_7_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 5])];
229
+ tensor<fp16, [512]> text_encoder_cnn_1_0_bias_to_fp16 = const()[name = tensor<string, []>("text_encoder_cnn_1_0_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10908928)))];
230
+ tensor<fp16, [1, 512, ?]> x_77_cast_fp16 = conv(bias = text_encoder_cnn_1_0_bias_to_fp16, dilations = x_77_dilations_0, groups = x_77_groups_0, pad = x_77_pad_0, pad_type = x_77_pad_type_0, strides = x_77_strides_0, weight = weight_7_to_fp16_palettized, x = input_29_cast_fp16)[name = tensor<string, []>("x_77_cast_fp16")];
231
+ tensor<int32, [3]> input_31_perm_0 = const()[name = tensor<string, []>("input_31_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
232
+ tensor<int32, [1]> x_79_axes_0 = const()[name = tensor<string, []>("x_79_axes_0"), val = tensor<int32, [1]>([-1])];
233
+ tensor<fp16, [512]> text_encoder_cnn_1_1_gamma_to_fp16 = const()[name = tensor<string, []>("text_encoder_cnn_1_1_gamma_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10910016)))];
234
+ tensor<fp16, [512]> text_encoder_cnn_1_1_beta_to_fp16 = const()[name = tensor<string, []>("text_encoder_cnn_1_1_beta_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10911104)))];
235
+ tensor<fp16, [1, ?, 512]> input_31_cast_fp16 = transpose(perm = input_31_perm_0, x = x_77_cast_fp16)[name = tensor<string, []>("transpose_23")];
236
+ tensor<fp16, [1, ?, 512]> x_79_cast_fp16 = layer_norm(axes = x_79_axes_0, beta = text_encoder_cnn_1_1_beta_to_fp16, epsilon = var_261_to_fp16, gamma = text_encoder_cnn_1_1_gamma_to_fp16, x = input_31_cast_fp16)[name = tensor<string, []>("x_79_cast_fp16")];
237
+ tensor<int32, [3]> input_33_perm_0 = const()[name = tensor<string, []>("input_33_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
238
+ tensor<fp16, [1, 512, ?]> input_33_cast_fp16 = transpose(perm = input_33_perm_0, x = x_79_cast_fp16)[name = tensor<string, []>("transpose_22")];
239
+ tensor<fp16, [1, 512, ?]> x_81_cast_fp16 = leaky_relu(alpha = var_263, x = input_33_cast_fp16)[name = tensor<string, []>("x_81_cast_fp16")];
240
+ tensor<fp16, [1, 512, ?]> input_35_cast_fp16 = select(a = var_265_to_fp16, b = x_81_cast_fp16, cond = m_unsq)[name = tensor<string, []>("input_35_cast_fp16")];
241
+ tensor<string, []> x_83_pad_type_0 = const()[name = tensor<string, []>("x_83_pad_type_0"), val = tensor<string, []>("custom")];
242
+ tensor<int32, [2]> x_83_pad_0 = const()[name = tensor<string, []>("x_83_pad_0"), val = tensor<int32, [2]>([2, 2])];
243
+ tensor<int32, [1]> x_83_strides_0 = const()[name = tensor<string, []>("x_83_strides_0"), val = tensor<int32, [1]>([1])];
244
+ tensor<int32, [1]> x_83_dilations_0 = const()[name = tensor<string, []>("x_83_dilations_0"), val = tensor<int32, [1]>([1])];
245
+ tensor<int32, []> x_83_groups_0 = const()[name = tensor<string, []>("x_83_groups_0"), val = tensor<int32, []>(1)];
246
+ tensor<fp16, [512, 512, 5]> weight_11_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [1310720]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10912192))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12222976))), name = tensor<string, []>("weight_11_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 5])];
247
+ tensor<fp16, [512]> text_encoder_cnn_2_0_bias_to_fp16 = const()[name = tensor<string, []>("text_encoder_cnn_2_0_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12223552)))];
248
+ tensor<fp16, [1, 512, ?]> x_83_cast_fp16 = conv(bias = text_encoder_cnn_2_0_bias_to_fp16, dilations = x_83_dilations_0, groups = x_83_groups_0, pad = x_83_pad_0, pad_type = x_83_pad_type_0, strides = x_83_strides_0, weight = weight_11_to_fp16_palettized, x = input_35_cast_fp16)[name = tensor<string, []>("x_83_cast_fp16")];
249
+ tensor<int32, [3]> input_37_perm_0 = const()[name = tensor<string, []>("input_37_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
250
+ tensor<int32, [1]> x_85_axes_0 = const()[name = tensor<string, []>("x_85_axes_0"), val = tensor<int32, [1]>([-1])];
251
+ tensor<fp16, [512]> text_encoder_cnn_2_1_gamma_to_fp16 = const()[name = tensor<string, []>("text_encoder_cnn_2_1_gamma_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12224640)))];
252
+ tensor<fp16, [512]> text_encoder_cnn_2_1_beta_to_fp16 = const()[name = tensor<string, []>("text_encoder_cnn_2_1_beta_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12225728)))];
253
+ tensor<fp16, [1, ?, 512]> input_37_cast_fp16 = transpose(perm = input_37_perm_0, x = x_83_cast_fp16)[name = tensor<string, []>("transpose_21")];
254
+ tensor<fp16, [1, ?, 512]> x_85_cast_fp16 = layer_norm(axes = x_85_axes_0, beta = text_encoder_cnn_2_1_beta_to_fp16, epsilon = var_261_to_fp16, gamma = text_encoder_cnn_2_1_gamma_to_fp16, x = input_37_cast_fp16)[name = tensor<string, []>("x_85_cast_fp16")];
255
+ tensor<int32, [3]> input_39_perm_0 = const()[name = tensor<string, []>("input_39_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
256
+ tensor<fp16, [1, 512, ?]> input_39_cast_fp16 = transpose(perm = input_39_perm_0, x = x_85_cast_fp16)[name = tensor<string, []>("transpose_20")];
257
+ tensor<fp16, [1, 512, ?]> x_87_cast_fp16 = leaky_relu(alpha = var_263, x = input_39_cast_fp16)[name = tensor<string, []>("x_87_cast_fp16")];
258
+ tensor<fp16, [1, 512, ?]> x_89_cast_fp16 = select(a = var_265_to_fp16, b = x_87_cast_fp16, cond = m_unsq)[name = tensor<string, []>("x_89_cast_fp16")];
259
+ tensor<int32, [3]> transpose_14_perm_0 = const()[name = tensor<string, []>("transpose_14_perm_0"), val = tensor<int32, [3]>([2, 0, 1])];
260
+ tensor<string, []> x_91_batch_first_direction_0 = const()[name = tensor<string, []>("x_91_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
261
+ tensor<bool, []> x_91_batch_first_output_sequence_0 = const()[name = tensor<string, []>("x_91_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
262
+ tensor<string, []> x_91_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("x_91_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
263
+ tensor<string, []> x_91_batch_first_cell_activation_0 = const()[name = tensor<string, []>("x_91_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
264
+ tensor<string, []> x_91_batch_first_activation_0 = const()[name = tensor<string, []>("x_91_batch_first_activation_0"), val = tensor<string, []>("tanh")];
265
+ tensor<fp16, [1024, 512]> concat_45_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12226816))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12751168))), name = tensor<string, []>("concat_45_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 512])];
266
+ tensor<fp16, [1024, 256]> concat_46_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12751744))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13013952))), name = tensor<string, []>("concat_46_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
267
+ tensor<fp16, [1024]> add_8_to_fp16 = const()[name = tensor<string, []>("add_8_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13014528)))];
268
+ tensor<fp16, [1024, 512]> concat_47_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13016640))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13540992))), name = tensor<string, []>("concat_47_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 512])];
269
+ tensor<fp16, [1024, 256]> concat_48_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13541568))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13803776))), name = tensor<string, []>("concat_48_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
270
+ tensor<fp16, [1024]> add_9_to_fp16 = const()[name = tensor<string, []>("add_9_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13804352)))];
271
+ tensor<fp16, [?, 1, 512]> transpose_14_cast_fp16 = transpose(perm = transpose_14_perm_0, x = x_89_cast_fp16)[name = tensor<string, []>("transpose_19")];
272
+ tensor<fp16, [?, 1, 512]> x_91_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> x_91_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> x_91_batch_first_cast_fp16_2 = lstm(activation = x_91_batch_first_activation_0, bias = add_8_to_fp16, bias_back = add_9_to_fp16, cell_activation = x_91_batch_first_cell_activation_0, direction = x_91_batch_first_direction_0, initial_c = input_3_batch_first_lstm_h0_reshaped_to_fp16, initial_h = input_3_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = x_91_batch_first_output_sequence_0, recurrent_activation = x_91_batch_first_recurrent_activation_0, weight_hh = concat_46_to_fp16_palettized, weight_hh_back = concat_48_to_fp16_palettized, weight_ih = concat_45_to_fp16_palettized, weight_ih_back = concat_47_to_fp16_palettized, x = transpose_14_cast_fp16)[name = tensor<string, []>("x_91_batch_first_cast_fp16")];
273
+ tensor<int32, [3]> transpose_15_perm_0 = const()[name = tensor<string, []>("transpose_15_perm_0"), val = tensor<int32, [3]>([1, 2, 0])];
274
+ tensor<fp16, [1, 512, ?]> transpose_15_cast_fp16 = transpose(perm = transpose_15_perm_0, x = x_91_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_18")];
275
+ tensor<fp16, [1, 512, ?]> t_en = select(a = var_265_to_fp16, b = transpose_15_cast_fp16, cond = m_unsq)[name = tensor<string, []>("t_en_cast_fp16")];
276
+ } -> (duration, d, t_en);
277
+ }
ANE/ANE-zh/KokoroPostAlbert.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bca73c9f402ccd18052fe8cc418741caa196c3caceeb3637187cdc5d67d8daf7
3
+ size 13806464
ANE/ANE-zh/KokoroPostAlbert.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5669010b4feed54b3e9431ae12983264a3e09be548ce6d08c3e02bbb5c1f6a0
3
+ size 49183
ANE/ANE-zh/KokoroPostAlbert.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bca73c9f402ccd18052fe8cc418741caa196c3caceeb3637187cdc5d67d8daf7
3
+ size 13806464
ANE/ANE-zh/KokoroPostAlbert.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "07F36BCD-E5BB-4939-9D39-45F59CBD8058": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Weights",
7
+ "name": "weights",
8
+ "path": "com.apple.CoreML/weights"
9
+ },
10
+ "DB17D18A-94F8-4785-9B53-34E1386398F3": {
11
+ "author": "com.apple.CoreML",
12
+ "description": "CoreML Model Specification",
13
+ "name": "model.mlmodel",
14
+ "path": "com.apple.CoreML/model.mlmodel"
15
+ }
16
+ },
17
+ "rootModelIdentifier": "DB17D18A-94F8-4785-9B53-34E1386398F3"
18
+ }
ANE/ANE-zh/KokoroProsody.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c90a008bfe021f8c24f39d26706a894cffb3cc6d5dca1ad131f54547f0554a1
3
+ size 243
ANE/ANE-zh/KokoroProsody.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f1932bc07f52171e0fff898ae0e9c289a8c73aaf0a2ac06b2352b8ef58afc61
3
+ size 422
ANE/ANE-zh/KokoroProsody.mlmodelc/metadata.json ADDED
@@ -0,0 +1,98 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "metadataOutputVersion" : "3.0",
4
+ "storagePrecision" : "Mixed (Float16, Palettized (8 bits))",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float16",
10
+ "formattedType" : "MultiArray (Float16)",
11
+ "shortDescription" : "",
12
+ "shape" : "[]",
13
+ "name" : "F0",
14
+ "type" : "MultiArray"
15
+ },
16
+ {
17
+ "hasShapeFlexibility" : "0",
18
+ "isOptional" : "0",
19
+ "dataType" : "Float16",
20
+ "formattedType" : "MultiArray (Float16)",
21
+ "shortDescription" : "",
22
+ "shape" : "[]",
23
+ "name" : "N",
24
+ "type" : "MultiArray"
25
+ }
26
+ ],
27
+ "modelParameters" : [
28
+
29
+ ],
30
+ "specificationVersion" : 8,
31
+ "mlProgramOperationTypeHistogram" : {
32
+ "Ios17.mul" : 18,
33
+ "Ios17.linear" : 12,
34
+ "Ios17.transpose" : 2,
35
+ "Ios16.constexprLutToDense" : 30,
36
+ "Ios17.conv" : 16,
37
+ "Ios17.leakyRelu" : 12,
38
+ "Ios17.add" : 30,
39
+ "Ios17.convTranspose" : 2,
40
+ "Ios17.lstm" : 1,
41
+ "Ios17.sliceByIndex" : 2,
42
+ "UpsampleNearestNeighbor" : 2,
43
+ "Ios17.expandDims" : 2,
44
+ "Ios17.instanceNorm" : 11,
45
+ "Ios17.reshape" : 12,
46
+ "Split" : 12,
47
+ "Ios17.squeeze" : 4
48
+ },
49
+ "computePrecision" : "Mixed (Float16, Float32, Int32)",
50
+ "isUpdatable" : "0",
51
+ "stateSchema" : [
52
+
53
+ ],
54
+ "availability" : {
55
+ "macOS" : "14.0",
56
+ "tvOS" : "17.0",
57
+ "visionOS" : "1.0",
58
+ "watchOS" : "10.0",
59
+ "iOS" : "17.0",
60
+ "macCatalyst" : "17.0"
61
+ },
62
+ "modelType" : {
63
+ "name" : "MLModelType_mlProgram"
64
+ },
65
+ "userDefinedMetadata" : {
66
+ "com.github.apple.coremltools.conversion_date" : "2026-05-03",
67
+ "com.github.apple.coremltools.source" : "torch==2.11.0",
68
+ "com.github.apple.coremltools.version" : "9.0",
69
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
70
+ },
71
+ "inputSchema" : [
72
+ {
73
+ "dataType" : "Float16",
74
+ "hasShapeFlexibility" : "1",
75
+ "isOptional" : "0",
76
+ "shapeFlexibility" : "1 × 640 × 1...2000",
77
+ "shapeRange" : "[[1, 1], [640, 640], [1, 2000]]",
78
+ "formattedType" : "MultiArray (Float16 1 × 640 × 133)",
79
+ "type" : "MultiArray",
80
+ "shape" : "[1, 640, 133]",
81
+ "name" : "en",
82
+ "shortDescription" : ""
83
+ },
84
+ {
85
+ "hasShapeFlexibility" : "0",
86
+ "isOptional" : "0",
87
+ "dataType" : "Float16",
88
+ "formattedType" : "MultiArray (Float16 1 × 128)",
89
+ "shortDescription" : "",
90
+ "shape" : "[1, 128]",
91
+ "name" : "style_s",
92
+ "type" : "MultiArray"
93
+ }
94
+ ],
95
+ "generatedClassName" : "KokoroProsody",
96
+ "method" : "predict"
97
+ }
98
+ ]
ANE/ANE-zh/KokoroProsody.mlmodelc/model.mil ADDED
@@ -0,0 +1,394 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ program(1.0)
2
+ [buildInfo = dict<tensor<string, []>, tensor<string, []>>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}})]
3
+ {
4
+ func main<ios17>(tensor<fp16, [1, 640, ?]> en, tensor<fp16, [1, 128]> style_s) [FlexibleShapeInformation = tuple<tuple<tensor<string, []>, dict<tensor<string, []>, tensor<int32, [?]>>>, tuple<tensor<string, []>, dict<tensor<string, []>, list<tensor<int32, [2]>, ?>>>>((("DefaultShapes", {{"en", [1, 640, 133]}}), ("RangeDims", {{"en", [[1, 1], [640, 640], [1, 2000]]}})))] {
5
+ tensor<int32, [3]> transpose_0_perm_0 = const()[name = tensor<string, []>("transpose_0_perm_0"), val = tensor<int32, [3]>([-1, 0, -2])];
6
+ tensor<string, []> x_batch_first_direction_0 = const()[name = tensor<string, []>("x_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
7
+ tensor<bool, []> x_batch_first_output_sequence_0 = const()[name = tensor<string, []>("x_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
8
+ tensor<string, []> x_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("x_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
9
+ tensor<string, []> x_batch_first_cell_activation_0 = const()[name = tensor<string, []>("x_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
10
+ tensor<string, []> x_batch_first_activation_0 = const()[name = tensor<string, []>("x_batch_first_activation_0"), val = tensor<string, []>("tanh")];
11
+ tensor<fp16, [1, 512]> x_batch_first_lstm_h0_reshaped_to_fp16 = const()[name = tensor<string, []>("x_batch_first_lstm_h0_reshaped_to_fp16"), val = tensor<fp16, [1, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64)))];
12
+ tensor<fp16, [1024, 640]> concat_4_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1152))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(656576))), name = tensor<string, []>("concat_4_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
13
+ tensor<fp16, [1024, 256]> concat_5_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(657152))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(919360))), name = tensor<string, []>("concat_5_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
14
+ tensor<fp16, [1024]> add_0_to_fp16 = const()[name = tensor<string, []>("add_0_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(919936)))];
15
+ tensor<fp16, [1024, 640]> concat_6_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(922048))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1577472))), name = tensor<string, []>("concat_6_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
16
+ tensor<fp16, [1024, 256]> concat_7_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1578048))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1840256))), name = tensor<string, []>("concat_7_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
17
+ tensor<fp16, [1024]> add_1_to_fp16 = const()[name = tensor<string, []>("add_1_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1840832)))];
18
+ tensor<fp16, [?, 1, 640]> transpose_0_cast_fp16 = transpose(perm = transpose_0_perm_0, x = en)[name = tensor<string, []>("transpose_2")];
19
+ tensor<fp16, [?, 1, 512]> x_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> x_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> x_batch_first_cast_fp16_2 = lstm(activation = x_batch_first_activation_0, bias = add_0_to_fp16, bias_back = add_1_to_fp16, cell_activation = x_batch_first_cell_activation_0, direction = x_batch_first_direction_0, initial_c = x_batch_first_lstm_h0_reshaped_to_fp16, initial_h = x_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = x_batch_first_output_sequence_0, recurrent_activation = x_batch_first_recurrent_activation_0, weight_hh = concat_5_to_fp16_palettized, weight_hh_back = concat_7_to_fp16_palettized, weight_ih = concat_4_to_fp16_palettized, weight_ih_back = concat_6_to_fp16_palettized, x = transpose_0_cast_fp16)[name = tensor<string, []>("x_batch_first_cast_fp16")];
20
+ tensor<fp32, []> var_53 = const()[name = tensor<string, []>("op_53"), val = tensor<fp32, []>(0x1.99999ap-3)];
21
+ tensor<fp16, [1024, 128]> F0_0_norm1_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1842944))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1974080))), name = tensor<string, []>("F0_0_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
22
+ tensor<fp16, [1024]> F0_0_norm1_fc_bias_to_fp16 = const()[name = tensor<string, []>("F0_0_norm1_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1974656)))];
23
+ tensor<fp16, [1, 1024]> linear_0_cast_fp16 = linear(bias = F0_0_norm1_fc_bias_to_fp16, weight = F0_0_norm1_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_0_cast_fp16")];
24
+ tensor<int32, [3]> var_79 = const()[name = tensor<string, []>("op_79"), val = tensor<int32, [3]>([1, 1024, 1])];
25
+ tensor<fp16, [1, 1024, 1]> h_3_cast_fp16 = reshape(shape = var_79, x = linear_0_cast_fp16)[name = tensor<string, []>("h_3_cast_fp16")];
26
+ tensor<int32, [2]> var_81_split_sizes_0 = const()[name = tensor<string, []>("op_81_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
27
+ tensor<int32, []> var_81_axis_0 = const()[name = tensor<string, []>("op_81_axis_0"), val = tensor<int32, []>(1)];
28
+ tensor<fp16, [1, 512, 1]> var_81_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_81_cast_fp16_1 = split(axis = var_81_axis_0, split_sizes = var_81_split_sizes_0, x = h_3_cast_fp16)[name = tensor<string, []>("op_81_cast_fp16")];
29
+ tensor<fp16, []> var_83_promoted_to_fp16 = const()[name = tensor<string, []>("op_83_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
30
+ tensor<fp16, [1, 512, 1]> var_84_cast_fp16 = add(x = var_81_cast_fp16_0, y = var_83_promoted_to_fp16)[name = tensor<string, []>("op_84_cast_fp16")];
31
+ tensor<fp16, [512]> F0_0_norm1_norm_weight_to_fp16 = const()[name = tensor<string, []>("F0_0_norm1_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1976768)))];
32
+ tensor<fp16, [512]> F0_0_norm1_norm_bias_to_fp16 = const()[name = tensor<string, []>("F0_0_norm1_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1977856)))];
33
+ tensor<fp16, []> var_56_to_fp16 = const()[name = tensor<string, []>("op_56_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
34
+ tensor<int32, [3]> transpose_0_perm_0_1 = const()[name = tensor<string, []>("transpose_0_perm_0_1"), val = tensor<int32, [3]>([1, 2, 0])];
35
+ tensor<fp16, [1, 512, ?]> transpose_0 = transpose(perm = transpose_0_perm_0_1, x = x_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_1")];
36
+ tensor<fp16, [1, 512, ?]> var_87_cast_fp16 = instance_norm(beta = F0_0_norm1_norm_bias_to_fp16, epsilon = var_56_to_fp16, gamma = F0_0_norm1_norm_weight_to_fp16, x = transpose_0)[name = tensor<string, []>("op_87_cast_fp16")];
37
+ tensor<fp16, [1, 512, ?]> var_88_cast_fp16 = mul(x = var_84_cast_fp16, y = var_87_cast_fp16)[name = tensor<string, []>("op_88_cast_fp16")];
38
+ tensor<fp16, [1, 512, ?]> input_5_cast_fp16 = add(x = var_88_cast_fp16, y = var_81_cast_fp16_1)[name = tensor<string, []>("input_5_cast_fp16")];
39
+ tensor<fp16, [1, 512, ?]> input_7_cast_fp16 = leaky_relu(alpha = var_53, x = input_5_cast_fp16)[name = tensor<string, []>("input_7_cast_fp16")];
40
+ tensor<string, []> input_9_pad_type_0 = const()[name = tensor<string, []>("input_9_pad_type_0"), val = tensor<string, []>("custom")];
41
+ tensor<int32, [2]> input_9_pad_0 = const()[name = tensor<string, []>("input_9_pad_0"), val = tensor<int32, [2]>([1, 1])];
42
+ tensor<int32, [1]> input_9_strides_0 = const()[name = tensor<string, []>("input_9_strides_0"), val = tensor<int32, [1]>([1])];
43
+ tensor<int32, [1]> input_9_dilations_0 = const()[name = tensor<string, []>("input_9_dilations_0"), val = tensor<int32, [1]>([1])];
44
+ tensor<int32, []> input_9_groups_0 = const()[name = tensor<string, []>("input_9_groups_0"), val = tensor<int32, []>(1)];
45
+ tensor<fp16, [512, 512, 3]> weight_3_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [786432]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1978944))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2765440))), name = tensor<string, []>("weight_3_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 3])];
46
+ tensor<fp16, [512]> F0_0_conv1_bias_to_fp16 = const()[name = tensor<string, []>("F0_0_conv1_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2766016)))];
47
+ tensor<fp16, [1, 512, ?]> input_9_cast_fp16 = conv(bias = F0_0_conv1_bias_to_fp16, dilations = input_9_dilations_0, groups = input_9_groups_0, pad = input_9_pad_0, pad_type = input_9_pad_type_0, strides = input_9_strides_0, weight = weight_3_to_fp16_palettized, x = input_7_cast_fp16)[name = tensor<string, []>("input_9_cast_fp16")];
48
+ tensor<fp16, [1024, 128]> F0_0_norm2_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2767104))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2898240))), name = tensor<string, []>("F0_0_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
49
+ tensor<fp16, [1024]> F0_0_norm2_fc_bias_to_fp16 = const()[name = tensor<string, []>("F0_0_norm2_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2898816)))];
50
+ tensor<fp16, [1, 1024]> linear_1_cast_fp16 = linear(bias = F0_0_norm2_fc_bias_to_fp16, weight = F0_0_norm2_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_1_cast_fp16")];
51
+ tensor<int32, [3]> var_105 = const()[name = tensor<string, []>("op_105"), val = tensor<int32, [3]>([1, 1024, 1])];
52
+ tensor<fp16, [1, 1024, 1]> h_7_cast_fp16 = reshape(shape = var_105, x = linear_1_cast_fp16)[name = tensor<string, []>("h_7_cast_fp16")];
53
+ tensor<int32, [2]> var_107_split_sizes_0 = const()[name = tensor<string, []>("op_107_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
54
+ tensor<int32, []> var_107_axis_0 = const()[name = tensor<string, []>("op_107_axis_0"), val = tensor<int32, []>(1)];
55
+ tensor<fp16, [1, 512, 1]> var_107_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_107_cast_fp16_1 = split(axis = var_107_axis_0, split_sizes = var_107_split_sizes_0, x = h_7_cast_fp16)[name = tensor<string, []>("op_107_cast_fp16")];
56
+ tensor<fp16, []> var_109_promoted_to_fp16 = const()[name = tensor<string, []>("op_109_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
57
+ tensor<fp16, [1, 512, 1]> var_110_cast_fp16 = add(x = var_107_cast_fp16_0, y = var_109_promoted_to_fp16)[name = tensor<string, []>("op_110_cast_fp16")];
58
+ tensor<fp16, [1, 512, ?]> var_113_cast_fp16 = instance_norm(beta = F0_0_norm1_norm_bias_to_fp16, epsilon = var_56_to_fp16, gamma = F0_0_norm1_norm_weight_to_fp16, x = input_9_cast_fp16)[name = tensor<string, []>("op_113_cast_fp16")];
59
+ tensor<fp16, [1, 512, ?]> var_114_cast_fp16 = mul(x = var_110_cast_fp16, y = var_113_cast_fp16)[name = tensor<string, []>("op_114_cast_fp16")];
60
+ tensor<fp16, [1, 512, ?]> input_11_cast_fp16 = add(x = var_114_cast_fp16, y = var_107_cast_fp16_1)[name = tensor<string, []>("input_11_cast_fp16")];
61
+ tensor<fp16, [1, 512, ?]> input_13_cast_fp16 = leaky_relu(alpha = var_53, x = input_11_cast_fp16)[name = tensor<string, []>("input_13_cast_fp16")];
62
+ tensor<string, []> out_1_pad_type_0 = const()[name = tensor<string, []>("out_1_pad_type_0"), val = tensor<string, []>("custom")];
63
+ tensor<int32, [2]> out_1_pad_0 = const()[name = tensor<string, []>("out_1_pad_0"), val = tensor<int32, [2]>([1, 1])];
64
+ tensor<int32, [1]> out_1_strides_0 = const()[name = tensor<string, []>("out_1_strides_0"), val = tensor<int32, [1]>([1])];
65
+ tensor<int32, [1]> out_1_dilations_0 = const()[name = tensor<string, []>("out_1_dilations_0"), val = tensor<int32, [1]>([1])];
66
+ tensor<int32, []> out_1_groups_0 = const()[name = tensor<string, []>("out_1_groups_0"), val = tensor<int32, []>(1)];
67
+ tensor<fp16, [512, 512, 3]> weight_7_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [786432]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2900928))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3687424))), name = tensor<string, []>("weight_7_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 3])];
68
+ tensor<fp16, [512]> F0_0_conv2_bias_to_fp16 = const()[name = tensor<string, []>("F0_0_conv2_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3688000)))];
69
+ tensor<fp16, [1, 512, ?]> out_1_cast_fp16 = conv(bias = F0_0_conv2_bias_to_fp16, dilations = out_1_dilations_0, groups = out_1_groups_0, pad = out_1_pad_0, pad_type = out_1_pad_type_0, strides = out_1_strides_0, weight = weight_7_to_fp16_palettized, x = input_13_cast_fp16)[name = tensor<string, []>("out_1_cast_fp16")];
70
+ tensor<fp16, [1, 512, ?]> var_124_cast_fp16 = add(x = out_1_cast_fp16, y = transpose_0)[name = tensor<string, []>("op_124_cast_fp16")];
71
+ tensor<fp16, []> var_125_to_fp16 = const()[name = tensor<string, []>("op_125_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
72
+ tensor<fp16, [1, 512, ?]> input_15_cast_fp16 = mul(x = var_124_cast_fp16, y = var_125_to_fp16)[name = tensor<string, []>("input_15_cast_fp16")];
73
+ tensor<fp32, []> var_130 = const()[name = tensor<string, []>("op_130"), val = tensor<fp32, []>(0x1.99999ap-3)];
74
+ tensor<fp16, [1024, 128]> F0_1_norm1_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3689088))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3820224))), name = tensor<string, []>("F0_1_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
75
+ tensor<fp16, [1024]> F0_1_norm1_fc_bias_to_fp16 = const()[name = tensor<string, []>("F0_1_norm1_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3820800)))];
76
+ tensor<fp16, [1, 1024]> linear_2_cast_fp16 = linear(bias = F0_1_norm1_fc_bias_to_fp16, weight = F0_1_norm1_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_2_cast_fp16")];
77
+ tensor<int32, [3]> var_166 = const()[name = tensor<string, []>("op_166"), val = tensor<int32, [3]>([1, 1024, 1])];
78
+ tensor<fp16, [1, 1024, 1]> h_11_cast_fp16 = reshape(shape = var_166, x = linear_2_cast_fp16)[name = tensor<string, []>("h_11_cast_fp16")];
79
+ tensor<int32, [2]> var_168_split_sizes_0 = const()[name = tensor<string, []>("op_168_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
80
+ tensor<int32, []> var_168_axis_0 = const()[name = tensor<string, []>("op_168_axis_0"), val = tensor<int32, []>(1)];
81
+ tensor<fp16, [1, 512, 1]> var_168_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_168_cast_fp16_1 = split(axis = var_168_axis_0, split_sizes = var_168_split_sizes_0, x = h_11_cast_fp16)[name = tensor<string, []>("op_168_cast_fp16")];
82
+ tensor<fp16, []> var_170_promoted_to_fp16 = const()[name = tensor<string, []>("op_170_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
83
+ tensor<fp16, [1, 512, 1]> var_171_cast_fp16 = add(x = var_168_cast_fp16_0, y = var_170_promoted_to_fp16)[name = tensor<string, []>("op_171_cast_fp16")];
84
+ tensor<fp16, []> var_134_to_fp16 = const()[name = tensor<string, []>("op_134_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
85
+ tensor<fp16, [1, 512, ?]> var_174_cast_fp16 = instance_norm(beta = F0_0_norm1_norm_bias_to_fp16, epsilon = var_134_to_fp16, gamma = F0_0_norm1_norm_weight_to_fp16, x = input_15_cast_fp16)[name = tensor<string, []>("op_174_cast_fp16")];
86
+ tensor<fp16, [1, 512, ?]> var_175_cast_fp16 = mul(x = var_171_cast_fp16, y = var_174_cast_fp16)[name = tensor<string, []>("op_175_cast_fp16")];
87
+ tensor<fp16, [1, 512, ?]> input_17_cast_fp16 = add(x = var_175_cast_fp16, y = var_168_cast_fp16_1)[name = tensor<string, []>("input_17_cast_fp16")];
88
+ tensor<fp16, [1, 512, ?]> input_19_cast_fp16 = leaky_relu(alpha = var_130, x = input_17_cast_fp16)[name = tensor<string, []>("input_19_cast_fp16")];
89
+ tensor<string, []> conv_transpose_0_pad_type_0 = const()[name = tensor<string, []>("conv_transpose_0_pad_type_0"), val = tensor<string, []>("custom")];
90
+ tensor<int32, [2]> conv_transpose_0_pad_0 = const()[name = tensor<string, []>("conv_transpose_0_pad_0"), val = tensor<int32, [2]>([0, 0])];
91
+ tensor<int32, [1]> conv_transpose_0_strides_0 = const()[name = tensor<string, []>("conv_transpose_0_strides_0"), val = tensor<int32, [1]>([2])];
92
+ tensor<int32, []> conv_transpose_0_groups_0 = const()[name = tensor<string, []>("conv_transpose_0_groups_0"), val = tensor<int32, []>(512)];
93
+ tensor<int32, [1]> conv_transpose_0_dilations_0 = const()[name = tensor<string, []>("conv_transpose_0_dilations_0"), val = tensor<int32, [1]>([1])];
94
+ tensor<fp16, [512, 1, 3]> var_178_to_fp16 = const()[name = tensor<string, []>("op_178_to_fp16"), val = tensor<fp16, [512, 1, 3]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3822912)))];
95
+ tensor<fp16, [512]> F0_1_pool_bias_to_fp16 = const()[name = tensor<string, []>("F0_1_pool_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3826048)))];
96
+ tensor<fp16, [1, 512, ?]> conv_transpose_0_cast_fp16 = conv_transpose(bias = F0_1_pool_bias_to_fp16, dilations = conv_transpose_0_dilations_0, groups = conv_transpose_0_groups_0, pad = conv_transpose_0_pad_0, pad_type = conv_transpose_0_pad_type_0, strides = conv_transpose_0_strides_0, weight = var_178_to_fp16, x = input_19_cast_fp16)[name = tensor<string, []>("conv_transpose_0_cast_fp16")];
97
+ tensor<int32, [3]> input_21_begin_0 = const()[name = tensor<string, []>("input_21_begin_0"), val = tensor<int32, [3]>([0, 0, 1])];
98
+ tensor<int32, [3]> input_21_end_0 = const()[name = tensor<string, []>("input_21_end_0"), val = tensor<int32, [3]>([0, 0, 0])];
99
+ tensor<bool, [3]> input_21_begin_mask_0 = const()[name = tensor<string, []>("input_21_begin_mask_0"), val = tensor<bool, [3]>([true, true, false])];
100
+ tensor<bool, [3]> input_21_end_mask_0 = const()[name = tensor<string, []>("input_21_end_mask_0"), val = tensor<bool, [3]>([true, true, true])];
101
+ tensor<fp16, [1, 512, ?]> input_21_cast_fp16 = slice_by_index(begin = input_21_begin_0, begin_mask = input_21_begin_mask_0, end = input_21_end_0, end_mask = input_21_end_mask_0, x = conv_transpose_0_cast_fp16)[name = tensor<string, []>("input_21_cast_fp16")];
102
+ tensor<string, []> input_23_pad_type_0 = const()[name = tensor<string, []>("input_23_pad_type_0"), val = tensor<string, []>("custom")];
103
+ tensor<int32, [2]> input_23_pad_0 = const()[name = tensor<string, []>("input_23_pad_0"), val = tensor<int32, [2]>([1, 1])];
104
+ tensor<int32, [1]> input_23_strides_0 = const()[name = tensor<string, []>("input_23_strides_0"), val = tensor<int32, [1]>([1])];
105
+ tensor<int32, [1]> input_23_dilations_0 = const()[name = tensor<string, []>("input_23_dilations_0"), val = tensor<int32, [1]>([1])];
106
+ tensor<int32, []> input_23_groups_0 = const()[name = tensor<string, []>("input_23_groups_0"), val = tensor<int32, []>(1)];
107
+ tensor<fp16, [256, 512, 3]> weight_11_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [393216]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3827136))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4220416))), name = tensor<string, []>("weight_11_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 512, 3])];
108
+ tensor<fp16, [256]> F0_1_conv1_bias_to_fp16 = const()[name = tensor<string, []>("F0_1_conv1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4220992)))];
109
+ tensor<fp16, [1, 256, ?]> input_23_cast_fp16 = conv(bias = F0_1_conv1_bias_to_fp16, dilations = input_23_dilations_0, groups = input_23_groups_0, pad = input_23_pad_0, pad_type = input_23_pad_type_0, strides = input_23_strides_0, weight = weight_11_to_fp16_palettized, x = input_21_cast_fp16)[name = tensor<string, []>("input_23_cast_fp16")];
110
+ tensor<fp16, [512, 128]> F0_1_norm2_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4221568))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4287168))), name = tensor<string, []>("F0_1_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
111
+ tensor<fp16, [512]> F0_1_norm2_fc_bias_to_fp16 = const()[name = tensor<string, []>("F0_1_norm2_fc_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4287744)))];
112
+ tensor<fp16, [1, 512]> linear_3_cast_fp16 = linear(bias = F0_1_norm2_fc_bias_to_fp16, weight = F0_1_norm2_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_3_cast_fp16")];
113
+ tensor<int32, [3]> var_199 = const()[name = tensor<string, []>("op_199"), val = tensor<int32, [3]>([1, 512, 1])];
114
+ tensor<fp16, [1, 512, 1]> h_15_cast_fp16 = reshape(shape = var_199, x = linear_3_cast_fp16)[name = tensor<string, []>("h_15_cast_fp16")];
115
+ tensor<int32, [2]> var_201_split_sizes_0 = const()[name = tensor<string, []>("op_201_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
116
+ tensor<int32, []> var_201_axis_0 = const()[name = tensor<string, []>("op_201_axis_0"), val = tensor<int32, []>(1)];
117
+ tensor<fp16, [1, 256, 1]> var_201_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_201_cast_fp16_1 = split(axis = var_201_axis_0, split_sizes = var_201_split_sizes_0, x = h_15_cast_fp16)[name = tensor<string, []>("op_201_cast_fp16")];
118
+ tensor<fp16, []> var_203_promoted_to_fp16 = const()[name = tensor<string, []>("op_203_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
119
+ tensor<fp16, [1, 256, 1]> var_204_cast_fp16 = add(x = var_201_cast_fp16_0, y = var_203_promoted_to_fp16)[name = tensor<string, []>("op_204_cast_fp16")];
120
+ tensor<fp16, [256]> F0_1_norm2_norm_weight_to_fp16 = const()[name = tensor<string, []>("F0_1_norm2_norm_weight_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4288832)))];
121
+ tensor<fp16, [256]> F0_1_norm2_norm_bias_to_fp16 = const()[name = tensor<string, []>("F0_1_norm2_norm_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4289408)))];
122
+ tensor<fp16, [1, 256, ?]> var_207_cast_fp16 = instance_norm(beta = F0_1_norm2_norm_bias_to_fp16, epsilon = var_134_to_fp16, gamma = F0_1_norm2_norm_weight_to_fp16, x = input_23_cast_fp16)[name = tensor<string, []>("op_207_cast_fp16")];
123
+ tensor<fp16, [1, 256, ?]> var_208_cast_fp16 = mul(x = var_204_cast_fp16, y = var_207_cast_fp16)[name = tensor<string, []>("op_208_cast_fp16")];
124
+ tensor<fp16, [1, 256, ?]> input_25_cast_fp16 = add(x = var_208_cast_fp16, y = var_201_cast_fp16_1)[name = tensor<string, []>("input_25_cast_fp16")];
125
+ tensor<fp16, [1, 256, ?]> input_27_cast_fp16 = leaky_relu(alpha = var_130, x = input_25_cast_fp16)[name = tensor<string, []>("input_27_cast_fp16")];
126
+ tensor<string, []> out_3_pad_type_0 = const()[name = tensor<string, []>("out_3_pad_type_0"), val = tensor<string, []>("custom")];
127
+ tensor<int32, [2]> out_3_pad_0 = const()[name = tensor<string, []>("out_3_pad_0"), val = tensor<int32, [2]>([1, 1])];
128
+ tensor<int32, [1]> out_3_strides_0 = const()[name = tensor<string, []>("out_3_strides_0"), val = tensor<int32, [1]>([1])];
129
+ tensor<int32, [1]> out_3_dilations_0 = const()[name = tensor<string, []>("out_3_dilations_0"), val = tensor<int32, [1]>([1])];
130
+ tensor<int32, []> out_3_groups_0 = const()[name = tensor<string, []>("out_3_groups_0"), val = tensor<int32, []>(1)];
131
+ tensor<fp16, [256, 256, 3]> weight_15_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4289984))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4486656))), name = tensor<string, []>("weight_15_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
132
+ tensor<fp16, [256]> F0_1_conv2_bias_to_fp16 = const()[name = tensor<string, []>("F0_1_conv2_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4487232)))];
133
+ tensor<fp16, [1, 256, ?]> out_3_cast_fp16 = conv(bias = F0_1_conv2_bias_to_fp16, dilations = out_3_dilations_0, groups = out_3_groups_0, pad = out_3_pad_0, pad_type = out_3_pad_type_0, strides = out_3_strides_0, weight = weight_15_to_fp16_palettized, x = input_27_cast_fp16)[name = tensor<string, []>("out_3_cast_fp16")];
134
+ tensor<int32, [1]> expand_dims_0_axes_0 = const()[name = tensor<string, []>("expand_dims_0_axes_0"), val = tensor<int32, [1]>([3])];
135
+ tensor<fp16, [1, 512, ?, 1]> expand_dims_0_cast_fp16 = expand_dims(axes = expand_dims_0_axes_0, x = input_15_cast_fp16)[name = tensor<string, []>("expand_dims_0_cast_fp16")];
136
+ tensor<int32, []> upsample_nearest_neighbor_0_scale_factor_height_0 = const()[name = tensor<string, []>("upsample_nearest_neighbor_0_scale_factor_height_0"), val = tensor<int32, []>(2)];
137
+ tensor<int32, []> upsample_nearest_neighbor_0_scale_factor_width_0 = const()[name = tensor<string, []>("upsample_nearest_neighbor_0_scale_factor_width_0"), val = tensor<int32, []>(1)];
138
+ tensor<fp16, [1, 512, ?, 1]> upsample_nearest_neighbor_0_cast_fp16 = upsample_nearest_neighbor(scale_factor_height = upsample_nearest_neighbor_0_scale_factor_height_0, scale_factor_width = upsample_nearest_neighbor_0_scale_factor_width_0, x = expand_dims_0_cast_fp16)[name = tensor<string, []>("upsample_nearest_neighbor_0_cast_fp16")];
139
+ tensor<int32, [1]> input_29_axes_0 = const()[name = tensor<string, []>("input_29_axes_0"), val = tensor<int32, [1]>([3])];
140
+ tensor<fp16, [1, 512, ?]> input_29_cast_fp16 = squeeze(axes = input_29_axes_0, x = upsample_nearest_neighbor_0_cast_fp16)[name = tensor<string, []>("input_29_cast_fp16")];
141
+ tensor<string, []> var_225_pad_type_0 = const()[name = tensor<string, []>("op_225_pad_type_0"), val = tensor<string, []>("valid")];
142
+ tensor<int32, [1]> var_225_strides_0 = const()[name = tensor<string, []>("op_225_strides_0"), val = tensor<int32, [1]>([1])];
143
+ tensor<int32, [2]> var_225_pad_0 = const()[name = tensor<string, []>("op_225_pad_0"), val = tensor<int32, [2]>([0, 0])];
144
+ tensor<int32, [1]> var_225_dilations_0 = const()[name = tensor<string, []>("op_225_dilations_0"), val = tensor<int32, [1]>([1])];
145
+ tensor<int32, []> var_225_groups_0 = const()[name = tensor<string, []>("op_225_groups_0"), val = tensor<int32, []>(1)];
146
+ tensor<fp16, [256, 512, 1]> weight_17_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4487808))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4618944))), name = tensor<string, []>("weight_17_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 512, 1])];
147
+ tensor<fp16, [1, 256, ?]> var_225_cast_fp16 = conv(dilations = var_225_dilations_0, groups = var_225_groups_0, pad = var_225_pad_0, pad_type = var_225_pad_type_0, strides = var_225_strides_0, weight = weight_17_to_fp16_palettized, x = input_29_cast_fp16)[name = tensor<string, []>("op_225_cast_fp16")];
148
+ tensor<fp16, [1, 256, ?]> var_226_cast_fp16 = add(x = out_3_cast_fp16, y = var_225_cast_fp16)[name = tensor<string, []>("op_226_cast_fp16")];
149
+ tensor<fp16, []> var_227_to_fp16 = const()[name = tensor<string, []>("op_227_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
150
+ tensor<fp16, [1, 256, ?]> input_31_cast_fp16 = mul(x = var_226_cast_fp16, y = var_227_to_fp16)[name = tensor<string, []>("input_31_cast_fp16")];
151
+ tensor<fp32, []> var_230 = const()[name = tensor<string, []>("op_230"), val = tensor<fp32, []>(0x1.99999ap-3)];
152
+ tensor<fp16, [512, 128]> F0_2_norm1_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4619520))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4685120))), name = tensor<string, []>("F0_2_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
153
+ tensor<fp16, [512]> F0_2_norm1_fc_bias_to_fp16 = const()[name = tensor<string, []>("F0_2_norm1_fc_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4685696)))];
154
+ tensor<fp16, [1, 512]> linear_4_cast_fp16 = linear(bias = F0_2_norm1_fc_bias_to_fp16, weight = F0_2_norm1_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_4_cast_fp16")];
155
+ tensor<int32, [3]> var_256 = const()[name = tensor<string, []>("op_256"), val = tensor<int32, [3]>([1, 512, 1])];
156
+ tensor<fp16, [1, 512, 1]> h_19_cast_fp16 = reshape(shape = var_256, x = linear_4_cast_fp16)[name = tensor<string, []>("h_19_cast_fp16")];
157
+ tensor<int32, [2]> var_258_split_sizes_0 = const()[name = tensor<string, []>("op_258_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
158
+ tensor<int32, []> var_258_axis_0 = const()[name = tensor<string, []>("op_258_axis_0"), val = tensor<int32, []>(1)];
159
+ tensor<fp16, [1, 256, 1]> var_258_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_258_cast_fp16_1 = split(axis = var_258_axis_0, split_sizes = var_258_split_sizes_0, x = h_19_cast_fp16)[name = tensor<string, []>("op_258_cast_fp16")];
160
+ tensor<fp16, []> var_260_promoted_to_fp16 = const()[name = tensor<string, []>("op_260_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
161
+ tensor<fp16, [1, 256, 1]> var_261_cast_fp16 = add(x = var_258_cast_fp16_0, y = var_260_promoted_to_fp16)[name = tensor<string, []>("op_261_cast_fp16")];
162
+ tensor<fp16, []> var_233_to_fp16 = const()[name = tensor<string, []>("op_233_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
163
+ tensor<fp16, [1, 256, ?]> var_264_cast_fp16 = instance_norm(beta = F0_1_norm2_norm_bias_to_fp16, epsilon = var_233_to_fp16, gamma = F0_1_norm2_norm_weight_to_fp16, x = input_31_cast_fp16)[name = tensor<string, []>("op_264_cast_fp16")];
164
+ tensor<fp16, [1, 256, ?]> var_265_cast_fp16 = mul(x = var_261_cast_fp16, y = var_264_cast_fp16)[name = tensor<string, []>("op_265_cast_fp16")];
165
+ tensor<fp16, [1, 256, ?]> input_33_cast_fp16 = add(x = var_265_cast_fp16, y = var_258_cast_fp16_1)[name = tensor<string, []>("input_33_cast_fp16")];
166
+ tensor<fp16, [1, 256, ?]> input_35_cast_fp16 = leaky_relu(alpha = var_230, x = input_33_cast_fp16)[name = tensor<string, []>("input_35_cast_fp16")];
167
+ tensor<string, []> input_37_pad_type_0 = const()[name = tensor<string, []>("input_37_pad_type_0"), val = tensor<string, []>("custom")];
168
+ tensor<int32, [2]> input_37_pad_0 = const()[name = tensor<string, []>("input_37_pad_0"), val = tensor<int32, [2]>([1, 1])];
169
+ tensor<int32, [1]> input_37_strides_0 = const()[name = tensor<string, []>("input_37_strides_0"), val = tensor<int32, [1]>([1])];
170
+ tensor<int32, [1]> input_37_dilations_0 = const()[name = tensor<string, []>("input_37_dilations_0"), val = tensor<int32, [1]>([1])];
171
+ tensor<int32, []> input_37_groups_0 = const()[name = tensor<string, []>("input_37_groups_0"), val = tensor<int32, []>(1)];
172
+ tensor<fp16, [256, 256, 3]> weight_21_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4686784))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4883456))), name = tensor<string, []>("weight_21_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
173
+ tensor<fp16, [256]> F0_2_conv1_bias_to_fp16 = const()[name = tensor<string, []>("F0_2_conv1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4884032)))];
174
+ tensor<fp16, [1, 256, ?]> input_37_cast_fp16 = conv(bias = F0_2_conv1_bias_to_fp16, dilations = input_37_dilations_0, groups = input_37_groups_0, pad = input_37_pad_0, pad_type = input_37_pad_type_0, strides = input_37_strides_0, weight = weight_21_to_fp16_palettized, x = input_35_cast_fp16)[name = tensor<string, []>("input_37_cast_fp16")];
175
+ tensor<fp16, [512, 128]> F0_2_norm2_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4884608))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4950208))), name = tensor<string, []>("F0_2_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
176
+ tensor<fp16, [512]> F0_2_norm2_fc_bias_to_fp16 = const()[name = tensor<string, []>("F0_2_norm2_fc_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4950784)))];
177
+ tensor<fp16, [1, 512]> linear_5_cast_fp16 = linear(bias = F0_2_norm2_fc_bias_to_fp16, weight = F0_2_norm2_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_5_cast_fp16")];
178
+ tensor<int32, [3]> var_282 = const()[name = tensor<string, []>("op_282"), val = tensor<int32, [3]>([1, 512, 1])];
179
+ tensor<fp16, [1, 512, 1]> h_23_cast_fp16 = reshape(shape = var_282, x = linear_5_cast_fp16)[name = tensor<string, []>("h_23_cast_fp16")];
180
+ tensor<int32, [2]> var_284_split_sizes_0 = const()[name = tensor<string, []>("op_284_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
181
+ tensor<int32, []> var_284_axis_0 = const()[name = tensor<string, []>("op_284_axis_0"), val = tensor<int32, []>(1)];
182
+ tensor<fp16, [1, 256, 1]> var_284_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_284_cast_fp16_1 = split(axis = var_284_axis_0, split_sizes = var_284_split_sizes_0, x = h_23_cast_fp16)[name = tensor<string, []>("op_284_cast_fp16")];
183
+ tensor<fp16, []> var_286_promoted_to_fp16 = const()[name = tensor<string, []>("op_286_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
184
+ tensor<fp16, [1, 256, 1]> var_287_cast_fp16 = add(x = var_284_cast_fp16_0, y = var_286_promoted_to_fp16)[name = tensor<string, []>("op_287_cast_fp16")];
185
+ tensor<fp16, [1, 256, ?]> var_290_cast_fp16 = instance_norm(beta = F0_1_norm2_norm_bias_to_fp16, epsilon = var_233_to_fp16, gamma = F0_1_norm2_norm_weight_to_fp16, x = input_37_cast_fp16)[name = tensor<string, []>("op_290_cast_fp16")];
186
+ tensor<fp16, [1, 256, ?]> var_291_cast_fp16 = mul(x = var_287_cast_fp16, y = var_290_cast_fp16)[name = tensor<string, []>("op_291_cast_fp16")];
187
+ tensor<fp16, [1, 256, ?]> input_39_cast_fp16 = add(x = var_291_cast_fp16, y = var_284_cast_fp16_1)[name = tensor<string, []>("input_39_cast_fp16")];
188
+ tensor<fp16, [1, 256, ?]> input_41_cast_fp16 = leaky_relu(alpha = var_230, x = input_39_cast_fp16)[name = tensor<string, []>("input_41_cast_fp16")];
189
+ tensor<string, []> out_5_pad_type_0 = const()[name = tensor<string, []>("out_5_pad_type_0"), val = tensor<string, []>("custom")];
190
+ tensor<int32, [2]> out_5_pad_0 = const()[name = tensor<string, []>("out_5_pad_0"), val = tensor<int32, [2]>([1, 1])];
191
+ tensor<int32, [1]> out_5_strides_0 = const()[name = tensor<string, []>("out_5_strides_0"), val = tensor<int32, [1]>([1])];
192
+ tensor<int32, [1]> out_5_dilations_0 = const()[name = tensor<string, []>("out_5_dilations_0"), val = tensor<int32, [1]>([1])];
193
+ tensor<int32, []> out_5_groups_0 = const()[name = tensor<string, []>("out_5_groups_0"), val = tensor<int32, []>(1)];
194
+ tensor<fp16, [256, 256, 3]> weight_25_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4951872))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5148544))), name = tensor<string, []>("weight_25_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
195
+ tensor<fp16, [256]> F0_2_conv2_bias_to_fp16 = const()[name = tensor<string, []>("F0_2_conv2_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5149120)))];
196
+ tensor<fp16, [1, 256, ?]> out_5_cast_fp16 = conv(bias = F0_2_conv2_bias_to_fp16, dilations = out_5_dilations_0, groups = out_5_groups_0, pad = out_5_pad_0, pad_type = out_5_pad_type_0, strides = out_5_strides_0, weight = weight_25_to_fp16_palettized, x = input_41_cast_fp16)[name = tensor<string, []>("out_5_cast_fp16")];
197
+ tensor<fp16, [1, 256, ?]> var_301_cast_fp16 = add(x = out_5_cast_fp16, y = input_31_cast_fp16)[name = tensor<string, []>("op_301_cast_fp16")];
198
+ tensor<fp16, []> var_302_to_fp16 = const()[name = tensor<string, []>("op_302_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
199
+ tensor<fp16, [1, 256, ?]> input_43_cast_fp16 = mul(x = var_301_cast_fp16, y = var_302_to_fp16)[name = tensor<string, []>("input_43_cast_fp16")];
200
+ tensor<string, []> var_314_pad_type_0 = const()[name = tensor<string, []>("op_314_pad_type_0"), val = tensor<string, []>("valid")];
201
+ tensor<int32, [1]> var_314_strides_0 = const()[name = tensor<string, []>("op_314_strides_0"), val = tensor<int32, [1]>([1])];
202
+ tensor<int32, [2]> var_314_pad_0 = const()[name = tensor<string, []>("op_314_pad_0"), val = tensor<int32, [2]>([0, 0])];
203
+ tensor<int32, [1]> var_314_dilations_0 = const()[name = tensor<string, []>("op_314_dilations_0"), val = tensor<int32, [1]>([1])];
204
+ tensor<int32, []> var_314_groups_0 = const()[name = tensor<string, []>("op_314_groups_0"), val = tensor<int32, []>(1)];
205
+ tensor<fp16, [1, 256, 1]> F0_proj_weight_to_fp16 = const()[name = tensor<string, []>("F0_proj_weight_to_fp16"), val = tensor<fp16, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5149696)))];
206
+ tensor<fp16, [1]> F0_proj_bias_to_fp16 = const()[name = tensor<string, []>("F0_proj_bias_to_fp16"), val = tensor<fp16, [1]>([0x1.f08p-3])];
207
+ tensor<fp16, [1, 1, ?]> var_314_cast_fp16 = conv(bias = F0_proj_bias_to_fp16, dilations = var_314_dilations_0, groups = var_314_groups_0, pad = var_314_pad_0, pad_type = var_314_pad_type_0, strides = var_314_strides_0, weight = F0_proj_weight_to_fp16, x = input_43_cast_fp16)[name = tensor<string, []>("op_314_cast_fp16")];
208
+ tensor<int32, [1]> var_316_axes_0 = const()[name = tensor<string, []>("op_316_axes_0"), val = tensor<int32, [1]>([1])];
209
+ tensor<fp16, [1, ?]> F0 = squeeze(axes = var_316_axes_0, x = var_314_cast_fp16)[name = tensor<string, []>("op_316_cast_fp16")];
210
+ tensor<fp32, []> var_321 = const()[name = tensor<string, []>("op_321"), val = tensor<fp32, []>(0x1.99999ap-3)];
211
+ tensor<fp16, [1024, 128]> N_0_norm1_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5150272))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5281408))), name = tensor<string, []>("N_0_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
212
+ tensor<fp16, [1024]> N_0_norm1_fc_bias_to_fp16 = const()[name = tensor<string, []>("N_0_norm1_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5281984)))];
213
+ tensor<fp16, [1, 1024]> linear_6_cast_fp16 = linear(bias = N_0_norm1_fc_bias_to_fp16, weight = N_0_norm1_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_6_cast_fp16")];
214
+ tensor<int32, [3]> var_347 = const()[name = tensor<string, []>("op_347"), val = tensor<int32, [3]>([1, 1024, 1])];
215
+ tensor<fp16, [1, 1024, 1]> h_27_cast_fp16 = reshape(shape = var_347, x = linear_6_cast_fp16)[name = tensor<string, []>("h_27_cast_fp16")];
216
+ tensor<int32, [2]> var_349_split_sizes_0 = const()[name = tensor<string, []>("op_349_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
217
+ tensor<int32, []> var_349_axis_0 = const()[name = tensor<string, []>("op_349_axis_0"), val = tensor<int32, []>(1)];
218
+ tensor<fp16, [1, 512, 1]> var_349_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_349_cast_fp16_1 = split(axis = var_349_axis_0, split_sizes = var_349_split_sizes_0, x = h_27_cast_fp16)[name = tensor<string, []>("op_349_cast_fp16")];
219
+ tensor<fp16, []> var_351_promoted_to_fp16 = const()[name = tensor<string, []>("op_351_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
220
+ tensor<fp16, [1, 512, 1]> var_352_cast_fp16 = add(x = var_349_cast_fp16_0, y = var_351_promoted_to_fp16)[name = tensor<string, []>("op_352_cast_fp16")];
221
+ tensor<fp16, [1, 512, ?]> var_356_cast_fp16 = mul(x = var_352_cast_fp16, y = var_87_cast_fp16)[name = tensor<string, []>("op_356_cast_fp16")];
222
+ tensor<fp16, [1, 512, ?]> input_47_cast_fp16 = add(x = var_356_cast_fp16, y = var_349_cast_fp16_1)[name = tensor<string, []>("input_47_cast_fp16")];
223
+ tensor<fp16, [1, 512, ?]> input_49_cast_fp16 = leaky_relu(alpha = var_321, x = input_47_cast_fp16)[name = tensor<string, []>("input_49_cast_fp16")];
224
+ tensor<string, []> input_51_pad_type_0 = const()[name = tensor<string, []>("input_51_pad_type_0"), val = tensor<string, []>("custom")];
225
+ tensor<int32, [2]> input_51_pad_0 = const()[name = tensor<string, []>("input_51_pad_0"), val = tensor<int32, [2]>([1, 1])];
226
+ tensor<int32, [1]> input_51_strides_0 = const()[name = tensor<string, []>("input_51_strides_0"), val = tensor<int32, [1]>([1])];
227
+ tensor<int32, [1]> input_51_dilations_0 = const()[name = tensor<string, []>("input_51_dilations_0"), val = tensor<int32, [1]>([1])];
228
+ tensor<int32, []> input_51_groups_0 = const()[name = tensor<string, []>("input_51_groups_0"), val = tensor<int32, []>(1)];
229
+ tensor<fp16, [512, 512, 3]> weight_31_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [786432]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5284096))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6070592))), name = tensor<string, []>("weight_31_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 3])];
230
+ tensor<fp16, [512]> N_0_conv1_bias_to_fp16 = const()[name = tensor<string, []>("N_0_conv1_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6071168)))];
231
+ tensor<fp16, [1, 512, ?]> input_51_cast_fp16 = conv(bias = N_0_conv1_bias_to_fp16, dilations = input_51_dilations_0, groups = input_51_groups_0, pad = input_51_pad_0, pad_type = input_51_pad_type_0, strides = input_51_strides_0, weight = weight_31_to_fp16_palettized, x = input_49_cast_fp16)[name = tensor<string, []>("input_51_cast_fp16")];
232
+ tensor<fp16, [1024, 128]> N_0_norm2_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6072256))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6203392))), name = tensor<string, []>("N_0_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
233
+ tensor<fp16, [1024]> N_0_norm2_fc_bias_to_fp16 = const()[name = tensor<string, []>("N_0_norm2_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6203968)))];
234
+ tensor<fp16, [1, 1024]> linear_7_cast_fp16 = linear(bias = N_0_norm2_fc_bias_to_fp16, weight = N_0_norm2_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_7_cast_fp16")];
235
+ tensor<int32, [3]> var_373 = const()[name = tensor<string, []>("op_373"), val = tensor<int32, [3]>([1, 1024, 1])];
236
+ tensor<fp16, [1, 1024, 1]> h_31_cast_fp16 = reshape(shape = var_373, x = linear_7_cast_fp16)[name = tensor<string, []>("h_31_cast_fp16")];
237
+ tensor<int32, [2]> var_375_split_sizes_0 = const()[name = tensor<string, []>("op_375_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
238
+ tensor<int32, []> var_375_axis_0 = const()[name = tensor<string, []>("op_375_axis_0"), val = tensor<int32, []>(1)];
239
+ tensor<fp16, [1, 512, 1]> var_375_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_375_cast_fp16_1 = split(axis = var_375_axis_0, split_sizes = var_375_split_sizes_0, x = h_31_cast_fp16)[name = tensor<string, []>("op_375_cast_fp16")];
240
+ tensor<fp16, []> var_377_promoted_to_fp16 = const()[name = tensor<string, []>("op_377_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
241
+ tensor<fp16, [1, 512, 1]> var_378_cast_fp16 = add(x = var_375_cast_fp16_0, y = var_377_promoted_to_fp16)[name = tensor<string, []>("op_378_cast_fp16")];
242
+ tensor<fp16, []> var_324_to_fp16 = const()[name = tensor<string, []>("op_324_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
243
+ tensor<fp16, [1, 512, ?]> var_381_cast_fp16 = instance_norm(beta = F0_0_norm1_norm_bias_to_fp16, epsilon = var_324_to_fp16, gamma = F0_0_norm1_norm_weight_to_fp16, x = input_51_cast_fp16)[name = tensor<string, []>("op_381_cast_fp16")];
244
+ tensor<fp16, [1, 512, ?]> var_382_cast_fp16 = mul(x = var_378_cast_fp16, y = var_381_cast_fp16)[name = tensor<string, []>("op_382_cast_fp16")];
245
+ tensor<fp16, [1, 512, ?]> input_53_cast_fp16 = add(x = var_382_cast_fp16, y = var_375_cast_fp16_1)[name = tensor<string, []>("input_53_cast_fp16")];
246
+ tensor<fp16, [1, 512, ?]> input_55_cast_fp16 = leaky_relu(alpha = var_321, x = input_53_cast_fp16)[name = tensor<string, []>("input_55_cast_fp16")];
247
+ tensor<string, []> out_7_pad_type_0 = const()[name = tensor<string, []>("out_7_pad_type_0"), val = tensor<string, []>("custom")];
248
+ tensor<int32, [2]> out_7_pad_0 = const()[name = tensor<string, []>("out_7_pad_0"), val = tensor<int32, [2]>([1, 1])];
249
+ tensor<int32, [1]> out_7_strides_0 = const()[name = tensor<string, []>("out_7_strides_0"), val = tensor<int32, [1]>([1])];
250
+ tensor<int32, [1]> out_7_dilations_0 = const()[name = tensor<string, []>("out_7_dilations_0"), val = tensor<int32, [1]>([1])];
251
+ tensor<int32, []> out_7_groups_0 = const()[name = tensor<string, []>("out_7_groups_0"), val = tensor<int32, []>(1)];
252
+ tensor<fp16, [512, 512, 3]> weight_35_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [786432]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6206080))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6992576))), name = tensor<string, []>("weight_35_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 3])];
253
+ tensor<fp16, [512]> N_0_conv2_bias_to_fp16 = const()[name = tensor<string, []>("N_0_conv2_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6993152)))];
254
+ tensor<fp16, [1, 512, ?]> out_7_cast_fp16 = conv(bias = N_0_conv2_bias_to_fp16, dilations = out_7_dilations_0, groups = out_7_groups_0, pad = out_7_pad_0, pad_type = out_7_pad_type_0, strides = out_7_strides_0, weight = weight_35_to_fp16_palettized, x = input_55_cast_fp16)[name = tensor<string, []>("out_7_cast_fp16")];
255
+ tensor<fp16, [1, 512, ?]> var_392_cast_fp16 = add(x = out_7_cast_fp16, y = transpose_0)[name = tensor<string, []>("op_392_cast_fp16")];
256
+ tensor<fp16, []> var_393_to_fp16 = const()[name = tensor<string, []>("op_393_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
257
+ tensor<fp16, [1, 512, ?]> input_57_cast_fp16 = mul(x = var_392_cast_fp16, y = var_393_to_fp16)[name = tensor<string, []>("input_57_cast_fp16")];
258
+ tensor<fp32, []> var_398 = const()[name = tensor<string, []>("op_398"), val = tensor<fp32, []>(0x1.99999ap-3)];
259
+ tensor<fp16, [1024, 128]> N_1_norm1_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6994240))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7125376))), name = tensor<string, []>("N_1_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
260
+ tensor<fp16, [1024]> N_1_norm1_fc_bias_to_fp16 = const()[name = tensor<string, []>("N_1_norm1_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7125952)))];
261
+ tensor<fp16, [1, 1024]> linear_8_cast_fp16 = linear(bias = N_1_norm1_fc_bias_to_fp16, weight = N_1_norm1_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_8_cast_fp16")];
262
+ tensor<int32, [3]> var_434 = const()[name = tensor<string, []>("op_434"), val = tensor<int32, [3]>([1, 1024, 1])];
263
+ tensor<fp16, [1, 1024, 1]> h_35_cast_fp16 = reshape(shape = var_434, x = linear_8_cast_fp16)[name = tensor<string, []>("h_35_cast_fp16")];
264
+ tensor<int32, [2]> var_436_split_sizes_0 = const()[name = tensor<string, []>("op_436_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
265
+ tensor<int32, []> var_436_axis_0 = const()[name = tensor<string, []>("op_436_axis_0"), val = tensor<int32, []>(1)];
266
+ tensor<fp16, [1, 512, 1]> var_436_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_436_cast_fp16_1 = split(axis = var_436_axis_0, split_sizes = var_436_split_sizes_0, x = h_35_cast_fp16)[name = tensor<string, []>("op_436_cast_fp16")];
267
+ tensor<fp16, []> var_438_promoted_to_fp16 = const()[name = tensor<string, []>("op_438_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
268
+ tensor<fp16, [1, 512, 1]> var_439_cast_fp16 = add(x = var_436_cast_fp16_0, y = var_438_promoted_to_fp16)[name = tensor<string, []>("op_439_cast_fp16")];
269
+ tensor<fp16, []> var_402_to_fp16 = const()[name = tensor<string, []>("op_402_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
270
+ tensor<fp16, [1, 512, ?]> var_442_cast_fp16 = instance_norm(beta = F0_0_norm1_norm_bias_to_fp16, epsilon = var_402_to_fp16, gamma = F0_0_norm1_norm_weight_to_fp16, x = input_57_cast_fp16)[name = tensor<string, []>("op_442_cast_fp16")];
271
+ tensor<fp16, [1, 512, ?]> var_443_cast_fp16 = mul(x = var_439_cast_fp16, y = var_442_cast_fp16)[name = tensor<string, []>("op_443_cast_fp16")];
272
+ tensor<fp16, [1, 512, ?]> input_59_cast_fp16 = add(x = var_443_cast_fp16, y = var_436_cast_fp16_1)[name = tensor<string, []>("input_59_cast_fp16")];
273
+ tensor<fp16, [1, 512, ?]> input_61_cast_fp16 = leaky_relu(alpha = var_398, x = input_59_cast_fp16)[name = tensor<string, []>("input_61_cast_fp16")];
274
+ tensor<string, []> conv_transpose_1_pad_type_0 = const()[name = tensor<string, []>("conv_transpose_1_pad_type_0"), val = tensor<string, []>("custom")];
275
+ tensor<int32, [2]> conv_transpose_1_pad_0 = const()[name = tensor<string, []>("conv_transpose_1_pad_0"), val = tensor<int32, [2]>([0, 0])];
276
+ tensor<int32, [1]> conv_transpose_1_strides_0 = const()[name = tensor<string, []>("conv_transpose_1_strides_0"), val = tensor<int32, [1]>([2])];
277
+ tensor<int32, []> conv_transpose_1_groups_0 = const()[name = tensor<string, []>("conv_transpose_1_groups_0"), val = tensor<int32, []>(512)];
278
+ tensor<int32, [1]> conv_transpose_1_dilations_0 = const()[name = tensor<string, []>("conv_transpose_1_dilations_0"), val = tensor<int32, [1]>([1])];
279
+ tensor<fp16, [512, 1, 3]> var_446_to_fp16 = const()[name = tensor<string, []>("op_446_to_fp16"), val = tensor<fp16, [512, 1, 3]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7128064)))];
280
+ tensor<fp16, [512]> N_1_pool_bias_to_fp16 = const()[name = tensor<string, []>("N_1_pool_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7131200)))];
281
+ tensor<fp16, [1, 512, ?]> conv_transpose_1_cast_fp16 = conv_transpose(bias = N_1_pool_bias_to_fp16, dilations = conv_transpose_1_dilations_0, groups = conv_transpose_1_groups_0, pad = conv_transpose_1_pad_0, pad_type = conv_transpose_1_pad_type_0, strides = conv_transpose_1_strides_0, weight = var_446_to_fp16, x = input_61_cast_fp16)[name = tensor<string, []>("conv_transpose_1_cast_fp16")];
282
+ tensor<int32, [3]> input_63_begin_0 = const()[name = tensor<string, []>("input_63_begin_0"), val = tensor<int32, [3]>([0, 0, 1])];
283
+ tensor<int32, [3]> input_63_end_0 = const()[name = tensor<string, []>("input_63_end_0"), val = tensor<int32, [3]>([0, 0, 0])];
284
+ tensor<bool, [3]> input_63_begin_mask_0 = const()[name = tensor<string, []>("input_63_begin_mask_0"), val = tensor<bool, [3]>([true, true, false])];
285
+ tensor<bool, [3]> input_63_end_mask_0 = const()[name = tensor<string, []>("input_63_end_mask_0"), val = tensor<bool, [3]>([true, true, true])];
286
+ tensor<fp16, [1, 512, ?]> input_63_cast_fp16 = slice_by_index(begin = input_63_begin_0, begin_mask = input_63_begin_mask_0, end = input_63_end_0, end_mask = input_63_end_mask_0, x = conv_transpose_1_cast_fp16)[name = tensor<string, []>("input_63_cast_fp16")];
287
+ tensor<string, []> input_65_pad_type_0 = const()[name = tensor<string, []>("input_65_pad_type_0"), val = tensor<string, []>("custom")];
288
+ tensor<int32, [2]> input_65_pad_0 = const()[name = tensor<string, []>("input_65_pad_0"), val = tensor<int32, [2]>([1, 1])];
289
+ tensor<int32, [1]> input_65_strides_0 = const()[name = tensor<string, []>("input_65_strides_0"), val = tensor<int32, [1]>([1])];
290
+ tensor<int32, [1]> input_65_dilations_0 = const()[name = tensor<string, []>("input_65_dilations_0"), val = tensor<int32, [1]>([1])];
291
+ tensor<int32, []> input_65_groups_0 = const()[name = tensor<string, []>("input_65_groups_0"), val = tensor<int32, []>(1)];
292
+ tensor<fp16, [256, 512, 3]> weight_39_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [393216]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7132288))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7525568))), name = tensor<string, []>("weight_39_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 512, 3])];
293
+ tensor<fp16, [256]> N_1_conv1_bias_to_fp16 = const()[name = tensor<string, []>("N_1_conv1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7526144)))];
294
+ tensor<fp16, [1, 256, ?]> input_65_cast_fp16 = conv(bias = N_1_conv1_bias_to_fp16, dilations = input_65_dilations_0, groups = input_65_groups_0, pad = input_65_pad_0, pad_type = input_65_pad_type_0, strides = input_65_strides_0, weight = weight_39_to_fp16_palettized, x = input_63_cast_fp16)[name = tensor<string, []>("input_65_cast_fp16")];
295
+ tensor<fp16, [512, 128]> N_1_norm2_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7526720))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7592320))), name = tensor<string, []>("N_1_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
296
+ tensor<fp16, [512]> N_1_norm2_fc_bias_to_fp16 = const()[name = tensor<string, []>("N_1_norm2_fc_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7592896)))];
297
+ tensor<fp16, [1, 512]> linear_9_cast_fp16 = linear(bias = N_1_norm2_fc_bias_to_fp16, weight = N_1_norm2_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_9_cast_fp16")];
298
+ tensor<int32, [3]> var_467 = const()[name = tensor<string, []>("op_467"), val = tensor<int32, [3]>([1, 512, 1])];
299
+ tensor<fp16, [1, 512, 1]> h_39_cast_fp16 = reshape(shape = var_467, x = linear_9_cast_fp16)[name = tensor<string, []>("h_39_cast_fp16")];
300
+ tensor<int32, [2]> var_469_split_sizes_0 = const()[name = tensor<string, []>("op_469_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
301
+ tensor<int32, []> var_469_axis_0 = const()[name = tensor<string, []>("op_469_axis_0"), val = tensor<int32, []>(1)];
302
+ tensor<fp16, [1, 256, 1]> var_469_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_469_cast_fp16_1 = split(axis = var_469_axis_0, split_sizes = var_469_split_sizes_0, x = h_39_cast_fp16)[name = tensor<string, []>("op_469_cast_fp16")];
303
+ tensor<fp16, []> var_471_promoted_to_fp16 = const()[name = tensor<string, []>("op_471_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
304
+ tensor<fp16, [1, 256, 1]> var_472_cast_fp16 = add(x = var_469_cast_fp16_0, y = var_471_promoted_to_fp16)[name = tensor<string, []>("op_472_cast_fp16")];
305
+ tensor<fp16, [1, 256, ?]> var_475_cast_fp16 = instance_norm(beta = F0_1_norm2_norm_bias_to_fp16, epsilon = var_402_to_fp16, gamma = F0_1_norm2_norm_weight_to_fp16, x = input_65_cast_fp16)[name = tensor<string, []>("op_475_cast_fp16")];
306
+ tensor<fp16, [1, 256, ?]> var_476_cast_fp16 = mul(x = var_472_cast_fp16, y = var_475_cast_fp16)[name = tensor<string, []>("op_476_cast_fp16")];
307
+ tensor<fp16, [1, 256, ?]> input_67_cast_fp16 = add(x = var_476_cast_fp16, y = var_469_cast_fp16_1)[name = tensor<string, []>("input_67_cast_fp16")];
308
+ tensor<fp16, [1, 256, ?]> input_69_cast_fp16 = leaky_relu(alpha = var_398, x = input_67_cast_fp16)[name = tensor<string, []>("input_69_cast_fp16")];
309
+ tensor<string, []> out_9_pad_type_0 = const()[name = tensor<string, []>("out_9_pad_type_0"), val = tensor<string, []>("custom")];
310
+ tensor<int32, [2]> out_9_pad_0 = const()[name = tensor<string, []>("out_9_pad_0"), val = tensor<int32, [2]>([1, 1])];
311
+ tensor<int32, [1]> out_9_strides_0 = const()[name = tensor<string, []>("out_9_strides_0"), val = tensor<int32, [1]>([1])];
312
+ tensor<int32, [1]> out_9_dilations_0 = const()[name = tensor<string, []>("out_9_dilations_0"), val = tensor<int32, [1]>([1])];
313
+ tensor<int32, []> out_9_groups_0 = const()[name = tensor<string, []>("out_9_groups_0"), val = tensor<int32, []>(1)];
314
+ tensor<fp16, [256, 256, 3]> weight_43_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7593984))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7790656))), name = tensor<string, []>("weight_43_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
315
+ tensor<fp16, [256]> N_1_conv2_bias_to_fp16 = const()[name = tensor<string, []>("N_1_conv2_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7791232)))];
316
+ tensor<fp16, [1, 256, ?]> out_9_cast_fp16 = conv(bias = N_1_conv2_bias_to_fp16, dilations = out_9_dilations_0, groups = out_9_groups_0, pad = out_9_pad_0, pad_type = out_9_pad_type_0, strides = out_9_strides_0, weight = weight_43_to_fp16_palettized, x = input_69_cast_fp16)[name = tensor<string, []>("out_9_cast_fp16")];
317
+ tensor<int32, [1]> expand_dims_1_axes_0 = const()[name = tensor<string, []>("expand_dims_1_axes_0"), val = tensor<int32, [1]>([3])];
318
+ tensor<fp16, [1, 512, ?, 1]> expand_dims_1_cast_fp16 = expand_dims(axes = expand_dims_1_axes_0, x = input_57_cast_fp16)[name = tensor<string, []>("expand_dims_1_cast_fp16")];
319
+ tensor<int32, []> upsample_nearest_neighbor_1_scale_factor_height_0 = const()[name = tensor<string, []>("upsample_nearest_neighbor_1_scale_factor_height_0"), val = tensor<int32, []>(2)];
320
+ tensor<int32, []> upsample_nearest_neighbor_1_scale_factor_width_0 = const()[name = tensor<string, []>("upsample_nearest_neighbor_1_scale_factor_width_0"), val = tensor<int32, []>(1)];
321
+ tensor<fp16, [1, 512, ?, 1]> upsample_nearest_neighbor_1_cast_fp16 = upsample_nearest_neighbor(scale_factor_height = upsample_nearest_neighbor_1_scale_factor_height_0, scale_factor_width = upsample_nearest_neighbor_1_scale_factor_width_0, x = expand_dims_1_cast_fp16)[name = tensor<string, []>("upsample_nearest_neighbor_1_cast_fp16")];
322
+ tensor<int32, [1]> input_71_axes_0 = const()[name = tensor<string, []>("input_71_axes_0"), val = tensor<int32, [1]>([3])];
323
+ tensor<fp16, [1, 512, ?]> input_71_cast_fp16 = squeeze(axes = input_71_axes_0, x = upsample_nearest_neighbor_1_cast_fp16)[name = tensor<string, []>("input_71_cast_fp16")];
324
+ tensor<string, []> var_493_pad_type_0 = const()[name = tensor<string, []>("op_493_pad_type_0"), val = tensor<string, []>("valid")];
325
+ tensor<int32, [1]> var_493_strides_0 = const()[name = tensor<string, []>("op_493_strides_0"), val = tensor<int32, [1]>([1])];
326
+ tensor<int32, [2]> var_493_pad_0 = const()[name = tensor<string, []>("op_493_pad_0"), val = tensor<int32, [2]>([0, 0])];
327
+ tensor<int32, [1]> var_493_dilations_0 = const()[name = tensor<string, []>("op_493_dilations_0"), val = tensor<int32, [1]>([1])];
328
+ tensor<int32, []> var_493_groups_0 = const()[name = tensor<string, []>("op_493_groups_0"), val = tensor<int32, []>(1)];
329
+ tensor<fp16, [256, 512, 1]> weight_45_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7791808))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7922944))), name = tensor<string, []>("weight_45_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 512, 1])];
330
+ tensor<fp16, [1, 256, ?]> var_493_cast_fp16 = conv(dilations = var_493_dilations_0, groups = var_493_groups_0, pad = var_493_pad_0, pad_type = var_493_pad_type_0, strides = var_493_strides_0, weight = weight_45_to_fp16_palettized, x = input_71_cast_fp16)[name = tensor<string, []>("op_493_cast_fp16")];
331
+ tensor<fp16, [1, 256, ?]> var_494_cast_fp16 = add(x = out_9_cast_fp16, y = var_493_cast_fp16)[name = tensor<string, []>("op_494_cast_fp16")];
332
+ tensor<fp16, []> var_495_to_fp16 = const()[name = tensor<string, []>("op_495_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
333
+ tensor<fp16, [1, 256, ?]> input_73_cast_fp16 = mul(x = var_494_cast_fp16, y = var_495_to_fp16)[name = tensor<string, []>("input_73_cast_fp16")];
334
+ tensor<fp32, []> var_498 = const()[name = tensor<string, []>("op_498"), val = tensor<fp32, []>(0x1.99999ap-3)];
335
+ tensor<fp16, [512, 128]> N_2_norm1_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7923520))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7989120))), name = tensor<string, []>("N_2_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
336
+ tensor<fp16, [512]> N_2_norm1_fc_bias_to_fp16 = const()[name = tensor<string, []>("N_2_norm1_fc_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7989696)))];
337
+ tensor<fp16, [1, 512]> linear_10_cast_fp16 = linear(bias = N_2_norm1_fc_bias_to_fp16, weight = N_2_norm1_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_10_cast_fp16")];
338
+ tensor<int32, [3]> var_524 = const()[name = tensor<string, []>("op_524"), val = tensor<int32, [3]>([1, 512, 1])];
339
+ tensor<fp16, [1, 512, 1]> h_43_cast_fp16 = reshape(shape = var_524, x = linear_10_cast_fp16)[name = tensor<string, []>("h_43_cast_fp16")];
340
+ tensor<int32, [2]> var_526_split_sizes_0 = const()[name = tensor<string, []>("op_526_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
341
+ tensor<int32, []> var_526_axis_0 = const()[name = tensor<string, []>("op_526_axis_0"), val = tensor<int32, []>(1)];
342
+ tensor<fp16, [1, 256, 1]> var_526_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_526_cast_fp16_1 = split(axis = var_526_axis_0, split_sizes = var_526_split_sizes_0, x = h_43_cast_fp16)[name = tensor<string, []>("op_526_cast_fp16")];
343
+ tensor<fp16, []> var_528_promoted_to_fp16 = const()[name = tensor<string, []>("op_528_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
344
+ tensor<fp16, [1, 256, 1]> var_529_cast_fp16 = add(x = var_526_cast_fp16_0, y = var_528_promoted_to_fp16)[name = tensor<string, []>("op_529_cast_fp16")];
345
+ tensor<fp16, []> var_501_to_fp16 = const()[name = tensor<string, []>("op_501_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
346
+ tensor<fp16, [1, 256, ?]> var_532_cast_fp16 = instance_norm(beta = F0_1_norm2_norm_bias_to_fp16, epsilon = var_501_to_fp16, gamma = F0_1_norm2_norm_weight_to_fp16, x = input_73_cast_fp16)[name = tensor<string, []>("op_532_cast_fp16")];
347
+ tensor<fp16, [1, 256, ?]> var_533_cast_fp16 = mul(x = var_529_cast_fp16, y = var_532_cast_fp16)[name = tensor<string, []>("op_533_cast_fp16")];
348
+ tensor<fp16, [1, 256, ?]> input_75_cast_fp16 = add(x = var_533_cast_fp16, y = var_526_cast_fp16_1)[name = tensor<string, []>("input_75_cast_fp16")];
349
+ tensor<fp16, [1, 256, ?]> input_77_cast_fp16 = leaky_relu(alpha = var_498, x = input_75_cast_fp16)[name = tensor<string, []>("input_77_cast_fp16")];
350
+ tensor<string, []> input_79_pad_type_0 = const()[name = tensor<string, []>("input_79_pad_type_0"), val = tensor<string, []>("custom")];
351
+ tensor<int32, [2]> input_79_pad_0 = const()[name = tensor<string, []>("input_79_pad_0"), val = tensor<int32, [2]>([1, 1])];
352
+ tensor<int32, [1]> input_79_strides_0 = const()[name = tensor<string, []>("input_79_strides_0"), val = tensor<int32, [1]>([1])];
353
+ tensor<int32, [1]> input_79_dilations_0 = const()[name = tensor<string, []>("input_79_dilations_0"), val = tensor<int32, [1]>([1])];
354
+ tensor<int32, []> input_79_groups_0 = const()[name = tensor<string, []>("input_79_groups_0"), val = tensor<int32, []>(1)];
355
+ tensor<fp16, [256, 256, 3]> weight_49_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7990784))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8187456))), name = tensor<string, []>("weight_49_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
356
+ tensor<fp16, [256]> N_2_conv1_bias_to_fp16 = const()[name = tensor<string, []>("N_2_conv1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8188032)))];
357
+ tensor<fp16, [1, 256, ?]> input_79_cast_fp16 = conv(bias = N_2_conv1_bias_to_fp16, dilations = input_79_dilations_0, groups = input_79_groups_0, pad = input_79_pad_0, pad_type = input_79_pad_type_0, strides = input_79_strides_0, weight = weight_49_to_fp16_palettized, x = input_77_cast_fp16)[name = tensor<string, []>("input_79_cast_fp16")];
358
+ tensor<fp16, [512, 128]> N_2_norm2_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [65536]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8188608))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8254208))), name = tensor<string, []>("N_2_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
359
+ tensor<fp16, [512]> N_2_norm2_fc_bias_to_fp16 = const()[name = tensor<string, []>("N_2_norm2_fc_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8254784)))];
360
+ tensor<fp16, [1, 512]> linear_11_cast_fp16 = linear(bias = N_2_norm2_fc_bias_to_fp16, weight = N_2_norm2_fc_weight_to_fp16_palettized, x = style_s)[name = tensor<string, []>("linear_11_cast_fp16")];
361
+ tensor<int32, [3]> var_550 = const()[name = tensor<string, []>("op_550"), val = tensor<int32, [3]>([1, 512, 1])];
362
+ tensor<fp16, [1, 512, 1]> h_cast_fp16 = reshape(shape = var_550, x = linear_11_cast_fp16)[name = tensor<string, []>("h_cast_fp16")];
363
+ tensor<int32, [2]> var_552_split_sizes_0 = const()[name = tensor<string, []>("op_552_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
364
+ tensor<int32, []> var_552_axis_0 = const()[name = tensor<string, []>("op_552_axis_0"), val = tensor<int32, []>(1)];
365
+ tensor<fp16, [1, 256, 1]> var_552_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_552_cast_fp16_1 = split(axis = var_552_axis_0, split_sizes = var_552_split_sizes_0, x = h_cast_fp16)[name = tensor<string, []>("op_552_cast_fp16")];
366
+ tensor<fp16, []> var_554_promoted_to_fp16 = const()[name = tensor<string, []>("op_554_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
367
+ tensor<fp16, [1, 256, 1]> var_555_cast_fp16 = add(x = var_552_cast_fp16_0, y = var_554_promoted_to_fp16)[name = tensor<string, []>("op_555_cast_fp16")];
368
+ tensor<fp16, [1, 256, ?]> var_558_cast_fp16 = instance_norm(beta = F0_1_norm2_norm_bias_to_fp16, epsilon = var_501_to_fp16, gamma = F0_1_norm2_norm_weight_to_fp16, x = input_79_cast_fp16)[name = tensor<string, []>("op_558_cast_fp16")];
369
+ tensor<fp16, [1, 256, ?]> var_559_cast_fp16 = mul(x = var_555_cast_fp16, y = var_558_cast_fp16)[name = tensor<string, []>("op_559_cast_fp16")];
370
+ tensor<fp16, [1, 256, ?]> input_81_cast_fp16 = add(x = var_559_cast_fp16, y = var_552_cast_fp16_1)[name = tensor<string, []>("input_81_cast_fp16")];
371
+ tensor<fp16, [1, 256, ?]> input_83_cast_fp16 = leaky_relu(alpha = var_498, x = input_81_cast_fp16)[name = tensor<string, []>("input_83_cast_fp16")];
372
+ tensor<string, []> out_pad_type_0 = const()[name = tensor<string, []>("out_pad_type_0"), val = tensor<string, []>("custom")];
373
+ tensor<int32, [2]> out_pad_0 = const()[name = tensor<string, []>("out_pad_0"), val = tensor<int32, [2]>([1, 1])];
374
+ tensor<int32, [1]> out_strides_0 = const()[name = tensor<string, []>("out_strides_0"), val = tensor<int32, [1]>([1])];
375
+ tensor<int32, [1]> out_dilations_0 = const()[name = tensor<string, []>("out_dilations_0"), val = tensor<int32, [1]>([1])];
376
+ tensor<int32, []> out_groups_0 = const()[name = tensor<string, []>("out_groups_0"), val = tensor<int32, []>(1)];
377
+ tensor<fp16, [256, 256, 3]> weight_53_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8255872))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8452544))), name = tensor<string, []>("weight_53_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
378
+ tensor<fp16, [256]> N_2_conv2_bias_to_fp16 = const()[name = tensor<string, []>("N_2_conv2_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8453120)))];
379
+ tensor<fp16, [1, 256, ?]> out_cast_fp16 = conv(bias = N_2_conv2_bias_to_fp16, dilations = out_dilations_0, groups = out_groups_0, pad = out_pad_0, pad_type = out_pad_type_0, strides = out_strides_0, weight = weight_53_to_fp16_palettized, x = input_83_cast_fp16)[name = tensor<string, []>("out_cast_fp16")];
380
+ tensor<fp16, [1, 256, ?]> var_569_cast_fp16 = add(x = out_cast_fp16, y = input_73_cast_fp16)[name = tensor<string, []>("op_569_cast_fp16")];
381
+ tensor<fp16, []> var_570_to_fp16 = const()[name = tensor<string, []>("op_570_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
382
+ tensor<fp16, [1, 256, ?]> input_cast_fp16 = mul(x = var_569_cast_fp16, y = var_570_to_fp16)[name = tensor<string, []>("input_cast_fp16")];
383
+ tensor<string, []> var_582_pad_type_0 = const()[name = tensor<string, []>("op_582_pad_type_0"), val = tensor<string, []>("valid")];
384
+ tensor<int32, [1]> var_582_strides_0 = const()[name = tensor<string, []>("op_582_strides_0"), val = tensor<int32, [1]>([1])];
385
+ tensor<int32, [2]> var_582_pad_0 = const()[name = tensor<string, []>("op_582_pad_0"), val = tensor<int32, [2]>([0, 0])];
386
+ tensor<int32, [1]> var_582_dilations_0 = const()[name = tensor<string, []>("op_582_dilations_0"), val = tensor<int32, [1]>([1])];
387
+ tensor<int32, []> var_582_groups_0 = const()[name = tensor<string, []>("op_582_groups_0"), val = tensor<int32, []>(1)];
388
+ tensor<fp16, [1, 256, 1]> N_proj_weight_to_fp16 = const()[name = tensor<string, []>("N_proj_weight_to_fp16"), val = tensor<fp16, [1, 256, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8453696)))];
389
+ tensor<fp16, [1]> N_proj_bias_to_fp16 = const()[name = tensor<string, []>("N_proj_bias_to_fp16"), val = tensor<fp16, [1]>([0x1.3fp-5])];
390
+ tensor<fp16, [1, 1, ?]> var_582_cast_fp16 = conv(bias = N_proj_bias_to_fp16, dilations = var_582_dilations_0, groups = var_582_groups_0, pad = var_582_pad_0, pad_type = var_582_pad_type_0, strides = var_582_strides_0, weight = N_proj_weight_to_fp16, x = input_cast_fp16)[name = tensor<string, []>("op_582_cast_fp16")];
391
+ tensor<int32, [1]> var_584_axes_0 = const()[name = tensor<string, []>("op_584_axes_0"), val = tensor<int32, [1]>([1])];
392
+ tensor<fp16, [1, ?]> N = squeeze(axes = var_584_axes_0, x = var_582_cast_fp16)[name = tensor<string, []>("op_584_cast_fp16")];
393
+ } -> (F0, N);
394
+ }
ANE/ANE-zh/KokoroProsody.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d7229d31a47e1d5c054c24b7aed8ce0df20460523472a55ff998f5939a75cf8
3
+ size 8454272
ANE/ANE-zh/KokoroProsody.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:801253b6ccdd10a0e8c991f66a34c6bbd05f98a1115c11836a5aa9778b7bed93
3
+ size 65723
ANE/ANE-zh/KokoroProsody.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d7229d31a47e1d5c054c24b7aed8ce0df20460523472a55ff998f5939a75cf8
3
+ size 8454272
ANE/ANE-zh/KokoroProsody.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "A584E1D6-B88E-412B-83F8-60F488E36916": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Specification",
7
+ "name": "model.mlmodel",
8
+ "path": "com.apple.CoreML/model.mlmodel"
9
+ },
10
+ "C7F601FE-E520-4ADA-B688-25A6DC60355C": {
11
+ "author": "com.apple.CoreML",
12
+ "description": "CoreML Model Weights",
13
+ "name": "weights",
14
+ "path": "com.apple.CoreML/weights"
15
+ }
16
+ },
17
+ "rootModelIdentifier": "A584E1D6-B88E-412B-83F8-60F488E36916"
18
+ }
ANE/ANE-zh/KokoroTail.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cbffea509dfcba72fae7a9dc7ae424e19f8af08eabc70360fe30fd7c1de09151
3
+ size 243
ANE/ANE-zh/KokoroTail.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94e611a84f91c7b135b031a0f978cc47b0edab42912e68a86c3f3e78f9edf6a0
3
+ size 392
ANE/ANE-zh/KokoroTail.mlmodelc/metadata.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "metadataOutputVersion" : "3.0",
4
+ "storagePrecision" : "Float32",
5
+ "outputSchema" : [
6
+ {
7
+ "hasShapeFlexibility" : "0",
8
+ "isOptional" : "0",
9
+ "dataType" : "Float32",
10
+ "formattedType" : "MultiArray (Float32)",
11
+ "shortDescription" : "",
12
+ "shape" : "[]",
13
+ "name" : "audio",
14
+ "type" : "MultiArray"
15
+ }
16
+ ],
17
+ "modelParameters" : [
18
+
19
+ ],
20
+ "specificationVersion" : 8,
21
+ "mlProgramOperationTypeHistogram" : {
22
+ "Ios17.sin" : 2,
23
+ "Ios17.convTranspose" : 2,
24
+ "Ios17.conv" : 1,
25
+ "Ios17.cos" : 1,
26
+ "Ios17.sliceByIndex" : 3,
27
+ "Ios17.mul" : 2,
28
+ "Ios17.sub" : 1,
29
+ "Ios17.exp" : 1
30
+ },
31
+ "computePrecision" : "Mixed (Float32, Int32)",
32
+ "isUpdatable" : "0",
33
+ "stateSchema" : [
34
+
35
+ ],
36
+ "availability" : {
37
+ "macOS" : "14.0",
38
+ "tvOS" : "17.0",
39
+ "visionOS" : "1.0",
40
+ "watchOS" : "10.0",
41
+ "iOS" : "17.0",
42
+ "macCatalyst" : "17.0"
43
+ },
44
+ "modelType" : {
45
+ "name" : "MLModelType_mlProgram"
46
+ },
47
+ "userDefinedMetadata" : {
48
+ "com.github.apple.coremltools.conversion_date" : "2026-05-03",
49
+ "com.github.apple.coremltools.source" : "torch==2.11.0",
50
+ "com.github.apple.coremltools.version" : "9.0",
51
+ "com.github.apple.coremltools.source_dialect" : "TorchScript"
52
+ },
53
+ "inputSchema" : [
54
+ {
55
+ "dataType" : "Float32",
56
+ "hasShapeFlexibility" : "1",
57
+ "isOptional" : "0",
58
+ "shapeFlexibility" : "1 × 128 × 100...240001",
59
+ "shapeRange" : "[[1, 1], [128, 128], [100, 240001]]",
60
+ "formattedType" : "MultiArray (Float32 1 × 128 × 15961)",
61
+ "type" : "MultiArray",
62
+ "shape" : "[1, 128, 15961]",
63
+ "name" : "x_pre",
64
+ "shortDescription" : ""
65
+ }
66
+ ],
67
+ "generatedClassName" : "KokoroTail",
68
+ "method" : "predict"
69
+ }
70
+ ]
ANE/ANE-zh/KokoroTail.mlmodelc/model.mil ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ program(1.0)
2
+ [buildInfo = dict<tensor<string, []>, tensor<string, []>>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.11.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
3
+ {
4
+ func main<ios17>(tensor<fp32, [1, 128, ?]> x_pre) [FlexibleShapeInformation = tuple<tuple<tensor<string, []>, dict<tensor<string, []>, tensor<int32, [?]>>>, tuple<tensor<string, []>, dict<tensor<string, []>, list<tensor<int32, [2]>, ?>>>>((("DefaultShapes", {{"x_pre", [1, 128, 15961]}}), ("RangeDims", {{"x_pre", [[1, 1], [128, 128], [100, 240001]]}})))] {
5
+ tensor<fp32, [22]> conv_post_bias = const()[name = tensor<string, []>("conv_post_bias"), val = tensor<fp32, [22]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64)))];
6
+ tensor<fp32, [11, 1, 20]> stft_deconv_real_weight = const()[name = tensor<string, []>("stft_deconv_real_weight"), val = tensor<fp32, [11, 1, 20]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(256)))];
7
+ tensor<fp32, [11, 1, 20]> stft_deconv_imag_weight = const()[name = tensor<string, []>("stft_deconv_imag_weight"), val = tensor<fp32, [11, 1, 20]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1216)))];
8
+ tensor<fp32, [22, 128, 7]> weight_1 = const()[name = tensor<string, []>("weight_1"), val = tensor<fp32, [22, 128, 7]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2176)))];
9
+ tensor<string, []> x_pad_type_0 = const()[name = tensor<string, []>("x_pad_type_0"), val = tensor<string, []>("custom")];
10
+ tensor<int32, [2]> x_pad_0 = const()[name = tensor<string, []>("x_pad_0"), val = tensor<int32, [2]>([3, 3])];
11
+ tensor<int32, [1]> x_strides_0 = const()[name = tensor<string, []>("x_strides_0"), val = tensor<int32, [1]>([1])];
12
+ tensor<int32, [1]> x_dilations_0 = const()[name = tensor<string, []>("x_dilations_0"), val = tensor<int32, [1]>([1])];
13
+ tensor<int32, []> x_groups_0 = const()[name = tensor<string, []>("x_groups_0"), val = tensor<int32, []>(1)];
14
+ tensor<fp32, [1, 22, ?]> x = conv(bias = conv_post_bias, dilations = x_dilations_0, groups = x_groups_0, pad = x_pad_0, pad_type = x_pad_type_0, strides = x_strides_0, weight = weight_1, x = x_pre)[name = tensor<string, []>("x")];
15
+ tensor<int32, [3]> var_33_begin_0 = const()[name = tensor<string, []>("op_33_begin_0"), val = tensor<int32, [3]>([0, 0, 0])];
16
+ tensor<int32, [3]> var_33_end_0 = const()[name = tensor<string, []>("op_33_end_0"), val = tensor<int32, [3]>([1, 11, 0])];
17
+ tensor<bool, [3]> var_33_end_mask_0 = const()[name = tensor<string, []>("op_33_end_mask_0"), val = tensor<bool, [3]>([true, false, true])];
18
+ tensor<fp32, [1, 11, ?]> var_33 = slice_by_index(begin = var_33_begin_0, end = var_33_end_0, end_mask = var_33_end_mask_0, x = x)[name = tensor<string, []>("op_33")];
19
+ tensor<fp32, [1, 11, ?]> magnitude = exp(x = var_33)[name = tensor<string, []>("magnitude")];
20
+ tensor<int32, [3]> var_49_begin_0 = const()[name = tensor<string, []>("op_49_begin_0"), val = tensor<int32, [3]>([0, 11, 0])];
21
+ tensor<int32, [3]> var_49_end_0 = const()[name = tensor<string, []>("op_49_end_0"), val = tensor<int32, [3]>([1, 22, 0])];
22
+ tensor<bool, [3]> var_49_end_mask_0 = const()[name = tensor<string, []>("op_49_end_mask_0"), val = tensor<bool, [3]>([true, true, true])];
23
+ tensor<fp32, [1, 11, ?]> var_49 = slice_by_index(begin = var_49_begin_0, end = var_49_end_0, end_mask = var_49_end_mask_0, x = x)[name = tensor<string, []>("op_49")];
24
+ tensor<fp32, [1, 11, ?]> phase = sin(x = var_49)[name = tensor<string, []>("phase")];
25
+ tensor<fp32, [1, 11, ?]> var_56 = cos(x = phase)[name = tensor<string, []>("op_56")];
26
+ tensor<fp32, [1, 11, ?]> input_1 = mul(x = magnitude, y = var_56)[name = tensor<string, []>("input_1")];
27
+ tensor<fp32, [1, 11, ?]> var_58 = sin(x = phase)[name = tensor<string, []>("op_58")];
28
+ tensor<fp32, [1, 11, ?]> input = mul(x = magnitude, y = var_58)[name = tensor<string, []>("input")];
29
+ tensor<string, []> var_71_pad_type_0 = const()[name = tensor<string, []>("op_71_pad_type_0"), val = tensor<string, []>("valid")];
30
+ tensor<int32, [1]> var_71_strides_0 = const()[name = tensor<string, []>("op_71_strides_0"), val = tensor<int32, [1]>([5])];
31
+ tensor<int32, [2]> var_71_pad_0 = const()[name = tensor<string, []>("op_71_pad_0"), val = tensor<int32, [2]>([0, 0])];
32
+ tensor<int32, [1]> var_71_dilations_0 = const()[name = tensor<string, []>("op_71_dilations_0"), val = tensor<int32, [1]>([1])];
33
+ tensor<int32, []> var_71_groups_0 = const()[name = tensor<string, []>("op_71_groups_0"), val = tensor<int32, []>(1)];
34
+ tensor<fp32, [1, 1, ?]> var_71 = conv_transpose(dilations = var_71_dilations_0, groups = var_71_groups_0, pad = var_71_pad_0, pad_type = var_71_pad_type_0, strides = var_71_strides_0, weight = stft_deconv_real_weight, x = input_1)[name = tensor<string, []>("op_71")];
35
+ tensor<string, []> var_83_pad_type_0 = const()[name = tensor<string, []>("op_83_pad_type_0"), val = tensor<string, []>("valid")];
36
+ tensor<int32, [1]> var_83_strides_0 = const()[name = tensor<string, []>("op_83_strides_0"), val = tensor<int32, [1]>([5])];
37
+ tensor<int32, [2]> var_83_pad_0 = const()[name = tensor<string, []>("op_83_pad_0"), val = tensor<int32, [2]>([0, 0])];
38
+ tensor<int32, [1]> var_83_dilations_0 = const()[name = tensor<string, []>("op_83_dilations_0"), val = tensor<int32, [1]>([1])];
39
+ tensor<int32, []> var_83_groups_0 = const()[name = tensor<string, []>("op_83_groups_0"), val = tensor<int32, []>(1)];
40
+ tensor<fp32, [1, 1, ?]> var_83 = conv_transpose(dilations = var_83_dilations_0, groups = var_83_groups_0, pad = var_83_pad_0, pad_type = var_83_pad_type_0, strides = var_83_strides_0, weight = stft_deconv_imag_weight, x = input)[name = tensor<string, []>("op_83")];
41
+ tensor<fp32, [1, 1, ?]> waveform = sub(x = var_71, y = var_83)[name = tensor<string, []>("waveform")];
42
+ tensor<int32, [3]> var_90_begin_0 = const()[name = tensor<string, []>("op_90_begin_0"), val = tensor<int32, [3]>([0, 0, 10])];
43
+ tensor<int32, [3]> var_90_end_0 = const()[name = tensor<string, []>("op_90_end_0"), val = tensor<int32, [3]>([1, 1, -10])];
44
+ tensor<bool, [3]> var_90_end_mask_0 = const()[name = tensor<string, []>("op_90_end_mask_0"), val = tensor<bool, [3]>([true, true, false])];
45
+ tensor<fp32, [1, 1, ?]> audio = slice_by_index(begin = var_90_begin_0, end = var_90_end_0, end_mask = var_90_end_mask_0, x = waveform)[name = tensor<string, []>("op_90")];
46
+ } -> (audio);
47
+ }
ANE/ANE-zh/KokoroTail.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1865207df8b7608f3fd443b5a3c744634a8942ccc917b5c8734818d569c0f4eb
3
+ size 81088
ANE/ANE-zh/KokoroTail.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c95acf7775be2c4832b10a81517b4b133dd01b4ed0b54a41f0d7689ffca8526
3
+ size 5964
ANE/ANE-zh/KokoroTail.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1865207df8b7608f3fd443b5a3c744634a8942ccc917b5c8734818d569c0f4eb
3
+ size 81088
ANE/ANE-zh/KokoroTail.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "18BB1DC6-314D-4189-A295-F891E896DE7B": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Weights",
7
+ "name": "weights",
8
+ "path": "com.apple.CoreML/weights"
9
+ },
10
+ "E288AD71-C613-4173-B1DC-4CEC042D94B8": {
11
+ "author": "com.apple.CoreML",
12
+ "description": "CoreML Model Specification",
13
+ "name": "model.mlmodel",
14
+ "path": "com.apple.CoreML/model.mlmodel"
15
+ }
16
+ },
17
+ "rootModelIdentifier": "E288AD71-C613-4173-B1DC-4CEC042D94B8"
18
+ }
ANE/ANE-zh/KokoroVocoder.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12bf39f5117a2fe2645b0158d75d71a99b328b237f1add55a1511d0e2ee3b456
3
+ size 243
ANE/ANE-zh/KokoroVocoder.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:24ad83fed32d94eff2b3d9d70159140057ebad500b2a7f8fb54b0be0204b1cd2
3
+ size 627