alexwengg commited on
Commit
34eaffa
·
verified ·
1 Parent(s): 2ad731b

Delete ANE

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/.DS_Store +0 -0
  2. ANE/styletts2_ane_alignment.mlmodelc/analytics/coremldata.bin +0 -3
  3. ANE/styletts2_ane_alignment.mlmodelc/coremldata.bin +0 -3
  4. ANE/styletts2_ane_alignment.mlmodelc/metadata.json +0 -104
  5. ANE/styletts2_ane_alignment.mlmodelc/model.mil +0 -33
  6. ANE/styletts2_ane_alignment.mlmodelc/weights/weight.bin +0 -3
  7. ANE/styletts2_ane_alignment.mlpackage/Data/com.apple.CoreML/model.mlmodel +0 -3
  8. ANE/styletts2_ane_alignment.mlpackage/Data/com.apple.CoreML/weights/weight.bin +0 -3
  9. ANE/styletts2_ane_alignment.mlpackage/Manifest.json +0 -18
  10. ANE/styletts2_ane_diffusion_step.mlmodelc/analytics/coremldata.bin +0 -3
  11. ANE/styletts2_ane_diffusion_step.mlmodelc/coremldata.bin +0 -3
  12. ANE/styletts2_ane_diffusion_step.mlmodelc/metadata.json +0 -115
  13. ANE/styletts2_ane_diffusion_step.mlmodelc/model.mil +0 -351
  14. ANE/styletts2_ane_diffusion_step.mlmodelc/weights/weight.bin +0 -3
  15. ANE/styletts2_ane_diffusion_step.mlpackage/Data/com.apple.CoreML/model.mlmodel +0 -3
  16. ANE/styletts2_ane_diffusion_step.mlpackage/Data/com.apple.CoreML/weights/weight.bin +0 -3
  17. ANE/styletts2_ane_diffusion_step.mlpackage/Manifest.json +0 -18
  18. ANE/styletts2_ane_noise.mlmodelc/analytics/coremldata.bin +0 -3
  19. ANE/styletts2_ane_noise.mlmodelc/coremldata.bin +0 -3
  20. ANE/styletts2_ane_noise.mlmodelc/metadata.json +0 -87
  21. ANE/styletts2_ane_noise.mlmodelc/model.mil +0 -99
  22. ANE/styletts2_ane_noise.mlmodelc/weights/weight.bin +0 -3
  23. ANE/styletts2_ane_noise.mlpackage/Data/com.apple.CoreML/model.mlmodel +0 -3
  24. ANE/styletts2_ane_noise.mlpackage/Data/com.apple.CoreML/weights/weight.bin +0 -3
  25. ANE/styletts2_ane_noise.mlpackage/Manifest.json +0 -18
  26. ANE/styletts2_ane_plbert.mlmodelc/analytics/coremldata.bin +0 -3
  27. ANE/styletts2_ane_plbert.mlmodelc/coremldata.bin +0 -3
  28. ANE/styletts2_ane_plbert.mlmodelc/metadata.json +0 -83
  29. ANE/styletts2_ane_plbert.mlmodelc/model.mil +0 -0
  30. ANE/styletts2_ane_plbert.mlmodelc/weights/weight.bin +0 -3
  31. ANE/styletts2_ane_plbert.mlpackage/Data/com.apple.CoreML/model.mlmodel +0 -3
  32. ANE/styletts2_ane_plbert.mlpackage/Data/com.apple.CoreML/weights/weight.bin +0 -3
  33. ANE/styletts2_ane_plbert.mlpackage/Manifest.json +0 -18
  34. ANE/styletts2_ane_postbert.mlmodelc/analytics/coremldata.bin +0 -3
  35. ANE/styletts2_ane_postbert.mlmodelc/coremldata.bin +0 -3
  36. ANE/styletts2_ane_postbert.mlmodelc/metadata.json +0 -134
  37. ANE/styletts2_ane_postbert.mlmodelc/model.mil +0 -265
  38. ANE/styletts2_ane_postbert.mlmodelc/weights/weight.bin +0 -3
  39. ANE/styletts2_ane_postbert.mlpackage/Data/com.apple.CoreML/model.mlmodel +0 -3
  40. ANE/styletts2_ane_postbert.mlpackage/Data/com.apple.CoreML/weights/weight.bin +0 -3
  41. ANE/styletts2_ane_postbert.mlpackage/Manifest.json +0 -18
  42. ANE/styletts2_ane_prosody.mlmodelc/analytics/coremldata.bin +0 -3
  43. ANE/styletts2_ane_prosody.mlmodelc/coremldata.bin +0 -3
  44. ANE/styletts2_ane_prosody.mlmodelc/metadata.json +0 -97
  45. ANE/styletts2_ane_prosody.mlmodelc/model.mil +0 -396
  46. ANE/styletts2_ane_prosody.mlmodelc/weights/weight.bin +0 -3
  47. ANE/styletts2_ane_prosody.mlpackage/Data/com.apple.CoreML/model.mlmodel +0 -3
  48. ANE/styletts2_ane_prosody.mlpackage/Data/com.apple.CoreML/weights/weight.bin +0 -3
  49. ANE/styletts2_ane_prosody.mlpackage/Manifest.json +0 -18
  50. ANE/styletts2_ane_vocoder.mlmodelc/analytics/coremldata.bin +0 -3
ANE/.DS_Store DELETED
Binary file (6.15 kB)
 
ANE/styletts2_ane_alignment.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:79b2d334b136d0db3149ae324b9423c3c3985d12512d8bb977d6e157bf37c7d4
3
- size 243
 
 
 
 
ANE/styletts2_ane_alignment.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:31fa9477923511300c42fe4a4477909ffa6c97ab70252d26f441c561d980f1ce
3
- size 519
 
 
 
 
ANE/styletts2_ane_alignment.mlmodelc/metadata.json DELETED
@@ -1,104 +0,0 @@
1
- [
2
- {
3
- "metadataOutputVersion" : "3.0",
4
- "shortDescription" : "StyleTTS2-ANE Stage 3: Alignment (cumsum+broadcast, T_a≤2000) — fp16, ANE.",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float16",
10
- "formattedType" : "MultiArray (Float16 1 × 640 × 2000)",
11
- "shortDescription" : "",
12
- "shape" : "[1, 640, 2000]",
13
- "name" : "en",
14
- "type" : "MultiArray"
15
- },
16
- {
17
- "hasShapeFlexibility" : "0",
18
- "isOptional" : "0",
19
- "dataType" : "Float16",
20
- "formattedType" : "MultiArray (Float16 1 × 512 × 2000)",
21
- "shortDescription" : "",
22
- "shape" : "[1, 512, 2000]",
23
- "name" : "asr",
24
- "type" : "MultiArray"
25
- }
26
- ],
27
- "storagePrecision" : "Float16",
28
- "modelParameters" : [
29
-
30
- ],
31
- "specificationVersion" : 8,
32
- "mlProgramOperationTypeHistogram" : {
33
- "Ios16.cumsum" : 1,
34
- "Ios17.logicalAnd" : 1,
35
- "Ios17.cast" : 4,
36
- "Ios17.less" : 1,
37
- "Ios17.matmul" : 2,
38
- "Ios17.greaterEqual" : 1,
39
- "Ios17.sub" : 1,
40
- "Ios17.expandDims" : 2
41
- },
42
- "computePrecision" : "Mixed (Float16, Int32)",
43
- "isUpdatable" : "0",
44
- "stateSchema" : [
45
-
46
- ],
47
- "availability" : {
48
- "macOS" : "14.0",
49
- "tvOS" : "17.0",
50
- "visionOS" : "1.0",
51
- "watchOS" : "10.0",
52
- "iOS" : "17.0",
53
- "macCatalyst" : "17.0"
54
- },
55
- "modelType" : {
56
- "name" : "MLModelType_mlProgram"
57
- },
58
- "inputSchema" : [
59
- {
60
- "dataType" : "Float32",
61
- "hasShapeFlexibility" : "1",
62
- "isOptional" : "0",
63
- "shapeFlexibility" : "1 × 2...512",
64
- "shapeRange" : "[[1, 1], [2, 512]]",
65
- "formattedType" : "MultiArray (Float32 1 × 64)",
66
- "type" : "MultiArray",
67
- "shape" : "[1, 64]",
68
- "name" : "pred_dur",
69
- "shortDescription" : ""
70
- },
71
- {
72
- "dataType" : "Float32",
73
- "hasShapeFlexibility" : "1",
74
- "isOptional" : "0",
75
- "shapeFlexibility" : "1 × 2...512 × 640",
76
- "shapeRange" : "[[1, 1], [2, 512], [640, 640]]",
77
- "formattedType" : "MultiArray (Float32 1 × 64 × 640)",
78
- "type" : "MultiArray",
79
- "shape" : "[1, 64, 640]",
80
- "name" : "d",
81
- "shortDescription" : ""
82
- },
83
- {
84
- "dataType" : "Float32",
85
- "hasShapeFlexibility" : "1",
86
- "isOptional" : "0",
87
- "shapeFlexibility" : "1 × 512 × 2...512",
88
- "shapeRange" : "[[1, 1], [512, 512], [2, 512]]",
89
- "formattedType" : "MultiArray (Float32 1 × 512 × 64)",
90
- "type" : "MultiArray",
91
- "shape" : "[1, 512, 64]",
92
- "name" : "t_en",
93
- "shortDescription" : ""
94
- }
95
- ],
96
- "userDefinedMetadata" : {
97
- "com.github.apple.coremltools.version" : "8.3.0",
98
- "com.github.apple.coremltools.source_dialect" : "TorchScript",
99
- "com.github.apple.coremltools.source" : "torch==2.11.0"
100
- },
101
- "generatedClassName" : "styletts2_ane_alignment",
102
- "method" : "predict"
103
- }
104
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_alignment.mlmodelc/model.mil DELETED
@@ -1,33 +0,0 @@
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, ?, 640]> d, tensor<fp32, [1, ?]> pred_dur, tensor<fp32, [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, 64, 640]}, {"pred_dur", [1, 64]}, {"t_en", [1, 512, 64]}}), ("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_9 = const()[name = tensor<string, []>("op_9"), val = tensor<int32, []>(-1)];
6
- tensor<bool, []> cum_exclusive_0 = const()[name = tensor<string, []>("cum_exclusive_0"), val = tensor<bool, []>(false)];
7
- tensor<bool, []> cum_reverse_0 = const()[name = tensor<string, []>("cum_reverse_0"), val = tensor<bool, []>(false)];
8
- tensor<string, []> pred_dur_to_fp16_dtype_0 = const()[name = tensor<string, []>("pred_dur_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
9
- tensor<fp16, [1, ?]> pred_dur_to_fp16 = cast(dtype = pred_dur_to_fp16_dtype_0, x = pred_dur)[name = tensor<string, []>("cast_3")];
10
- tensor<fp16, [1, ?]> cum_cast_fp16 = cumsum(axis = var_9, exclusive = cum_exclusive_0, reverse = cum_reverse_0, x = pred_dur_to_fp16)[name = tensor<string, []>("cum_cast_fp16")];
11
- tensor<fp16, [1, ?]> starts_cast_fp16 = sub(x = cum_cast_fp16, y = pred_dur_to_fp16)[name = tensor<string, []>("starts_cast_fp16")];
12
- tensor<int32, [1]> var_25_axes_0 = const()[name = tensor<string, []>("op_25_axes_0"), val = tensor<int32, [1]>([-1])];
13
- tensor<fp16, [1, ?, 1]> var_25_cast_fp16 = expand_dims(axes = var_25_axes_0, x = starts_cast_fp16)[name = tensor<string, []>("op_25_cast_fp16")];
14
- tensor<fp16, [1, 1, 2000]> frames_promoted_to_fp16 = const()[name = tensor<string, []>("frames_promoted_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_26_cast_fp16 = greater_equal(x = frames_promoted_to_fp16, y = var_25_cast_fp16)[name = tensor<string, []>("op_26_cast_fp16")];
16
- tensor<int32, [1]> var_28_axes_0 = const()[name = tensor<string, []>("op_28_axes_0"), val = tensor<int32, [1]>([-1])];
17
- tensor<fp16, [1, ?, 1]> var_28_cast_fp16 = expand_dims(axes = var_28_axes_0, x = cum_cast_fp16)[name = tensor<string, []>("op_28_cast_fp16")];
18
- tensor<bool, [1, ?, 2000]> var_29_cast_fp16 = less(x = frames_promoted_to_fp16, y = var_28_cast_fp16)[name = tensor<string, []>("op_29_cast_fp16")];
19
- tensor<bool, [1, ?, 2000]> in_seg = logical_and(x = var_26_cast_fp16, y = var_29_cast_fp16)[name = tensor<string, []>("in_seg")];
20
- tensor<string, []> d_to_fp16_dtype_0 = const()[name = tensor<string, []>("d_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
21
- tensor<bool, []> var_44_transpose_x_1 = const()[name = tensor<string, []>("op_44_transpose_x_1"), val = tensor<bool, []>(true)];
22
- tensor<bool, []> var_44_transpose_y_1 = const()[name = tensor<string, []>("op_44_transpose_y_1"), val = tensor<bool, []>(false)];
23
- tensor<string, []> cast_3_to_fp16_dtype_0 = const()[name = tensor<string, []>("cast_3_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
24
- tensor<fp16, [1, ?, 640]> d_to_fp16 = cast(dtype = d_to_fp16_dtype_0, x = d)[name = tensor<string, []>("cast_1")];
25
- tensor<fp16, [1, ?, 2000]> in_seg_to_fp16 = cast(dtype = cast_3_to_fp16_dtype_0, x = in_seg)[name = tensor<string, []>("cast_2")];
26
- tensor<fp16, [1, 640, 2000]> en = matmul(transpose_x = var_44_transpose_x_1, transpose_y = var_44_transpose_y_1, x = d_to_fp16, y = in_seg_to_fp16)[name = tensor<string, []>("op_44_cast_fp16")];
27
- tensor<bool, []> var_50_transpose_x_0 = const()[name = tensor<string, []>("op_50_transpose_x_0"), val = tensor<bool, []>(false)];
28
- tensor<bool, []> var_50_transpose_y_0 = const()[name = tensor<string, []>("op_50_transpose_y_0"), val = tensor<bool, []>(false)];
29
- tensor<string, []> t_en_to_fp16_dtype_0 = const()[name = tensor<string, []>("t_en_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
30
- tensor<fp16, [1, 512, ?]> t_en_to_fp16 = cast(dtype = t_en_to_fp16_dtype_0, x = t_en)[name = tensor<string, []>("cast_0")];
31
- tensor<fp16, [1, 512, 2000]> asr = matmul(transpose_x = var_50_transpose_x_0, transpose_y = var_50_transpose_y_0, x = t_en_to_fp16, y = in_seg_to_fp16)[name = tensor<string, []>("op_50_cast_fp16")];
32
- } -> (en, asr);
33
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_alignment.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:2e7d69128b59d615fc3d3cf85637a687235fc086b1eb136359adb11a61615f6b
3
- size 4128
 
 
 
 
ANE/styletts2_ane_alignment.mlpackage/Data/com.apple.CoreML/model.mlmodel DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:0a866791eb7d8e7fe4a54551d9c98d7e8a10bd442799862fa52dd81ac9c550e6
3
- size 4162
 
 
 
 
ANE/styletts2_ane_alignment.mlpackage/Data/com.apple.CoreML/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:2e7d69128b59d615fc3d3cf85637a687235fc086b1eb136359adb11a61615f6b
3
- size 4128
 
 
 
 
ANE/styletts2_ane_alignment.mlpackage/Manifest.json DELETED
@@ -1,18 +0,0 @@
1
- {
2
- "fileFormatVersion": "1.0.0",
3
- "itemInfoEntries": {
4
- "B5C37E4A-EDAD-42AA-A450-1BF7CC840C90": {
5
- "author": "com.apple.CoreML",
6
- "description": "CoreML Model Weights",
7
- "name": "weights",
8
- "path": "com.apple.CoreML/weights"
9
- },
10
- "DBA3FF95-5E96-49D4-A75D-BD281A186EE8": {
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": "DBA3FF95-5E96-49D4-A75D-BD281A186EE8"
18
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_diffusion_step.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:78d70f424ec17bfb8c861db787faf81ca410d2011942e5f50a7e2a251108d2f3
3
- size 243
 
 
 
 
ANE/styletts2_ane_diffusion_step.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:56190538745103a0483d4249d07ce77618a2e23ec537d6a06103ed9f33fa27b0
3
- size 472
 
 
 
 
ANE/styletts2_ane_diffusion_step.mlmodelc/metadata.json DELETED
@@ -1,115 +0,0 @@
1
- [
2
- {
3
- "metadataOutputVersion" : "3.0",
4
- "shortDescription" : "StyleTTS2-ANE Stage 4: ADPM2 single denoise step (static shapes) — fp16, ANE.",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float16",
10
- "formattedType" : "MultiArray (Float16 1 × 1 × 256)",
11
- "shortDescription" : "",
12
- "shape" : "[1, 1, 256]",
13
- "name" : "denoised",
14
- "type" : "MultiArray"
15
- }
16
- ],
17
- "storagePrecision" : "Mixed (Float16, Palettized (8 bits))",
18
- "modelParameters" : [
19
-
20
- ],
21
- "specificationVersion" : 8,
22
- "mlProgramOperationTypeHistogram" : {
23
- "Ios17.cos" : 1,
24
- "Ios17.reshape" : 20,
25
- "Ios17.inverse" : 2,
26
- "Ios16.softmax" : 3,
27
- "Ios17.matmul" : 6,
28
- "Ios17.sin" : 1,
29
- "Ios17.transpose" : 27,
30
- "Split" : 9,
31
- "Ios16.reduceMean" : 1,
32
- "Ios17.expandDims" : 2,
33
- "Ios17.add" : 23,
34
- "Tile" : 2,
35
- "Ios16.reduceSum" : 1,
36
- "Ios17.log" : 1,
37
- "Ios17.layerNorm" : 3,
38
- "Ios17.sqrt" : 1,
39
- "Ios16.gelu" : 7,
40
- "Ios16.constexprLutToDense" : 26,
41
- "Ios17.cast" : 4,
42
- "Ios17.realDiv" : 1,
43
- "Ios17.linear" : 25,
44
- "Stack" : 1,
45
- "Ios17.concat" : 2,
46
- "Ios17.conv" : 1,
47
- "Ios17.mul" : 19
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
- "inputSchema" : [
66
- {
67
- "hasShapeFlexibility" : "0",
68
- "isOptional" : "0",
69
- "dataType" : "Float32",
70
- "formattedType" : "MultiArray (Float32 1 × 1 × 256)",
71
- "shortDescription" : "",
72
- "shape" : "[1, 1, 256]",
73
- "name" : "x_noisy",
74
- "type" : "MultiArray"
75
- },
76
- {
77
- "hasShapeFlexibility" : "0",
78
- "isOptional" : "0",
79
- "dataType" : "Float32",
80
- "formattedType" : "MultiArray (Float32 1)",
81
- "shortDescription" : "",
82
- "shape" : "[1]",
83
- "name" : "sigma",
84
- "type" : "MultiArray"
85
- },
86
- {
87
- "hasShapeFlexibility" : "0",
88
- "isOptional" : "0",
89
- "dataType" : "Float32",
90
- "formattedType" : "MultiArray (Float32 1 × 512 × 768)",
91
- "shortDescription" : "",
92
- "shape" : "[1, 512, 768]",
93
- "name" : "embedding",
94
- "type" : "MultiArray"
95
- },
96
- {
97
- "hasShapeFlexibility" : "0",
98
- "isOptional" : "0",
99
- "dataType" : "Float32",
100
- "formattedType" : "MultiArray (Float32 1 × 256)",
101
- "shortDescription" : "",
102
- "shape" : "[1, 256]",
103
- "name" : "features",
104
- "type" : "MultiArray"
105
- }
106
- ],
107
- "userDefinedMetadata" : {
108
- "com.github.apple.coremltools.source_dialect" : "TorchScript",
109
- "com.github.apple.coremltools.source" : "torch==2.11.0",
110
- "com.github.apple.coremltools.version" : "8.3.0"
111
- },
112
- "generatedClassName" : "styletts2_ane_diffusion_step",
113
- "method" : "predict"
114
- }
115
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_diffusion_step.mlmodelc/model.mil DELETED
@@ -1,351 +0,0 @@
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, 512, 768]> embedding, tensor<fp32, [1, 256]> features, tensor<fp32, [1]> sigma, tensor<fp32, [1, 1, 256]> x_noisy) {
5
- tensor<int32, [3]> var_25 = const()[name = tensor<string, []>("op_25"), val = tensor<int32, [3]>([-1, 1, 1])];
6
- tensor<string, []> sigma_to_fp16_dtype_0 = const()[name = tensor<string, []>("sigma_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
7
- tensor<fp16, [1]> sigma_to_fp16 = cast(dtype = sigma_to_fp16_dtype_0, x = sigma)[name = tensor<string, []>("cast_3")];
8
- tensor<fp16, [1, 1, 1]> s_cast_fp16 = reshape(shape = var_25, x = sigma_to_fp16)[name = tensor<string, []>("s_cast_fp16")];
9
- tensor<fp16, [1, 1, 1]> var_27_cast_fp16 = mul(x = s_cast_fp16, y = s_cast_fp16)[name = tensor<string, []>("op_27_cast_fp16")];
10
- tensor<fp16, []> var_29_to_fp16 = const()[name = tensor<string, []>("op_29_to_fp16"), val = tensor<fp16, []>(0x1.47cp-5)];
11
- tensor<fp16, [1, 1, 1]> var_30_cast_fp16 = add(x = var_27_cast_fp16, y = var_29_to_fp16)[name = tensor<string, []>("op_30_cast_fp16")];
12
- tensor<fp32, []> var_31_epsilon_0 = const()[name = tensor<string, []>("op_31_epsilon_0"), val = tensor<fp32, []>(0x1.a36e2ep-14)];
13
- tensor<fp16, [1, 1, 1]> var_31_cast_fp16 = inverse(epsilon = var_31_epsilon_0, x = var_30_cast_fp16)[name = tensor<string, []>("op_31_cast_fp16")];
14
- tensor<fp16, []> var_32_to_fp16 = const()[name = tensor<string, []>("op_32_to_fp16"), val = tensor<fp16, []>(0x1.47cp-5)];
15
- tensor<fp16, [1, 1, 1]> c_skip_cast_fp16 = mul(x = var_31_cast_fp16, y = var_32_to_fp16)[name = tensor<string, []>("c_skip_cast_fp16")];
16
- tensor<fp16, []> var_34_to_fp16 = const()[name = tensor<string, []>("op_34_to_fp16"), val = tensor<fp16, []>(0x1.998p-3)];
17
- tensor<fp16, [1, 1, 1]> var_35_cast_fp16 = mul(x = s_cast_fp16, y = var_34_to_fp16)[name = tensor<string, []>("op_35_cast_fp16")];
18
- tensor<fp16, [1, 1, 1]> var_40_cast_fp16 = sqrt(x = var_30_cast_fp16)[name = tensor<string, []>("op_40_cast_fp16")];
19
- tensor<fp16, [1, 1, 1]> c_out_cast_fp16 = real_div(x = var_35_cast_fp16, y = var_40_cast_fp16)[name = tensor<string, []>("c_out_cast_fp16")];
20
- tensor<fp32, []> var_47_epsilon_0 = const()[name = tensor<string, []>("op_47_epsilon_0"), val = tensor<fp32, []>(0x1.a36e2ep-14)];
21
- tensor<fp16, [1, 1, 1]> var_47_cast_fp16 = inverse(epsilon = var_47_epsilon_0, x = var_40_cast_fp16)[name = tensor<string, []>("op_47_cast_fp16")];
22
- tensor<fp32, []> var_50_epsilon_0 = const()[name = tensor<string, []>("op_50_epsilon_0"), val = tensor<fp32, []>(0x1p-149)];
23
- tensor<fp16, [1]> var_50_cast_fp16 = log(epsilon = var_50_epsilon_0, x = sigma_to_fp16)[name = tensor<string, []>("op_50_cast_fp16")];
24
- tensor<fp16, []> var_51_to_fp16 = const()[name = tensor<string, []>("op_51_to_fp16"), val = tensor<fp16, []>(0x1p-2)];
25
- tensor<fp16, [1]> x_1_cast_fp16 = mul(x = var_50_cast_fp16, y = var_51_to_fp16)[name = tensor<string, []>("x_1_cast_fp16")];
26
- tensor<string, []> x_noisy_to_fp16_dtype_0 = const()[name = tensor<string, []>("x_noisy_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
27
- tensor<fp16, [1, 1, 256]> x_noisy_to_fp16 = cast(dtype = x_noisy_to_fp16_dtype_0, x = x_noisy)[name = tensor<string, []>("cast_2")];
28
- tensor<fp16, [1, 1, 256]> x_11_cast_fp16 = mul(x = var_47_cast_fp16, y = x_noisy_to_fp16)[name = tensor<string, []>("x_11_cast_fp16")];
29
- tensor<int32, []> var_55 = const()[name = tensor<string, []>("op_55"), val = tensor<int32, []>(-1)];
30
- tensor<int32, [2]> var_67 = const()[name = tensor<string, []>("op_67"), val = tensor<int32, [2]>([1, 1])];
31
- tensor<fp16, [1, 1]> x_5_cast_fp16 = reshape(shape = var_67, x = x_1_cast_fp16)[name = tensor<string, []>("x_5_cast_fp16")];
32
- tensor<fp16, [1, 128]> var_75_to_fp16 = const()[name = tensor<string, []>("op_75_to_fp16"), val = tensor<fp16, [1, 128]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64)))];
33
- tensor<fp16, [1, 128]> var_76_cast_fp16 = mul(x = x_5_cast_fp16, y = var_75_to_fp16)[name = tensor<string, []>("op_76_cast_fp16")];
34
- tensor<fp16, []> var_77_promoted_to_fp16 = const()[name = tensor<string, []>("op_77_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+1)];
35
- tensor<fp16, [1, 128]> var_78_cast_fp16 = mul(x = var_76_cast_fp16, y = var_77_promoted_to_fp16)[name = tensor<string, []>("op_78_cast_fp16")];
36
- tensor<fp16, []> var_79_to_fp16 = const()[name = tensor<string, []>("op_79_to_fp16"), val = tensor<fp16, []>(0x1.92p+1)];
37
- tensor<fp16, [1, 128]> freqs_cast_fp16 = mul(x = var_78_cast_fp16, y = var_79_to_fp16)[name = tensor<string, []>("freqs_cast_fp16")];
38
- tensor<fp16, [1, 128]> var_81_cast_fp16 = sin(x = freqs_cast_fp16)[name = tensor<string, []>("op_81_cast_fp16")];
39
- tensor<fp16, [1, 128]> var_82_cast_fp16 = cos(x = freqs_cast_fp16)[name = tensor<string, []>("op_82_cast_fp16")];
40
- tensor<bool, []> input_1_interleave_0 = const()[name = tensor<string, []>("input_1_interleave_0"), val = tensor<bool, []>(false)];
41
- tensor<fp16, [1, 257]> input_1_cast_fp16 = concat(axis = var_55, interleave = input_1_interleave_0, values = (x_5_cast_fp16, var_81_cast_fp16, var_82_cast_fp16))[name = tensor<string, []>("input_1_cast_fp16")];
42
- tensor<fp16, [1024, 257]> transformer_to_time_0_1_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [263168]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(384))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(263616))), name = tensor<string, []>("transformer_to_time_0_1_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 257])];
43
- tensor<fp16, [1024]> transformer_to_time_0_1_bias_to_fp16 = const()[name = tensor<string, []>("transformer_to_time_0_1_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(264192)))];
44
- tensor<fp16, [1, 1024]> linear_0_cast_fp16 = linear(bias = transformer_to_time_0_1_bias_to_fp16, weight = transformer_to_time_0_1_weight_to_fp16_palettized, x = input_1_cast_fp16)[name = tensor<string, []>("linear_0_cast_fp16")];
45
- tensor<string, []> var_88_mode_0 = const()[name = tensor<string, []>("op_88_mode_0"), val = tensor<string, []>("EXACT")];
46
- tensor<fp16, [1, 1024]> var_88_cast_fp16 = gelu(mode = var_88_mode_0, x = linear_0_cast_fp16)[name = tensor<string, []>("op_88_cast_fp16")];
47
- tensor<string, []> features_to_fp16_dtype_0 = const()[name = tensor<string, []>("features_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
48
- tensor<fp16, [1024, 256]> transformer_to_features_0_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(266304))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(528512))), name = tensor<string, []>("transformer_to_features_0_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
49
- tensor<fp16, [1024]> transformer_to_features_0_bias_to_fp16 = const()[name = tensor<string, []>("transformer_to_features_0_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(529088)))];
50
- tensor<fp16, [1, 256]> features_to_fp16 = cast(dtype = features_to_fp16_dtype_0, x = features)[name = tensor<string, []>("cast_1")];
51
- tensor<fp16, [1, 1024]> linear_1_cast_fp16 = linear(bias = transformer_to_features_0_bias_to_fp16, weight = transformer_to_features_0_weight_to_fp16_palettized, x = features_to_fp16)[name = tensor<string, []>("linear_1_cast_fp16")];
52
- tensor<string, []> var_94_mode_0 = const()[name = tensor<string, []>("op_94_mode_0"), val = tensor<string, []>("EXACT")];
53
- tensor<fp16, [1, 1024]> var_94_cast_fp16 = gelu(mode = var_94_mode_0, x = linear_1_cast_fp16)[name = tensor<string, []>("op_94_cast_fp16")];
54
- tensor<int32, []> x_7_axis_0 = const()[name = tensor<string, []>("x_7_axis_0"), val = tensor<int32, []>(0)];
55
- tensor<fp16, [2, 1, 1024]> x_7_cast_fp16 = stack(axis = x_7_axis_0, values = (var_88_cast_fp16, var_94_cast_fp16))[name = tensor<string, []>("x_7_cast_fp16")];
56
- tensor<int32, [3]> var_101 = const()[name = tensor<string, []>("op_101"), val = tensor<int32, [3]>([1, 2, 0])];
57
- tensor<int32, [1]> input_7_axes_0 = const()[name = tensor<string, []>("input_7_axes_0"), val = tensor<int32, [1]>([2])];
58
- tensor<bool, []> input_7_keep_dims_0 = const()[name = tensor<string, []>("input_7_keep_dims_0"), val = tensor<bool, []>(false)];
59
- tensor<fp16, [1, 1024, 2]> x_9_cast_fp16 = transpose(perm = var_101, x = x_7_cast_fp16)[name = tensor<string, []>("transpose_44")];
60
- tensor<fp16, [1, 1024]> input_7_cast_fp16 = reduce_sum(axes = input_7_axes_0, keep_dims = input_7_keep_dims_0, x = x_9_cast_fp16)[name = tensor<string, []>("input_7_cast_fp16")];
61
- tensor<fp16, [1024, 1024]> transformer_to_mapping_0_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [1048576]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(531200))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1579840))), name = tensor<string, []>("transformer_to_mapping_0_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 1024])];
62
- tensor<fp16, [1024]> transformer_to_mapping_0_bias_to_fp16 = const()[name = tensor<string, []>("transformer_to_mapping_0_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1580416)))];
63
- tensor<fp16, [1, 1024]> linear_2_cast_fp16 = linear(bias = transformer_to_mapping_0_bias_to_fp16, weight = transformer_to_mapping_0_weight_to_fp16_palettized, x = input_7_cast_fp16)[name = tensor<string, []>("linear_2_cast_fp16")];
64
- tensor<string, []> input_11_mode_0 = const()[name = tensor<string, []>("input_11_mode_0"), val = tensor<string, []>("EXACT")];
65
- tensor<fp16, [1, 1024]> input_11_cast_fp16 = gelu(mode = input_11_mode_0, x = linear_2_cast_fp16)[name = tensor<string, []>("input_11_cast_fp16")];
66
- tensor<fp16, [1024, 1024]> transformer_to_mapping_2_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [1048576]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1582528))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2631168))), name = tensor<string, []>("transformer_to_mapping_2_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 1024])];
67
- tensor<fp16, [1024]> transformer_to_mapping_2_bias_to_fp16 = const()[name = tensor<string, []>("transformer_to_mapping_2_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2631744)))];
68
- tensor<fp16, [1, 1024]> linear_3_cast_fp16 = linear(bias = transformer_to_mapping_2_bias_to_fp16, weight = transformer_to_mapping_2_weight_to_fp16_palettized, x = input_11_cast_fp16)[name = tensor<string, []>("linear_3_cast_fp16")];
69
- tensor<string, []> mapping_1_mode_0 = const()[name = tensor<string, []>("mapping_1_mode_0"), val = tensor<string, []>("EXACT")];
70
- tensor<fp16, [1, 1024]> mapping_1_cast_fp16 = gelu(mode = mapping_1_mode_0, x = linear_3_cast_fp16)[name = tensor<string, []>("mapping_1_cast_fp16")];
71
- tensor<int32, [3]> var_127_reps_0 = const()[name = tensor<string, []>("op_127_reps_0"), val = tensor<int32, [3]>([1, 512, 1])];
72
- tensor<fp16, [1, 512, 256]> var_127_cast_fp16 = tile(reps = var_127_reps_0, x = x_11_cast_fp16)[name = tensor<string, []>("op_127_cast_fp16")];
73
- tensor<int32, []> var_129 = const()[name = tensor<string, []>("op_129"), val = tensor<int32, []>(-1)];
74
- tensor<bool, []> x_13_interleave_0 = const()[name = tensor<string, []>("x_13_interleave_0"), val = tensor<bool, []>(false)];
75
- tensor<string, []> embedding_to_fp16_dtype_0 = const()[name = tensor<string, []>("embedding_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
76
- tensor<fp16, [1, 512, 768]> embedding_to_fp16 = cast(dtype = embedding_to_fp16_dtype_0, x = embedding)[name = tensor<string, []>("cast_0")];
77
- tensor<fp16, [1, 512, 1024]> x_13_cast_fp16 = concat(axis = var_129, interleave = x_13_interleave_0, values = (var_127_cast_fp16, embedding_to_fp16))[name = tensor<string, []>("x_13_cast_fp16")];
78
- tensor<int32, [1]> var_132_axes_0 = const()[name = tensor<string, []>("op_132_axes_0"), val = tensor<int32, [1]>([1])];
79
- tensor<fp16, [1, 1, 1024]> var_132_cast_fp16 = expand_dims(axes = var_132_axes_0, x = mapping_1_cast_fp16)[name = tensor<string, []>("op_132_cast_fp16")];
80
- tensor<int32, [3]> mapping_reps_0 = const()[name = tensor<string, []>("mapping_reps_0"), val = tensor<int32, [3]>([1, 512, 1])];
81
- tensor<fp16, [1, 512, 1024]> mapping_cast_fp16 = tile(reps = mapping_reps_0, x = var_132_cast_fp16)[name = tensor<string, []>("mapping_cast_fp16")];
82
- tensor<fp16, [1, 512, 1024]> x_15_cast_fp16 = add(x = x_13_cast_fp16, y = mapping_cast_fp16)[name = tensor<string, []>("x_15_cast_fp16")];
83
- tensor<int32, []> var_153 = const()[name = tensor<string, []>("op_153"), val = tensor<int32, []>(-1)];
84
- tensor<fp16, [2048, 256]> transformer_blocks_0_attention_norm_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2633856))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3158208))), name = tensor<string, []>("transformer_blocks_0_attention_norm_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([2048, 256])];
85
- tensor<fp16, [2048]> transformer_blocks_0_attention_norm_fc_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_0_attention_norm_fc_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3158784)))];
86
- tensor<fp16, [1, 2048]> linear_4_cast_fp16 = linear(bias = transformer_blocks_0_attention_norm_fc_bias_to_fp16, weight = transformer_blocks_0_attention_norm_fc_weight_to_fp16_palettized, x = features_to_fp16)[name = tensor<string, []>("linear_4_cast_fp16")];
87
- tensor<int32, [3]> var_172 = const()[name = tensor<string, []>("op_172"), val = tensor<int32, [3]>([1, 2048, 1])];
88
- tensor<fp16, [1, 2048, 1]> h_3_cast_fp16 = reshape(shape = var_172, x = linear_4_cast_fp16)[name = tensor<string, []>("h_3_cast_fp16")];
89
- tensor<int32, [2]> var_174_split_sizes_0 = const()[name = tensor<string, []>("op_174_split_sizes_0"), val = tensor<int32, [2]>([1024, 1024])];
90
- tensor<int32, []> var_174_axis_0 = const()[name = tensor<string, []>("op_174_axis_0"), val = tensor<int32, []>(1)];
91
- tensor<fp16, [1, 1024, 1]> var_174_cast_fp16_0, tensor<fp16, [1, 1024, 1]> var_174_cast_fp16_1 = split(axis = var_174_axis_0, split_sizes = var_174_split_sizes_0, x = h_3_cast_fp16)[name = tensor<string, []>("op_174_cast_fp16")];
92
- tensor<int32, [3]> gamma_3_perm_0 = const()[name = tensor<string, []>("gamma_3_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
93
- tensor<int32, [3]> beta_3_perm_0 = const()[name = tensor<string, []>("beta_3_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
94
- tensor<int32, [1]> x_19_axes_0 = const()[name = tensor<string, []>("x_19_axes_0"), val = tensor<int32, [1]>([-1])];
95
- tensor<fp16, []> var_146_to_fp16 = const()[name = tensor<string, []>("op_146_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
96
- tensor<fp16, [1, 512, 1024]> x_19_cast_fp16 = layer_norm(axes = x_19_axes_0, epsilon = var_146_to_fp16, x = x_15_cast_fp16)[name = tensor<string, []>("x_19_cast_fp16")];
97
- tensor<fp16, []> var_180_promoted_to_fp16 = const()[name = tensor<string, []>("op_180_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
98
- tensor<fp16, [1, 1, 1024]> gamma_3_cast_fp16 = transpose(perm = gamma_3_perm_0, x = var_174_cast_fp16_0)[name = tensor<string, []>("transpose_43")];
99
- tensor<fp16, [1, 1, 1024]> var_181_cast_fp16 = add(x = gamma_3_cast_fp16, y = var_180_promoted_to_fp16)[name = tensor<string, []>("op_181_cast_fp16")];
100
- tensor<fp16, [1, 512, 1024]> var_182_cast_fp16 = mul(x = var_181_cast_fp16, y = x_19_cast_fp16)[name = tensor<string, []>("op_182_cast_fp16")];
101
- tensor<fp16, [1, 1, 1024]> beta_3_cast_fp16 = transpose(perm = beta_3_perm_0, x = var_174_cast_fp16_1)[name = tensor<string, []>("transpose_42")];
102
- tensor<fp16, [1, 512, 1024]> x_21_cast_fp16 = add(x = var_182_cast_fp16, y = beta_3_cast_fp16)[name = tensor<string, []>("x_21_cast_fp16")];
103
- tensor<fp16, [2048, 256]> transformer_blocks_0_attention_norm_context_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3162944))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3687296))), name = tensor<string, []>("transformer_blocks_0_attention_norm_context_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([2048, 256])];
104
- tensor<fp16, [2048]> transformer_blocks_0_attention_norm_context_fc_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_0_attention_norm_context_fc_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3687872)))];
105
- tensor<fp16, [1, 2048]> linear_5_cast_fp16 = linear(bias = transformer_blocks_0_attention_norm_context_fc_bias_to_fp16, weight = transformer_blocks_0_attention_norm_context_fc_weight_to_fp16_palettized, x = features_to_fp16)[name = tensor<string, []>("linear_5_cast_fp16")];
106
- tensor<int32, [3]> var_194 = const()[name = tensor<string, []>("op_194"), val = tensor<int32, [3]>([1, 2048, 1])];
107
- tensor<fp16, [1, 2048, 1]> h_7_cast_fp16 = reshape(shape = var_194, x = linear_5_cast_fp16)[name = tensor<string, []>("h_7_cast_fp16")];
108
- tensor<int32, [2]> var_196_split_sizes_0 = const()[name = tensor<string, []>("op_196_split_sizes_0"), val = tensor<int32, [2]>([1024, 1024])];
109
- tensor<int32, []> var_196_axis_0 = const()[name = tensor<string, []>("op_196_axis_0"), val = tensor<int32, []>(1)];
110
- tensor<fp16, [1, 1024, 1]> var_196_cast_fp16_0, tensor<fp16, [1, 1024, 1]> var_196_cast_fp16_1 = split(axis = var_196_axis_0, split_sizes = var_196_split_sizes_0, x = h_7_cast_fp16)[name = tensor<string, []>("op_196_cast_fp16")];
111
- tensor<int32, [3]> gamma_7_perm_0 = const()[name = tensor<string, []>("gamma_7_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
112
- tensor<int32, [3]> beta_7_perm_0 = const()[name = tensor<string, []>("beta_7_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
113
- tensor<fp16, []> var_202_promoted_to_fp16 = const()[name = tensor<string, []>("op_202_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
114
- tensor<fp16, [1, 1, 1024]> gamma_7_cast_fp16 = transpose(perm = gamma_7_perm_0, x = var_196_cast_fp16_0)[name = tensor<string, []>("transpose_41")];
115
- tensor<fp16, [1, 1, 1024]> var_203_cast_fp16 = add(x = gamma_7_cast_fp16, y = var_202_promoted_to_fp16)[name = tensor<string, []>("op_203_cast_fp16")];
116
- tensor<fp16, [1, 512, 1024]> var_204_cast_fp16 = mul(x = var_203_cast_fp16, y = x_19_cast_fp16)[name = tensor<string, []>("op_204_cast_fp16")];
117
- tensor<fp16, [1, 1, 1024]> beta_7_cast_fp16 = transpose(perm = beta_7_perm_0, x = var_196_cast_fp16_1)[name = tensor<string, []>("transpose_40")];
118
- tensor<fp16, [1, 512, 1024]> x_27_cast_fp16 = add(x = var_204_cast_fp16, y = beta_7_cast_fp16)[name = tensor<string, []>("x_27_cast_fp16")];
119
- tensor<fp16, [512, 1024]> transformer_blocks_0_attention_to_q_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3692032))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4216384))), name = tensor<string, []>("transformer_blocks_0_attention_to_q_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 1024])];
120
- tensor<fp16, [512]> linear_6_bias_0_to_fp16 = const()[name = tensor<string, []>("linear_6_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4216960)))];
121
- tensor<fp16, [1, 512, 512]> linear_6_cast_fp16 = linear(bias = linear_6_bias_0_to_fp16, weight = transformer_blocks_0_attention_to_q_weight_to_fp16_palettized, x = x_21_cast_fp16)[name = tensor<string, []>("linear_6_cast_fp16")];
122
- tensor<fp16, [1024, 1024]> transformer_blocks_0_attention_to_kv_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [1048576]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4218048))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5266688))), name = tensor<string, []>("transformer_blocks_0_attention_to_kv_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 1024])];
123
- tensor<fp16, [1024]> linear_7_bias_0_to_fp16 = const()[name = tensor<string, []>("linear_7_bias_0_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5267264)))];
124
- tensor<fp16, [1, 512, 1024]> linear_7_cast_fp16 = linear(bias = linear_7_bias_0_to_fp16, weight = transformer_blocks_0_attention_to_kv_weight_to_fp16_palettized, x = x_27_cast_fp16)[name = tensor<string, []>("linear_7_cast_fp16")];
125
- tensor<int32, [2]> var_212_split_sizes_0 = const()[name = tensor<string, []>("op_212_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
126
- tensor<int32, []> var_212_axis_0 = const()[name = tensor<string, []>("op_212_axis_0"), val = tensor<int32, []>(-1)];
127
- tensor<fp16, [1, 512, 512]> var_212_cast_fp16_0, tensor<fp16, [1, 512, 512]> var_212_cast_fp16_1 = split(axis = var_212_axis_0, split_sizes = var_212_split_sizes_0, x = linear_7_cast_fp16)[name = tensor<string, []>("op_212_cast_fp16")];
128
- tensor<int32, [4]> var_221 = const()[name = tensor<string, []>("op_221"), val = tensor<int32, [4]>([1, 512, 8, 64])];
129
- tensor<fp16, [1, 512, 8, 64]> x_31_cast_fp16 = reshape(shape = var_221, x = linear_6_cast_fp16)[name = tensor<string, []>("x_31_cast_fp16")];
130
- tensor<int32, [4]> var_231 = const()[name = tensor<string, []>("op_231"), val = tensor<int32, [4]>([1, 512, 8, 64])];
131
- tensor<fp16, [1, 512, 8, 64]> x_35_cast_fp16 = reshape(shape = var_231, x = var_212_cast_fp16_0)[name = tensor<string, []>("x_35_cast_fp16")];
132
- tensor<int32, [4]> var_241 = const()[name = tensor<string, []>("op_241"), val = tensor<int32, [4]>([1, 512, 8, 64])];
133
- tensor<fp16, [1, 512, 8, 64]> x_39_cast_fp16 = reshape(shape = var_241, x = var_212_cast_fp16_1)[name = tensor<string, []>("x_39_cast_fp16")];
134
- tensor<int32, [4]> var_243 = const()[name = tensor<string, []>("op_243"), val = tensor<int32, [4]>([0, 2, -3, -1])];
135
- tensor<bool, []> sim_1_transpose_x_0 = const()[name = tensor<string, []>("sim_1_transpose_x_0"), val = tensor<bool, []>(false)];
136
- tensor<bool, []> sim_1_transpose_y_0 = const()[name = tensor<string, []>("sim_1_transpose_y_0"), val = tensor<bool, []>(false)];
137
- tensor<int32, [4]> transpose_12_perm_0 = const()[name = tensor<string, []>("transpose_12_perm_0"), val = tensor<int32, [4]>([0, 2, -3, -1])];
138
- tensor<int32, [4]> transpose_13_perm_0 = const()[name = tensor<string, []>("transpose_13_perm_0"), val = tensor<int32, [4]>([0, 2, -1, -3])];
139
- tensor<fp16, [1, 8, 64, 512]> transpose_13 = transpose(perm = transpose_13_perm_0, x = x_35_cast_fp16)[name = tensor<string, []>("transpose_38")];
140
- tensor<fp16, [1, 8, 512, 64]> transpose_12 = transpose(perm = transpose_12_perm_0, x = x_31_cast_fp16)[name = tensor<string, []>("transpose_39")];
141
- tensor<fp16, [1, 8, 512, 512]> sim_1_cast_fp16 = matmul(transpose_x = sim_1_transpose_x_0, transpose_y = sim_1_transpose_y_0, x = transpose_12, y = transpose_13)[name = tensor<string, []>("sim_1_cast_fp16")];
142
- tensor<fp16, []> var_247_to_fp16 = const()[name = tensor<string, []>("op_247_to_fp16"), val = tensor<fp16, []>(0x1p-3)];
143
- tensor<fp16, [1, 8, 512, 512]> sim_3_cast_fp16 = mul(x = sim_1_cast_fp16, y = var_247_to_fp16)[name = tensor<string, []>("sim_3_cast_fp16")];
144
- tensor<fp16, [1, 8, 512, 512]> attn_1_cast_fp16 = softmax(axis = var_153, x = sim_3_cast_fp16)[name = tensor<string, []>("attn_1_cast_fp16")];
145
- tensor<bool, []> x_41_transpose_x_0 = const()[name = tensor<string, []>("x_41_transpose_x_0"), val = tensor<bool, []>(false)];
146
- tensor<bool, []> x_41_transpose_y_0 = const()[name = tensor<string, []>("x_41_transpose_y_0"), val = tensor<bool, []>(false)];
147
- tensor<fp16, [1, 8, 512, 64]> v_1_cast_fp16 = transpose(perm = var_243, x = x_39_cast_fp16)[name = tensor<string, []>("transpose_37")];
148
- tensor<fp16, [1, 8, 512, 64]> x_41_cast_fp16 = matmul(transpose_x = x_41_transpose_x_0, transpose_y = x_41_transpose_y_0, x = attn_1_cast_fp16, y = v_1_cast_fp16)[name = tensor<string, []>("x_41_cast_fp16")];
149
- tensor<int32, [4]> var_269 = const()[name = tensor<string, []>("op_269"), val = tensor<int32, [4]>([0, 2, 1, 3])];
150
- tensor<int32, [3]> var_271 = const()[name = tensor<string, []>("op_271"), val = tensor<int32, [3]>([1, 512, 512])];
151
- tensor<fp16, [1, 512, 8, 64]> x_43_cast_fp16 = transpose(perm = var_269, x = x_41_cast_fp16)[name = tensor<string, []>("transpose_36")];
152
- tensor<fp16, [1, 512, 512]> input_23_cast_fp16 = reshape(shape = var_271, x = x_43_cast_fp16)[name = tensor<string, []>("input_23_cast_fp16")];
153
- tensor<fp16, [1024, 512]> transformer_blocks_0_attention_attention_to_out_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5269376))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5793728))), name = tensor<string, []>("transformer_blocks_0_attention_attention_to_out_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 512])];
154
- tensor<fp16, [1024]> transformer_blocks_0_attention_attention_to_out_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_0_attention_attention_to_out_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5794304)))];
155
- tensor<fp16, [1, 512, 1024]> linear_8_cast_fp16 = linear(bias = transformer_blocks_0_attention_attention_to_out_bias_to_fp16, weight = transformer_blocks_0_attention_attention_to_out_weight_to_fp16_palettized, x = input_23_cast_fp16)[name = tensor<string, []>("linear_8_cast_fp16")];
156
- tensor<fp16, [1, 512, 1024]> input_25_cast_fp16 = add(x = linear_8_cast_fp16, y = x_15_cast_fp16)[name = tensor<string, []>("input_25_cast_fp16")];
157
- tensor<fp16, [2048, 1024]> transformer_blocks_0_feed_forward_0_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [2097152]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5796416))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7893632))), name = tensor<string, []>("transformer_blocks_0_feed_forward_0_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([2048, 1024])];
158
- tensor<fp16, [2048]> transformer_blocks_0_feed_forward_0_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_0_feed_forward_0_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7894208)))];
159
- tensor<fp16, [1, 512, 2048]> linear_9_cast_fp16 = linear(bias = transformer_blocks_0_feed_forward_0_bias_to_fp16, weight = transformer_blocks_0_feed_forward_0_weight_to_fp16_palettized, x = input_25_cast_fp16)[name = tensor<string, []>("linear_9_cast_fp16")];
160
- tensor<string, []> input_29_mode_0 = const()[name = tensor<string, []>("input_29_mode_0"), val = tensor<string, []>("EXACT")];
161
- tensor<fp16, [1, 512, 2048]> input_29_cast_fp16 = gelu(mode = input_29_mode_0, x = linear_9_cast_fp16)[name = tensor<string, []>("input_29_cast_fp16")];
162
- tensor<fp16, [1024, 2048]> transformer_blocks_0_feed_forward_2_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [2097152]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7898368))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9995584))), name = tensor<string, []>("transformer_blocks_0_feed_forward_2_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 2048])];
163
- tensor<fp16, [1024]> transformer_blocks_0_feed_forward_2_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_0_feed_forward_2_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9996160)))];
164
- tensor<fp16, [1, 512, 1024]> linear_10_cast_fp16 = linear(bias = transformer_blocks_0_feed_forward_2_bias_to_fp16, weight = transformer_blocks_0_feed_forward_2_weight_to_fp16_palettized, x = input_29_cast_fp16)[name = tensor<string, []>("linear_10_cast_fp16")];
165
- tensor<fp16, [1, 512, 1024]> x_45_cast_fp16 = add(x = linear_10_cast_fp16, y = input_25_cast_fp16)[name = tensor<string, []>("x_45_cast_fp16")];
166
- tensor<fp16, [1, 512, 1024]> x_47_cast_fp16 = add(x = x_45_cast_fp16, y = mapping_cast_fp16)[name = tensor<string, []>("x_47_cast_fp16")];
167
- tensor<int32, []> var_298 = const()[name = tensor<string, []>("op_298"), val = tensor<int32, []>(-1)];
168
- tensor<fp16, [2048, 256]> transformer_blocks_1_attention_norm_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9998272))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10522624))), name = tensor<string, []>("transformer_blocks_1_attention_norm_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([2048, 256])];
169
- tensor<fp16, [2048]> transformer_blocks_1_attention_norm_fc_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_1_attention_norm_fc_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10523200)))];
170
- tensor<fp16, [1, 2048]> linear_11_cast_fp16 = linear(bias = transformer_blocks_1_attention_norm_fc_bias_to_fp16, weight = transformer_blocks_1_attention_norm_fc_weight_to_fp16_palettized, x = features_to_fp16)[name = tensor<string, []>("linear_11_cast_fp16")];
171
- tensor<int32, [3]> var_317 = const()[name = tensor<string, []>("op_317"), val = tensor<int32, [3]>([1, 2048, 1])];
172
- tensor<fp16, [1, 2048, 1]> h_11_cast_fp16 = reshape(shape = var_317, x = linear_11_cast_fp16)[name = tensor<string, []>("h_11_cast_fp16")];
173
- tensor<int32, [2]> var_319_split_sizes_0 = const()[name = tensor<string, []>("op_319_split_sizes_0"), val = tensor<int32, [2]>([1024, 1024])];
174
- tensor<int32, []> var_319_axis_0 = const()[name = tensor<string, []>("op_319_axis_0"), val = tensor<int32, []>(1)];
175
- tensor<fp16, [1, 1024, 1]> var_319_cast_fp16_0, tensor<fp16, [1, 1024, 1]> var_319_cast_fp16_1 = split(axis = var_319_axis_0, split_sizes = var_319_split_sizes_0, x = h_11_cast_fp16)[name = tensor<string, []>("op_319_cast_fp16")];
176
- tensor<int32, [3]> gamma_11_perm_0 = const()[name = tensor<string, []>("gamma_11_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
177
- tensor<int32, [3]> beta_11_perm_0 = const()[name = tensor<string, []>("beta_11_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
178
- tensor<int32, [1]> x_51_axes_0 = const()[name = tensor<string, []>("x_51_axes_0"), val = tensor<int32, [1]>([-1])];
179
- tensor<fp16, []> var_291_to_fp16 = const()[name = tensor<string, []>("op_291_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
180
- tensor<fp16, [1, 512, 1024]> x_51_cast_fp16 = layer_norm(axes = x_51_axes_0, epsilon = var_291_to_fp16, x = x_47_cast_fp16)[name = tensor<string, []>("x_51_cast_fp16")];
181
- tensor<fp16, []> var_325_promoted_to_fp16 = const()[name = tensor<string, []>("op_325_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
182
- tensor<fp16, [1, 1, 1024]> gamma_11_cast_fp16 = transpose(perm = gamma_11_perm_0, x = var_319_cast_fp16_0)[name = tensor<string, []>("transpose_35")];
183
- tensor<fp16, [1, 1, 1024]> var_326_cast_fp16 = add(x = gamma_11_cast_fp16, y = var_325_promoted_to_fp16)[name = tensor<string, []>("op_326_cast_fp16")];
184
- tensor<fp16, [1, 512, 1024]> var_327_cast_fp16 = mul(x = var_326_cast_fp16, y = x_51_cast_fp16)[name = tensor<string, []>("op_327_cast_fp16")];
185
- tensor<fp16, [1, 1, 1024]> beta_11_cast_fp16 = transpose(perm = beta_11_perm_0, x = var_319_cast_fp16_1)[name = tensor<string, []>("transpose_34")];
186
- tensor<fp16, [1, 512, 1024]> x_53_cast_fp16 = add(x = var_327_cast_fp16, y = beta_11_cast_fp16)[name = tensor<string, []>("x_53_cast_fp16")];
187
- tensor<fp16, [2048, 256]> transformer_blocks_1_attention_norm_context_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10527360))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11051712))), name = tensor<string, []>("transformer_blocks_1_attention_norm_context_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([2048, 256])];
188
- tensor<fp16, [2048]> transformer_blocks_1_attention_norm_context_fc_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_1_attention_norm_context_fc_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11052288)))];
189
- tensor<fp16, [1, 2048]> linear_12_cast_fp16 = linear(bias = transformer_blocks_1_attention_norm_context_fc_bias_to_fp16, weight = transformer_blocks_1_attention_norm_context_fc_weight_to_fp16_palettized, x = features_to_fp16)[name = tensor<string, []>("linear_12_cast_fp16")];
190
- tensor<int32, [3]> var_339 = const()[name = tensor<string, []>("op_339"), val = tensor<int32, [3]>([1, 2048, 1])];
191
- tensor<fp16, [1, 2048, 1]> h_15_cast_fp16 = reshape(shape = var_339, x = linear_12_cast_fp16)[name = tensor<string, []>("h_15_cast_fp16")];
192
- tensor<int32, [2]> var_341_split_sizes_0 = const()[name = tensor<string, []>("op_341_split_sizes_0"), val = tensor<int32, [2]>([1024, 1024])];
193
- tensor<int32, []> var_341_axis_0 = const()[name = tensor<string, []>("op_341_axis_0"), val = tensor<int32, []>(1)];
194
- tensor<fp16, [1, 1024, 1]> var_341_cast_fp16_0, tensor<fp16, [1, 1024, 1]> var_341_cast_fp16_1 = split(axis = var_341_axis_0, split_sizes = var_341_split_sizes_0, x = h_15_cast_fp16)[name = tensor<string, []>("op_341_cast_fp16")];
195
- tensor<int32, [3]> gamma_15_perm_0 = const()[name = tensor<string, []>("gamma_15_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
196
- tensor<int32, [3]> beta_15_perm_0 = const()[name = tensor<string, []>("beta_15_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
197
- tensor<fp16, []> var_347_promoted_to_fp16 = const()[name = tensor<string, []>("op_347_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
198
- tensor<fp16, [1, 1, 1024]> gamma_15_cast_fp16 = transpose(perm = gamma_15_perm_0, x = var_341_cast_fp16_0)[name = tensor<string, []>("transpose_33")];
199
- tensor<fp16, [1, 1, 1024]> var_348_cast_fp16 = add(x = gamma_15_cast_fp16, y = var_347_promoted_to_fp16)[name = tensor<string, []>("op_348_cast_fp16")];
200
- tensor<fp16, [1, 512, 1024]> var_349_cast_fp16 = mul(x = var_348_cast_fp16, y = x_51_cast_fp16)[name = tensor<string, []>("op_349_cast_fp16")];
201
- tensor<fp16, [1, 1, 1024]> beta_15_cast_fp16 = transpose(perm = beta_15_perm_0, x = var_341_cast_fp16_1)[name = tensor<string, []>("transpose_32")];
202
- tensor<fp16, [1, 512, 1024]> x_59_cast_fp16 = add(x = var_349_cast_fp16, y = beta_15_cast_fp16)[name = tensor<string, []>("x_59_cast_fp16")];
203
- tensor<fp16, [512, 1024]> transformer_blocks_1_attention_to_q_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11056448))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11580800))), name = tensor<string, []>("transformer_blocks_1_attention_to_q_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 1024])];
204
- tensor<fp16, [1, 512, 512]> linear_13_cast_fp16 = linear(bias = linear_6_bias_0_to_fp16, weight = transformer_blocks_1_attention_to_q_weight_to_fp16_palettized, x = x_53_cast_fp16)[name = tensor<string, []>("linear_13_cast_fp16")];
205
- tensor<fp16, [1024, 1024]> transformer_blocks_1_attention_to_kv_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [1048576]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11581376))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12630016))), name = tensor<string, []>("transformer_blocks_1_attention_to_kv_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 1024])];
206
- tensor<fp16, [1, 512, 1024]> linear_14_cast_fp16 = linear(bias = linear_7_bias_0_to_fp16, weight = transformer_blocks_1_attention_to_kv_weight_to_fp16_palettized, x = x_59_cast_fp16)[name = tensor<string, []>("linear_14_cast_fp16")];
207
- tensor<int32, [2]> var_357_split_sizes_0 = const()[name = tensor<string, []>("op_357_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
208
- tensor<int32, []> var_357_axis_0 = const()[name = tensor<string, []>("op_357_axis_0"), val = tensor<int32, []>(-1)];
209
- tensor<fp16, [1, 512, 512]> var_357_cast_fp16_0, tensor<fp16, [1, 512, 512]> var_357_cast_fp16_1 = split(axis = var_357_axis_0, split_sizes = var_357_split_sizes_0, x = linear_14_cast_fp16)[name = tensor<string, []>("op_357_cast_fp16")];
210
- tensor<int32, [4]> var_366 = const()[name = tensor<string, []>("op_366"), val = tensor<int32, [4]>([1, 512, 8, 64])];
211
- tensor<fp16, [1, 512, 8, 64]> x_63_cast_fp16 = reshape(shape = var_366, x = linear_13_cast_fp16)[name = tensor<string, []>("x_63_cast_fp16")];
212
- tensor<int32, [4]> var_376 = const()[name = tensor<string, []>("op_376"), val = tensor<int32, [4]>([1, 512, 8, 64])];
213
- tensor<fp16, [1, 512, 8, 64]> x_67_cast_fp16 = reshape(shape = var_376, x = var_357_cast_fp16_0)[name = tensor<string, []>("x_67_cast_fp16")];
214
- tensor<int32, [4]> var_386 = const()[name = tensor<string, []>("op_386"), val = tensor<int32, [4]>([1, 512, 8, 64])];
215
- tensor<fp16, [1, 512, 8, 64]> x_71_cast_fp16 = reshape(shape = var_386, x = var_357_cast_fp16_1)[name = tensor<string, []>("x_71_cast_fp16")];
216
- tensor<int32, [4]> var_388 = const()[name = tensor<string, []>("op_388"), val = tensor<int32, [4]>([0, 2, -3, -1])];
217
- tensor<bool, []> sim_5_transpose_x_0 = const()[name = tensor<string, []>("sim_5_transpose_x_0"), val = tensor<bool, []>(false)];
218
- tensor<bool, []> sim_5_transpose_y_0 = const()[name = tensor<string, []>("sim_5_transpose_y_0"), val = tensor<bool, []>(false)];
219
- tensor<int32, [4]> transpose_14_perm_0 = const()[name = tensor<string, []>("transpose_14_perm_0"), val = tensor<int32, [4]>([0, 2, -3, -1])];
220
- tensor<int32, [4]> transpose_15_perm_0 = const()[name = tensor<string, []>("transpose_15_perm_0"), val = tensor<int32, [4]>([0, 2, -1, -3])];
221
- tensor<fp16, [1, 8, 64, 512]> transpose_15 = transpose(perm = transpose_15_perm_0, x = x_67_cast_fp16)[name = tensor<string, []>("transpose_30")];
222
- tensor<fp16, [1, 8, 512, 64]> transpose_14 = transpose(perm = transpose_14_perm_0, x = x_63_cast_fp16)[name = tensor<string, []>("transpose_31")];
223
- tensor<fp16, [1, 8, 512, 512]> sim_5_cast_fp16 = matmul(transpose_x = sim_5_transpose_x_0, transpose_y = sim_5_transpose_y_0, x = transpose_14, y = transpose_15)[name = tensor<string, []>("sim_5_cast_fp16")];
224
- tensor<fp16, []> var_392_to_fp16 = const()[name = tensor<string, []>("op_392_to_fp16"), val = tensor<fp16, []>(0x1p-3)];
225
- tensor<fp16, [1, 8, 512, 512]> sim_7_cast_fp16 = mul(x = sim_5_cast_fp16, y = var_392_to_fp16)[name = tensor<string, []>("sim_7_cast_fp16")];
226
- tensor<fp16, [1, 8, 512, 512]> attn_3_cast_fp16 = softmax(axis = var_298, x = sim_7_cast_fp16)[name = tensor<string, []>("attn_3_cast_fp16")];
227
- tensor<bool, []> x_73_transpose_x_0 = const()[name = tensor<string, []>("x_73_transpose_x_0"), val = tensor<bool, []>(false)];
228
- tensor<bool, []> x_73_transpose_y_0 = const()[name = tensor<string, []>("x_73_transpose_y_0"), val = tensor<bool, []>(false)];
229
- tensor<fp16, [1, 8, 512, 64]> v_3_cast_fp16 = transpose(perm = var_388, x = x_71_cast_fp16)[name = tensor<string, []>("transpose_29")];
230
- tensor<fp16, [1, 8, 512, 64]> x_73_cast_fp16 = matmul(transpose_x = x_73_transpose_x_0, transpose_y = x_73_transpose_y_0, x = attn_3_cast_fp16, y = v_3_cast_fp16)[name = tensor<string, []>("x_73_cast_fp16")];
231
- tensor<int32, [4]> var_414 = const()[name = tensor<string, []>("op_414"), val = tensor<int32, [4]>([0, 2, 1, 3])];
232
- tensor<int32, [3]> var_416 = const()[name = tensor<string, []>("op_416"), val = tensor<int32, [3]>([1, 512, 512])];
233
- tensor<fp16, [1, 512, 8, 64]> x_75_cast_fp16 = transpose(perm = var_414, x = x_73_cast_fp16)[name = tensor<string, []>("transpose_28")];
234
- tensor<fp16, [1, 512, 512]> input_39_cast_fp16 = reshape(shape = var_416, x = x_75_cast_fp16)[name = tensor<string, []>("input_39_cast_fp16")];
235
- tensor<fp16, [1024, 512]> transformer_blocks_1_attention_attention_to_out_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12630592))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13154944))), name = tensor<string, []>("transformer_blocks_1_attention_attention_to_out_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 512])];
236
- tensor<fp16, [1024]> transformer_blocks_1_attention_attention_to_out_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_1_attention_attention_to_out_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13155520)))];
237
- tensor<fp16, [1, 512, 1024]> linear_15_cast_fp16 = linear(bias = transformer_blocks_1_attention_attention_to_out_bias_to_fp16, weight = transformer_blocks_1_attention_attention_to_out_weight_to_fp16_palettized, x = input_39_cast_fp16)[name = tensor<string, []>("linear_15_cast_fp16")];
238
- tensor<fp16, [1, 512, 1024]> input_41_cast_fp16 = add(x = linear_15_cast_fp16, y = x_47_cast_fp16)[name = tensor<string, []>("input_41_cast_fp16")];
239
- tensor<fp16, [2048, 1024]> transformer_blocks_1_feed_forward_0_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [2097152]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13157632))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(15254848))), name = tensor<string, []>("transformer_blocks_1_feed_forward_0_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([2048, 1024])];
240
- tensor<fp16, [2048]> transformer_blocks_1_feed_forward_0_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_1_feed_forward_0_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(15255424)))];
241
- tensor<fp16, [1, 512, 2048]> linear_16_cast_fp16 = linear(bias = transformer_blocks_1_feed_forward_0_bias_to_fp16, weight = transformer_blocks_1_feed_forward_0_weight_to_fp16_palettized, x = input_41_cast_fp16)[name = tensor<string, []>("linear_16_cast_fp16")];
242
- tensor<string, []> input_45_mode_0 = const()[name = tensor<string, []>("input_45_mode_0"), val = tensor<string, []>("EXACT")];
243
- tensor<fp16, [1, 512, 2048]> input_45_cast_fp16 = gelu(mode = input_45_mode_0, x = linear_16_cast_fp16)[name = tensor<string, []>("input_45_cast_fp16")];
244
- tensor<fp16, [1024, 2048]> transformer_blocks_1_feed_forward_2_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [2097152]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(15259584))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(17356800))), name = tensor<string, []>("transformer_blocks_1_feed_forward_2_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 2048])];
245
- tensor<fp16, [1024]> transformer_blocks_1_feed_forward_2_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_1_feed_forward_2_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(17357376)))];
246
- tensor<fp16, [1, 512, 1024]> linear_17_cast_fp16 = linear(bias = transformer_blocks_1_feed_forward_2_bias_to_fp16, weight = transformer_blocks_1_feed_forward_2_weight_to_fp16_palettized, x = input_45_cast_fp16)[name = tensor<string, []>("linear_17_cast_fp16")];
247
- tensor<fp16, [1, 512, 1024]> x_77_cast_fp16 = add(x = linear_17_cast_fp16, y = input_41_cast_fp16)[name = tensor<string, []>("x_77_cast_fp16")];
248
- tensor<fp16, [1, 512, 1024]> x_79_cast_fp16 = add(x = x_77_cast_fp16, y = mapping_cast_fp16)[name = tensor<string, []>("x_79_cast_fp16")];
249
- tensor<int32, []> var_443 = const()[name = tensor<string, []>("op_443"), val = tensor<int32, []>(-1)];
250
- tensor<fp16, [2048, 256]> transformer_blocks_2_attention_norm_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(17359488))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(17883840))), name = tensor<string, []>("transformer_blocks_2_attention_norm_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([2048, 256])];
251
- tensor<fp16, [2048]> transformer_blocks_2_attention_norm_fc_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_2_attention_norm_fc_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(17884416)))];
252
- tensor<fp16, [1, 2048]> linear_18_cast_fp16 = linear(bias = transformer_blocks_2_attention_norm_fc_bias_to_fp16, weight = transformer_blocks_2_attention_norm_fc_weight_to_fp16_palettized, x = features_to_fp16)[name = tensor<string, []>("linear_18_cast_fp16")];
253
- tensor<int32, [3]> var_462 = const()[name = tensor<string, []>("op_462"), val = tensor<int32, [3]>([1, 2048, 1])];
254
- tensor<fp16, [1, 2048, 1]> h_19_cast_fp16 = reshape(shape = var_462, x = linear_18_cast_fp16)[name = tensor<string, []>("h_19_cast_fp16")];
255
- tensor<int32, [2]> var_464_split_sizes_0 = const()[name = tensor<string, []>("op_464_split_sizes_0"), val = tensor<int32, [2]>([1024, 1024])];
256
- tensor<int32, []> var_464_axis_0 = const()[name = tensor<string, []>("op_464_axis_0"), val = tensor<int32, []>(1)];
257
- tensor<fp16, [1, 1024, 1]> var_464_cast_fp16_0, tensor<fp16, [1, 1024, 1]> var_464_cast_fp16_1 = split(axis = var_464_axis_0, split_sizes = var_464_split_sizes_0, x = h_19_cast_fp16)[name = tensor<string, []>("op_464_cast_fp16")];
258
- tensor<int32, [3]> gamma_19_perm_0 = const()[name = tensor<string, []>("gamma_19_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
259
- tensor<int32, [3]> beta_19_perm_0 = const()[name = tensor<string, []>("beta_19_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
260
- tensor<int32, [1]> x_83_axes_0 = const()[name = tensor<string, []>("x_83_axes_0"), val = tensor<int32, [1]>([-1])];
261
- tensor<fp16, []> var_436_to_fp16 = const()[name = tensor<string, []>("op_436_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
262
- tensor<fp16, [1, 512, 1024]> x_83_cast_fp16 = layer_norm(axes = x_83_axes_0, epsilon = var_436_to_fp16, x = x_79_cast_fp16)[name = tensor<string, []>("x_83_cast_fp16")];
263
- tensor<fp16, []> var_470_promoted_to_fp16 = const()[name = tensor<string, []>("op_470_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
264
- tensor<fp16, [1, 1, 1024]> gamma_19_cast_fp16 = transpose(perm = gamma_19_perm_0, x = var_464_cast_fp16_0)[name = tensor<string, []>("transpose_27")];
265
- tensor<fp16, [1, 1, 1024]> var_471_cast_fp16 = add(x = gamma_19_cast_fp16, y = var_470_promoted_to_fp16)[name = tensor<string, []>("op_471_cast_fp16")];
266
- tensor<fp16, [1, 512, 1024]> var_472_cast_fp16 = mul(x = var_471_cast_fp16, y = x_83_cast_fp16)[name = tensor<string, []>("op_472_cast_fp16")];
267
- tensor<fp16, [1, 1, 1024]> beta_19_cast_fp16 = transpose(perm = beta_19_perm_0, x = var_464_cast_fp16_1)[name = tensor<string, []>("transpose_26")];
268
- tensor<fp16, [1, 512, 1024]> x_85_cast_fp16 = add(x = var_472_cast_fp16, y = beta_19_cast_fp16)[name = tensor<string, []>("x_85_cast_fp16")];
269
- tensor<fp16, [2048, 256]> transformer_blocks_2_attention_norm_context_fc_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(17888576))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(18412928))), name = tensor<string, []>("transformer_blocks_2_attention_norm_context_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([2048, 256])];
270
- tensor<fp16, [2048]> transformer_blocks_2_attention_norm_context_fc_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_2_attention_norm_context_fc_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(18413504)))];
271
- tensor<fp16, [1, 2048]> linear_19_cast_fp16 = linear(bias = transformer_blocks_2_attention_norm_context_fc_bias_to_fp16, weight = transformer_blocks_2_attention_norm_context_fc_weight_to_fp16_palettized, x = features_to_fp16)[name = tensor<string, []>("linear_19_cast_fp16")];
272
- tensor<int32, [3]> var_484 = const()[name = tensor<string, []>("op_484"), val = tensor<int32, [3]>([1, 2048, 1])];
273
- tensor<fp16, [1, 2048, 1]> h_cast_fp16 = reshape(shape = var_484, x = linear_19_cast_fp16)[name = tensor<string, []>("h_cast_fp16")];
274
- tensor<int32, [2]> var_486_split_sizes_0 = const()[name = tensor<string, []>("op_486_split_sizes_0"), val = tensor<int32, [2]>([1024, 1024])];
275
- tensor<int32, []> var_486_axis_0 = const()[name = tensor<string, []>("op_486_axis_0"), val = tensor<int32, []>(1)];
276
- tensor<fp16, [1, 1024, 1]> var_486_cast_fp16_0, tensor<fp16, [1, 1024, 1]> var_486_cast_fp16_1 = split(axis = var_486_axis_0, split_sizes = var_486_split_sizes_0, x = h_cast_fp16)[name = tensor<string, []>("op_486_cast_fp16")];
277
- tensor<int32, [3]> gamma_perm_0 = const()[name = tensor<string, []>("gamma_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
278
- tensor<int32, [3]> beta_perm_0 = const()[name = tensor<string, []>("beta_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
279
- tensor<fp16, []> var_492_promoted_to_fp16 = const()[name = tensor<string, []>("op_492_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
280
- tensor<fp16, [1, 1, 1024]> gamma_cast_fp16 = transpose(perm = gamma_perm_0, x = var_486_cast_fp16_0)[name = tensor<string, []>("transpose_25")];
281
- tensor<fp16, [1, 1, 1024]> var_493_cast_fp16 = add(x = gamma_cast_fp16, y = var_492_promoted_to_fp16)[name = tensor<string, []>("op_493_cast_fp16")];
282
- tensor<fp16, [1, 512, 1024]> var_494_cast_fp16 = mul(x = var_493_cast_fp16, y = x_83_cast_fp16)[name = tensor<string, []>("op_494_cast_fp16")];
283
- tensor<fp16, [1, 1, 1024]> beta_cast_fp16 = transpose(perm = beta_perm_0, x = var_486_cast_fp16_1)[name = tensor<string, []>("transpose_24")];
284
- tensor<fp16, [1, 512, 1024]> x_91_cast_fp16 = add(x = var_494_cast_fp16, y = beta_cast_fp16)[name = tensor<string, []>("x_91_cast_fp16")];
285
- tensor<fp16, [512, 1024]> transformer_blocks_2_attention_to_q_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(18417664))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(18942016))), name = tensor<string, []>("transformer_blocks_2_attention_to_q_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 1024])];
286
- tensor<fp16, [1, 512, 512]> linear_20_cast_fp16 = linear(bias = linear_6_bias_0_to_fp16, weight = transformer_blocks_2_attention_to_q_weight_to_fp16_palettized, x = x_85_cast_fp16)[name = tensor<string, []>("linear_20_cast_fp16")];
287
- tensor<fp16, [1024, 1024]> transformer_blocks_2_attention_to_kv_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [1048576]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(18942592))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(19991232))), name = tensor<string, []>("transformer_blocks_2_attention_to_kv_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 1024])];
288
- tensor<fp16, [1, 512, 1024]> linear_21_cast_fp16 = linear(bias = linear_7_bias_0_to_fp16, weight = transformer_blocks_2_attention_to_kv_weight_to_fp16_palettized, x = x_91_cast_fp16)[name = tensor<string, []>("linear_21_cast_fp16")];
289
- tensor<int32, [2]> var_502_split_sizes_0 = const()[name = tensor<string, []>("op_502_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
290
- tensor<int32, []> var_502_axis_0 = const()[name = tensor<string, []>("op_502_axis_0"), val = tensor<int32, []>(-1)];
291
- tensor<fp16, [1, 512, 512]> var_502_cast_fp16_0, tensor<fp16, [1, 512, 512]> var_502_cast_fp16_1 = split(axis = var_502_axis_0, split_sizes = var_502_split_sizes_0, x = linear_21_cast_fp16)[name = tensor<string, []>("op_502_cast_fp16")];
292
- tensor<int32, [4]> var_511 = const()[name = tensor<string, []>("op_511"), val = tensor<int32, [4]>([1, 512, 8, 64])];
293
- tensor<fp16, [1, 512, 8, 64]> x_95_cast_fp16 = reshape(shape = var_511, x = linear_20_cast_fp16)[name = tensor<string, []>("x_95_cast_fp16")];
294
- tensor<int32, [4]> var_521 = const()[name = tensor<string, []>("op_521"), val = tensor<int32, [4]>([1, 512, 8, 64])];
295
- tensor<fp16, [1, 512, 8, 64]> x_99_cast_fp16 = reshape(shape = var_521, x = var_502_cast_fp16_0)[name = tensor<string, []>("x_99_cast_fp16")];
296
- tensor<int32, [4]> var_531 = const()[name = tensor<string, []>("op_531"), val = tensor<int32, [4]>([1, 512, 8, 64])];
297
- tensor<fp16, [1, 512, 8, 64]> x_103_cast_fp16 = reshape(shape = var_531, x = var_502_cast_fp16_1)[name = tensor<string, []>("x_103_cast_fp16")];
298
- tensor<int32, [4]> var_533 = const()[name = tensor<string, []>("op_533"), val = tensor<int32, [4]>([0, 2, -3, -1])];
299
- tensor<bool, []> sim_9_transpose_x_0 = const()[name = tensor<string, []>("sim_9_transpose_x_0"), val = tensor<bool, []>(false)];
300
- tensor<bool, []> sim_9_transpose_y_0 = const()[name = tensor<string, []>("sim_9_transpose_y_0"), val = tensor<bool, []>(false)];
301
- tensor<int32, [4]> transpose_16_perm_0 = const()[name = tensor<string, []>("transpose_16_perm_0"), val = tensor<int32, [4]>([0, 2, -3, -1])];
302
- tensor<int32, [4]> transpose_17_perm_0 = const()[name = tensor<string, []>("transpose_17_perm_0"), val = tensor<int32, [4]>([0, 2, -1, -3])];
303
- tensor<fp16, [1, 8, 64, 512]> transpose_17 = transpose(perm = transpose_17_perm_0, x = x_99_cast_fp16)[name = tensor<string, []>("transpose_22")];
304
- tensor<fp16, [1, 8, 512, 64]> transpose_16 = transpose(perm = transpose_16_perm_0, x = x_95_cast_fp16)[name = tensor<string, []>("transpose_23")];
305
- tensor<fp16, [1, 8, 512, 512]> sim_9_cast_fp16 = matmul(transpose_x = sim_9_transpose_x_0, transpose_y = sim_9_transpose_y_0, x = transpose_16, y = transpose_17)[name = tensor<string, []>("sim_9_cast_fp16")];
306
- tensor<fp16, []> var_537_to_fp16 = const()[name = tensor<string, []>("op_537_to_fp16"), val = tensor<fp16, []>(0x1p-3)];
307
- tensor<fp16, [1, 8, 512, 512]> sim_cast_fp16 = mul(x = sim_9_cast_fp16, y = var_537_to_fp16)[name = tensor<string, []>("sim_cast_fp16")];
308
- tensor<fp16, [1, 8, 512, 512]> attn_cast_fp16 = softmax(axis = var_443, x = sim_cast_fp16)[name = tensor<string, []>("attn_cast_fp16")];
309
- tensor<bool, []> x_105_transpose_x_0 = const()[name = tensor<string, []>("x_105_transpose_x_0"), val = tensor<bool, []>(false)];
310
- tensor<bool, []> x_105_transpose_y_0 = const()[name = tensor<string, []>("x_105_transpose_y_0"), val = tensor<bool, []>(false)];
311
- tensor<fp16, [1, 8, 512, 64]> v_cast_fp16 = transpose(perm = var_533, x = x_103_cast_fp16)[name = tensor<string, []>("transpose_21")];
312
- tensor<fp16, [1, 8, 512, 64]> x_105_cast_fp16 = matmul(transpose_x = x_105_transpose_x_0, transpose_y = x_105_transpose_y_0, x = attn_cast_fp16, y = v_cast_fp16)[name = tensor<string, []>("x_105_cast_fp16")];
313
- tensor<int32, [4]> var_559 = const()[name = tensor<string, []>("op_559"), val = tensor<int32, [4]>([0, 2, 1, 3])];
314
- tensor<int32, [3]> var_561 = const()[name = tensor<string, []>("op_561"), val = tensor<int32, [3]>([1, 512, 512])];
315
- tensor<fp16, [1, 512, 8, 64]> x_107_cast_fp16 = transpose(perm = var_559, x = x_105_cast_fp16)[name = tensor<string, []>("transpose_20")];
316
- tensor<fp16, [1, 512, 512]> input_55_cast_fp16 = reshape(shape = var_561, x = x_107_cast_fp16)[name = tensor<string, []>("input_55_cast_fp16")];
317
- tensor<fp16, [1024, 512]> transformer_blocks_2_attention_attention_to_out_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(19991808))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(20516160))), name = tensor<string, []>("transformer_blocks_2_attention_attention_to_out_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 512])];
318
- tensor<fp16, [1024]> transformer_blocks_2_attention_attention_to_out_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_2_attention_attention_to_out_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(20516736)))];
319
- tensor<fp16, [1, 512, 1024]> linear_22_cast_fp16 = linear(bias = transformer_blocks_2_attention_attention_to_out_bias_to_fp16, weight = transformer_blocks_2_attention_attention_to_out_weight_to_fp16_palettized, x = input_55_cast_fp16)[name = tensor<string, []>("linear_22_cast_fp16")];
320
- tensor<fp16, [1, 512, 1024]> input_57_cast_fp16 = add(x = linear_22_cast_fp16, y = x_79_cast_fp16)[name = tensor<string, []>("input_57_cast_fp16")];
321
- tensor<fp16, [2048, 1024]> transformer_blocks_2_feed_forward_0_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [2097152]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(20518848))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(22616064))), name = tensor<string, []>("transformer_blocks_2_feed_forward_0_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([2048, 1024])];
322
- tensor<fp16, [2048]> transformer_blocks_2_feed_forward_0_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_2_feed_forward_0_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(22616640)))];
323
- tensor<fp16, [1, 512, 2048]> linear_23_cast_fp16 = linear(bias = transformer_blocks_2_feed_forward_0_bias_to_fp16, weight = transformer_blocks_2_feed_forward_0_weight_to_fp16_palettized, x = input_57_cast_fp16)[name = tensor<string, []>("linear_23_cast_fp16")];
324
- tensor<string, []> input_61_mode_0 = const()[name = tensor<string, []>("input_61_mode_0"), val = tensor<string, []>("EXACT")];
325
- tensor<fp16, [1, 512, 2048]> input_61_cast_fp16 = gelu(mode = input_61_mode_0, x = linear_23_cast_fp16)[name = tensor<string, []>("input_61_cast_fp16")];
326
- tensor<fp16, [1024, 2048]> transformer_blocks_2_feed_forward_2_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [2097152]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(22620800))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(24718016))), name = tensor<string, []>("transformer_blocks_2_feed_forward_2_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 2048])];
327
- tensor<fp16, [1024]> transformer_blocks_2_feed_forward_2_bias_to_fp16 = const()[name = tensor<string, []>("transformer_blocks_2_feed_forward_2_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(24718592)))];
328
- tensor<fp16, [1, 512, 1024]> linear_24_cast_fp16 = linear(bias = transformer_blocks_2_feed_forward_2_bias_to_fp16, weight = transformer_blocks_2_feed_forward_2_weight_to_fp16_palettized, x = input_61_cast_fp16)[name = tensor<string, []>("linear_24_cast_fp16")];
329
- tensor<fp16, [1, 512, 1024]> x_109_cast_fp16 = add(x = linear_24_cast_fp16, y = input_57_cast_fp16)[name = tensor<string, []>("x_109_cast_fp16")];
330
- tensor<int32, [1]> var_581_axes_0 = const()[name = tensor<string, []>("op_581_axes_0"), val = tensor<int32, [1]>([1])];
331
- tensor<bool, []> var_581_keep_dims_0 = const()[name = tensor<string, []>("op_581_keep_dims_0"), val = tensor<bool, []>(false)];
332
- tensor<fp16, [1, 1024]> var_581_cast_fp16 = reduce_mean(axes = var_581_axes_0, keep_dims = var_581_keep_dims_0, x = x_109_cast_fp16)[name = tensor<string, []>("op_581_cast_fp16")];
333
- tensor<int32, [1]> x_111_axes_0 = const()[name = tensor<string, []>("x_111_axes_0"), val = tensor<int32, [1]>([1])];
334
- tensor<fp16, [1, 1, 1024]> x_111_cast_fp16 = expand_dims(axes = x_111_axes_0, x = var_581_cast_fp16)[name = tensor<string, []>("x_111_cast_fp16")];
335
- tensor<int32, [3]> var_590 = const()[name = tensor<string, []>("op_590"), val = tensor<int32, [3]>([0, 2, 1])];
336
- tensor<string, []> x_pad_type_0 = const()[name = tensor<string, []>("x_pad_type_0"), val = tensor<string, []>("valid")];
337
- tensor<int32, [1]> x_strides_0 = const()[name = tensor<string, []>("x_strides_0"), val = tensor<int32, [1]>([1])];
338
- tensor<int32, [2]> x_pad_0 = const()[name = tensor<string, []>("x_pad_0"), val = tensor<int32, [2]>([0, 0])];
339
- tensor<int32, [1]> x_dilations_0 = const()[name = tensor<string, []>("x_dilations_0"), val = tensor<int32, [1]>([1])];
340
- tensor<int32, []> x_groups_0 = const()[name = tensor<string, []>("x_groups_0"), val = tensor<int32, []>(1)];
341
- tensor<fp16, [256, 1024, 1]> transformer_to_out_1_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(24720704))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(24982912))), name = tensor<string, []>("transformer_to_out_1_weight_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 1024, 1])];
342
- tensor<fp16, [256]> transformer_to_out_1_bias_to_fp16 = const()[name = tensor<string, []>("transformer_to_out_1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(24983488)))];
343
- tensor<fp16, [1, 1024, 1]> input_cast_fp16 = transpose(perm = var_590, x = x_111_cast_fp16)[name = tensor<string, []>("transpose_19")];
344
- tensor<fp16, [1, 256, 1]> x_cast_fp16 = conv(bias = transformer_to_out_1_bias_to_fp16, dilations = x_dilations_0, groups = x_groups_0, pad = x_pad_0, pad_type = x_pad_type_0, strides = x_strides_0, weight = transformer_to_out_1_weight_to_fp16_palettized, x = input_cast_fp16)[name = tensor<string, []>("x_cast_fp16")];
345
- tensor<int32, [3]> x_pred_perm_0 = const()[name = tensor<string, []>("x_pred_perm_0"), val = tensor<int32, [3]>([0, -1, -2])];
346
- tensor<fp16, [1, 1, 256]> var_602_cast_fp16 = mul(x = c_skip_cast_fp16, y = x_noisy_to_fp16)[name = tensor<string, []>("op_602_cast_fp16")];
347
- tensor<fp16, [1, 1, 256]> x_pred_cast_fp16 = transpose(perm = x_pred_perm_0, x = x_cast_fp16)[name = tensor<string, []>("transpose_18")];
348
- tensor<fp16, [1, 1, 256]> var_603_cast_fp16 = mul(x = c_out_cast_fp16, y = x_pred_cast_fp16)[name = tensor<string, []>("op_603_cast_fp16")];
349
- tensor<fp16, [1, 1, 256]> denoised = add(x = var_602_cast_fp16, y = var_603_cast_fp16)[name = tensor<string, []>("op_605_cast_fp16")];
350
- } -> (denoised);
351
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_diffusion_step.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:c7ec58d48cbfd5094ca3c0a9de778a789878e9d2cae35cc4ad277cff669d538d
3
- size 24984064
 
 
 
 
ANE/styletts2_ane_diffusion_step.mlpackage/Data/com.apple.CoreML/model.mlmodel DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f7012e3db65ab67a3ad9a85a421d83b9ceb2363b370c7fc3f27d23271af6ca9e
3
- size 58407
 
 
 
 
ANE/styletts2_ane_diffusion_step.mlpackage/Data/com.apple.CoreML/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:c7ec58d48cbfd5094ca3c0a9de778a789878e9d2cae35cc4ad277cff669d538d
3
- size 24984064
 
 
 
 
ANE/styletts2_ane_diffusion_step.mlpackage/Manifest.json DELETED
@@ -1,18 +0,0 @@
1
- {
2
- "fileFormatVersion": "1.0.0",
3
- "itemInfoEntries": {
4
- "0D2868A2-D1E7-4D21-BDBB-2078B73600CD": {
5
- "author": "com.apple.CoreML",
6
- "description": "CoreML Model Specification",
7
- "name": "model.mlmodel",
8
- "path": "com.apple.CoreML/model.mlmodel"
9
- },
10
- "99F44601-8B02-4B68-8756-183CA2AB7B21": {
11
- "author": "com.apple.CoreML",
12
- "description": "CoreML Model Weights",
13
- "name": "weights",
14
- "path": "com.apple.CoreML/weights"
15
- }
16
- },
17
- "rootModelIdentifier": "0D2868A2-D1E7-4D21-BDBB-2078B73600CD"
18
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_noise.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:4f4c8e0ce15ecb4ac16b403361160287c7638ac0dfdd8327835be795979b6395
3
- size 243
 
 
 
 
ANE/styletts2_ane_noise.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:70c88a08bda3ad733ac2cad2cb90012b51ea81e2b8eef7680a8dfe7ff6731d54
3
- size 407
 
 
 
 
ANE/styletts2_ane_noise.mlmodelc/metadata.json DELETED
@@ -1,87 +0,0 @@
1
- [
2
- {
3
- "metadataOutputVersion" : "3.0",
4
- "shortDescription" : "StyleTTS2-ANE Stage 6: SineGen (static T_a=2000) — fp32, ALL.",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float32",
10
- "formattedType" : "MultiArray (Float32 1 × 1200000 × 9)",
11
- "shortDescription" : "",
12
- "shape" : "[1, 1200000, 9]",
13
- "name" : "sine_waves",
14
- "type" : "MultiArray"
15
- },
16
- {
17
- "hasShapeFlexibility" : "0",
18
- "isOptional" : "0",
19
- "dataType" : "Float32",
20
- "formattedType" : "MultiArray (Float32 1 × 1200000 × 1)",
21
- "shortDescription" : "",
22
- "shape" : "[1, 1200000, 1]",
23
- "name" : "uv",
24
- "type" : "MultiArray"
25
- }
26
- ],
27
- "storagePrecision" : "Palettized (8 bits)",
28
- "modelParameters" : [
29
-
30
- ],
31
- "specificationVersion" : 8,
32
- "mlProgramOperationTypeHistogram" : {
33
- "Ios17.mul" : 14,
34
- "Ios17.floor" : 1,
35
- "Ios17.transpose" : 7,
36
- "Ios17.sub" : 2,
37
- "Ios16.constexprLutToDense" : 2,
38
- "Ios17.sin" : 2,
39
- "Ios16.cumsum" : 1,
40
- "Ios17.greater" : 1,
41
- "Ios17.concat" : 1,
42
- "Ios17.sliceByIndex" : 3,
43
- "UpsampleNearestNeighbor" : 1,
44
- "Ios17.add" : 3,
45
- "Tile" : 2,
46
- "Ios17.expandDims" : 2,
47
- "Ios17.squeeze" : 1,
48
- "Ios17.reshape" : 4,
49
- "Ios17.cast" : 1
50
- },
51
- "computePrecision" : "Mixed (Float32, Int32)",
52
- "isUpdatable" : "0",
53
- "stateSchema" : [
54
-
55
- ],
56
- "availability" : {
57
- "macOS" : "14.0",
58
- "tvOS" : "17.0",
59
- "visionOS" : "1.0",
60
- "watchOS" : "10.0",
61
- "iOS" : "17.0",
62
- "macCatalyst" : "17.0"
63
- },
64
- "modelType" : {
65
- "name" : "MLModelType_mlProgram"
66
- },
67
- "inputSchema" : [
68
- {
69
- "hasShapeFlexibility" : "0",
70
- "isOptional" : "0",
71
- "dataType" : "Float32",
72
- "formattedType" : "MultiArray (Float32 1 × 4000)",
73
- "shortDescription" : "",
74
- "shape" : "[1, 4000]",
75
- "name" : "F0_curve",
76
- "type" : "MultiArray"
77
- }
78
- ],
79
- "userDefinedMetadata" : {
80
- "com.github.apple.coremltools.version" : "8.3.0",
81
- "com.github.apple.coremltools.source_dialect" : "TorchScript",
82
- "com.github.apple.coremltools.source" : "torch==2.11.0"
83
- },
84
- "generatedClassName" : "styletts2_ane_noise",
85
- "method" : "predict"
86
- }
87
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_noise.mlmodelc/model.mil DELETED
@@ -1,99 +0,0 @@
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, 4000]> F0_curve) {
5
- tensor<int32, [1]> input_axes_0 = const()[name = tensor<string, []>("input_axes_0"), val = tensor<int32, [1]>([1])];
6
- tensor<fp32, [1, 1, 4000]> input = expand_dims(axes = input_axes_0, x = F0_curve)[name = tensor<string, []>("input")];
7
- tensor<int32, [1]> expand_dims_0_axes_0 = const()[name = tensor<string, []>("expand_dims_0_axes_0"), val = tensor<int32, [1]>([3])];
8
- tensor<fp32, [1, 1, 4000, 1]> expand_dims_0 = expand_dims(axes = expand_dims_0_axes_0, x = input)[name = tensor<string, []>("expand_dims_0")];
9
- 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)];
10
- 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)];
11
- tensor<fp32, [1, 1, 1200000, 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")];
12
- tensor<int32, [1]> var_12_axes_0 = const()[name = tensor<string, []>("op_12_axes_0"), val = tensor<int32, [1]>([3])];
13
- tensor<fp32, [1, 1, 1200000]> var_12 = squeeze(axes = var_12_axes_0, x = upsample_nearest_neighbor_0)[name = tensor<string, []>("op_12")];
14
- tensor<int32, [3]> f0_perm_0 = const()[name = tensor<string, []>("f0_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
15
- tensor<int32, []> var_23 = const()[name = tensor<string, []>("op_23"), val = tensor<int32, []>(1)];
16
- tensor<fp32, [1, 1, 9]> const_1 = const()[name = tensor<string, []>("const_1"), val = tensor<fp32, [1, 1, 9]>([[[0x1p+0, 0x1p+1, 0x1.8p+1, 0x1p+2, 0x1.4p+2, 0x1.8p+2, 0x1.cp+2, 0x1p+3, 0x1.2p+3]]])];
17
- tensor<fp32, [1, 1200000, 1]> f0 = transpose(perm = f0_perm_0, x = var_12)[name = tensor<string, []>("transpose_6")];
18
- tensor<fp32, [1, 1200000, 9]> f0_values = mul(x = f0, y = const_1)[name = tensor<string, []>("f0_values")];
19
- tensor<fp32, []> _inversed_rad_div_y_0 = const()[name = tensor<string, []>("_inversed_rad_div_y_0"), val = tensor<fp32, []>(0x1.5d867cp-15)];
20
- tensor<fp32, [1, 1200000, 9]> _inversed_rad_div = mul(x = f0_values, y = _inversed_rad_div_y_0)[name = tensor<string, []>("_inversed_rad_div")];
21
- tensor<fp32, [1, 1200000, 9]> var_35 = floor(x = _inversed_rad_div)[name = tensor<string, []>("op_35")];
22
- tensor<fp32, [1, 1200000, 9]> rad_mod = sub(x = _inversed_rad_div, y = var_35)[name = tensor<string, []>("rad_mod")];
23
- tensor<int32, [3]> var_38_begin_0 = const()[name = tensor<string, []>("op_38_begin_0"), val = tensor<int32, [3]>([0, 0, 0])];
24
- tensor<int32, [3]> var_38_end_0 = const()[name = tensor<string, []>("op_38_end_0"), val = tensor<int32, [3]>([1, 1200000, 9])];
25
- tensor<bool, [3]> var_38_end_mask_0 = const()[name = tensor<string, []>("op_38_end_mask_0"), val = tensor<bool, [3]>([true, true, true])];
26
- tensor<int32, [3]> var_38_stride_0 = const()[name = tensor<string, []>("op_38_stride_0"), val = tensor<int32, [3]>([1, 300, 1])];
27
- tensor<fp32, [1, 4000, 9]> var_38 = slice_by_index(begin = var_38_begin_0, end = var_38_end_0, end_mask = var_38_end_mask_0, stride = var_38_stride_0, x = rad_mod)[name = tensor<string, []>("op_38")];
28
- tensor<bool, []> var_40_exclusive_0 = const()[name = tensor<string, []>("op_40_exclusive_0"), val = tensor<bool, []>(false)];
29
- tensor<bool, []> var_40_reverse_0 = const()[name = tensor<string, []>("op_40_reverse_0"), val = tensor<bool, []>(false)];
30
- tensor<fp32, [1, 4000, 9]> var_40 = cumsum(axis = var_23, exclusive = var_40_exclusive_0, reverse = var_40_reverse_0, x = var_38)[name = tensor<string, []>("op_40")];
31
- tensor<fp32, []> var_41_promoted = const()[name = tensor<string, []>("op_41_promoted"), val = tensor<fp32, []>(0x1p+1)];
32
- tensor<fp32, [1, 4000, 9]> var_42 = mul(x = var_40, y = var_41_promoted)[name = tensor<string, []>("op_42")];
33
- tensor<fp32, []> var_43 = const()[name = tensor<string, []>("op_43"), val = tensor<fp32, []>(0x1.921fb6p+1)];
34
- tensor<fp32, [1, 4000, 9]> phase = mul(x = var_42, y = var_43)[name = tensor<string, []>("phase")];
35
- tensor<fp32, []> var_45_promoted = const()[name = tensor<string, []>("op_45_promoted"), val = tensor<fp32, []>(0x1.2cp+8)];
36
- tensor<fp32, [1, 4000, 9]> phase_scaled = mul(x = phase, y = var_45_promoted)[name = tensor<string, []>("phase_scaled")];
37
- tensor<int32, [3]> transpose_0_perm_0 = const()[name = tensor<string, []>("transpose_0_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
38
- tensor<int32, [3]> tile_0_reps_0 = const()[name = tensor<string, []>("tile_0_reps_0"), val = tensor<int32, [3]>([300, 1, 1])];
39
- tensor<fp32, [4000, 1, 9]> transpose_0 = transpose(perm = transpose_0_perm_0, x = phase_scaled)[name = tensor<string, []>("transpose_5")];
40
- tensor<fp32, [1200000, 1, 9]> tile_0 = tile(reps = tile_0_reps_0, x = transpose_0)[name = tensor<string, []>("tile_0")];
41
- tensor<int32, [4]> concat_0 = const()[name = tensor<string, []>("concat_0"), val = tensor<int32, [4]>([300, 4000, 1, 9])];
42
- tensor<fp32, [300, 4000, 1, 9]> reshape_0 = reshape(shape = concat_0, x = tile_0)[name = tensor<string, []>("reshape_0")];
43
- tensor<int32, [4]> transpose_1_perm_0 = const()[name = tensor<string, []>("transpose_1_perm_0"), val = tensor<int32, [4]>([1, 0, 2, 3])];
44
- tensor<int32, [3]> concat_1 = const()[name = tensor<string, []>("concat_1"), val = tensor<int32, [3]>([-1, 1, 9])];
45
- tensor<fp32, [4000, 300, 1, 9]> transpose_1 = transpose(perm = transpose_1_perm_0, x = reshape_0)[name = tensor<string, []>("transpose_4")];
46
- tensor<fp32, [1200000, 1, 9]> reshape_1 = reshape(shape = concat_1, x = transpose_1)[name = tensor<string, []>("reshape_1")];
47
- tensor<int32, [3]> left_perm_0 = const()[name = tensor<string, []>("left_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
48
- tensor<int32, [3]> var_49_begin_0 = const()[name = tensor<string, []>("op_49_begin_0"), val = tensor<int32, [3]>([0, -1, 0])];
49
- tensor<int32, [3]> var_49_end_0 = const()[name = tensor<string, []>("op_49_end_0"), val = tensor<int32, [3]>([1, 4000, 9])];
50
- tensor<bool, [3]> var_49_end_mask_0 = const()[name = tensor<string, []>("op_49_end_mask_0"), val = tensor<bool, [3]>([true, true, true])];
51
- tensor<fp32, [1, 1, 9]> var_49 = slice_by_index(begin = var_49_begin_0, end = var_49_end_0, end_mask = var_49_end_mask_0, x = phase_scaled)[name = tensor<string, []>("op_49")];
52
- tensor<int32, [3]> var_52_begin_0 = const()[name = tensor<string, []>("op_52_begin_0"), val = tensor<int32, [3]>([0, 1, 0])];
53
- tensor<int32, [3]> var_52_end_0 = const()[name = tensor<string, []>("op_52_end_0"), val = tensor<int32, [3]>([1, 4000, 9])];
54
- tensor<bool, [3]> var_52_end_mask_0 = const()[name = tensor<string, []>("op_52_end_mask_0"), val = tensor<bool, [3]>([true, true, true])];
55
- tensor<fp32, [1, 3999, 9]> var_52 = slice_by_index(begin = var_52_begin_0, end = var_52_end_0, end_mask = var_52_end_mask_0, x = phase_scaled)[name = tensor<string, []>("op_52")];
56
- tensor<bool, []> shifted_interleave_0 = const()[name = tensor<string, []>("shifted_interleave_0"), val = tensor<bool, []>(false)];
57
- tensor<fp32, [1, 4000, 9]> shifted = concat(axis = var_23, interleave = shifted_interleave_0, values = (var_52, var_49))[name = tensor<string, []>("shifted")];
58
- tensor<int32, [3]> transpose_2_perm_0 = const()[name = tensor<string, []>("transpose_2_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
59
- tensor<int32, [3]> tile_1_reps_0 = const()[name = tensor<string, []>("tile_1_reps_0"), val = tensor<int32, [3]>([300, 1, 1])];
60
- tensor<fp32, [4000, 1, 9]> transpose_2 = transpose(perm = transpose_2_perm_0, x = shifted)[name = tensor<string, []>("transpose_3")];
61
- tensor<fp32, [1200000, 1, 9]> tile_1 = tile(reps = tile_1_reps_0, x = transpose_2)[name = tensor<string, []>("tile_1")];
62
- tensor<int32, [4]> concat_2 = const()[name = tensor<string, []>("concat_2"), val = tensor<int32, [4]>([300, 4000, 1, 9])];
63
- tensor<fp32, [300, 4000, 1, 9]> reshape_2 = reshape(shape = concat_2, x = tile_1)[name = tensor<string, []>("reshape_2")];
64
- tensor<int32, [4]> transpose_3_perm_0 = const()[name = tensor<string, []>("transpose_3_perm_0"), val = tensor<int32, [4]>([1, 0, 2, 3])];
65
- tensor<int32, [3]> concat_3 = const()[name = tensor<string, []>("concat_3"), val = tensor<int32, [3]>([-1, 1, 9])];
66
- tensor<fp32, [4000, 300, 1, 9]> transpose_3 = transpose(perm = transpose_3_perm_0, x = reshape_2)[name = tensor<string, []>("transpose_2")];
67
- tensor<fp32, [1200000, 1, 9]> reshape_3 = reshape(shape = concat_3, x = transpose_3)[name = tensor<string, []>("reshape_3")];
68
- tensor<int32, [3]> right_perm_0 = const()[name = tensor<string, []>("right_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
69
- tensor<fp32, [1, 1200000, 1]> const_2_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [1200000]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1200128))), name = tensor<string, []>("const_2_palettized"), shape = tensor<uint32, [3]>([1, 1200000, 1])];
70
- tensor<fp32, [1, 1200000, 1]> op_58_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [1200000]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1201216))), lut = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2401280))), name = tensor<string, []>("op_58_palettized"), shape = tensor<uint32, [3]>([1, 1200000, 1])];
71
- tensor<fp32, [1, 1200000, 9]> left = transpose(perm = left_perm_0, x = reshape_1)[name = tensor<string, []>("transpose_1")];
72
- tensor<fp32, [1, 1200000, 9]> var_59 = mul(x = left, y = op_58_palettized)[name = tensor<string, []>("op_59")];
73
- tensor<fp32, [1, 1200000, 9]> right = transpose(perm = right_perm_0, x = reshape_3)[name = tensor<string, []>("transpose_0")];
74
- tensor<fp32, [1, 1200000, 9]> var_60 = mul(x = right, y = const_2_palettized)[name = tensor<string, []>("op_60")];
75
- tensor<fp32, [1, 1200000, 9]> phase_up = add(x = var_59, y = var_60)[name = tensor<string, []>("phase_up")];
76
- tensor<fp32, [1, 1200000, 9]> sines = sin(x = phase_up)[name = tensor<string, []>("sines")];
77
- tensor<fp32, []> var_63 = const()[name = tensor<string, []>("op_63"), val = tensor<fp32, []>(0x1.99999ap-4)];
78
- tensor<fp32, [1, 1200000, 9]> sine_waves_1 = mul(x = sines, y = var_63)[name = tensor<string, []>("sine_waves")];
79
- tensor<fp32, []> var_16_promoted = const()[name = tensor<string, []>("op_16_promoted"), val = tensor<fp32, []>(0x1.4p+3)];
80
- tensor<bool, [1, 1200000, 1]> var_65 = greater(x = f0, y = var_16_promoted)[name = tensor<string, []>("op_65")];
81
- tensor<string, []> cast_2_dtype_0 = const()[name = tensor<string, []>("cast_2_dtype_0"), val = tensor<string, []>("fp32")];
82
- tensor<fp32, []> var_67 = const()[name = tensor<string, []>("op_67"), val = tensor<fp32, []>(0x1.89374cp-9)];
83
- tensor<fp32, [1, 1200000, 1]> uv = cast(dtype = cast_2_dtype_0, x = var_65)[name = tensor<string, []>("cast_0")];
84
- tensor<fp32, [1, 1200000, 1]> var_68 = mul(x = uv, y = var_67)[name = tensor<string, []>("op_68")];
85
- tensor<fp32, []> var_23_promoted = const()[name = tensor<string, []>("op_23_promoted"), val = tensor<fp32, []>(0x1p+0)];
86
- tensor<fp32, [1, 1200000, 1]> var_69 = sub(x = var_23_promoted, y = uv)[name = tensor<string, []>("op_69")];
87
- tensor<fp32, []> var_70 = const()[name = tensor<string, []>("op_70"), val = tensor<fp32, []>(0x1.99999ap-4)];
88
- tensor<fp32, [1, 1200000, 1]> var_71 = mul(x = var_69, y = var_70)[name = tensor<string, []>("op_71")];
89
- tensor<fp32, []> _inversed_73_y_0 = const()[name = tensor<string, []>("_inversed_73_y_0"), val = tensor<fp32, []>(0x1.555556p-2)];
90
- tensor<fp32, [1, 1200000, 1]> _inversed_73 = mul(x = var_71, y = _inversed_73_y_0)[name = tensor<string, []>("_inversed_73")];
91
- tensor<fp32, [1, 1200000, 1]> noise_amp = add(x = var_68, y = _inversed_73)[name = tensor<string, []>("noise_amp")];
92
- tensor<fp32, []> var_75 = const()[name = tensor<string, []>("op_75"), val = tensor<fp32, []>(0x1.9p+6)];
93
- tensor<fp32, [1, 1200000, 9]> var_76 = mul(x = f0_values, y = var_75)[name = tensor<string, []>("op_76")];
94
- tensor<fp32, [1, 1200000, 9]> var_77 = sin(x = var_76)[name = tensor<string, []>("op_77")];
95
- tensor<fp32, [1, 1200000, 9]> noise = mul(x = noise_amp, y = var_77)[name = tensor<string, []>("noise")];
96
- tensor<fp32, [1, 1200000, 9]> var_79 = mul(x = sine_waves_1, y = uv)[name = tensor<string, []>("op_79")];
97
- tensor<fp32, [1, 1200000, 9]> sine_waves = add(x = var_79, y = noise)[name = tensor<string, []>("op_80")];
98
- } -> (sine_waves, uv);
99
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_noise.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:1b988942b88ad44925a6cf81bf3d1108b1da6c2f2c84edc19091e75cd29e53d6
3
- size 2402368
 
 
 
 
ANE/styletts2_ane_noise.mlpackage/Data/com.apple.CoreML/model.mlmodel DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:54d28925e686100531855b225d53c4c50f94bac6eac7a2d0193340d4dcecb0bc
3
- size 12114
 
 
 
 
ANE/styletts2_ane_noise.mlpackage/Data/com.apple.CoreML/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:1b988942b88ad44925a6cf81bf3d1108b1da6c2f2c84edc19091e75cd29e53d6
3
- size 2402368
 
 
 
 
ANE/styletts2_ane_noise.mlpackage/Manifest.json DELETED
@@ -1,18 +0,0 @@
1
- {
2
- "fileFormatVersion": "1.0.0",
3
- "itemInfoEntries": {
4
- "47473672-28F9-487C-9ED6-63C6D82ABB57": {
5
- "author": "com.apple.CoreML",
6
- "description": "CoreML Model Weights",
7
- "name": "weights",
8
- "path": "com.apple.CoreML/weights"
9
- },
10
- "88C15F91-FFC7-4B76-B3DF-CFFD20CF2957": {
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": "88C15F91-FFC7-4B76-B3DF-CFFD20CF2957"
18
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_plbert.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:a828a4caef0ea1661731eb81369d07f3c0bcbfa9bbb9d2aaebe0c75fd1136221
3
- size 243
 
 
 
 
ANE/styletts2_ane_plbert.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8a36584b83542236b10da229c5cb36a30d9ead114ee086e11fe7023d2a54164d
3
- size 380
 
 
 
 
ANE/styletts2_ane_plbert.mlmodelc/metadata.json DELETED
@@ -1,83 +0,0 @@
1
- [
2
- {
3
- "metadataOutputVersion" : "3.0",
4
- "shortDescription" : "StyleTTS2-ANE Stage 1: PLBERT (Albert) — fp16, ANE.",
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
- "storagePrecision" : "Mixed (Float16, Palettized (8 bits))",
18
- "modelParameters" : [
19
-
20
- ],
21
- "specificationVersion" : 8,
22
- "mlProgramOperationTypeHistogram" : {
23
- "Fill" : 1,
24
- "Ios17.reshape" : 48,
25
- "Ios16.softmax" : 12,
26
- "Ios17.matmul" : 24,
27
- "Ios17.transpose" : 48,
28
- "Select" : 1,
29
- "Ios17.expandDims" : 2,
30
- "Ios17.add" : 38,
31
- "Tile" : 2,
32
- "Ios17.sliceByIndex" : 2,
33
- "Shape" : 16,
34
- "Ios17.gather" : 29,
35
- "Ios17.layerNorm" : 25,
36
- "Ios17.cast" : 44,
37
- "Ios17.sub" : 1,
38
- "Ios16.constexprLutToDense" : 9,
39
- "Ios16.gelu" : 12,
40
- "Ios17.realDiv" : 2,
41
- "Ios17.linear" : 73,
42
- "Ios17.concat" : 26,
43
- "Ios17.mul" : 12
44
- },
45
- "computePrecision" : "Mixed (Float16, Int16, Int32, UInt16)",
46
- "isUpdatable" : "0",
47
- "stateSchema" : [
48
-
49
- ],
50
- "availability" : {
51
- "macOS" : "14.0",
52
- "tvOS" : "17.0",
53
- "visionOS" : "1.0",
54
- "watchOS" : "10.0",
55
- "iOS" : "17.0",
56
- "macCatalyst" : "17.0"
57
- },
58
- "modelType" : {
59
- "name" : "MLModelType_mlProgram"
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 × 64)",
69
- "type" : "MultiArray",
70
- "shape" : "[1, 64]",
71
- "name" : "tokens",
72
- "shortDescription" : ""
73
- }
74
- ],
75
- "userDefinedMetadata" : {
76
- "com.github.apple.coremltools.version" : "8.3.0",
77
- "com.github.apple.coremltools.source_dialect" : "TorchScript",
78
- "com.github.apple.coremltools.source" : "torch==2.11.0"
79
- },
80
- "generatedClassName" : "styletts2_ane_plbert",
81
- "method" : "predict"
82
- }
83
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_plbert.mlmodelc/model.mil DELETED
The diff for this file is too large to render. See raw diff
 
ANE/styletts2_ane_plbert.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:59c7da8562b26c0842a3941543ebf48d94629105022320c9d8cd590761b6fcab
3
- size 5718848
 
 
 
 
ANE/styletts2_ane_plbert.mlpackage/Data/com.apple.CoreML/model.mlmodel DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8a91ff072efb2aa10a26f76e280108f110a877429d74ec76769aa560c3ebbc7e
3
- size 148365
 
 
 
 
ANE/styletts2_ane_plbert.mlpackage/Data/com.apple.CoreML/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:59c7da8562b26c0842a3941543ebf48d94629105022320c9d8cd590761b6fcab
3
- size 5718848
 
 
 
 
ANE/styletts2_ane_plbert.mlpackage/Manifest.json DELETED
@@ -1,18 +0,0 @@
1
- {
2
- "fileFormatVersion": "1.0.0",
3
- "itemInfoEntries": {
4
- "3B613FCB-B965-4C2D-94A7-4D5C80EDE23F": {
5
- "author": "com.apple.CoreML",
6
- "description": "CoreML Model Weights",
7
- "name": "weights",
8
- "path": "com.apple.CoreML/weights"
9
- },
10
- "440AF525-7EC6-4E07-8019-7BF58173E928": {
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": "440AF525-7EC6-4E07-8019-7BF58173E928"
18
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_postbert.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:a52227f7c26ca3ef5fa3069d2efc614c652412125fad1d2838ecccb4c0e293e0
3
- size 243
 
 
 
 
ANE/styletts2_ane_postbert.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:19aff2a2236725dc1051a37174fe4380e7231c609591527522b114294a86932f
3
- size 542
 
 
 
 
ANE/styletts2_ane_postbert.mlmodelc/metadata.json DELETED
@@ -1,134 +0,0 @@
1
- [
2
- {
3
- "metadataOutputVersion" : "3.0",
4
- "shortDescription" : "StyleTTS2-ANE Stage 2: TextEncoder + DurationEncoder + duration head — fp16, ANE.",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float16",
10
- "formattedType" : "MultiArray (Float16)",
11
- "shortDescription" : "",
12
- "shape" : "[]",
13
- "name" : "t_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" : "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" : "pred_dur_log",
34
- "type" : "MultiArray"
35
- },
36
- {
37
- "hasShapeFlexibility" : "0",
38
- "isOptional" : "0",
39
- "dataType" : "Float16",
40
- "formattedType" : "MultiArray (Float16)",
41
- "shortDescription" : "",
42
- "shape" : "[]",
43
- "name" : "fixed_embedding",
44
- "type" : "MultiArray"
45
- }
46
- ],
47
- "storagePrecision" : "Mixed (Float16, Palettized (8 bits))",
48
- "modelParameters" : [
49
-
50
- ],
51
- "specificationVersion" : 8,
52
- "mlProgramOperationTypeHistogram" : {
53
- "Ios17.reshape" : 3,
54
- "Ios17.transpose" : 27,
55
- "Split" : 3,
56
- "Ios17.expandDims" : 2,
57
- "Ios17.add" : 7,
58
- "Ios17.sliceByIndex" : 3,
59
- "Ios16.fillLike" : 1,
60
- "Ios17.squeeze" : 1,
61
- "Ios17.gather" : 2,
62
- "Ios17.layerNorm" : 6,
63
- "Ios17.conv" : 3,
64
- "Ios17.sub" : 2,
65
- "Ios16.constexprLutToDense" : 30,
66
- "Ios17.lstm" : 5,
67
- "Ios17.cast" : 3,
68
- "Ios17.linear" : 5,
69
- "Ios17.leakyRelu" : 3,
70
- "Ios17.concat" : 4,
71
- "Ios16.cumsum" : 1,
72
- "Ios17.mul" : 3
73
- },
74
- "computePrecision" : "Mixed (Float16, Float32, Int16, Int32, UInt16)",
75
- "isUpdatable" : "0",
76
- "stateSchema" : [
77
-
78
- ],
79
- "availability" : {
80
- "macOS" : "14.0",
81
- "tvOS" : "17.0",
82
- "visionOS" : "1.0",
83
- "watchOS" : "10.0",
84
- "iOS" : "17.0",
85
- "macCatalyst" : "17.0"
86
- },
87
- "modelType" : {
88
- "name" : "MLModelType_mlProgram"
89
- },
90
- "inputSchema" : [
91
- {
92
- "dataType" : "Float16",
93
- "hasShapeFlexibility" : "1",
94
- "isOptional" : "0",
95
- "shapeFlexibility" : "1 × 2...512 × 768",
96
- "shapeRange" : "[[1, 1], [2, 512], [768, 768]]",
97
- "formattedType" : "MultiArray (Float16 1 × 64 × 768)",
98
- "type" : "MultiArray",
99
- "shape" : "[1, 64, 768]",
100
- "name" : "bert_dur",
101
- "shortDescription" : ""
102
- },
103
- {
104
- "dataType" : "Int32",
105
- "hasShapeFlexibility" : "1",
106
- "isOptional" : "0",
107
- "shapeFlexibility" : "1 × 2...512",
108
- "shapeRange" : "[[1, 1], [2, 512]]",
109
- "formattedType" : "MultiArray (Int32 1 × 64)",
110
- "type" : "MultiArray",
111
- "shape" : "[1, 64]",
112
- "name" : "tokens",
113
- "shortDescription" : ""
114
- },
115
- {
116
- "hasShapeFlexibility" : "0",
117
- "isOptional" : "0",
118
- "dataType" : "Float16",
119
- "formattedType" : "MultiArray (Float16 1 × 256)",
120
- "shortDescription" : "",
121
- "shape" : "[1, 256]",
122
- "name" : "style",
123
- "type" : "MultiArray"
124
- }
125
- ],
126
- "userDefinedMetadata" : {
127
- "com.github.apple.coremltools.version" : "8.3.0",
128
- "com.github.apple.coremltools.source" : "torch==2.11.0",
129
- "com.github.apple.coremltools.source_dialect" : "TorchScript"
130
- },
131
- "generatedClassName" : "styletts2_ane_postbert",
132
- "method" : "predict"
133
- }
134
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_postbert.mlmodelc/model.mil DELETED
@@ -1,265 +0,0 @@
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, ?, 768]> bert_dur, tensor<fp16, [1, 256]> style, tensor<int32, [1, ?]> tokens) [FlexibleShapeInformation = tuple<tuple<tensor<string, []>, dict<tensor<string, []>, tensor<int32, [?]>>>, tuple<tensor<string, []>, dict<tensor<string, []>, list<tensor<int32, [2]>, ?>>>>((("DefaultShapes", {{"bert_dur", [1, 64, 768]}, {"tokens", [1, 64]}}), ("RangeDims", {{"bert_dur", [[1, 1], [2, 512], [768, 768]]}, {"tokens", [[1, 1], [2, 512]]}})))] {
5
- tensor<int32, [2]> style_begin_0 = const()[name = tensor<string, []>("style_begin_0"), val = tensor<int32, [2]>([0, 128])];
6
- tensor<int32, [2]> style_end_0 = const()[name = tensor<string, []>("style_end_0"), val = tensor<int32, [2]>([1, 256])];
7
- tensor<bool, [2]> style_end_mask_0 = const()[name = tensor<string, []>("style_end_mask_0"), val = tensor<bool, [2]>([true, true])];
8
- tensor<fp16, [1, 128]> style_cast_fp16 = slice_by_index(begin = style_begin_0, end = style_end_0, end_mask = style_end_mask_0, x = style)[name = tensor<string, []>("style_cast_fp16")];
9
- tensor<fp32, []> var_30 = const()[name = tensor<string, []>("op_30"), val = tensor<fp32, []>(0x1.99999ap-3)];
10
- tensor<int32, []> x_1_axis_0 = const()[name = tensor<string, []>("x_1_axis_0"), val = tensor<int32, []>(0)];
11
- tensor<int32, []> x_1_batch_dims_0 = const()[name = tensor<string, []>("x_1_batch_dims_0"), val = tensor<int32, []>(0)];
12
- tensor<bool, []> x_1_validate_indices_0 = const()[name = tensor<string, []>("x_1_validate_indices_0"), val = tensor<bool, []>(false)];
13
- 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, []>(64))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(91264))), name = tensor<string, []>("text_encoder_embedding_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([178, 512])];
14
- tensor<string, []> tokens_to_int16_dtype_0 = const()[name = tensor<string, []>("tokens_to_int16_dtype_0"), val = tensor<string, []>("int16")];
15
- tensor<int16, [1, ?]> tokens_to_int16 = cast(dtype = tokens_to_int16_dtype_0, x = tokens)[name = tensor<string, []>("cast_2")];
16
- tensor<fp16, [1, ?, 512]> x_1_cast_fp16_cast_uint16 = gather(axis = x_1_axis_0, batch_dims = x_1_batch_dims_0, indices = tokens_to_int16, validate_indices = x_1_validate_indices_0, x = text_encoder_embedding_weight_to_fp16_palettized)[name = tensor<string, []>("x_1_cast_fp16_cast_uint16")];
17
- tensor<int32, [3]> input_1_perm_0 = const()[name = tensor<string, []>("input_1_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
18
- tensor<string, []> x_3_pad_type_0 = const()[name = tensor<string, []>("x_3_pad_type_0"), val = tensor<string, []>("custom")];
19
- tensor<int32, [2]> x_3_pad_0 = const()[name = tensor<string, []>("x_3_pad_0"), val = tensor<int32, [2]>([2, 2])];
20
- tensor<int32, [1]> x_3_strides_0 = const()[name = tensor<string, []>("x_3_strides_0"), val = tensor<int32, [1]>([1])];
21
- tensor<int32, [1]> x_3_dilations_0 = const()[name = tensor<string, []>("x_3_dilations_0"), val = tensor<int32, [1]>([1])];
22
- tensor<int32, []> x_3_groups_0 = const()[name = tensor<string, []>("x_3_groups_0"), val = tensor<int32, []>(1)];
23
- 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, []>(91840))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1402624))), name = tensor<string, []>("weight_3_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 5])];
24
- 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, []>(1403200)))];
25
- tensor<fp16, [1, 512, ?]> input_1_cast_fp16 = transpose(perm = input_1_perm_0, x = x_1_cast_fp16_cast_uint16)[name = tensor<string, []>("transpose_26")];
26
- tensor<fp16, [1, 512, ?]> x_3_cast_fp16 = conv(bias = text_encoder_cnn_0_0_bias_to_fp16, dilations = x_3_dilations_0, groups = x_3_groups_0, pad = x_3_pad_0, pad_type = x_3_pad_type_0, strides = x_3_strides_0, weight = weight_3_to_fp16_palettized, x = input_1_cast_fp16)[name = tensor<string, []>("x_3_cast_fp16")];
27
- tensor<int32, [3]> input_3_perm_0 = const()[name = tensor<string, []>("input_3_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
28
- tensor<int32, [1]> x_5_axes_0 = const()[name = tensor<string, []>("x_5_axes_0"), val = tensor<int32, [1]>([-1])];
29
- 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, []>(1404288)))];
30
- 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, []>(1405376)))];
31
- tensor<fp16, []> var_28_to_fp16 = const()[name = tensor<string, []>("op_28_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
32
- tensor<fp16, [1, ?, 512]> input_3_cast_fp16 = transpose(perm = input_3_perm_0, x = x_3_cast_fp16)[name = tensor<string, []>("transpose_25")];
33
- tensor<fp16, [1, ?, 512]> x_5_cast_fp16 = layer_norm(axes = x_5_axes_0, beta = text_encoder_cnn_0_1_beta_to_fp16, epsilon = var_28_to_fp16, gamma = text_encoder_cnn_0_1_gamma_to_fp16, x = input_3_cast_fp16)[name = tensor<string, []>("x_5_cast_fp16")];
34
- tensor<int32, [3]> input_5_perm_0 = const()[name = tensor<string, []>("input_5_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
35
- tensor<fp16, [1, 512, ?]> input_5_cast_fp16 = transpose(perm = input_5_perm_0, x = x_5_cast_fp16)[name = tensor<string, []>("transpose_24")];
36
- tensor<fp16, [1, 512, ?]> input_7_cast_fp16 = leaky_relu(alpha = var_30, x = input_5_cast_fp16)[name = tensor<string, []>("input_7_cast_fp16")];
37
- tensor<string, []> x_7_pad_type_0 = const()[name = tensor<string, []>("x_7_pad_type_0"), val = tensor<string, []>("custom")];
38
- tensor<int32, [2]> x_7_pad_0 = const()[name = tensor<string, []>("x_7_pad_0"), val = tensor<int32, [2]>([2, 2])];
39
- tensor<int32, [1]> x_7_strides_0 = const()[name = tensor<string, []>("x_7_strides_0"), val = tensor<int32, [1]>([1])];
40
- tensor<int32, [1]> x_7_dilations_0 = const()[name = tensor<string, []>("x_7_dilations_0"), val = tensor<int32, [1]>([1])];
41
- tensor<int32, []> x_7_groups_0 = const()[name = tensor<string, []>("x_7_groups_0"), val = tensor<int32, []>(1)];
42
- 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, []>(1406464))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2717248))), name = tensor<string, []>("weight_7_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 5])];
43
- 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, []>(2717824)))];
44
- tensor<fp16, [1, 512, ?]> x_7_cast_fp16 = conv(bias = text_encoder_cnn_1_0_bias_to_fp16, dilations = x_7_dilations_0, groups = x_7_groups_0, pad = x_7_pad_0, pad_type = x_7_pad_type_0, strides = x_7_strides_0, weight = weight_7_to_fp16_palettized, x = input_7_cast_fp16)[name = tensor<string, []>("x_7_cast_fp16")];
45
- tensor<int32, [3]> input_11_perm_0 = const()[name = tensor<string, []>("input_11_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
46
- tensor<int32, [1]> x_9_axes_0 = const()[name = tensor<string, []>("x_9_axes_0"), val = tensor<int32, [1]>([-1])];
47
- 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, []>(2718912)))];
48
- 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, []>(2720000)))];
49
- tensor<fp16, [1, ?, 512]> input_11_cast_fp16 = transpose(perm = input_11_perm_0, x = x_7_cast_fp16)[name = tensor<string, []>("transpose_23")];
50
- tensor<fp16, [1, ?, 512]> x_9_cast_fp16 = layer_norm(axes = x_9_axes_0, beta = text_encoder_cnn_1_1_beta_to_fp16, epsilon = var_28_to_fp16, gamma = text_encoder_cnn_1_1_gamma_to_fp16, x = input_11_cast_fp16)[name = tensor<string, []>("x_9_cast_fp16")];
51
- tensor<int32, [3]> input_13_perm_0 = const()[name = tensor<string, []>("input_13_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
52
- tensor<fp16, [1, 512, ?]> input_13_cast_fp16 = transpose(perm = input_13_perm_0, x = x_9_cast_fp16)[name = tensor<string, []>("transpose_22")];
53
- tensor<fp16, [1, 512, ?]> input_15_cast_fp16 = leaky_relu(alpha = var_30, x = input_13_cast_fp16)[name = tensor<string, []>("input_15_cast_fp16")];
54
- tensor<string, []> x_11_pad_type_0 = const()[name = tensor<string, []>("x_11_pad_type_0"), val = tensor<string, []>("custom")];
55
- tensor<int32, [2]> x_11_pad_0 = const()[name = tensor<string, []>("x_11_pad_0"), val = tensor<int32, [2]>([2, 2])];
56
- tensor<int32, [1]> x_11_strides_0 = const()[name = tensor<string, []>("x_11_strides_0"), val = tensor<int32, [1]>([1])];
57
- tensor<int32, [1]> x_11_dilations_0 = const()[name = tensor<string, []>("x_11_dilations_0"), val = tensor<int32, [1]>([1])];
58
- tensor<int32, []> x_11_groups_0 = const()[name = tensor<string, []>("x_11_groups_0"), val = tensor<int32, []>(1)];
59
- 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, []>(2721088))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4031872))), name = tensor<string, []>("weight_11_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 5])];
60
- 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, []>(4032448)))];
61
- tensor<fp16, [1, 512, ?]> x_11_cast_fp16 = conv(bias = text_encoder_cnn_2_0_bias_to_fp16, dilations = x_11_dilations_0, groups = x_11_groups_0, pad = x_11_pad_0, pad_type = x_11_pad_type_0, strides = x_11_strides_0, weight = weight_11_to_fp16_palettized, x = input_15_cast_fp16)[name = tensor<string, []>("x_11_cast_fp16")];
62
- tensor<int32, [3]> input_19_perm_0 = const()[name = tensor<string, []>("input_19_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
63
- tensor<int32, [1]> x_13_axes_0 = const()[name = tensor<string, []>("x_13_axes_0"), val = tensor<int32, [1]>([-1])];
64
- 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, []>(4033536)))];
65
- 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, []>(4034624)))];
66
- tensor<fp16, [1, ?, 512]> input_19_cast_fp16 = transpose(perm = input_19_perm_0, x = x_11_cast_fp16)[name = tensor<string, []>("transpose_21")];
67
- tensor<fp16, [1, ?, 512]> x_13_cast_fp16 = layer_norm(axes = x_13_axes_0, beta = text_encoder_cnn_2_1_beta_to_fp16, epsilon = var_28_to_fp16, gamma = text_encoder_cnn_2_1_gamma_to_fp16, x = input_19_cast_fp16)[name = tensor<string, []>("x_13_cast_fp16")];
68
- tensor<fp16, [1, ?, 512]> input_23_cast_fp16 = leaky_relu(alpha = var_30, x = x_13_cast_fp16)[name = tensor<string, []>("input_23_cast_fp16")];
69
- tensor<int32, [3]> input_25_batch_first_transpose_perm_0 = const()[name = tensor<string, []>("input_25_batch_first_transpose_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
70
- tensor<string, []> x_17_batch_first_direction_0 = const()[name = tensor<string, []>("x_17_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
71
- tensor<bool, []> x_17_batch_first_output_sequence_0 = const()[name = tensor<string, []>("x_17_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
72
- tensor<string, []> x_17_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("x_17_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
73
- tensor<string, []> x_17_batch_first_cell_activation_0 = const()[name = tensor<string, []>("x_17_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
74
- tensor<string, []> x_17_batch_first_activation_0 = const()[name = tensor<string, []>("x_17_batch_first_activation_0"), val = tensor<string, []>("tanh")];
75
- tensor<fp16, [1, 512]> x_17_batch_first_lstm_h0_reshaped_to_fp16 = const()[name = tensor<string, []>("x_17_batch_first_lstm_h0_reshaped_to_fp16"), val = tensor<fp16, [1, 512]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4035712)))];
76
- tensor<fp16, [1024, 512]> concat_4_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4036800))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4561152))), name = tensor<string, []>("concat_4_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 512])];
77
- 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, []>(4561728))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4823936))), name = tensor<string, []>("concat_5_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
78
- 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, []>(4824512)))];
79
- tensor<fp16, [1024, 512]> concat_6_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [524288]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4826624))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5350976))), name = tensor<string, []>("concat_6_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 512])];
80
- 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, []>(5351552))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5613760))), name = tensor<string, []>("concat_7_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
81
- 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, []>(5614336)))];
82
- tensor<fp16, [?, 1, 512]> input_25_batch_first_transpose_cast_fp16 = transpose(perm = input_25_batch_first_transpose_perm_0, x = input_23_cast_fp16)[name = tensor<string, []>("transpose_20")];
83
- tensor<fp16, [?, 1, 512]> x_17_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> x_17_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> x_17_batch_first_cast_fp16_2 = lstm(activation = x_17_batch_first_activation_0, bias = add_1_to_fp16, bias_back = add_2_to_fp16, cell_activation = x_17_batch_first_cell_activation_0, direction = x_17_batch_first_direction_0, initial_c = x_17_batch_first_lstm_h0_reshaped_to_fp16, initial_h = x_17_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = x_17_batch_first_output_sequence_0, recurrent_activation = x_17_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 = input_25_batch_first_transpose_cast_fp16)[name = tensor<string, []>("x_17_batch_first_cast_fp16")];
84
- tensor<int32, [3]> transpose_0_perm_0 = const()[name = tensor<string, []>("transpose_0_perm_0"), val = tensor<int32, [3]>([1, 2, 0])];
85
- 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, []>(5616448))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6009728))), name = tensor<string, []>("bert_encoder_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 768])];
86
- 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, []>(6010304)))];
87
- 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")];
88
- tensor<int32, [3]> x_19_perm_0 = const()[name = tensor<string, []>("x_19_perm_0"), val = tensor<int32, [3]>([0, -1, -2])];
89
- tensor<int32, []> var_147 = const()[name = tensor<string, []>("op_147"), val = tensor<int32, []>(1)];
90
- tensor<int32, [3]> var_163_begin_0 = const()[name = tensor<string, []>("op_163_begin_0"), val = tensor<int32, [3]>([0, 0, 0])];
91
- tensor<int32, [3]> var_163_end_0 = const()[name = tensor<string, []>("op_163_end_0"), val = tensor<int32, [3]>([1, 1, 0])];
92
- tensor<bool, [3]> var_163_end_mask_0 = const()[name = tensor<string, []>("op_163_end_mask_0"), val = tensor<bool, [3]>([true, false, true])];
93
- tensor<fp16, [1, 512, ?]> x_19_cast_fp16 = transpose(perm = x_19_perm_0, x = linear_0_cast_fp16)[name = tensor<string, []>("transpose_19")];
94
- tensor<fp16, [1, 1, ?]> var_163_cast_fp16 = slice_by_index(begin = var_163_begin_0, end = var_163_end_0, end_mask = var_163_end_mask_0, x = x_19_cast_fp16)[name = tensor<string, []>("op_163_cast_fp16")];
95
- tensor<fp16, [1, 1, ?]> sub_0_cast_fp16 = sub(x = var_163_cast_fp16, y = var_163_cast_fp16)[name = tensor<string, []>("sub_0_cast_fp16")];
96
- tensor<int32, [1]> var_166_axes_0 = const()[name = tensor<string, []>("op_166_axes_0"), val = tensor<int32, [1]>([-1])];
97
- tensor<fp16, [1, 128, 1]> var_166_cast_fp16 = expand_dims(axes = var_166_axes_0, x = style_cast_fp16)[name = tensor<string, []>("op_166_cast_fp16")];
98
- tensor<fp16, [1, 128, ?]> s_chan_cast_fp16 = add(x = var_166_cast_fp16, y = sub_0_cast_fp16)[name = tensor<string, []>("s_chan_cast_fp16")];
99
- tensor<bool, []> x_21_interleave_0 = const()[name = tensor<string, []>("x_21_interleave_0"), val = tensor<bool, []>(false)];
100
- tensor<fp16, [1, 640, ?]> x_21_cast_fp16 = concat(axis = var_147, interleave = x_21_interleave_0, values = (x_19_cast_fp16, s_chan_cast_fp16))[name = tensor<string, []>("x_21_cast_fp16")];
101
- tensor<int32, [3]> transpose_1_perm_0 = const()[name = tensor<string, []>("transpose_1_perm_0"), val = tensor<int32, [3]>([-1, 0, -2])];
102
- tensor<string, []> input_29_batch_first_direction_0 = const()[name = tensor<string, []>("input_29_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
103
- tensor<bool, []> input_29_batch_first_output_sequence_0 = const()[name = tensor<string, []>("input_29_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
104
- tensor<string, []> input_29_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("input_29_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
105
- tensor<string, []> input_29_batch_first_cell_activation_0 = const()[name = tensor<string, []>("input_29_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
106
- tensor<string, []> input_29_batch_first_activation_0 = const()[name = tensor<string, []>("input_29_batch_first_activation_0"), val = tensor<string, []>("tanh")];
107
- tensor<fp16, [1024, 640]> concat_14_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6011392))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6666816))), name = tensor<string, []>("concat_14_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
108
- tensor<fp16, [1024, 256]> concat_15_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6667392))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6929600))), name = tensor<string, []>("concat_15_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
109
- 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, []>(6930176)))];
110
- tensor<fp16, [1024, 640]> concat_16_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6932288))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7587712))), name = tensor<string, []>("concat_16_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
111
- tensor<fp16, [1024, 256]> concat_17_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7588288))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7850496))), name = tensor<string, []>("concat_17_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
112
- 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, []>(7851072)))];
113
- tensor<fp16, [?, 1, 640]> transpose_1_cast_fp16 = transpose(perm = transpose_1_perm_0, x = x_21_cast_fp16)[name = tensor<string, []>("transpose_18")];
114
- tensor<fp16, [?, 1, 512]> input_29_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> input_29_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> input_29_batch_first_cast_fp16_2 = lstm(activation = input_29_batch_first_activation_0, bias = add_4_to_fp16, bias_back = add_5_to_fp16, cell_activation = input_29_batch_first_cell_activation_0, direction = input_29_batch_first_direction_0, initial_c = x_17_batch_first_lstm_h0_reshaped_to_fp16, initial_h = x_17_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = input_29_batch_first_output_sequence_0, recurrent_activation = input_29_batch_first_recurrent_activation_0, weight_hh = concat_15_to_fp16_palettized, weight_hh_back = concat_17_to_fp16_palettized, weight_ih = concat_14_to_fp16_palettized, weight_ih_back = concat_16_to_fp16_palettized, x = transpose_1_cast_fp16)[name = tensor<string, []>("input_29_batch_first_cast_fp16")];
115
- tensor<int32, [3]> transpose_7_perm_0 = const()[name = tensor<string, []>("transpose_7_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
116
- tensor<fp16, [1024, 128]> duration_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, []>(7853184))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7984320))), name = tensor<string, []>("duration_encoder_norms_0_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
117
- tensor<fp16, [1024]> duration_encoder_norms_0_fc_bias_to_fp16 = const()[name = tensor<string, []>("duration_encoder_norms_0_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7984896)))];
118
- tensor<fp16, [1, 1024]> linear_1_cast_fp16 = linear(bias = duration_encoder_norms_0_fc_bias_to_fp16, weight = duration_encoder_norms_0_fc_weight_to_fp16_palettized, x = style_cast_fp16)[name = tensor<string, []>("linear_1_cast_fp16")];
119
- tensor<int32, [3]> var_200 = const()[name = tensor<string, []>("op_200"), val = tensor<int32, [3]>([1, 1024, 1])];
120
- tensor<fp16, [1, 1024, 1]> h_3_cast_fp16 = reshape(shape = var_200, x = linear_1_cast_fp16)[name = tensor<string, []>("h_3_cast_fp16")];
121
- tensor<int32, [2]> var_202_split_sizes_0 = const()[name = tensor<string, []>("op_202_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
122
- tensor<int32, []> var_202_axis_0 = const()[name = tensor<string, []>("op_202_axis_0"), val = tensor<int32, []>(1)];
123
- tensor<fp16, [1, 512, 1]> var_202_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_202_cast_fp16_1 = split(axis = var_202_axis_0, split_sizes = var_202_split_sizes_0, x = h_3_cast_fp16)[name = tensor<string, []>("op_202_cast_fp16")];
124
- tensor<int32, [3]> gamma_3_perm_0 = const()[name = tensor<string, []>("gamma_3_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
125
- tensor<int32, [3]> beta_3_perm_0 = const()[name = tensor<string, []>("beta_3_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
126
- tensor<int32, [1]> x_31_axes_0 = const()[name = tensor<string, []>("x_31_axes_0"), val = tensor<int32, [1]>([-1])];
127
- tensor<fp16, []> var_136_to_fp16 = const()[name = tensor<string, []>("op_136_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
128
- tensor<fp16, [1, ?, 512]> transpose_7_cast_fp16 = transpose(perm = transpose_7_perm_0, x = input_29_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_17")];
129
- tensor<fp16, [1, ?, 512]> x_31_cast_fp16 = layer_norm(axes = x_31_axes_0, epsilon = var_136_to_fp16, x = transpose_7_cast_fp16)[name = tensor<string, []>("x_31_cast_fp16")];
130
- tensor<fp16, []> var_208_promoted_to_fp16 = const()[name = tensor<string, []>("op_208_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
131
- tensor<fp16, [1, 1, 512]> gamma_3_cast_fp16 = transpose(perm = gamma_3_perm_0, x = var_202_cast_fp16_0)[name = tensor<string, []>("transpose_16")];
132
- tensor<fp16, [1, 1, 512]> var_209_cast_fp16 = add(x = gamma_3_cast_fp16, y = var_208_promoted_to_fp16)[name = tensor<string, []>("op_209_cast_fp16")];
133
- tensor<fp16, [1, ?, 512]> var_210_cast_fp16 = mul(x = var_209_cast_fp16, y = x_31_cast_fp16)[name = tensor<string, []>("op_210_cast_fp16")];
134
- tensor<fp16, [1, 1, 512]> beta_3_cast_fp16 = transpose(perm = beta_3_perm_0, x = var_202_cast_fp16_1)[name = tensor<string, []>("transpose_15")];
135
- tensor<fp16, [1, ?, 512]> x_33_cast_fp16 = add(x = var_210_cast_fp16, y = beta_3_cast_fp16)[name = tensor<string, []>("x_33_cast_fp16")];
136
- tensor<int32, [3]> x_35_perm_0 = const()[name = tensor<string, []>("x_35_perm_0"), val = tensor<int32, [3]>([0, -1, -2])];
137
- tensor<bool, []> x_37_interleave_0 = const()[name = tensor<string, []>("x_37_interleave_0"), val = tensor<bool, []>(false)];
138
- tensor<fp16, [1, 512, ?]> x_35_cast_fp16 = transpose(perm = x_35_perm_0, x = x_33_cast_fp16)[name = tensor<string, []>("transpose_14")];
139
- tensor<fp16, [1, 640, ?]> x_37_cast_fp16 = concat(axis = var_147, interleave = x_37_interleave_0, values = (x_35_cast_fp16, s_chan_cast_fp16))[name = tensor<string, []>("x_37_cast_fp16")];
140
- tensor<int32, [3]> transpose_3_perm_0 = const()[name = tensor<string, []>("transpose_3_perm_0"), val = tensor<int32, [3]>([-1, 0, -2])];
141
- tensor<string, []> input_35_batch_first_direction_0 = const()[name = tensor<string, []>("input_35_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
142
- tensor<bool, []> input_35_batch_first_output_sequence_0 = const()[name = tensor<string, []>("input_35_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
143
- tensor<string, []> input_35_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("input_35_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
144
- tensor<string, []> input_35_batch_first_cell_activation_0 = const()[name = tensor<string, []>("input_35_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
145
- tensor<string, []> input_35_batch_first_activation_0 = const()[name = tensor<string, []>("input_35_batch_first_activation_0"), val = tensor<string, []>("tanh")];
146
- tensor<fp16, [1024, 640]> concat_24_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7987008))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8642432))), name = tensor<string, []>("concat_24_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
147
- tensor<fp16, [1024, 256]> concat_25_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8643008))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8905216))), name = tensor<string, []>("concat_25_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
148
- 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, []>(8905792)))];
149
- tensor<fp16, [1024, 640]> concat_26_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8907904))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9563328))), name = tensor<string, []>("concat_26_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
150
- tensor<fp16, [1024, 256]> concat_27_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9563904))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9826112))), name = tensor<string, []>("concat_27_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
151
- 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, []>(9826688)))];
152
- tensor<fp16, [?, 1, 640]> transpose_3_cast_fp16 = transpose(perm = transpose_3_perm_0, x = x_37_cast_fp16)[name = tensor<string, []>("transpose_13")];
153
- tensor<fp16, [?, 1, 512]> input_35_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> input_35_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> input_35_batch_first_cast_fp16_2 = lstm(activation = input_35_batch_first_activation_0, bias = add_7_to_fp16, bias_back = add_8_to_fp16, cell_activation = input_35_batch_first_cell_activation_0, direction = input_35_batch_first_direction_0, initial_c = x_17_batch_first_lstm_h0_reshaped_to_fp16, initial_h = x_17_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = input_35_batch_first_output_sequence_0, recurrent_activation = input_35_batch_first_recurrent_activation_0, weight_hh = concat_25_to_fp16_palettized, weight_hh_back = concat_27_to_fp16_palettized, weight_ih = concat_24_to_fp16_palettized, weight_ih_back = concat_26_to_fp16_palettized, x = transpose_3_cast_fp16)[name = tensor<string, []>("input_35_batch_first_cast_fp16")];
154
- tensor<int32, [3]> transpose_8_perm_0 = const()[name = tensor<string, []>("transpose_8_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
155
- tensor<fp16, [1024, 128]> duration_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, []>(9828800))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9959936))), name = tensor<string, []>("duration_encoder_norms_1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
156
- tensor<fp16, [1024]> duration_encoder_norms_1_fc_bias_to_fp16 = const()[name = tensor<string, []>("duration_encoder_norms_1_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9960512)))];
157
- tensor<fp16, [1, 1024]> linear_2_cast_fp16 = linear(bias = duration_encoder_norms_1_fc_bias_to_fp16, weight = duration_encoder_norms_1_fc_weight_to_fp16_palettized, x = style_cast_fp16)[name = tensor<string, []>("linear_2_cast_fp16")];
158
- tensor<int32, [3]> var_247 = const()[name = tensor<string, []>("op_247"), val = tensor<int32, [3]>([1, 1024, 1])];
159
- tensor<fp16, [1, 1024, 1]> h_7_cast_fp16 = reshape(shape = var_247, x = linear_2_cast_fp16)[name = tensor<string, []>("h_7_cast_fp16")];
160
- tensor<int32, [2]> var_249_split_sizes_0 = const()[name = tensor<string, []>("op_249_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
161
- tensor<int32, []> var_249_axis_0 = const()[name = tensor<string, []>("op_249_axis_0"), val = tensor<int32, []>(1)];
162
- tensor<fp16, [1, 512, 1]> var_249_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_249_cast_fp16_1 = split(axis = var_249_axis_0, split_sizes = var_249_split_sizes_0, x = h_7_cast_fp16)[name = tensor<string, []>("op_249_cast_fp16")];
163
- tensor<int32, [3]> gamma_7_perm_0 = const()[name = tensor<string, []>("gamma_7_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
164
- tensor<int32, [3]> beta_7_perm_0 = const()[name = tensor<string, []>("beta_7_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
165
- tensor<int32, [1]> x_47_axes_0 = const()[name = tensor<string, []>("x_47_axes_0"), val = tensor<int32, [1]>([-1])];
166
- tensor<fp16, [1, ?, 512]> transpose_8_cast_fp16 = transpose(perm = transpose_8_perm_0, x = input_35_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_12")];
167
- tensor<fp16, [1, ?, 512]> x_47_cast_fp16 = layer_norm(axes = x_47_axes_0, epsilon = var_136_to_fp16, x = transpose_8_cast_fp16)[name = tensor<string, []>("x_47_cast_fp16")];
168
- tensor<fp16, []> var_255_promoted_to_fp16 = const()[name = tensor<string, []>("op_255_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
169
- tensor<fp16, [1, 1, 512]> gamma_7_cast_fp16 = transpose(perm = gamma_7_perm_0, x = var_249_cast_fp16_0)[name = tensor<string, []>("transpose_11")];
170
- tensor<fp16, [1, 1, 512]> var_256_cast_fp16 = add(x = gamma_7_cast_fp16, y = var_255_promoted_to_fp16)[name = tensor<string, []>("op_256_cast_fp16")];
171
- tensor<fp16, [1, ?, 512]> var_257_cast_fp16 = mul(x = var_256_cast_fp16, y = x_47_cast_fp16)[name = tensor<string, []>("op_257_cast_fp16")];
172
- tensor<fp16, [1, 1, 512]> beta_7_cast_fp16 = transpose(perm = beta_7_perm_0, x = var_249_cast_fp16_1)[name = tensor<string, []>("transpose_10")];
173
- tensor<fp16, [1, ?, 512]> x_49_cast_fp16 = add(x = var_257_cast_fp16, y = beta_7_cast_fp16)[name = tensor<string, []>("x_49_cast_fp16")];
174
- tensor<int32, [3]> x_51_perm_0 = const()[name = tensor<string, []>("x_51_perm_0"), val = tensor<int32, [3]>([0, -1, -2])];
175
- tensor<bool, []> x_53_interleave_0 = const()[name = tensor<string, []>("x_53_interleave_0"), val = tensor<bool, []>(false)];
176
- tensor<fp16, [1, 512, ?]> x_51_cast_fp16 = transpose(perm = x_51_perm_0, x = x_49_cast_fp16)[name = tensor<string, []>("transpose_9")];
177
- tensor<fp16, [1, 640, ?]> x_53_cast_fp16 = concat(axis = var_147, interleave = x_53_interleave_0, values = (x_51_cast_fp16, s_chan_cast_fp16))[name = tensor<string, []>("x_53_cast_fp16")];
178
- tensor<int32, [3]> transpose_5_perm_0 = const()[name = tensor<string, []>("transpose_5_perm_0"), val = tensor<int32, [3]>([-1, 0, -2])];
179
- tensor<string, []> input_41_batch_first_direction_0 = const()[name = tensor<string, []>("input_41_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
180
- tensor<bool, []> input_41_batch_first_output_sequence_0 = const()[name = tensor<string, []>("input_41_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
181
- tensor<string, []> input_41_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("input_41_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
182
- tensor<string, []> input_41_batch_first_cell_activation_0 = const()[name = tensor<string, []>("input_41_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
183
- tensor<string, []> input_41_batch_first_activation_0 = const()[name = tensor<string, []>("input_41_batch_first_activation_0"), val = tensor<string, []>("tanh")];
184
- tensor<fp16, [1024, 640]> concat_34_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(9962624))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10618048))), name = tensor<string, []>("concat_34_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
185
- tensor<fp16, [1024, 256]> concat_35_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10618624))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10880832))), name = tensor<string, []>("concat_35_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
186
- tensor<fp16, [1024]> add_10_to_fp16 = const()[name = tensor<string, []>("add_10_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10881408)))];
187
- tensor<fp16, [1024, 640]> concat_36_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(10883520))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11538944))), name = tensor<string, []>("concat_36_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
188
- tensor<fp16, [1024, 256]> concat_37_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11539520))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11801728))), name = tensor<string, []>("concat_37_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
189
- tensor<fp16, [1024]> add_11_to_fp16 = const()[name = tensor<string, []>("add_11_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11802304)))];
190
- tensor<fp16, [?, 1, 640]> transpose_5_cast_fp16 = transpose(perm = transpose_5_perm_0, x = x_53_cast_fp16)[name = tensor<string, []>("transpose_8")];
191
- tensor<fp16, [?, 1, 512]> input_41_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> input_41_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> input_41_batch_first_cast_fp16_2 = lstm(activation = input_41_batch_first_activation_0, bias = add_10_to_fp16, bias_back = add_11_to_fp16, cell_activation = input_41_batch_first_cell_activation_0, direction = input_41_batch_first_direction_0, initial_c = x_17_batch_first_lstm_h0_reshaped_to_fp16, initial_h = x_17_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = input_41_batch_first_output_sequence_0, recurrent_activation = input_41_batch_first_recurrent_activation_0, weight_hh = concat_35_to_fp16_palettized, weight_hh_back = concat_37_to_fp16_palettized, weight_ih = concat_34_to_fp16_palettized, weight_ih_back = concat_36_to_fp16_palettized, x = transpose_5_cast_fp16)[name = tensor<string, []>("input_41_batch_first_cast_fp16")];
192
- tensor<int32, [3]> transpose_9_perm_0 = const()[name = tensor<string, []>("transpose_9_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
193
- tensor<fp16, [1024, 128]> duration_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, []>(11804416))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11935552))), name = tensor<string, []>("duration_encoder_norms_2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
194
- tensor<fp16, [1024]> duration_encoder_norms_2_fc_bias_to_fp16 = const()[name = tensor<string, []>("duration_encoder_norms_2_fc_bias_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11936128)))];
195
- tensor<fp16, [1, 1024]> linear_3_cast_fp16 = linear(bias = duration_encoder_norms_2_fc_bias_to_fp16, weight = duration_encoder_norms_2_fc_weight_to_fp16_palettized, x = style_cast_fp16)[name = tensor<string, []>("linear_3_cast_fp16")];
196
- tensor<int32, [3]> var_294 = const()[name = tensor<string, []>("op_294"), val = tensor<int32, [3]>([1, 1024, 1])];
197
- tensor<fp16, [1, 1024, 1]> h_cast_fp16 = reshape(shape = var_294, x = linear_3_cast_fp16)[name = tensor<string, []>("h_cast_fp16")];
198
- tensor<int32, [2]> var_296_split_sizes_0 = const()[name = tensor<string, []>("op_296_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
199
- tensor<int32, []> var_296_axis_0 = const()[name = tensor<string, []>("op_296_axis_0"), val = tensor<int32, []>(1)];
200
- tensor<fp16, [1, 512, 1]> var_296_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_296_cast_fp16_1 = split(axis = var_296_axis_0, split_sizes = var_296_split_sizes_0, x = h_cast_fp16)[name = tensor<string, []>("op_296_cast_fp16")];
201
- tensor<int32, [3]> gamma_11_perm_0 = const()[name = tensor<string, []>("gamma_11_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
202
- tensor<int32, [3]> beta_11_perm_0 = const()[name = tensor<string, []>("beta_11_perm_0"), val = tensor<int32, [3]>([0, -1, 1])];
203
- tensor<int32, [1]> x_63_axes_0 = const()[name = tensor<string, []>("x_63_axes_0"), val = tensor<int32, [1]>([-1])];
204
- tensor<fp16, [1, ?, 512]> transpose_9_cast_fp16 = transpose(perm = transpose_9_perm_0, x = input_41_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_7")];
205
- tensor<fp16, [1, ?, 512]> x_63_cast_fp16 = layer_norm(axes = x_63_axes_0, epsilon = var_136_to_fp16, x = transpose_9_cast_fp16)[name = tensor<string, []>("x_63_cast_fp16")];
206
- tensor<fp16, []> var_302_promoted_to_fp16 = const()[name = tensor<string, []>("op_302_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
207
- tensor<fp16, [1, 1, 512]> gamma_11_cast_fp16 = transpose(perm = gamma_11_perm_0, x = var_296_cast_fp16_0)[name = tensor<string, []>("transpose_6")];
208
- tensor<fp16, [1, 1, 512]> var_303_cast_fp16 = add(x = gamma_11_cast_fp16, y = var_302_promoted_to_fp16)[name = tensor<string, []>("op_303_cast_fp16")];
209
- tensor<fp16, [1, ?, 512]> var_304_cast_fp16 = mul(x = var_303_cast_fp16, y = x_63_cast_fp16)[name = tensor<string, []>("op_304_cast_fp16")];
210
- tensor<fp16, [1, 1, 512]> beta_11_cast_fp16 = transpose(perm = beta_11_perm_0, x = var_296_cast_fp16_1)[name = tensor<string, []>("transpose_5")];
211
- tensor<fp16, [1, ?, 512]> x_65_cast_fp16 = add(x = var_304_cast_fp16, y = beta_11_cast_fp16)[name = tensor<string, []>("x_65_cast_fp16")];
212
- tensor<int32, [3]> x_67_perm_0 = const()[name = tensor<string, []>("x_67_perm_0"), val = tensor<int32, [3]>([0, -1, -2])];
213
- tensor<bool, []> x_interleave_0 = const()[name = tensor<string, []>("x_interleave_0"), val = tensor<bool, []>(false)];
214
- tensor<fp16, [1, 512, ?]> x_67_cast_fp16 = transpose(perm = x_67_perm_0, x = x_65_cast_fp16)[name = tensor<string, []>("transpose_4")];
215
- tensor<fp16, [1, 640, ?]> x_cast_fp16 = concat(axis = var_147, interleave = x_interleave_0, values = (x_67_cast_fp16, s_chan_cast_fp16))[name = tensor<string, []>("x_cast_fp16")];
216
- tensor<int32, [3]> input_45_perm_0 = const()[name = tensor<string, []>("input_45_perm_0"), val = tensor<int32, [3]>([0, -1, -2])];
217
- tensor<int32, [3]> input_45_batch_first_transpose_perm_0 = const()[name = tensor<string, []>("input_45_batch_first_transpose_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
218
- tensor<string, []> input_47_batch_first_direction_0 = const()[name = tensor<string, []>("input_47_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
219
- tensor<bool, []> input_47_batch_first_output_sequence_0 = const()[name = tensor<string, []>("input_47_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
220
- tensor<string, []> input_47_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("input_47_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
221
- tensor<string, []> input_47_batch_first_cell_activation_0 = const()[name = tensor<string, []>("input_47_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
222
- tensor<string, []> input_47_batch_first_activation_0 = const()[name = tensor<string, []>("input_47_batch_first_activation_0"), val = tensor<string, []>("tanh")];
223
- tensor<fp16, [1024, 640]> concat_44_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(11938240))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12593664))), name = tensor<string, []>("concat_44_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
224
- tensor<fp16, [1024, 256]> concat_45_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12594240))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12856448))), name = tensor<string, []>("concat_45_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
225
- tensor<fp16, [1024]> add_13_to_fp16 = const()[name = tensor<string, []>("add_13_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12857024)))];
226
- tensor<fp16, [1024, 640]> concat_46_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [655360]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(12859136))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13514560))), name = tensor<string, []>("concat_46_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 640])];
227
- tensor<fp16, [1024, 256]> concat_47_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [262144]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13515136))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13777344))), name = tensor<string, []>("concat_47_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 256])];
228
- tensor<fp16, [1024]> add_14_to_fp16 = const()[name = tensor<string, []>("add_14_to_fp16"), val = tensor<fp16, [1024]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13777920)))];
229
- tensor<fp16, [1, ?, 640]> d = transpose(perm = input_45_perm_0, x = x_cast_fp16)[name = tensor<string, []>("transpose_3")];
230
- tensor<fp16, [?, 1, 640]> input_45_batch_first_transpose_cast_fp16 = transpose(perm = input_45_batch_first_transpose_perm_0, x = d)[name = tensor<string, []>("transpose_2")];
231
- tensor<fp16, [?, 1, 512]> input_47_batch_first_cast_fp16_0, tensor<fp16, [1, 512]> input_47_batch_first_cast_fp16_1, tensor<fp16, [1, 512]> input_47_batch_first_cast_fp16_2 = lstm(activation = input_47_batch_first_activation_0, bias = add_13_to_fp16, bias_back = add_14_to_fp16, cell_activation = input_47_batch_first_cell_activation_0, direction = input_47_batch_first_direction_0, initial_c = x_17_batch_first_lstm_h0_reshaped_to_fp16, initial_h = x_17_batch_first_lstm_h0_reshaped_to_fp16, output_sequence = input_47_batch_first_output_sequence_0, recurrent_activation = input_47_batch_first_recurrent_activation_0, weight_hh = concat_45_to_fp16_palettized, weight_hh_back = concat_47_to_fp16_palettized, weight_ih = concat_44_to_fp16_palettized, weight_ih_back = concat_46_to_fp16_palettized, x = input_45_batch_first_transpose_cast_fp16)[name = tensor<string, []>("input_47_batch_first_cast_fp16")];
232
- tensor<int32, [3]> input_47_perm_0 = const()[name = tensor<string, []>("input_47_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
233
- 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, []>(13780032))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13805696))), name = tensor<string, []>("duration_proj_linear_layer_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([50, 512])];
234
- 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, []>(13806272)))];
235
- tensor<fp16, [1, ?, 512]> input_47_cast_fp16 = transpose(perm = input_47_perm_0, x = input_47_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_1")];
236
- tensor<fp16, [1, ?, 50]> pred_dur_log = linear(bias = duration_proj_linear_layer_bias_to_fp16, weight = duration_proj_linear_layer_weight_to_fp16_palettized, x = input_47_cast_fp16)[name = tensor<string, []>("linear_4_cast_fp16")];
237
- tensor<int32, []> var_345 = const()[name = tensor<string, []>("op_345"), val = tensor<int32, []>(-1)];
238
- tensor<int32, [3]> var_355_begin_0 = const()[name = tensor<string, []>("op_355_begin_0"), val = tensor<int32, [3]>([0, 0, 0])];
239
- tensor<int32, [3]> var_355_end_0 = const()[name = tensor<string, []>("op_355_end_0"), val = tensor<int32, [3]>([1, 0, 1])];
240
- tensor<bool, [3]> var_355_end_mask_0 = const()[name = tensor<string, []>("op_355_end_mask_0"), val = tensor<bool, [3]>([true, true, false])];
241
- tensor<bool, [3]> var_355_squeeze_mask_0 = const()[name = tensor<string, []>("op_355_squeeze_mask_0"), val = tensor<bool, [3]>([false, false, true])];
242
- tensor<fp16, [1, ?]> var_355_cast_fp16 = slice_by_index(begin = var_355_begin_0, end = var_355_end_0, end_mask = var_355_end_mask_0, squeeze_mask = var_355_squeeze_mask_0, x = bert_dur)[name = tensor<string, []>("op_355_cast_fp16")];
243
- tensor<fp16, []> fill_like_0_value_0_to_fp16 = const()[name = tensor<string, []>("fill_like_0_value_0_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
244
- tensor<fp16, [1, ?]> fill_like_0_cast_fp16 = fill_like(ref_tensor = var_355_cast_fp16, value = fill_like_0_value_0_to_fp16)[name = tensor<string, []>("fill_like_0_cast_fp16")];
245
- tensor<bool, []> var_357_exclusive_0 = const()[name = tensor<string, []>("op_357_exclusive_0"), val = tensor<bool, []>(false)];
246
- tensor<bool, []> var_357_reverse_0 = const()[name = tensor<string, []>("op_357_reverse_0"), val = tensor<bool, []>(false)];
247
- tensor<fp16, [1, ?]> var_357_cast_fp16 = cumsum(axis = var_345, exclusive = var_357_exclusive_0, reverse = var_357_reverse_0, x = fill_like_0_cast_fp16)[name = tensor<string, []>("op_357_cast_fp16")];
248
- tensor<string, []> cast_6_dtype_0 = const()[name = tensor<string, []>("cast_6_dtype_0"), val = tensor<string, []>("int32")];
249
- tensor<int32, []> var_359 = const()[name = tensor<string, []>("op_359"), val = tensor<int32, []>(1)];
250
- tensor<int32, [1, ?]> var_357_cast_fp16_to_int32 = cast(dtype = cast_6_dtype_0, x = var_357_cast_fp16)[name = tensor<string, []>("cast_1")];
251
- tensor<int32, [1, ?]> indices = sub(x = var_357_cast_fp16_to_int32, y = var_359)[name = tensor<string, []>("indices")];
252
- tensor<int32, [1]> input_axes_0 = const()[name = tensor<string, []>("input_axes_0"), val = tensor<int32, [1]>([0])];
253
- tensor<string, []> indices_to_uint16_dtype_0 = const()[name = tensor<string, []>("indices_to_uint16_dtype_0"), val = tensor<string, []>("uint16")];
254
- tensor<uint16, [1, ?]> indices_to_uint16 = cast(dtype = indices_to_uint16_dtype_0, x = indices)[name = tensor<string, []>("cast_0")];
255
- tensor<uint16, [?]> input_cast_uint16 = squeeze(axes = input_axes_0, x = indices_to_uint16)[name = tensor<string, []>("input_cast_uint16")];
256
- tensor<int32, []> fe_axis_0 = const()[name = tensor<string, []>("fe_axis_0"), val = tensor<int32, []>(0)];
257
- tensor<int32, []> fe_batch_dims_0 = const()[name = tensor<string, []>("fe_batch_dims_0"), val = tensor<int32, []>(0)];
258
- tensor<bool, []> fe_validate_indices_0 = const()[name = tensor<string, []>("fe_validate_indices_0"), val = tensor<bool, []>(false)];
259
- tensor<fp16, [512, 768]> fixed_embedding_embedding_weight_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [393216]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(13806464))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(14199744))), name = tensor<string, []>("fixed_embedding_embedding_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 768])];
260
- tensor<fp16, [?, 768]> fe_cast_fp16_cast_uint16 = gather(axis = fe_axis_0, batch_dims = fe_batch_dims_0, indices = input_cast_uint16, validate_indices = fe_validate_indices_0, x = fixed_embedding_embedding_weight_to_fp16_palettized)[name = tensor<string, []>("fe_cast_fp16_cast_uint16")];
261
- tensor<int32, [1]> var_364_axes_0 = const()[name = tensor<string, []>("op_364_axes_0"), val = tensor<int32, [1]>([0])];
262
- tensor<fp16, [1, ?, 768]> fixed_embedding = expand_dims(axes = var_364_axes_0, x = fe_cast_fp16_cast_uint16)[name = tensor<string, []>("op_364_cast_fp16")];
263
- tensor<fp16, [1, 512, ?]> t_en = transpose(perm = transpose_0_perm_0, x = x_17_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_0")];
264
- } -> (t_en, d, pred_dur_log, fixed_embedding);
265
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_postbert.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:cf0fad98760e4fda11d7c11114e6448ee2443240265720a55080d40ee97b346f
3
- size 14200320
 
 
 
 
ANE/styletts2_ane_postbert.mlpackage/Data/com.apple.CoreML/model.mlmodel DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:b454e92bf029364391c1456bf4f69a5d0981f905423b86fa7eb955bf99217d0b
3
- size 47924
 
 
 
 
ANE/styletts2_ane_postbert.mlpackage/Data/com.apple.CoreML/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:cf0fad98760e4fda11d7c11114e6448ee2443240265720a55080d40ee97b346f
3
- size 14200320
 
 
 
 
ANE/styletts2_ane_postbert.mlpackage/Manifest.json DELETED
@@ -1,18 +0,0 @@
1
- {
2
- "fileFormatVersion": "1.0.0",
3
- "itemInfoEntries": {
4
- "579EF757-64B2-4FC5-B6F3-E89145D74BE1": {
5
- "author": "com.apple.CoreML",
6
- "description": "CoreML Model Weights",
7
- "name": "weights",
8
- "path": "com.apple.CoreML/weights"
9
- },
10
- "86717364-E608-461C-A6FF-7DFA98056399": {
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": "86717364-E608-461C-A6FF-7DFA98056399"
18
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_prosody.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:a569de5bd6b2a4cb48a7ec8c1544f80e7c6484933bb9160dc7c88b462c23c9cc
3
- size 243
 
 
 
 
ANE/styletts2_ane_prosody.mlmodelc/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:21ec9c806bdf3191a486aec9bb495ac9dcedfe1983565c6453b96ee4455340a2
3
- size 417
 
 
 
 
ANE/styletts2_ane_prosody.mlmodelc/metadata.json DELETED
@@ -1,97 +0,0 @@
1
- [
2
- {
3
- "metadataOutputVersion" : "3.0",
4
- "shortDescription" : "StyleTTS2-ANE Stage 5: Prosody F0Ntrain (static T_a=2000) — fp16, ANE.",
5
- "outputSchema" : [
6
- {
7
- "hasShapeFlexibility" : "0",
8
- "isOptional" : "0",
9
- "dataType" : "Float16",
10
- "formattedType" : "MultiArray (Float16 1 × 4000)",
11
- "shortDescription" : "",
12
- "shape" : "[1, 4000]",
13
- "name" : "F0",
14
- "type" : "MultiArray"
15
- },
16
- {
17
- "hasShapeFlexibility" : "0",
18
- "isOptional" : "0",
19
- "dataType" : "Float16",
20
- "formattedType" : "MultiArray (Float16 1 × 4000)",
21
- "shortDescription" : "",
22
- "shape" : "[1, 4000]",
23
- "name" : "N",
24
- "type" : "MultiArray"
25
- }
26
- ],
27
- "storagePrecision" : "Mixed (Float16, Palettized (8 bits))",
28
- "modelParameters" : [
29
-
30
- ],
31
- "specificationVersion" : 8,
32
- "mlProgramOperationTypeHistogram" : {
33
- "Ios17.squeeze" : 4,
34
- "Ios17.mul" : 18,
35
- "Ios17.linear" : 12,
36
- "Ios17.transpose" : 2,
37
- "Ios16.constexprLutToDense" : 30,
38
- "Ios17.conv" : 16,
39
- "Ios17.leakyRelu" : 12,
40
- "Ios17.add" : 30,
41
- "Ios17.lstm" : 1,
42
- "Ios17.convTranspose" : 2,
43
- "Ios17.sliceByIndex" : 2,
44
- "UpsampleNearestNeighbor" : 2,
45
- "Ios17.expandDims" : 2,
46
- "Ios17.instanceNorm" : 11,
47
- "Ios17.reshape" : 12,
48
- "Ios17.cast" : 2,
49
- "Split" : 12
50
- },
51
- "computePrecision" : "Mixed (Float16, Float32, Int32)",
52
- "isUpdatable" : "0",
53
- "stateSchema" : [
54
-
55
- ],
56
- "availability" : {
57
- "macOS" : "14.0",
58
- "tvOS" : "17.0",
59
- "visionOS" : "1.0",
60
- "watchOS" : "10.0",
61
- "iOS" : "17.0",
62
- "macCatalyst" : "17.0"
63
- },
64
- "modelType" : {
65
- "name" : "MLModelType_mlProgram"
66
- },
67
- "inputSchema" : [
68
- {
69
- "hasShapeFlexibility" : "0",
70
- "isOptional" : "0",
71
- "dataType" : "Float32",
72
- "formattedType" : "MultiArray (Float32 1 × 640 × 2000)",
73
- "shortDescription" : "",
74
- "shape" : "[1, 640, 2000]",
75
- "name" : "en",
76
- "type" : "MultiArray"
77
- },
78
- {
79
- "hasShapeFlexibility" : "0",
80
- "isOptional" : "0",
81
- "dataType" : "Float32",
82
- "formattedType" : "MultiArray (Float32 1 × 128)",
83
- "shortDescription" : "",
84
- "shape" : "[1, 128]",
85
- "name" : "s",
86
- "type" : "MultiArray"
87
- }
88
- ],
89
- "userDefinedMetadata" : {
90
- "com.github.apple.coremltools.version" : "8.3.0",
91
- "com.github.apple.coremltools.source" : "torch==2.11.0",
92
- "com.github.apple.coremltools.source_dialect" : "TorchScript"
93
- },
94
- "generatedClassName" : "styletts2_ane_prosody",
95
- "method" : "predict"
96
- }
97
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_prosody.mlmodelc/model.mil DELETED
@@ -1,396 +0,0 @@
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, 640, 2000]> en, tensor<fp32, [1, 128]> s) {
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, []> en_to_fp16_dtype_0 = const()[name = tensor<string, []>("en_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
7
- tensor<string, []> x_batch_first_direction_0 = const()[name = tensor<string, []>("x_batch_first_direction_0"), val = tensor<string, []>("bidirectional")];
8
- tensor<bool, []> x_batch_first_output_sequence_0 = const()[name = tensor<string, []>("x_batch_first_output_sequence_0"), val = tensor<bool, []>(true)];
9
- tensor<string, []> x_batch_first_recurrent_activation_0 = const()[name = tensor<string, []>("x_batch_first_recurrent_activation_0"), val = tensor<string, []>("sigmoid")];
10
- tensor<string, []> x_batch_first_cell_activation_0 = const()[name = tensor<string, []>("x_batch_first_cell_activation_0"), val = tensor<string, []>("tanh")];
11
- tensor<string, []> x_batch_first_activation_0 = const()[name = tensor<string, []>("x_batch_first_activation_0"), val = tensor<string, []>("tanh")];
12
- 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)))];
13
- 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])];
14
- 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])];
15
- 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)))];
16
- 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])];
17
- 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])];
18
- 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)))];
19
- tensor<fp16, [1, 640, 2000]> en_to_fp16 = cast(dtype = en_to_fp16_dtype_0, x = en)[name = tensor<string, []>("cast_1")];
20
- tensor<fp16, [2000, 1, 640]> transpose_0_cast_fp16 = transpose(perm = transpose_0_perm_0, x = en_to_fp16)[name = tensor<string, []>("transpose_2")];
21
- tensor<fp16, [2000, 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")];
22
- tensor<fp32, []> var_53 = const()[name = tensor<string, []>("op_53"), val = tensor<fp32, []>(0x1.99999ap-3)];
23
- tensor<string, []> s_to_fp16_dtype_0 = const()[name = tensor<string, []>("s_to_fp16_dtype_0"), val = tensor<string, []>("fp16")];
24
- 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])];
25
- 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)))];
26
- tensor<fp16, [1, 128]> s_to_fp16 = cast(dtype = s_to_fp16_dtype_0, x = s)[name = tensor<string, []>("cast_0")];
27
- 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 = s_to_fp16)[name = tensor<string, []>("linear_0_cast_fp16")];
28
- tensor<int32, [3]> var_78 = const()[name = tensor<string, []>("op_78"), val = tensor<int32, [3]>([1, 1024, 1])];
29
- tensor<fp16, [1, 1024, 1]> h_3_cast_fp16 = reshape(shape = var_78, x = linear_0_cast_fp16)[name = tensor<string, []>("h_3_cast_fp16")];
30
- tensor<int32, [2]> var_80_split_sizes_0 = const()[name = tensor<string, []>("op_80_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
31
- tensor<int32, []> var_80_axis_0 = const()[name = tensor<string, []>("op_80_axis_0"), val = tensor<int32, []>(1)];
32
- tensor<fp16, [1, 512, 1]> var_80_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_80_cast_fp16_1 = split(axis = var_80_axis_0, split_sizes = var_80_split_sizes_0, x = h_3_cast_fp16)[name = tensor<string, []>("op_80_cast_fp16")];
33
- tensor<fp16, []> var_82_promoted_to_fp16 = const()[name = tensor<string, []>("op_82_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
34
- tensor<fp16, [1, 512, 1]> var_83_cast_fp16 = add(x = var_80_cast_fp16_0, y = var_82_promoted_to_fp16)[name = tensor<string, []>("op_83_cast_fp16")];
35
- tensor<fp16, []> var_56_to_fp16 = const()[name = tensor<string, []>("op_56_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
36
- tensor<int32, [3]> transpose_0_perm_0_1 = const()[name = tensor<string, []>("transpose_0_perm_0_1"), val = tensor<int32, [3]>([1, 2, 0])];
37
- tensor<fp16, [1, 512, 2000]> transpose_0 = transpose(perm = transpose_0_perm_0_1, x = x_batch_first_cast_fp16_0)[name = tensor<string, []>("transpose_1")];
38
- tensor<fp16, [1, 512, 2000]> var_84_cast_fp16 = instance_norm(epsilon = var_56_to_fp16, x = transpose_0)[name = tensor<string, []>("op_84_cast_fp16")];
39
- tensor<fp16, [1, 512, 2000]> var_85_cast_fp16 = mul(x = var_83_cast_fp16, y = var_84_cast_fp16)[name = tensor<string, []>("op_85_cast_fp16")];
40
- tensor<fp16, [1, 512, 2000]> input_5_cast_fp16 = add(x = var_85_cast_fp16, y = var_80_cast_fp16_1)[name = tensor<string, []>("input_5_cast_fp16")];
41
- tensor<fp16, [1, 512, 2000]> input_7_cast_fp16 = leaky_relu(alpha = var_53, x = input_5_cast_fp16)[name = tensor<string, []>("input_7_cast_fp16")];
42
- tensor<string, []> input_11_pad_type_0 = const()[name = tensor<string, []>("input_11_pad_type_0"), val = tensor<string, []>("custom")];
43
- tensor<int32, [2]> input_11_pad_0 = const()[name = tensor<string, []>("input_11_pad_0"), val = tensor<int32, [2]>([1, 1])];
44
- tensor<int32, [1]> input_11_strides_0 = const()[name = tensor<string, []>("input_11_strides_0"), val = tensor<int32, [1]>([1])];
45
- tensor<int32, [1]> input_11_dilations_0 = const()[name = tensor<string, []>("input_11_dilations_0"), val = tensor<int32, [1]>([1])];
46
- tensor<int32, []> input_11_groups_0 = const()[name = tensor<string, []>("input_11_groups_0"), val = tensor<int32, []>(1)];
47
- tensor<fp16, [512, 512, 3]> weight_1_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [786432]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1976768))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2763264))), name = tensor<string, []>("weight_1_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 3])];
48
- 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, []>(2763840)))];
49
- tensor<fp16, [1, 512, 2000]> input_11_cast_fp16 = conv(bias = F0_0_conv1_bias_to_fp16, dilations = input_11_dilations_0, groups = input_11_groups_0, pad = input_11_pad_0, pad_type = input_11_pad_type_0, strides = input_11_strides_0, weight = weight_1_to_fp16_palettized, x = input_7_cast_fp16)[name = tensor<string, []>("input_11_cast_fp16")];
50
- 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, []>(2764928))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2896064))), name = tensor<string, []>("F0_0_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
51
- 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, []>(2896640)))];
52
- 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 = s_to_fp16)[name = tensor<string, []>("linear_1_cast_fp16")];
53
- tensor<int32, [3]> var_102 = const()[name = tensor<string, []>("op_102"), val = tensor<int32, [3]>([1, 1024, 1])];
54
- tensor<fp16, [1, 1024, 1]> h_7_cast_fp16 = reshape(shape = var_102, x = linear_1_cast_fp16)[name = tensor<string, []>("h_7_cast_fp16")];
55
- tensor<int32, [2]> var_104_split_sizes_0 = const()[name = tensor<string, []>("op_104_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
56
- tensor<int32, []> var_104_axis_0 = const()[name = tensor<string, []>("op_104_axis_0"), val = tensor<int32, []>(1)];
57
- tensor<fp16, [1, 512, 1]> var_104_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_104_cast_fp16_1 = split(axis = var_104_axis_0, split_sizes = var_104_split_sizes_0, x = h_7_cast_fp16)[name = tensor<string, []>("op_104_cast_fp16")];
58
- tensor<fp16, []> var_106_promoted_to_fp16 = const()[name = tensor<string, []>("op_106_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
59
- tensor<fp16, [1, 512, 1]> var_107_cast_fp16 = add(x = var_104_cast_fp16_0, y = var_106_promoted_to_fp16)[name = tensor<string, []>("op_107_cast_fp16")];
60
- tensor<fp16, [1, 512, 2000]> var_108_cast_fp16 = instance_norm(epsilon = var_56_to_fp16, x = input_11_cast_fp16)[name = tensor<string, []>("op_108_cast_fp16")];
61
- tensor<fp16, [1, 512, 2000]> var_109_cast_fp16 = mul(x = var_107_cast_fp16, y = var_108_cast_fp16)[name = tensor<string, []>("op_109_cast_fp16")];
62
- tensor<fp16, [1, 512, 2000]> input_13_cast_fp16 = add(x = var_109_cast_fp16, y = var_104_cast_fp16_1)[name = tensor<string, []>("input_13_cast_fp16")];
63
- tensor<fp16, [1, 512, 2000]> input_15_cast_fp16 = leaky_relu(alpha = var_53, x = input_13_cast_fp16)[name = tensor<string, []>("input_15_cast_fp16")];
64
- tensor<string, []> out_1_pad_type_0 = const()[name = tensor<string, []>("out_1_pad_type_0"), val = tensor<string, []>("custom")];
65
- tensor<int32, [2]> out_1_pad_0 = const()[name = tensor<string, []>("out_1_pad_0"), val = tensor<int32, [2]>([1, 1])];
66
- tensor<int32, [1]> out_1_strides_0 = const()[name = tensor<string, []>("out_1_strides_0"), val = tensor<int32, [1]>([1])];
67
- tensor<int32, [1]> out_1_dilations_0 = const()[name = tensor<string, []>("out_1_dilations_0"), val = tensor<int32, [1]>([1])];
68
- tensor<int32, []> out_1_groups_0 = const()[name = tensor<string, []>("out_1_groups_0"), val = tensor<int32, []>(1)];
69
- 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, []>(2898752))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3685248))), name = tensor<string, []>("weight_3_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 3])];
70
- 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, []>(3685824)))];
71
- tensor<fp16, [1, 512, 2000]> 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_3_to_fp16_palettized, x = input_15_cast_fp16)[name = tensor<string, []>("out_1_cast_fp16")];
72
- tensor<fp16, [1, 512, 2000]> var_120_cast_fp16 = add(x = out_1_cast_fp16, y = transpose_0)[name = tensor<string, []>("op_120_cast_fp16")];
73
- tensor<fp16, []> _inversed_input_19_y_0_to_fp16 = const()[name = tensor<string, []>("_inversed_input_19_y_0_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
74
- tensor<fp16, [1, 512, 2000]> _inversed_input_19_cast_fp16 = mul(x = var_120_cast_fp16, y = _inversed_input_19_y_0_to_fp16)[name = tensor<string, []>("_inversed_input_19_cast_fp16")];
75
- tensor<fp32, []> var_126 = const()[name = tensor<string, []>("op_126"), val = tensor<fp32, []>(0x1.99999ap-3)];
76
- 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, []>(3686912))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3818048))), name = tensor<string, []>("F0_1_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
77
- 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, []>(3818624)))];
78
- 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 = s_to_fp16)[name = tensor<string, []>("linear_2_cast_fp16")];
79
- tensor<int32, [3]> var_161 = const()[name = tensor<string, []>("op_161"), val = tensor<int32, [3]>([1, 1024, 1])];
80
- tensor<fp16, [1, 1024, 1]> h_11_cast_fp16 = reshape(shape = var_161, x = linear_2_cast_fp16)[name = tensor<string, []>("h_11_cast_fp16")];
81
- tensor<int32, [2]> var_163_split_sizes_0 = const()[name = tensor<string, []>("op_163_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
82
- tensor<int32, []> var_163_axis_0 = const()[name = tensor<string, []>("op_163_axis_0"), val = tensor<int32, []>(1)];
83
- tensor<fp16, [1, 512, 1]> var_163_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_163_cast_fp16_1 = split(axis = var_163_axis_0, split_sizes = var_163_split_sizes_0, x = h_11_cast_fp16)[name = tensor<string, []>("op_163_cast_fp16")];
84
- tensor<fp16, []> var_165_promoted_to_fp16 = const()[name = tensor<string, []>("op_165_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
85
- tensor<fp16, [1, 512, 1]> var_166_cast_fp16 = add(x = var_163_cast_fp16_0, y = var_165_promoted_to_fp16)[name = tensor<string, []>("op_166_cast_fp16")];
86
- tensor<fp16, []> var_130_to_fp16 = const()[name = tensor<string, []>("op_130_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
87
- tensor<fp16, [1, 512, 2000]> var_167_cast_fp16 = instance_norm(epsilon = var_130_to_fp16, x = _inversed_input_19_cast_fp16)[name = tensor<string, []>("op_167_cast_fp16")];
88
- tensor<fp16, [1, 512, 2000]> var_168_cast_fp16 = mul(x = var_166_cast_fp16, y = var_167_cast_fp16)[name = tensor<string, []>("op_168_cast_fp16")];
89
- tensor<fp16, [1, 512, 2000]> input_21_cast_fp16 = add(x = var_168_cast_fp16, y = var_163_cast_fp16_1)[name = tensor<string, []>("input_21_cast_fp16")];
90
- tensor<fp16, [1, 512, 2000]> input_23_cast_fp16 = leaky_relu(alpha = var_126, x = input_21_cast_fp16)[name = tensor<string, []>("input_23_cast_fp16")];
91
- tensor<string, []> conv_transpose_0_pad_type_0 = const()[name = tensor<string, []>("conv_transpose_0_pad_type_0"), val = tensor<string, []>("custom")];
92
- tensor<int32, [2]> conv_transpose_0_pad_0 = const()[name = tensor<string, []>("conv_transpose_0_pad_0"), val = tensor<int32, [2]>([0, 0])];
93
- tensor<int32, [1]> conv_transpose_0_strides_0 = const()[name = tensor<string, []>("conv_transpose_0_strides_0"), val = tensor<int32, [1]>([2])];
94
- tensor<int32, []> conv_transpose_0_groups_0 = const()[name = tensor<string, []>("conv_transpose_0_groups_0"), val = tensor<int32, []>(512)];
95
- tensor<int32, [1]> conv_transpose_0_dilations_0 = const()[name = tensor<string, []>("conv_transpose_0_dilations_0"), val = tensor<int32, [1]>([1])];
96
- tensor<int32, [3]> conv_transpose_0_has_output_shape_output_shape_0 = const()[name = tensor<string, []>("conv_transpose_0_has_output_shape_output_shape_0"), val = tensor<int32, [3]>([1, 512, 4001])];
97
- tensor<fp16, [512, 1, 3]> var_171_to_fp16 = const()[name = tensor<string, []>("op_171_to_fp16"), val = tensor<fp16, [512, 1, 3]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3820736)))];
98
- 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, []>(3823872)))];
99
- tensor<fp16, [1, 512, 4001]> conv_transpose_0_has_output_shape_cast_fp16 = conv_transpose(bias = F0_1_pool_bias_to_fp16, dilations = conv_transpose_0_dilations_0, groups = conv_transpose_0_groups_0, output_shape = conv_transpose_0_has_output_shape_output_shape_0, pad = conv_transpose_0_pad_0, pad_type = conv_transpose_0_pad_type_0, strides = conv_transpose_0_strides_0, weight = var_171_to_fp16, x = input_23_cast_fp16)[name = tensor<string, []>("conv_transpose_0_has_output_shape_cast_fp16")];
100
- tensor<int32, [3]> input_25_begin_0 = const()[name = tensor<string, []>("input_25_begin_0"), val = tensor<int32, [3]>([0, 0, 1])];
101
- tensor<int32, [3]> input_25_end_0 = const()[name = tensor<string, []>("input_25_end_0"), val = tensor<int32, [3]>([0, 0, 0])];
102
- tensor<bool, [3]> input_25_begin_mask_0 = const()[name = tensor<string, []>("input_25_begin_mask_0"), val = tensor<bool, [3]>([true, true, false])];
103
- tensor<bool, [3]> input_25_end_mask_0 = const()[name = tensor<string, []>("input_25_end_mask_0"), val = tensor<bool, [3]>([true, true, true])];
104
- tensor<fp16, [1, 512, 4000]> input_25_cast_fp16 = slice_by_index(begin = input_25_begin_0, begin_mask = input_25_begin_mask_0, end = input_25_end_0, end_mask = input_25_end_mask_0, x = conv_transpose_0_has_output_shape_cast_fp16)[name = tensor<string, []>("input_25_cast_fp16")];
105
- tensor<string, []> input_29_pad_type_0 = const()[name = tensor<string, []>("input_29_pad_type_0"), val = tensor<string, []>("custom")];
106
- tensor<int32, [2]> input_29_pad_0 = const()[name = tensor<string, []>("input_29_pad_0"), val = tensor<int32, [2]>([1, 1])];
107
- tensor<int32, [1]> input_29_strides_0 = const()[name = tensor<string, []>("input_29_strides_0"), val = tensor<int32, [1]>([1])];
108
- tensor<int32, [1]> input_29_dilations_0 = const()[name = tensor<string, []>("input_29_dilations_0"), val = tensor<int32, [1]>([1])];
109
- tensor<int32, []> input_29_groups_0 = const()[name = tensor<string, []>("input_29_groups_0"), val = tensor<int32, []>(1)];
110
- tensor<fp16, [256, 512, 3]> weight_5_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [393216]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3824960))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4218240))), name = tensor<string, []>("weight_5_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 512, 3])];
111
- 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, []>(4218816)))];
112
- tensor<fp16, [1, 256, 4000]> input_29_cast_fp16 = conv(bias = F0_1_conv1_bias_to_fp16, dilations = input_29_dilations_0, groups = input_29_groups_0, pad = input_29_pad_0, pad_type = input_29_pad_type_0, strides = input_29_strides_0, weight = weight_5_to_fp16_palettized, x = input_25_cast_fp16)[name = tensor<string, []>("input_29_cast_fp16")];
113
- 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, []>(4219392))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4284992))), name = tensor<string, []>("F0_1_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
114
- 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, []>(4285568)))];
115
- 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 = s_to_fp16)[name = tensor<string, []>("linear_3_cast_fp16")];
116
- tensor<int32, [3]> var_192 = const()[name = tensor<string, []>("op_192"), val = tensor<int32, [3]>([1, 512, 1])];
117
- tensor<fp16, [1, 512, 1]> h_15_cast_fp16 = reshape(shape = var_192, x = linear_3_cast_fp16)[name = tensor<string, []>("h_15_cast_fp16")];
118
- tensor<int32, [2]> var_194_split_sizes_0 = const()[name = tensor<string, []>("op_194_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
119
- tensor<int32, []> var_194_axis_0 = const()[name = tensor<string, []>("op_194_axis_0"), val = tensor<int32, []>(1)];
120
- tensor<fp16, [1, 256, 1]> var_194_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_194_cast_fp16_1 = split(axis = var_194_axis_0, split_sizes = var_194_split_sizes_0, x = h_15_cast_fp16)[name = tensor<string, []>("op_194_cast_fp16")];
121
- tensor<fp16, []> var_196_promoted_to_fp16 = const()[name = tensor<string, []>("op_196_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
122
- tensor<fp16, [1, 256, 1]> var_197_cast_fp16 = add(x = var_194_cast_fp16_0, y = var_196_promoted_to_fp16)[name = tensor<string, []>("op_197_cast_fp16")];
123
- tensor<fp16, [1, 256, 4000]> var_198_cast_fp16 = instance_norm(epsilon = var_130_to_fp16, x = input_29_cast_fp16)[name = tensor<string, []>("op_198_cast_fp16")];
124
- tensor<fp16, [1, 256, 4000]> var_199_cast_fp16 = mul(x = var_197_cast_fp16, y = var_198_cast_fp16)[name = tensor<string, []>("op_199_cast_fp16")];
125
- tensor<fp16, [1, 256, 4000]> input_31_cast_fp16 = add(x = var_199_cast_fp16, y = var_194_cast_fp16_1)[name = tensor<string, []>("input_31_cast_fp16")];
126
- tensor<fp16, [1, 256, 4000]> input_33_cast_fp16 = leaky_relu(alpha = var_126, x = input_31_cast_fp16)[name = tensor<string, []>("input_33_cast_fp16")];
127
- tensor<string, []> out_3_pad_type_0 = const()[name = tensor<string, []>("out_3_pad_type_0"), val = tensor<string, []>("custom")];
128
- tensor<int32, [2]> out_3_pad_0 = const()[name = tensor<string, []>("out_3_pad_0"), val = tensor<int32, [2]>([1, 1])];
129
- tensor<int32, [1]> out_3_strides_0 = const()[name = tensor<string, []>("out_3_strides_0"), val = tensor<int32, [1]>([1])];
130
- tensor<int32, [1]> out_3_dilations_0 = const()[name = tensor<string, []>("out_3_dilations_0"), val = tensor<int32, [1]>([1])];
131
- tensor<int32, []> out_3_groups_0 = const()[name = tensor<string, []>("out_3_groups_0"), val = tensor<int32, []>(1)];
132
- tensor<fp16, [256, 256, 3]> weight_7_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4286656))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4483328))), name = tensor<string, []>("weight_7_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
133
- 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, []>(4483904)))];
134
- tensor<fp16, [1, 256, 4000]> 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_7_to_fp16_palettized, x = input_33_cast_fp16)[name = tensor<string, []>("out_3_cast_fp16")];
135
- tensor<int32, [1]> expand_dims_0_axes_0 = const()[name = tensor<string, []>("expand_dims_0_axes_0"), val = tensor<int32, [1]>([3])];
136
- tensor<fp16, [1, 512, 2000, 1]> expand_dims_0_cast_fp16 = expand_dims(axes = expand_dims_0_axes_0, x = _inversed_input_19_cast_fp16)[name = tensor<string, []>("expand_dims_0_cast_fp16")];
137
- 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)];
138
- 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)];
139
- tensor<fp16, [1, 512, 4000, 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")];
140
- tensor<int32, [1]> input_37_axes_0 = const()[name = tensor<string, []>("input_37_axes_0"), val = tensor<int32, [1]>([3])];
141
- tensor<fp16, [1, 512, 4000]> input_37_cast_fp16 = squeeze(axes = input_37_axes_0, x = upsample_nearest_neighbor_0_cast_fp16)[name = tensor<string, []>("input_37_cast_fp16")];
142
- tensor<string, []> var_217_pad_type_0 = const()[name = tensor<string, []>("op_217_pad_type_0"), val = tensor<string, []>("valid")];
143
- tensor<int32, [1]> var_217_strides_0 = const()[name = tensor<string, []>("op_217_strides_0"), val = tensor<int32, [1]>([1])];
144
- tensor<int32, [2]> var_217_pad_0 = const()[name = tensor<string, []>("op_217_pad_0"), val = tensor<int32, [2]>([0, 0])];
145
- tensor<int32, [1]> var_217_dilations_0 = const()[name = tensor<string, []>("op_217_dilations_0"), val = tensor<int32, [1]>([1])];
146
- tensor<int32, []> var_217_groups_0 = const()[name = tensor<string, []>("op_217_groups_0"), val = tensor<int32, []>(1)];
147
- tensor<fp16, [256, 512, 1]> weight_9_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4484480))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4615616))), name = tensor<string, []>("weight_9_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 512, 1])];
148
- tensor<fp16, [1, 256, 4000]> var_217_cast_fp16 = conv(dilations = var_217_dilations_0, groups = var_217_groups_0, pad = var_217_pad_0, pad_type = var_217_pad_type_0, strides = var_217_strides_0, weight = weight_9_to_fp16_palettized, x = input_37_cast_fp16)[name = tensor<string, []>("op_217_cast_fp16")];
149
- tensor<fp16, [1, 256, 4000]> var_218_cast_fp16 = add(x = out_3_cast_fp16, y = var_217_cast_fp16)[name = tensor<string, []>("op_218_cast_fp16")];
150
- tensor<fp16, []> _inversed_input_39_y_0_to_fp16 = const()[name = tensor<string, []>("_inversed_input_39_y_0_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
151
- tensor<fp16, [1, 256, 4000]> _inversed_input_39_cast_fp16 = mul(x = var_218_cast_fp16, y = _inversed_input_39_y_0_to_fp16)[name = tensor<string, []>("_inversed_input_39_cast_fp16")];
152
- tensor<fp32, []> var_222 = const()[name = tensor<string, []>("op_222"), val = tensor<fp32, []>(0x1.99999ap-3)];
153
- 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, []>(4616192))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4681792))), name = tensor<string, []>("F0_2_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
154
- 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, []>(4682368)))];
155
- 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 = s_to_fp16)[name = tensor<string, []>("linear_4_cast_fp16")];
156
- tensor<int32, [3]> var_247 = const()[name = tensor<string, []>("op_247"), val = tensor<int32, [3]>([1, 512, 1])];
157
- tensor<fp16, [1, 512, 1]> h_19_cast_fp16 = reshape(shape = var_247, x = linear_4_cast_fp16)[name = tensor<string, []>("h_19_cast_fp16")];
158
- tensor<int32, [2]> var_249_split_sizes_0 = const()[name = tensor<string, []>("op_249_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
159
- tensor<int32, []> var_249_axis_0 = const()[name = tensor<string, []>("op_249_axis_0"), val = tensor<int32, []>(1)];
160
- tensor<fp16, [1, 256, 1]> var_249_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_249_cast_fp16_1 = split(axis = var_249_axis_0, split_sizes = var_249_split_sizes_0, x = h_19_cast_fp16)[name = tensor<string, []>("op_249_cast_fp16")];
161
- tensor<fp16, []> var_251_promoted_to_fp16 = const()[name = tensor<string, []>("op_251_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
162
- tensor<fp16, [1, 256, 1]> var_252_cast_fp16 = add(x = var_249_cast_fp16_0, y = var_251_promoted_to_fp16)[name = tensor<string, []>("op_252_cast_fp16")];
163
- tensor<fp16, []> var_225_to_fp16 = const()[name = tensor<string, []>("op_225_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
164
- tensor<fp16, [1, 256, 4000]> var_253_cast_fp16 = instance_norm(epsilon = var_225_to_fp16, x = _inversed_input_39_cast_fp16)[name = tensor<string, []>("op_253_cast_fp16")];
165
- tensor<fp16, [1, 256, 4000]> var_254_cast_fp16 = mul(x = var_252_cast_fp16, y = var_253_cast_fp16)[name = tensor<string, []>("op_254_cast_fp16")];
166
- tensor<fp16, [1, 256, 4000]> input_41_cast_fp16 = add(x = var_254_cast_fp16, y = var_249_cast_fp16_1)[name = tensor<string, []>("input_41_cast_fp16")];
167
- tensor<fp16, [1, 256, 4000]> input_43_cast_fp16 = leaky_relu(alpha = var_222, x = input_41_cast_fp16)[name = tensor<string, []>("input_43_cast_fp16")];
168
- tensor<string, []> input_47_pad_type_0 = const()[name = tensor<string, []>("input_47_pad_type_0"), val = tensor<string, []>("custom")];
169
- tensor<int32, [2]> input_47_pad_0 = const()[name = tensor<string, []>("input_47_pad_0"), val = tensor<int32, [2]>([1, 1])];
170
- tensor<int32, [1]> input_47_strides_0 = const()[name = tensor<string, []>("input_47_strides_0"), val = tensor<int32, [1]>([1])];
171
- tensor<int32, [1]> input_47_dilations_0 = const()[name = tensor<string, []>("input_47_dilations_0"), val = tensor<int32, [1]>([1])];
172
- tensor<int32, []> input_47_groups_0 = const()[name = tensor<string, []>("input_47_groups_0"), val = tensor<int32, []>(1)];
173
- tensor<fp16, [256, 256, 3]> weight_11_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4683456))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4880128))), name = tensor<string, []>("weight_11_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
174
- 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, []>(4880704)))];
175
- tensor<fp16, [1, 256, 4000]> input_47_cast_fp16 = conv(bias = F0_2_conv1_bias_to_fp16, dilations = input_47_dilations_0, groups = input_47_groups_0, pad = input_47_pad_0, pad_type = input_47_pad_type_0, strides = input_47_strides_0, weight = weight_11_to_fp16_palettized, x = input_43_cast_fp16)[name = tensor<string, []>("input_47_cast_fp16")];
176
- 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, []>(4881280))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4946880))), name = tensor<string, []>("F0_2_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
177
- 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, []>(4947456)))];
178
- 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 = s_to_fp16)[name = tensor<string, []>("linear_5_cast_fp16")];
179
- tensor<int32, [3]> var_271 = const()[name = tensor<string, []>("op_271"), val = tensor<int32, [3]>([1, 512, 1])];
180
- tensor<fp16, [1, 512, 1]> h_23_cast_fp16 = reshape(shape = var_271, x = linear_5_cast_fp16)[name = tensor<string, []>("h_23_cast_fp16")];
181
- tensor<int32, [2]> var_273_split_sizes_0 = const()[name = tensor<string, []>("op_273_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
182
- tensor<int32, []> var_273_axis_0 = const()[name = tensor<string, []>("op_273_axis_0"), val = tensor<int32, []>(1)];
183
- tensor<fp16, [1, 256, 1]> var_273_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_273_cast_fp16_1 = split(axis = var_273_axis_0, split_sizes = var_273_split_sizes_0, x = h_23_cast_fp16)[name = tensor<string, []>("op_273_cast_fp16")];
184
- tensor<fp16, []> var_275_promoted_to_fp16 = const()[name = tensor<string, []>("op_275_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
185
- tensor<fp16, [1, 256, 1]> var_276_cast_fp16 = add(x = var_273_cast_fp16_0, y = var_275_promoted_to_fp16)[name = tensor<string, []>("op_276_cast_fp16")];
186
- tensor<fp16, [1, 256, 4000]> var_277_cast_fp16 = instance_norm(epsilon = var_225_to_fp16, x = input_47_cast_fp16)[name = tensor<string, []>("op_277_cast_fp16")];
187
- tensor<fp16, [1, 256, 4000]> var_278_cast_fp16 = mul(x = var_276_cast_fp16, y = var_277_cast_fp16)[name = tensor<string, []>("op_278_cast_fp16")];
188
- tensor<fp16, [1, 256, 4000]> input_49_cast_fp16 = add(x = var_278_cast_fp16, y = var_273_cast_fp16_1)[name = tensor<string, []>("input_49_cast_fp16")];
189
- tensor<fp16, [1, 256, 4000]> input_51_cast_fp16 = leaky_relu(alpha = var_222, x = input_49_cast_fp16)[name = tensor<string, []>("input_51_cast_fp16")];
190
- tensor<string, []> out_5_pad_type_0 = const()[name = tensor<string, []>("out_5_pad_type_0"), val = tensor<string, []>("custom")];
191
- tensor<int32, [2]> out_5_pad_0 = const()[name = tensor<string, []>("out_5_pad_0"), val = tensor<int32, [2]>([1, 1])];
192
- tensor<int32, [1]> out_5_strides_0 = const()[name = tensor<string, []>("out_5_strides_0"), val = tensor<int32, [1]>([1])];
193
- tensor<int32, [1]> out_5_dilations_0 = const()[name = tensor<string, []>("out_5_dilations_0"), val = tensor<int32, [1]>([1])];
194
- tensor<int32, []> out_5_groups_0 = const()[name = tensor<string, []>("out_5_groups_0"), val = tensor<int32, []>(1)];
195
- tensor<fp16, [256, 256, 3]> weight_13_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(4948544))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5145216))), name = tensor<string, []>("weight_13_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
196
- 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, []>(5145792)))];
197
- tensor<fp16, [1, 256, 4000]> 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_13_to_fp16_palettized, x = input_51_cast_fp16)[name = tensor<string, []>("out_5_cast_fp16")];
198
- tensor<fp16, [1, 256, 4000]> var_289_cast_fp16 = add(x = out_5_cast_fp16, y = _inversed_input_39_cast_fp16)[name = tensor<string, []>("op_289_cast_fp16")];
199
- tensor<fp16, []> _inversed_input_55_y_0_to_fp16 = const()[name = tensor<string, []>("_inversed_input_55_y_0_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
200
- tensor<fp16, [1, 256, 4000]> _inversed_input_55_cast_fp16 = mul(x = var_289_cast_fp16, y = _inversed_input_55_y_0_to_fp16)[name = tensor<string, []>("_inversed_input_55_cast_fp16")];
201
- tensor<string, []> var_302_pad_type_0 = const()[name = tensor<string, []>("op_302_pad_type_0"), val = tensor<string, []>("valid")];
202
- tensor<int32, [1]> var_302_strides_0 = const()[name = tensor<string, []>("op_302_strides_0"), val = tensor<int32, [1]>([1])];
203
- tensor<int32, [2]> var_302_pad_0 = const()[name = tensor<string, []>("op_302_pad_0"), val = tensor<int32, [2]>([0, 0])];
204
- tensor<int32, [1]> var_302_dilations_0 = const()[name = tensor<string, []>("op_302_dilations_0"), val = tensor<int32, [1]>([1])];
205
- tensor<int32, []> var_302_groups_0 = const()[name = tensor<string, []>("op_302_groups_0"), val = tensor<int32, []>(1)];
206
- 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, []>(5146368)))];
207
- tensor<fp16, [1]> F0_proj_bias_to_fp16 = const()[name = tensor<string, []>("F0_proj_bias_to_fp16"), val = tensor<fp16, [1]>([0x1.adp-4])];
208
- tensor<fp16, [1, 1, 4000]> var_302_cast_fp16 = conv(bias = F0_proj_bias_to_fp16, dilations = var_302_dilations_0, groups = var_302_groups_0, pad = var_302_pad_0, pad_type = var_302_pad_type_0, strides = var_302_strides_0, weight = F0_proj_weight_to_fp16, x = _inversed_input_55_cast_fp16)[name = tensor<string, []>("op_302_cast_fp16")];
209
- tensor<int32, [1]> var_304_axes_0 = const()[name = tensor<string, []>("op_304_axes_0"), val = tensor<int32, [1]>([1])];
210
- tensor<fp16, [1, 4000]> F0 = squeeze(axes = var_304_axes_0, x = var_302_cast_fp16)[name = tensor<string, []>("op_304_cast_fp16")];
211
- tensor<fp32, []> var_309 = const()[name = tensor<string, []>("op_309"), val = tensor<fp32, []>(0x1.99999ap-3)];
212
- 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, []>(5146944))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5278080))), name = tensor<string, []>("N_0_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
213
- 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, []>(5278656)))];
214
- 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 = s_to_fp16)[name = tensor<string, []>("linear_6_cast_fp16")];
215
- tensor<int32, [3]> var_334 = const()[name = tensor<string, []>("op_334"), val = tensor<int32, [3]>([1, 1024, 1])];
216
- tensor<fp16, [1, 1024, 1]> h_27_cast_fp16 = reshape(shape = var_334, x = linear_6_cast_fp16)[name = tensor<string, []>("h_27_cast_fp16")];
217
- tensor<int32, [2]> var_336_split_sizes_0 = const()[name = tensor<string, []>("op_336_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
218
- tensor<int32, []> var_336_axis_0 = const()[name = tensor<string, []>("op_336_axis_0"), val = tensor<int32, []>(1)];
219
- tensor<fp16, [1, 512, 1]> var_336_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_336_cast_fp16_1 = split(axis = var_336_axis_0, split_sizes = var_336_split_sizes_0, x = h_27_cast_fp16)[name = tensor<string, []>("op_336_cast_fp16")];
220
- tensor<fp16, []> var_338_promoted_to_fp16 = const()[name = tensor<string, []>("op_338_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
221
- tensor<fp16, [1, 512, 1]> var_339_cast_fp16 = add(x = var_336_cast_fp16_0, y = var_338_promoted_to_fp16)[name = tensor<string, []>("op_339_cast_fp16")];
222
- tensor<fp16, [1, 512, 2000]> var_341_cast_fp16 = mul(x = var_339_cast_fp16, y = var_84_cast_fp16)[name = tensor<string, []>("op_341_cast_fp16")];
223
- tensor<fp16, [1, 512, 2000]> input_59_cast_fp16 = add(x = var_341_cast_fp16, y = var_336_cast_fp16_1)[name = tensor<string, []>("input_59_cast_fp16")];
224
- tensor<fp16, [1, 512, 2000]> input_61_cast_fp16 = leaky_relu(alpha = var_309, x = input_59_cast_fp16)[name = tensor<string, []>("input_61_cast_fp16")];
225
- tensor<string, []> input_65_pad_type_0 = const()[name = tensor<string, []>("input_65_pad_type_0"), val = tensor<string, []>("custom")];
226
- tensor<int32, [2]> input_65_pad_0 = const()[name = tensor<string, []>("input_65_pad_0"), val = tensor<int32, [2]>([1, 1])];
227
- tensor<int32, [1]> input_65_strides_0 = const()[name = tensor<string, []>("input_65_strides_0"), val = tensor<int32, [1]>([1])];
228
- tensor<int32, [1]> input_65_dilations_0 = const()[name = tensor<string, []>("input_65_dilations_0"), val = tensor<int32, [1]>([1])];
229
- tensor<int32, []> input_65_groups_0 = const()[name = tensor<string, []>("input_65_groups_0"), val = tensor<int32, []>(1)];
230
- tensor<fp16, [512, 512, 3]> weight_17_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [786432]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(5280768))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6067264))), name = tensor<string, []>("weight_17_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 3])];
231
- 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, []>(6067840)))];
232
- tensor<fp16, [1, 512, 2000]> input_65_cast_fp16 = conv(bias = N_0_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_17_to_fp16_palettized, x = input_61_cast_fp16)[name = tensor<string, []>("input_65_cast_fp16")];
233
- 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, []>(6068928))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6200064))), name = tensor<string, []>("N_0_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
234
- 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, []>(6200640)))];
235
- 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 = s_to_fp16)[name = tensor<string, []>("linear_7_cast_fp16")];
236
- tensor<int32, [3]> var_358 = const()[name = tensor<string, []>("op_358"), val = tensor<int32, [3]>([1, 1024, 1])];
237
- tensor<fp16, [1, 1024, 1]> h_31_cast_fp16 = reshape(shape = var_358, x = linear_7_cast_fp16)[name = tensor<string, []>("h_31_cast_fp16")];
238
- tensor<int32, [2]> var_360_split_sizes_0 = const()[name = tensor<string, []>("op_360_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
239
- tensor<int32, []> var_360_axis_0 = const()[name = tensor<string, []>("op_360_axis_0"), val = tensor<int32, []>(1)];
240
- tensor<fp16, [1, 512, 1]> var_360_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_360_cast_fp16_1 = split(axis = var_360_axis_0, split_sizes = var_360_split_sizes_0, x = h_31_cast_fp16)[name = tensor<string, []>("op_360_cast_fp16")];
241
- tensor<fp16, []> var_362_promoted_to_fp16 = const()[name = tensor<string, []>("op_362_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
242
- tensor<fp16, [1, 512, 1]> var_363_cast_fp16 = add(x = var_360_cast_fp16_0, y = var_362_promoted_to_fp16)[name = tensor<string, []>("op_363_cast_fp16")];
243
- tensor<fp16, []> var_312_to_fp16 = const()[name = tensor<string, []>("op_312_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
244
- tensor<fp16, [1, 512, 2000]> var_364_cast_fp16 = instance_norm(epsilon = var_312_to_fp16, x = input_65_cast_fp16)[name = tensor<string, []>("op_364_cast_fp16")];
245
- tensor<fp16, [1, 512, 2000]> var_365_cast_fp16 = mul(x = var_363_cast_fp16, y = var_364_cast_fp16)[name = tensor<string, []>("op_365_cast_fp16")];
246
- tensor<fp16, [1, 512, 2000]> input_67_cast_fp16 = add(x = var_365_cast_fp16, y = var_360_cast_fp16_1)[name = tensor<string, []>("input_67_cast_fp16")];
247
- tensor<fp16, [1, 512, 2000]> input_69_cast_fp16 = leaky_relu(alpha = var_309, x = input_67_cast_fp16)[name = tensor<string, []>("input_69_cast_fp16")];
248
- tensor<string, []> out_7_pad_type_0 = const()[name = tensor<string, []>("out_7_pad_type_0"), val = tensor<string, []>("custom")];
249
- tensor<int32, [2]> out_7_pad_0 = const()[name = tensor<string, []>("out_7_pad_0"), val = tensor<int32, [2]>([1, 1])];
250
- tensor<int32, [1]> out_7_strides_0 = const()[name = tensor<string, []>("out_7_strides_0"), val = tensor<int32, [1]>([1])];
251
- tensor<int32, [1]> out_7_dilations_0 = const()[name = tensor<string, []>("out_7_dilations_0"), val = tensor<int32, [1]>([1])];
252
- tensor<int32, []> out_7_groups_0 = const()[name = tensor<string, []>("out_7_groups_0"), val = tensor<int32, []>(1)];
253
- tensor<fp16, [512, 512, 3]> weight_19_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [786432]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6202752))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(6989248))), name = tensor<string, []>("weight_19_to_fp16_palettized"), shape = tensor<uint32, [3]>([512, 512, 3])];
254
- 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, []>(6989824)))];
255
- tensor<fp16, [1, 512, 2000]> 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_19_to_fp16_palettized, x = input_69_cast_fp16)[name = tensor<string, []>("out_7_cast_fp16")];
256
- tensor<fp16, [1, 512, 2000]> var_376_cast_fp16 = add(x = out_7_cast_fp16, y = transpose_0)[name = tensor<string, []>("op_376_cast_fp16")];
257
- tensor<fp16, []> _inversed_input_73_y_0_to_fp16 = const()[name = tensor<string, []>("_inversed_input_73_y_0_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
258
- tensor<fp16, [1, 512, 2000]> _inversed_input_73_cast_fp16 = mul(x = var_376_cast_fp16, y = _inversed_input_73_y_0_to_fp16)[name = tensor<string, []>("_inversed_input_73_cast_fp16")];
259
- tensor<fp32, []> var_382 = const()[name = tensor<string, []>("op_382"), val = tensor<fp32, []>(0x1.99999ap-3)];
260
- 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, []>(6990912))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7122048))), name = tensor<string, []>("N_1_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([1024, 128])];
261
- 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, []>(7122624)))];
262
- 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 = s_to_fp16)[name = tensor<string, []>("linear_8_cast_fp16")];
263
- tensor<int32, [3]> var_417 = const()[name = tensor<string, []>("op_417"), val = tensor<int32, [3]>([1, 1024, 1])];
264
- tensor<fp16, [1, 1024, 1]> h_35_cast_fp16 = reshape(shape = var_417, x = linear_8_cast_fp16)[name = tensor<string, []>("h_35_cast_fp16")];
265
- tensor<int32, [2]> var_419_split_sizes_0 = const()[name = tensor<string, []>("op_419_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
266
- tensor<int32, []> var_419_axis_0 = const()[name = tensor<string, []>("op_419_axis_0"), val = tensor<int32, []>(1)];
267
- tensor<fp16, [1, 512, 1]> var_419_cast_fp16_0, tensor<fp16, [1, 512, 1]> var_419_cast_fp16_1 = split(axis = var_419_axis_0, split_sizes = var_419_split_sizes_0, x = h_35_cast_fp16)[name = tensor<string, []>("op_419_cast_fp16")];
268
- tensor<fp16, []> var_421_promoted_to_fp16 = const()[name = tensor<string, []>("op_421_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
269
- tensor<fp16, [1, 512, 1]> var_422_cast_fp16 = add(x = var_419_cast_fp16_0, y = var_421_promoted_to_fp16)[name = tensor<string, []>("op_422_cast_fp16")];
270
- tensor<fp16, []> var_386_to_fp16 = const()[name = tensor<string, []>("op_386_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
271
- tensor<fp16, [1, 512, 2000]> var_423_cast_fp16 = instance_norm(epsilon = var_386_to_fp16, x = _inversed_input_73_cast_fp16)[name = tensor<string, []>("op_423_cast_fp16")];
272
- tensor<fp16, [1, 512, 2000]> var_424_cast_fp16 = mul(x = var_422_cast_fp16, y = var_423_cast_fp16)[name = tensor<string, []>("op_424_cast_fp16")];
273
- tensor<fp16, [1, 512, 2000]> input_75_cast_fp16 = add(x = var_424_cast_fp16, y = var_419_cast_fp16_1)[name = tensor<string, []>("input_75_cast_fp16")];
274
- tensor<fp16, [1, 512, 2000]> input_77_cast_fp16 = leaky_relu(alpha = var_382, x = input_75_cast_fp16)[name = tensor<string, []>("input_77_cast_fp16")];
275
- tensor<string, []> conv_transpose_1_pad_type_0 = const()[name = tensor<string, []>("conv_transpose_1_pad_type_0"), val = tensor<string, []>("custom")];
276
- tensor<int32, [2]> conv_transpose_1_pad_0 = const()[name = tensor<string, []>("conv_transpose_1_pad_0"), val = tensor<int32, [2]>([0, 0])];
277
- tensor<int32, [1]> conv_transpose_1_strides_0 = const()[name = tensor<string, []>("conv_transpose_1_strides_0"), val = tensor<int32, [1]>([2])];
278
- tensor<int32, []> conv_transpose_1_groups_0 = const()[name = tensor<string, []>("conv_transpose_1_groups_0"), val = tensor<int32, []>(512)];
279
- tensor<int32, [1]> conv_transpose_1_dilations_0 = const()[name = tensor<string, []>("conv_transpose_1_dilations_0"), val = tensor<int32, [1]>([1])];
280
- tensor<int32, [3]> conv_transpose_1_has_output_shape_output_shape_0 = const()[name = tensor<string, []>("conv_transpose_1_has_output_shape_output_shape_0"), val = tensor<int32, [3]>([1, 512, 4001])];
281
- tensor<fp16, [512, 1, 3]> var_427_to_fp16 = const()[name = tensor<string, []>("op_427_to_fp16"), val = tensor<fp16, [512, 1, 3]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7124736)))];
282
- 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, []>(7127872)))];
283
- tensor<fp16, [1, 512, 4001]> conv_transpose_1_has_output_shape_cast_fp16 = conv_transpose(bias = N_1_pool_bias_to_fp16, dilations = conv_transpose_1_dilations_0, groups = conv_transpose_1_groups_0, output_shape = conv_transpose_1_has_output_shape_output_shape_0, pad = conv_transpose_1_pad_0, pad_type = conv_transpose_1_pad_type_0, strides = conv_transpose_1_strides_0, weight = var_427_to_fp16, x = input_77_cast_fp16)[name = tensor<string, []>("conv_transpose_1_has_output_shape_cast_fp16")];
284
- tensor<int32, [3]> input_79_begin_0 = const()[name = tensor<string, []>("input_79_begin_0"), val = tensor<int32, [3]>([0, 0, 1])];
285
- tensor<int32, [3]> input_79_end_0 = const()[name = tensor<string, []>("input_79_end_0"), val = tensor<int32, [3]>([0, 0, 0])];
286
- tensor<bool, [3]> input_79_begin_mask_0 = const()[name = tensor<string, []>("input_79_begin_mask_0"), val = tensor<bool, [3]>([true, true, false])];
287
- tensor<bool, [3]> input_79_end_mask_0 = const()[name = tensor<string, []>("input_79_end_mask_0"), val = tensor<bool, [3]>([true, true, true])];
288
- tensor<fp16, [1, 512, 4000]> input_79_cast_fp16 = slice_by_index(begin = input_79_begin_0, begin_mask = input_79_begin_mask_0, end = input_79_end_0, end_mask = input_79_end_mask_0, x = conv_transpose_1_has_output_shape_cast_fp16)[name = tensor<string, []>("input_79_cast_fp16")];
289
- tensor<string, []> input_83_pad_type_0 = const()[name = tensor<string, []>("input_83_pad_type_0"), val = tensor<string, []>("custom")];
290
- tensor<int32, [2]> input_83_pad_0 = const()[name = tensor<string, []>("input_83_pad_0"), val = tensor<int32, [2]>([1, 1])];
291
- tensor<int32, [1]> input_83_strides_0 = const()[name = tensor<string, []>("input_83_strides_0"), val = tensor<int32, [1]>([1])];
292
- tensor<int32, [1]> input_83_dilations_0 = const()[name = tensor<string, []>("input_83_dilations_0"), val = tensor<int32, [1]>([1])];
293
- tensor<int32, []> input_83_groups_0 = const()[name = tensor<string, []>("input_83_groups_0"), val = tensor<int32, []>(1)];
294
- tensor<fp16, [256, 512, 3]> weight_21_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [393216]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7128960))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7522240))), name = tensor<string, []>("weight_21_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 512, 3])];
295
- 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, []>(7522816)))];
296
- tensor<fp16, [1, 256, 4000]> input_83_cast_fp16 = conv(bias = N_1_conv1_bias_to_fp16, dilations = input_83_dilations_0, groups = input_83_groups_0, pad = input_83_pad_0, pad_type = input_83_pad_type_0, strides = input_83_strides_0, weight = weight_21_to_fp16_palettized, x = input_79_cast_fp16)[name = tensor<string, []>("input_83_cast_fp16")];
297
- 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, []>(7523392))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7588992))), name = tensor<string, []>("N_1_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
298
- 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, []>(7589568)))];
299
- 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 = s_to_fp16)[name = tensor<string, []>("linear_9_cast_fp16")];
300
- tensor<int32, [3]> var_448 = const()[name = tensor<string, []>("op_448"), val = tensor<int32, [3]>([1, 512, 1])];
301
- tensor<fp16, [1, 512, 1]> h_39_cast_fp16 = reshape(shape = var_448, x = linear_9_cast_fp16)[name = tensor<string, []>("h_39_cast_fp16")];
302
- tensor<int32, [2]> var_450_split_sizes_0 = const()[name = tensor<string, []>("op_450_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
303
- tensor<int32, []> var_450_axis_0 = const()[name = tensor<string, []>("op_450_axis_0"), val = tensor<int32, []>(1)];
304
- tensor<fp16, [1, 256, 1]> var_450_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_450_cast_fp16_1 = split(axis = var_450_axis_0, split_sizes = var_450_split_sizes_0, x = h_39_cast_fp16)[name = tensor<string, []>("op_450_cast_fp16")];
305
- tensor<fp16, []> var_452_promoted_to_fp16 = const()[name = tensor<string, []>("op_452_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
306
- tensor<fp16, [1, 256, 1]> var_453_cast_fp16 = add(x = var_450_cast_fp16_0, y = var_452_promoted_to_fp16)[name = tensor<string, []>("op_453_cast_fp16")];
307
- tensor<fp16, [1, 256, 4000]> var_454_cast_fp16 = instance_norm(epsilon = var_386_to_fp16, x = input_83_cast_fp16)[name = tensor<string, []>("op_454_cast_fp16")];
308
- tensor<fp16, [1, 256, 4000]> var_455_cast_fp16 = mul(x = var_453_cast_fp16, y = var_454_cast_fp16)[name = tensor<string, []>("op_455_cast_fp16")];
309
- tensor<fp16, [1, 256, 4000]> input_85_cast_fp16 = add(x = var_455_cast_fp16, y = var_450_cast_fp16_1)[name = tensor<string, []>("input_85_cast_fp16")];
310
- tensor<fp16, [1, 256, 4000]> input_87_cast_fp16 = leaky_relu(alpha = var_382, x = input_85_cast_fp16)[name = tensor<string, []>("input_87_cast_fp16")];
311
- tensor<string, []> out_9_pad_type_0 = const()[name = tensor<string, []>("out_9_pad_type_0"), val = tensor<string, []>("custom")];
312
- tensor<int32, [2]> out_9_pad_0 = const()[name = tensor<string, []>("out_9_pad_0"), val = tensor<int32, [2]>([1, 1])];
313
- tensor<int32, [1]> out_9_strides_0 = const()[name = tensor<string, []>("out_9_strides_0"), val = tensor<int32, [1]>([1])];
314
- tensor<int32, [1]> out_9_dilations_0 = const()[name = tensor<string, []>("out_9_dilations_0"), val = tensor<int32, [1]>([1])];
315
- tensor<int32, []> out_9_groups_0 = const()[name = tensor<string, []>("out_9_groups_0"), val = tensor<int32, []>(1)];
316
- tensor<fp16, [256, 256, 3]> weight_23_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7590656))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7787328))), name = tensor<string, []>("weight_23_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
317
- 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, []>(7787904)))];
318
- tensor<fp16, [1, 256, 4000]> 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_23_to_fp16_palettized, x = input_87_cast_fp16)[name = tensor<string, []>("out_9_cast_fp16")];
319
- tensor<int32, [1]> expand_dims_1_axes_0 = const()[name = tensor<string, []>("expand_dims_1_axes_0"), val = tensor<int32, [1]>([3])];
320
- tensor<fp16, [1, 512, 2000, 1]> expand_dims_1_cast_fp16 = expand_dims(axes = expand_dims_1_axes_0, x = _inversed_input_73_cast_fp16)[name = tensor<string, []>("expand_dims_1_cast_fp16")];
321
- 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)];
322
- 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)];
323
- tensor<fp16, [1, 512, 4000, 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")];
324
- tensor<int32, [1]> input_91_axes_0 = const()[name = tensor<string, []>("input_91_axes_0"), val = tensor<int32, [1]>([3])];
325
- tensor<fp16, [1, 512, 4000]> input_91_cast_fp16 = squeeze(axes = input_91_axes_0, x = upsample_nearest_neighbor_1_cast_fp16)[name = tensor<string, []>("input_91_cast_fp16")];
326
- tensor<string, []> var_473_pad_type_0 = const()[name = tensor<string, []>("op_473_pad_type_0"), val = tensor<string, []>("valid")];
327
- tensor<int32, [1]> var_473_strides_0 = const()[name = tensor<string, []>("op_473_strides_0"), val = tensor<int32, [1]>([1])];
328
- tensor<int32, [2]> var_473_pad_0 = const()[name = tensor<string, []>("op_473_pad_0"), val = tensor<int32, [2]>([0, 0])];
329
- tensor<int32, [1]> var_473_dilations_0 = const()[name = tensor<string, []>("op_473_dilations_0"), val = tensor<int32, [1]>([1])];
330
- tensor<int32, []> var_473_groups_0 = const()[name = tensor<string, []>("op_473_groups_0"), val = tensor<int32, []>(1)];
331
- tensor<fp16, [256, 512, 1]> weight_25_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [131072]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7788480))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7919616))), name = tensor<string, []>("weight_25_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 512, 1])];
332
- tensor<fp16, [1, 256, 4000]> var_473_cast_fp16 = conv(dilations = var_473_dilations_0, groups = var_473_groups_0, pad = var_473_pad_0, pad_type = var_473_pad_type_0, strides = var_473_strides_0, weight = weight_25_to_fp16_palettized, x = input_91_cast_fp16)[name = tensor<string, []>("op_473_cast_fp16")];
333
- tensor<fp16, [1, 256, 4000]> var_474_cast_fp16 = add(x = out_9_cast_fp16, y = var_473_cast_fp16)[name = tensor<string, []>("op_474_cast_fp16")];
334
- tensor<fp16, []> _inversed_input_93_y_0_to_fp16 = const()[name = tensor<string, []>("_inversed_input_93_y_0_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
335
- tensor<fp16, [1, 256, 4000]> _inversed_input_93_cast_fp16 = mul(x = var_474_cast_fp16, y = _inversed_input_93_y_0_to_fp16)[name = tensor<string, []>("_inversed_input_93_cast_fp16")];
336
- tensor<fp32, []> var_478 = const()[name = tensor<string, []>("op_478"), val = tensor<fp32, []>(0x1.99999ap-3)];
337
- 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, []>(7920192))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7985792))), name = tensor<string, []>("N_2_norm1_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
338
- 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, []>(7986368)))];
339
- 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 = s_to_fp16)[name = tensor<string, []>("linear_10_cast_fp16")];
340
- tensor<int32, [3]> var_503 = const()[name = tensor<string, []>("op_503"), val = tensor<int32, [3]>([1, 512, 1])];
341
- tensor<fp16, [1, 512, 1]> h_43_cast_fp16 = reshape(shape = var_503, x = linear_10_cast_fp16)[name = tensor<string, []>("h_43_cast_fp16")];
342
- tensor<int32, [2]> var_505_split_sizes_0 = const()[name = tensor<string, []>("op_505_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
343
- tensor<int32, []> var_505_axis_0 = const()[name = tensor<string, []>("op_505_axis_0"), val = tensor<int32, []>(1)];
344
- tensor<fp16, [1, 256, 1]> var_505_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_505_cast_fp16_1 = split(axis = var_505_axis_0, split_sizes = var_505_split_sizes_0, x = h_43_cast_fp16)[name = tensor<string, []>("op_505_cast_fp16")];
345
- tensor<fp16, []> var_507_promoted_to_fp16 = const()[name = tensor<string, []>("op_507_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
346
- tensor<fp16, [1, 256, 1]> var_508_cast_fp16 = add(x = var_505_cast_fp16_0, y = var_507_promoted_to_fp16)[name = tensor<string, []>("op_508_cast_fp16")];
347
- tensor<fp16, []> var_481_to_fp16 = const()[name = tensor<string, []>("op_481_to_fp16"), val = tensor<fp16, []>(0x1.5p-17)];
348
- tensor<fp16, [1, 256, 4000]> var_509_cast_fp16 = instance_norm(epsilon = var_481_to_fp16, x = _inversed_input_93_cast_fp16)[name = tensor<string, []>("op_509_cast_fp16")];
349
- tensor<fp16, [1, 256, 4000]> var_510_cast_fp16 = mul(x = var_508_cast_fp16, y = var_509_cast_fp16)[name = tensor<string, []>("op_510_cast_fp16")];
350
- tensor<fp16, [1, 256, 4000]> input_95_cast_fp16 = add(x = var_510_cast_fp16, y = var_505_cast_fp16_1)[name = tensor<string, []>("input_95_cast_fp16")];
351
- tensor<fp16, [1, 256, 4000]> input_97_cast_fp16 = leaky_relu(alpha = var_478, x = input_95_cast_fp16)[name = tensor<string, []>("input_97_cast_fp16")];
352
- tensor<string, []> input_101_pad_type_0 = const()[name = tensor<string, []>("input_101_pad_type_0"), val = tensor<string, []>("custom")];
353
- tensor<int32, [2]> input_101_pad_0 = const()[name = tensor<string, []>("input_101_pad_0"), val = tensor<int32, [2]>([1, 1])];
354
- tensor<int32, [1]> input_101_strides_0 = const()[name = tensor<string, []>("input_101_strides_0"), val = tensor<int32, [1]>([1])];
355
- tensor<int32, [1]> input_101_dilations_0 = const()[name = tensor<string, []>("input_101_dilations_0"), val = tensor<int32, [1]>([1])];
356
- tensor<int32, []> input_101_groups_0 = const()[name = tensor<string, []>("input_101_groups_0"), val = tensor<int32, []>(1)];
357
- tensor<fp16, [256, 256, 3]> weight_27_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(7987456))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8184128))), name = tensor<string, []>("weight_27_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
358
- 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, []>(8184704)))];
359
- tensor<fp16, [1, 256, 4000]> input_101_cast_fp16 = conv(bias = N_2_conv1_bias_to_fp16, dilations = input_101_dilations_0, groups = input_101_groups_0, pad = input_101_pad_0, pad_type = input_101_pad_type_0, strides = input_101_strides_0, weight = weight_27_to_fp16_palettized, x = input_97_cast_fp16)[name = tensor<string, []>("input_101_cast_fp16")];
360
- 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, []>(8185280))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8250880))), name = tensor<string, []>("N_2_norm2_fc_weight_to_fp16_palettized"), shape = tensor<uint32, [2]>([512, 128])];
361
- 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, []>(8251456)))];
362
- 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 = s_to_fp16)[name = tensor<string, []>("linear_11_cast_fp16")];
363
- tensor<int32, [3]> var_527 = const()[name = tensor<string, []>("op_527"), val = tensor<int32, [3]>([1, 512, 1])];
364
- tensor<fp16, [1, 512, 1]> h_cast_fp16 = reshape(shape = var_527, x = linear_11_cast_fp16)[name = tensor<string, []>("h_cast_fp16")];
365
- tensor<int32, [2]> var_529_split_sizes_0 = const()[name = tensor<string, []>("op_529_split_sizes_0"), val = tensor<int32, [2]>([256, 256])];
366
- tensor<int32, []> var_529_axis_0 = const()[name = tensor<string, []>("op_529_axis_0"), val = tensor<int32, []>(1)];
367
- tensor<fp16, [1, 256, 1]> var_529_cast_fp16_0, tensor<fp16, [1, 256, 1]> var_529_cast_fp16_1 = split(axis = var_529_axis_0, split_sizes = var_529_split_sizes_0, x = h_cast_fp16)[name = tensor<string, []>("op_529_cast_fp16")];
368
- tensor<fp16, []> var_531_promoted_to_fp16 = const()[name = tensor<string, []>("op_531_promoted_to_fp16"), val = tensor<fp16, []>(0x1p+0)];
369
- tensor<fp16, [1, 256, 1]> var_532_cast_fp16 = add(x = var_529_cast_fp16_0, y = var_531_promoted_to_fp16)[name = tensor<string, []>("op_532_cast_fp16")];
370
- tensor<fp16, [1, 256, 4000]> var_533_cast_fp16 = instance_norm(epsilon = var_481_to_fp16, x = input_101_cast_fp16)[name = tensor<string, []>("op_533_cast_fp16")];
371
- tensor<fp16, [1, 256, 4000]> var_534_cast_fp16 = mul(x = var_532_cast_fp16, y = var_533_cast_fp16)[name = tensor<string, []>("op_534_cast_fp16")];
372
- tensor<fp16, [1, 256, 4000]> input_103_cast_fp16 = add(x = var_534_cast_fp16, y = var_529_cast_fp16_1)[name = tensor<string, []>("input_103_cast_fp16")];
373
- tensor<fp16, [1, 256, 4000]> input_105_cast_fp16 = leaky_relu(alpha = var_478, x = input_103_cast_fp16)[name = tensor<string, []>("input_105_cast_fp16")];
374
- tensor<string, []> out_pad_type_0 = const()[name = tensor<string, []>("out_pad_type_0"), val = tensor<string, []>("custom")];
375
- tensor<int32, [2]> out_pad_0 = const()[name = tensor<string, []>("out_pad_0"), val = tensor<int32, [2]>([1, 1])];
376
- tensor<int32, [1]> out_strides_0 = const()[name = tensor<string, []>("out_strides_0"), val = tensor<int32, [1]>([1])];
377
- tensor<int32, [1]> out_dilations_0 = const()[name = tensor<string, []>("out_dilations_0"), val = tensor<int32, [1]>([1])];
378
- tensor<int32, []> out_groups_0 = const()[name = tensor<string, []>("out_groups_0"), val = tensor<int32, []>(1)];
379
- tensor<fp16, [256, 256, 3]> weight_29_to_fp16_palettized = constexpr_lut_to_dense()[indices = tensor<uint8, [196608]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8252544))), lut = tensor<fp16, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(8449216))), name = tensor<string, []>("weight_29_to_fp16_palettized"), shape = tensor<uint32, [3]>([256, 256, 3])];
380
- 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, []>(8449792)))];
381
- tensor<fp16, [1, 256, 4000]> 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_29_to_fp16_palettized, x = input_105_cast_fp16)[name = tensor<string, []>("out_cast_fp16")];
382
- tensor<fp16, [1, 256, 4000]> var_545_cast_fp16 = add(x = out_cast_fp16, y = _inversed_input_93_cast_fp16)[name = tensor<string, []>("op_545_cast_fp16")];
383
- tensor<fp16, []> _inversed_input_y_0_to_fp16 = const()[name = tensor<string, []>("_inversed_input_y_0_to_fp16"), val = tensor<fp16, []>(0x1.6ap-1)];
384
- tensor<fp16, [1, 256, 4000]> _inversed_input_cast_fp16 = mul(x = var_545_cast_fp16, y = _inversed_input_y_0_to_fp16)[name = tensor<string, []>("_inversed_input_cast_fp16")];
385
- tensor<string, []> var_558_pad_type_0 = const()[name = tensor<string, []>("op_558_pad_type_0"), val = tensor<string, []>("valid")];
386
- tensor<int32, [1]> var_558_strides_0 = const()[name = tensor<string, []>("op_558_strides_0"), val = tensor<int32, [1]>([1])];
387
- tensor<int32, [2]> var_558_pad_0 = const()[name = tensor<string, []>("op_558_pad_0"), val = tensor<int32, [2]>([0, 0])];
388
- tensor<int32, [1]> var_558_dilations_0 = const()[name = tensor<string, []>("op_558_dilations_0"), val = tensor<int32, [1]>([1])];
389
- tensor<int32, []> var_558_groups_0 = const()[name = tensor<string, []>("op_558_groups_0"), val = tensor<int32, []>(1)];
390
- 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, []>(8450368)))];
391
- tensor<fp16, [1]> N_proj_bias_to_fp16 = const()[name = tensor<string, []>("N_proj_bias_to_fp16"), val = tensor<fp16, [1]>([0x1.6ep-4])];
392
- tensor<fp16, [1, 1, 4000]> var_558_cast_fp16 = conv(bias = N_proj_bias_to_fp16, dilations = var_558_dilations_0, groups = var_558_groups_0, pad = var_558_pad_0, pad_type = var_558_pad_type_0, strides = var_558_strides_0, weight = N_proj_weight_to_fp16, x = _inversed_input_cast_fp16)[name = tensor<string, []>("op_558_cast_fp16")];
393
- tensor<int32, [1]> var_560_axes_0 = const()[name = tensor<string, []>("op_560_axes_0"), val = tensor<int32, [1]>([1])];
394
- tensor<fp16, [1, 4000]> N = squeeze(axes = var_560_axes_0, x = var_558_cast_fp16)[name = tensor<string, []>("op_560_cast_fp16")];
395
- } -> (F0, N);
396
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_prosody.mlmodelc/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:4242f6c193f8d1c995e9da98e63cda3f9559daaa9154be4d7a325c976d347ad8
3
- size 8450944
 
 
 
 
ANE/styletts2_ane_prosody.mlpackage/Data/com.apple.CoreML/model.mlmodel DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:4c8bb395a1d0b34fceb6da4e42c4cd201fe6c4121f788475940abaeb7ccd9e6f
3
- size 65948
 
 
 
 
ANE/styletts2_ane_prosody.mlpackage/Data/com.apple.CoreML/weights/weight.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:4242f6c193f8d1c995e9da98e63cda3f9559daaa9154be4d7a325c976d347ad8
3
- size 8450944
 
 
 
 
ANE/styletts2_ane_prosody.mlpackage/Manifest.json DELETED
@@ -1,18 +0,0 @@
1
- {
2
- "fileFormatVersion": "1.0.0",
3
- "itemInfoEntries": {
4
- "5239C997-4433-48DE-B7F7-8F278231936E": {
5
- "author": "com.apple.CoreML",
6
- "description": "CoreML Model Weights",
7
- "name": "weights",
8
- "path": "com.apple.CoreML/weights"
9
- },
10
- "EAD2ABD5-31E9-4A23-A7AB-80FE05510307": {
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": "EAD2ABD5-31E9-4A23-A7AB-80FE05510307"
18
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ANE/styletts2_ane_vocoder.mlmodelc/analytics/coremldata.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:66edac0a49a0eb82d4f02ad1bcab582cc3e6e0aae0a47cf9b5d669a0963ab5fd
3
- size 243