alexwengg commited on
Commit
f962799
·
verified ·
1 Parent(s): 1e40a12

Upload 33 files

Browse files
Files changed (33) hide show
  1. DurationPredictor.mlmodelc/analytics/coremldata.bin +3 -0
  2. DurationPredictor.mlmodelc/coremldata.bin +3 -0
  3. DurationPredictor.mlmodelc/model.mil +616 -0
  4. DurationPredictor.mlmodelc/weights/weight.bin +3 -0
  5. DurationPredictor.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  6. DurationPredictor.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  7. DurationPredictor.mlpackage/Manifest.json +18 -0
  8. README.md +202 -0
  9. TextEncoder.mlmodelc/analytics/coremldata.bin +3 -0
  10. TextEncoder.mlmodelc/coremldata.bin +3 -0
  11. TextEncoder.mlmodelc/model.mil +0 -0
  12. TextEncoder.mlmodelc/weights/weight.bin +3 -0
  13. TextEncoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  14. TextEncoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  15. TextEncoder.mlpackage/Manifest.json +18 -0
  16. VectorEstimator.mlmodelc/analytics/coremldata.bin +3 -0
  17. VectorEstimator.mlmodelc/coremldata.bin +3 -0
  18. VectorEstimator.mlmodelc/model.mil +0 -0
  19. VectorEstimator.mlmodelc/weights/weight.bin +3 -0
  20. VectorEstimator.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  21. VectorEstimator.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  22. VectorEstimator.mlpackage/Manifest.json +18 -0
  23. Vocoder.mlmodelc/analytics/coremldata.bin +3 -0
  24. Vocoder.mlmodelc/coremldata.bin +3 -0
  25. Vocoder.mlmodelc/model.mil +465 -0
  26. Vocoder.mlmodelc/weights/weight.bin +3 -0
  27. Vocoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  28. Vocoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  29. Vocoder.mlpackage/Manifest.json +18 -0
  30. manifest.json +250 -0
  31. tts.json +311 -0
  32. unicode_indexer.json +0 -0
  33. voice_styles/M1.json +0 -0
DurationPredictor.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b836b72b388c2bc009a13bcc495cd08278a1742b1a866cdaf7faab7d80786bcb
3
+ size 243
DurationPredictor.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ee71b864377bd54479f6b07d141ed37c9fc49718aba6dba21c56cc5cb975be0
3
+ size 425
DurationPredictor.mlmodelc/model.mil ADDED
@@ -0,0 +1,616 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ program(1.3)
2
+ [buildInfo = dict<string, string>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.5.1"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
3
+ {
4
+ func main<ios18>(tensor<fp32, [1, 8, 16]> style_dp, tensor<int32, [1, 128]> text_ids, tensor<fp32, [1, 1, 128]> text_mask) {
5
+ int32 var_25 = const()[name = string("op_25"), val = int32(2)];
6
+ int32 var_26 = const()[name = string("op_26"), val = int32(1)];
7
+ int32 var_27 = const()[name = string("op_27"), val = int32(-1)];
8
+ int32 x_1_batch_dims_0 = const()[name = string("x_1_batch_dims_0"), val = int32(0)];
9
+ bool x_1_validate_indices_0 = const()[name = string("x_1_validate_indices_0"), val = bool(false)];
10
+ tensor<fp16, [8322, 64]> m_char_embedder_weight_to_fp16 = const()[name = string("m_char_embedder_weight_to_fp16"), val = tensor<fp16, [8322, 64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
11
+ string text_ids_to_int16_dtype_0 = const()[name = string("text_ids_to_int16_dtype_0"), val = string("int16")];
12
+ string cast_19_dtype_0 = const()[name = string("cast_19_dtype_0"), val = string("int32")];
13
+ int32 greater_equal_0_y_0 = const()[name = string("greater_equal_0_y_0"), val = int32(0)];
14
+ tensor<int16, [1, 128]> text_ids_to_int16 = cast(dtype = text_ids_to_int16_dtype_0, x = text_ids)[name = string("cast_25")];
15
+ tensor<int32, [1, 128]> cast_19 = cast(dtype = cast_19_dtype_0, x = text_ids_to_int16)[name = string("cast_24")];
16
+ tensor<bool, [1, 128]> greater_equal_0 = greater_equal(x = cast_19, y = greater_equal_0_y_0)[name = string("greater_equal_0")];
17
+ int32 slice_by_index_0 = const()[name = string("slice_by_index_0"), val = int32(8322)];
18
+ tensor<int32, [1, 128]> add_0 = add(x = cast_19, y = slice_by_index_0)[name = string("add_0")];
19
+ tensor<int32, [1, 128]> select_0 = select(a = cast_19, b = add_0, cond = greater_equal_0)[name = string("select_0")];
20
+ int32 x_1_cast_fp16_cast_uint16_axis_0 = const()[name = string("x_1_cast_fp16_cast_uint16_axis_0"), val = int32(0)];
21
+ string select_0_to_int16_dtype_0 = const()[name = string("select_0_to_int16_dtype_0"), val = string("int16")];
22
+ tensor<int16, [1, 128]> select_0_to_int16 = cast(dtype = select_0_to_int16_dtype_0, x = select_0)[name = string("cast_23")];
23
+ tensor<fp16, [1, 128, 64]> x_1_cast_fp16_cast_uint16_cast_uint16 = gather(axis = x_1_cast_fp16_cast_uint16_axis_0, batch_dims = x_1_batch_dims_0, indices = select_0_to_int16, validate_indices = x_1_validate_indices_0, x = m_char_embedder_weight_to_fp16)[name = string("x_1_cast_fp16_cast_uint16_cast_uint16")];
24
+ tensor<int32, [3]> var_53_perm_0 = const()[name = string("op_53_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
25
+ string text_mask_to_fp16_dtype_0 = const()[name = string("text_mask_to_fp16_dtype_0"), val = string("fp16")];
26
+ tensor<fp16, [1, 1, 128]> text_mask_to_fp16 = cast(dtype = text_mask_to_fp16_dtype_0, x = text_mask)[name = string("cast_22")];
27
+ tensor<fp16, [1, 64, 128]> var_53_cast_fp16 = transpose(perm = var_53_perm_0, x = x_1_cast_fp16_cast_uint16_cast_uint16)[name = string("transpose_24")];
28
+ tensor<fp16, [1, 64, 128]> x_3_cast_fp16 = mul(x = var_53_cast_fp16, y = text_mask_to_fp16)[name = string("x_3_cast_fp16")];
29
+ bool x_5_interleave_0 = const()[name = string("x_5_interleave_0"), val = bool(false)];
30
+ tensor<fp16, [1, 64, 1]> m_sentence_token_to_fp16 = const()[name = string("m_sentence_token_to_fp16"), val = tensor<fp16, [1, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1065344)))];
31
+ tensor<fp16, [1, 64, 129]> x_5_cast_fp16 = concat(axis = var_25, interleave = x_5_interleave_0, values = (m_sentence_token_to_fp16, x_3_cast_fp16))[name = string("x_5_cast_fp16")];
32
+ bool mask_interleave_0 = const()[name = string("mask_interleave_0"), val = bool(false)];
33
+ tensor<fp16, [1, 1, 1]> fill_0_to_fp16 = const()[name = string("fill_0_to_fp16"), val = tensor<fp16, [1, 1, 1]>([[[0x1p+0]]])];
34
+ tensor<fp16, [1, 1, 129]> mask_cast_fp16 = concat(axis = var_25, interleave = mask_interleave_0, values = (fill_0_to_fp16, text_mask_to_fp16))[name = string("mask_cast_fp16")];
35
+ tensor<fp16, [1, 64, 129]> input_1_cast_fp16 = mul(x = x_5_cast_fp16, y = mask_cast_fp16)[name = string("input_1_cast_fp16")];
36
+ tensor<int32, [6]> input_3_pad_0 = const()[name = string("input_3_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 2, 2])];
37
+ string input_3_mode_0 = const()[name = string("input_3_mode_0"), val = string("replicate")];
38
+ fp16 const_1_to_fp16 = const()[name = string("const_1_to_fp16"), val = fp16(0x0p+0)];
39
+ tensor<fp16, [1, 64, 133]> input_3_cast_fp16 = pad(constant_val = const_1_to_fp16, mode = input_3_mode_0, pad = input_3_pad_0, x = input_1_cast_fp16)[name = string("input_3_cast_fp16")];
40
+ string h_1_pad_type_0 = const()[name = string("h_1_pad_type_0"), val = string("valid")];
41
+ int32 h_1_groups_0 = const()[name = string("h_1_groups_0"), val = int32(64)];
42
+ tensor<int32, [1]> h_1_strides_0 = const()[name = string("h_1_strides_0"), val = tensor<int32, [1]>([1])];
43
+ tensor<int32, [2]> h_1_pad_0 = const()[name = string("h_1_pad_0"), val = tensor<int32, [2]>([0, 0])];
44
+ tensor<int32, [1]> h_1_dilations_0 = const()[name = string("h_1_dilations_0"), val = tensor<int32, [1]>([1])];
45
+ tensor<fp16, [64, 1, 5]> m_convnext_0_dwconv__conv_weight_to_fp16 = const()[name = string("m_convnext_0_dwconv__conv_weight_to_fp16"), val = tensor<fp16, [64, 1, 5]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1065536)))];
46
+ tensor<fp16, [64]> m_convnext_0_dwconv__conv_bias_to_fp16 = const()[name = string("m_convnext_0_dwconv__conv_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1066240)))];
47
+ tensor<fp16, [1, 64, 129]> h_1_cast_fp16 = conv(bias = m_convnext_0_dwconv__conv_bias_to_fp16, dilations = h_1_dilations_0, groups = h_1_groups_0, pad = h_1_pad_0, pad_type = h_1_pad_type_0, strides = h_1_strides_0, weight = m_convnext_0_dwconv__conv_weight_to_fp16, x = input_3_cast_fp16)[name = string("h_1_cast_fp16")];
48
+ tensor<fp16, [1, 64, 129]> x_7_cast_fp16 = mul(x = h_1_cast_fp16, y = mask_cast_fp16)[name = string("x_7_cast_fp16")];
49
+ tensor<int32, [3]> input_5_perm_0 = const()[name = string("input_5_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
50
+ tensor<int32, [1]> var_85_axes_0 = const()[name = string("op_85_axes_0"), val = tensor<int32, [1]>([-1])];
51
+ tensor<fp16, [64]> m_convnext_0_norm_norm_weight_to_fp16 = const()[name = string("m_convnext_0_norm_norm_weight_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1066432)))];
52
+ tensor<fp16, [64]> m_convnext_0_norm_norm_bias_to_fp16 = const()[name = string("m_convnext_0_norm_norm_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1066624)))];
53
+ fp16 var_20_to_fp16 = const()[name = string("op_20_to_fp16"), val = fp16(0x1.5p-17)];
54
+ tensor<fp16, [1, 129, 64]> input_5_cast_fp16 = transpose(perm = input_5_perm_0, x = x_7_cast_fp16)[name = string("transpose_23")];
55
+ tensor<fp16, [1, 129, 64]> var_85_cast_fp16 = layer_norm(axes = var_85_axes_0, beta = m_convnext_0_norm_norm_bias_to_fp16, epsilon = var_20_to_fp16, gamma = m_convnext_0_norm_norm_weight_to_fp16, x = input_5_cast_fp16)[name = string("op_85_cast_fp16")];
56
+ tensor<int32, [3]> input_7_perm_0 = const()[name = string("input_7_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
57
+ string h_3_pad_type_0 = const()[name = string("h_3_pad_type_0"), val = string("valid")];
58
+ tensor<int32, [1]> h_3_strides_0 = const()[name = string("h_3_strides_0"), val = tensor<int32, [1]>([1])];
59
+ tensor<int32, [2]> h_3_pad_0 = const()[name = string("h_3_pad_0"), val = tensor<int32, [2]>([0, 0])];
60
+ tensor<int32, [1]> h_3_dilations_0 = const()[name = string("h_3_dilations_0"), val = tensor<int32, [1]>([1])];
61
+ int32 h_3_groups_0 = const()[name = string("h_3_groups_0"), val = int32(1)];
62
+ tensor<fp16, [256, 64, 1]> m_convnext_0_pwconv1_weight_to_fp16 = const()[name = string("m_convnext_0_pwconv1_weight_to_fp16"), val = tensor<fp16, [256, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1066816)))];
63
+ tensor<fp16, [256]> m_convnext_0_pwconv1_bias_to_fp16 = const()[name = string("m_convnext_0_pwconv1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1099648)))];
64
+ tensor<fp16, [1, 64, 129]> input_7_cast_fp16 = transpose(perm = input_7_perm_0, x = var_85_cast_fp16)[name = string("transpose_22")];
65
+ tensor<fp16, [1, 256, 129]> h_3_cast_fp16 = conv(bias = m_convnext_0_pwconv1_bias_to_fp16, dilations = h_3_dilations_0, groups = h_3_groups_0, pad = h_3_pad_0, pad_type = h_3_pad_type_0, strides = h_3_strides_0, weight = m_convnext_0_pwconv1_weight_to_fp16, x = input_7_cast_fp16)[name = string("h_3_cast_fp16")];
66
+ string input_9_mode_0 = const()[name = string("input_9_mode_0"), val = string("EXACT")];
67
+ tensor<fp16, [1, 256, 129]> input_9_cast_fp16 = gelu(mode = input_9_mode_0, x = h_3_cast_fp16)[name = string("input_9_cast_fp16")];
68
+ string h_5_pad_type_0 = const()[name = string("h_5_pad_type_0"), val = string("valid")];
69
+ tensor<int32, [1]> h_5_strides_0 = const()[name = string("h_5_strides_0"), val = tensor<int32, [1]>([1])];
70
+ tensor<int32, [2]> h_5_pad_0 = const()[name = string("h_5_pad_0"), val = tensor<int32, [2]>([0, 0])];
71
+ tensor<int32, [1]> h_5_dilations_0 = const()[name = string("h_5_dilations_0"), val = tensor<int32, [1]>([1])];
72
+ int32 h_5_groups_0 = const()[name = string("h_5_groups_0"), val = int32(1)];
73
+ tensor<fp16, [64, 256, 1]> var_102_weight_0_to_fp16 = const()[name = string("op_102_weight_0_to_fp16"), val = tensor<fp16, [64, 256, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1100224)))];
74
+ tensor<fp16, [64]> var_102_bias_0_to_fp16 = const()[name = string("op_102_bias_0_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1133056)))];
75
+ tensor<fp16, [1, 64, 129]> var_102_cast_fp16 = conv(bias = var_102_bias_0_to_fp16, dilations = h_5_dilations_0, groups = h_5_groups_0, pad = h_5_pad_0, pad_type = h_5_pad_type_0, strides = h_5_strides_0, weight = var_102_weight_0_to_fp16, x = input_9_cast_fp16)[name = string("op_102_cast_fp16")];
76
+ tensor<fp16, [1, 64, 129]> out_1_cast_fp16 = add(x = input_1_cast_fp16, y = var_102_cast_fp16)[name = string("out_1_cast_fp16")];
77
+ tensor<fp16, [1, 64, 129]> x_9_cast_fp16 = mul(x = out_1_cast_fp16, y = mask_cast_fp16)[name = string("x_9_cast_fp16")];
78
+ tensor<fp16, [1, 64, 129]> input_11_cast_fp16 = mul(x = x_9_cast_fp16, y = mask_cast_fp16)[name = string("input_11_cast_fp16")];
79
+ tensor<int32, [6]> input_13_pad_0 = const()[name = string("input_13_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 2, 2])];
80
+ string input_13_mode_0 = const()[name = string("input_13_mode_0"), val = string("replicate")];
81
+ fp16 const_2_to_fp16 = const()[name = string("const_2_to_fp16"), val = fp16(0x0p+0)];
82
+ tensor<fp16, [1, 64, 133]> input_13_cast_fp16 = pad(constant_val = const_2_to_fp16, mode = input_13_mode_0, pad = input_13_pad_0, x = input_11_cast_fp16)[name = string("input_13_cast_fp16")];
83
+ string h_7_pad_type_0 = const()[name = string("h_7_pad_type_0"), val = string("valid")];
84
+ int32 h_7_groups_0 = const()[name = string("h_7_groups_0"), val = int32(64)];
85
+ tensor<int32, [1]> h_7_strides_0 = const()[name = string("h_7_strides_0"), val = tensor<int32, [1]>([1])];
86
+ tensor<int32, [2]> h_7_pad_0 = const()[name = string("h_7_pad_0"), val = tensor<int32, [2]>([0, 0])];
87
+ tensor<int32, [1]> h_7_dilations_0 = const()[name = string("h_7_dilations_0"), val = tensor<int32, [1]>([1])];
88
+ tensor<fp16, [64, 1, 5]> m_convnext_1_dwconv__conv_weight_to_fp16 = const()[name = string("m_convnext_1_dwconv__conv_weight_to_fp16"), val = tensor<fp16, [64, 1, 5]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1133248)))];
89
+ tensor<fp16, [64]> m_convnext_1_dwconv__conv_bias_to_fp16 = const()[name = string("m_convnext_1_dwconv__conv_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1133952)))];
90
+ tensor<fp16, [1, 64, 129]> h_7_cast_fp16 = conv(bias = m_convnext_1_dwconv__conv_bias_to_fp16, dilations = h_7_dilations_0, groups = h_7_groups_0, pad = h_7_pad_0, pad_type = h_7_pad_type_0, strides = h_7_strides_0, weight = m_convnext_1_dwconv__conv_weight_to_fp16, x = input_13_cast_fp16)[name = string("h_7_cast_fp16")];
91
+ tensor<fp16, [1, 64, 129]> x_11_cast_fp16 = mul(x = h_7_cast_fp16, y = mask_cast_fp16)[name = string("x_11_cast_fp16")];
92
+ tensor<int32, [3]> input_15_perm_0 = const()[name = string("input_15_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
93
+ tensor<int32, [1]> var_127_axes_0 = const()[name = string("op_127_axes_0"), val = tensor<int32, [1]>([-1])];
94
+ tensor<fp16, [64]> m_convnext_1_norm_norm_weight_to_fp16 = const()[name = string("m_convnext_1_norm_norm_weight_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1134144)))];
95
+ tensor<fp16, [64]> m_convnext_1_norm_norm_bias_to_fp16 = const()[name = string("m_convnext_1_norm_norm_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1134336)))];
96
+ tensor<fp16, [1, 129, 64]> input_15_cast_fp16 = transpose(perm = input_15_perm_0, x = x_11_cast_fp16)[name = string("transpose_21")];
97
+ tensor<fp16, [1, 129, 64]> var_127_cast_fp16 = layer_norm(axes = var_127_axes_0, beta = m_convnext_1_norm_norm_bias_to_fp16, epsilon = var_20_to_fp16, gamma = m_convnext_1_norm_norm_weight_to_fp16, x = input_15_cast_fp16)[name = string("op_127_cast_fp16")];
98
+ tensor<int32, [3]> input_17_perm_0 = const()[name = string("input_17_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
99
+ string h_9_pad_type_0 = const()[name = string("h_9_pad_type_0"), val = string("valid")];
100
+ tensor<int32, [1]> h_9_strides_0 = const()[name = string("h_9_strides_0"), val = tensor<int32, [1]>([1])];
101
+ tensor<int32, [2]> h_9_pad_0 = const()[name = string("h_9_pad_0"), val = tensor<int32, [2]>([0, 0])];
102
+ tensor<int32, [1]> h_9_dilations_0 = const()[name = string("h_9_dilations_0"), val = tensor<int32, [1]>([1])];
103
+ int32 h_9_groups_0 = const()[name = string("h_9_groups_0"), val = int32(1)];
104
+ tensor<fp16, [256, 64, 1]> m_convnext_1_pwconv1_weight_to_fp16 = const()[name = string("m_convnext_1_pwconv1_weight_to_fp16"), val = tensor<fp16, [256, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1134528)))];
105
+ tensor<fp16, [256]> m_convnext_1_pwconv1_bias_to_fp16 = const()[name = string("m_convnext_1_pwconv1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1167360)))];
106
+ tensor<fp16, [1, 64, 129]> input_17_cast_fp16 = transpose(perm = input_17_perm_0, x = var_127_cast_fp16)[name = string("transpose_20")];
107
+ tensor<fp16, [1, 256, 129]> h_9_cast_fp16 = conv(bias = m_convnext_1_pwconv1_bias_to_fp16, dilations = h_9_dilations_0, groups = h_9_groups_0, pad = h_9_pad_0, pad_type = h_9_pad_type_0, strides = h_9_strides_0, weight = m_convnext_1_pwconv1_weight_to_fp16, x = input_17_cast_fp16)[name = string("h_9_cast_fp16")];
108
+ string input_19_mode_0 = const()[name = string("input_19_mode_0"), val = string("EXACT")];
109
+ tensor<fp16, [1, 256, 129]> input_19_cast_fp16 = gelu(mode = input_19_mode_0, x = h_9_cast_fp16)[name = string("input_19_cast_fp16")];
110
+ string h_11_pad_type_0 = const()[name = string("h_11_pad_type_0"), val = string("valid")];
111
+ tensor<int32, [1]> h_11_strides_0 = const()[name = string("h_11_strides_0"), val = tensor<int32, [1]>([1])];
112
+ tensor<int32, [2]> h_11_pad_0 = const()[name = string("h_11_pad_0"), val = tensor<int32, [2]>([0, 0])];
113
+ tensor<int32, [1]> h_11_dilations_0 = const()[name = string("h_11_dilations_0"), val = tensor<int32, [1]>([1])];
114
+ int32 h_11_groups_0 = const()[name = string("h_11_groups_0"), val = int32(1)];
115
+ tensor<fp16, [64, 256, 1]> var_144_weight_0_to_fp16 = const()[name = string("op_144_weight_0_to_fp16"), val = tensor<fp16, [64, 256, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1167936)))];
116
+ tensor<fp16, [64]> var_144_bias_0_to_fp16 = const()[name = string("op_144_bias_0_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1200768)))];
117
+ tensor<fp16, [1, 64, 129]> var_144_cast_fp16 = conv(bias = var_144_bias_0_to_fp16, dilations = h_11_dilations_0, groups = h_11_groups_0, pad = h_11_pad_0, pad_type = h_11_pad_type_0, strides = h_11_strides_0, weight = var_144_weight_0_to_fp16, x = input_19_cast_fp16)[name = string("op_144_cast_fp16")];
118
+ tensor<fp16, [1, 64, 129]> out_3_cast_fp16 = add(x = input_11_cast_fp16, y = var_144_cast_fp16)[name = string("out_3_cast_fp16")];
119
+ tensor<fp16, [1, 64, 129]> x_13_cast_fp16 = mul(x = out_3_cast_fp16, y = mask_cast_fp16)[name = string("x_13_cast_fp16")];
120
+ tensor<fp16, [1, 64, 129]> input_21_cast_fp16 = mul(x = x_13_cast_fp16, y = mask_cast_fp16)[name = string("input_21_cast_fp16")];
121
+ tensor<int32, [6]> input_23_pad_0 = const()[name = string("input_23_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 2, 2])];
122
+ string input_23_mode_0 = const()[name = string("input_23_mode_0"), val = string("replicate")];
123
+ fp16 const_3_to_fp16 = const()[name = string("const_3_to_fp16"), val = fp16(0x0p+0)];
124
+ tensor<fp16, [1, 64, 133]> input_23_cast_fp16 = pad(constant_val = const_3_to_fp16, mode = input_23_mode_0, pad = input_23_pad_0, x = input_21_cast_fp16)[name = string("input_23_cast_fp16")];
125
+ string h_13_pad_type_0 = const()[name = string("h_13_pad_type_0"), val = string("valid")];
126
+ int32 h_13_groups_0 = const()[name = string("h_13_groups_0"), val = int32(64)];
127
+ tensor<int32, [1]> h_13_strides_0 = const()[name = string("h_13_strides_0"), val = tensor<int32, [1]>([1])];
128
+ tensor<int32, [2]> h_13_pad_0 = const()[name = string("h_13_pad_0"), val = tensor<int32, [2]>([0, 0])];
129
+ tensor<int32, [1]> h_13_dilations_0 = const()[name = string("h_13_dilations_0"), val = tensor<int32, [1]>([1])];
130
+ tensor<fp16, [64, 1, 5]> m_convnext_2_dwconv__conv_weight_to_fp16 = const()[name = string("m_convnext_2_dwconv__conv_weight_to_fp16"), val = tensor<fp16, [64, 1, 5]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1200960)))];
131
+ tensor<fp16, [64]> m_convnext_2_dwconv__conv_bias_to_fp16 = const()[name = string("m_convnext_2_dwconv__conv_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1201664)))];
132
+ tensor<fp16, [1, 64, 129]> h_13_cast_fp16 = conv(bias = m_convnext_2_dwconv__conv_bias_to_fp16, dilations = h_13_dilations_0, groups = h_13_groups_0, pad = h_13_pad_0, pad_type = h_13_pad_type_0, strides = h_13_strides_0, weight = m_convnext_2_dwconv__conv_weight_to_fp16, x = input_23_cast_fp16)[name = string("h_13_cast_fp16")];
133
+ tensor<fp16, [1, 64, 129]> x_15_cast_fp16 = mul(x = h_13_cast_fp16, y = mask_cast_fp16)[name = string("x_15_cast_fp16")];
134
+ tensor<int32, [3]> input_25_perm_0 = const()[name = string("input_25_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
135
+ tensor<int32, [1]> var_169_axes_0 = const()[name = string("op_169_axes_0"), val = tensor<int32, [1]>([-1])];
136
+ tensor<fp16, [64]> m_convnext_2_norm_norm_weight_to_fp16 = const()[name = string("m_convnext_2_norm_norm_weight_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1201856)))];
137
+ tensor<fp16, [64]> m_convnext_2_norm_norm_bias_to_fp16 = const()[name = string("m_convnext_2_norm_norm_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1202048)))];
138
+ tensor<fp16, [1, 129, 64]> input_25_cast_fp16 = transpose(perm = input_25_perm_0, x = x_15_cast_fp16)[name = string("transpose_19")];
139
+ tensor<fp16, [1, 129, 64]> var_169_cast_fp16 = layer_norm(axes = var_169_axes_0, beta = m_convnext_2_norm_norm_bias_to_fp16, epsilon = var_20_to_fp16, gamma = m_convnext_2_norm_norm_weight_to_fp16, x = input_25_cast_fp16)[name = string("op_169_cast_fp16")];
140
+ tensor<int32, [3]> input_27_perm_0 = const()[name = string("input_27_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
141
+ string h_15_pad_type_0 = const()[name = string("h_15_pad_type_0"), val = string("valid")];
142
+ tensor<int32, [1]> h_15_strides_0 = const()[name = string("h_15_strides_0"), val = tensor<int32, [1]>([1])];
143
+ tensor<int32, [2]> h_15_pad_0 = const()[name = string("h_15_pad_0"), val = tensor<int32, [2]>([0, 0])];
144
+ tensor<int32, [1]> h_15_dilations_0 = const()[name = string("h_15_dilations_0"), val = tensor<int32, [1]>([1])];
145
+ int32 h_15_groups_0 = const()[name = string("h_15_groups_0"), val = int32(1)];
146
+ tensor<fp16, [256, 64, 1]> m_convnext_2_pwconv1_weight_to_fp16 = const()[name = string("m_convnext_2_pwconv1_weight_to_fp16"), val = tensor<fp16, [256, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1202240)))];
147
+ tensor<fp16, [256]> m_convnext_2_pwconv1_bias_to_fp16 = const()[name = string("m_convnext_2_pwconv1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1235072)))];
148
+ tensor<fp16, [1, 64, 129]> input_27_cast_fp16 = transpose(perm = input_27_perm_0, x = var_169_cast_fp16)[name = string("transpose_18")];
149
+ tensor<fp16, [1, 256, 129]> h_15_cast_fp16 = conv(bias = m_convnext_2_pwconv1_bias_to_fp16, dilations = h_15_dilations_0, groups = h_15_groups_0, pad = h_15_pad_0, pad_type = h_15_pad_type_0, strides = h_15_strides_0, weight = m_convnext_2_pwconv1_weight_to_fp16, x = input_27_cast_fp16)[name = string("h_15_cast_fp16")];
150
+ string input_29_mode_0 = const()[name = string("input_29_mode_0"), val = string("EXACT")];
151
+ tensor<fp16, [1, 256, 129]> input_29_cast_fp16 = gelu(mode = input_29_mode_0, x = h_15_cast_fp16)[name = string("input_29_cast_fp16")];
152
+ string h_17_pad_type_0 = const()[name = string("h_17_pad_type_0"), val = string("valid")];
153
+ tensor<int32, [1]> h_17_strides_0 = const()[name = string("h_17_strides_0"), val = tensor<int32, [1]>([1])];
154
+ tensor<int32, [2]> h_17_pad_0 = const()[name = string("h_17_pad_0"), val = tensor<int32, [2]>([0, 0])];
155
+ tensor<int32, [1]> h_17_dilations_0 = const()[name = string("h_17_dilations_0"), val = tensor<int32, [1]>([1])];
156
+ int32 h_17_groups_0 = const()[name = string("h_17_groups_0"), val = int32(1)];
157
+ tensor<fp16, [64, 256, 1]> var_186_weight_0_to_fp16 = const()[name = string("op_186_weight_0_to_fp16"), val = tensor<fp16, [64, 256, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1235648)))];
158
+ tensor<fp16, [64]> var_186_bias_0_to_fp16 = const()[name = string("op_186_bias_0_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1268480)))];
159
+ tensor<fp16, [1, 64, 129]> var_186_cast_fp16 = conv(bias = var_186_bias_0_to_fp16, dilations = h_17_dilations_0, groups = h_17_groups_0, pad = h_17_pad_0, pad_type = h_17_pad_type_0, strides = h_17_strides_0, weight = var_186_weight_0_to_fp16, x = input_29_cast_fp16)[name = string("op_186_cast_fp16")];
160
+ tensor<fp16, [1, 64, 129]> out_5_cast_fp16 = add(x = input_21_cast_fp16, y = var_186_cast_fp16)[name = string("out_5_cast_fp16")];
161
+ tensor<fp16, [1, 64, 129]> x_17_cast_fp16 = mul(x = out_5_cast_fp16, y = mask_cast_fp16)[name = string("x_17_cast_fp16")];
162
+ tensor<fp16, [1, 64, 129]> input_31_cast_fp16 = mul(x = x_17_cast_fp16, y = mask_cast_fp16)[name = string("input_31_cast_fp16")];
163
+ tensor<int32, [6]> input_33_pad_0 = const()[name = string("input_33_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 2, 2])];
164
+ string input_33_mode_0 = const()[name = string("input_33_mode_0"), val = string("replicate")];
165
+ fp16 const_4_to_fp16 = const()[name = string("const_4_to_fp16"), val = fp16(0x0p+0)];
166
+ tensor<fp16, [1, 64, 133]> input_33_cast_fp16 = pad(constant_val = const_4_to_fp16, mode = input_33_mode_0, pad = input_33_pad_0, x = input_31_cast_fp16)[name = string("input_33_cast_fp16")];
167
+ string h_19_pad_type_0 = const()[name = string("h_19_pad_type_0"), val = string("valid")];
168
+ int32 h_19_groups_0 = const()[name = string("h_19_groups_0"), val = int32(64)];
169
+ tensor<int32, [1]> h_19_strides_0 = const()[name = string("h_19_strides_0"), val = tensor<int32, [1]>([1])];
170
+ tensor<int32, [2]> h_19_pad_0 = const()[name = string("h_19_pad_0"), val = tensor<int32, [2]>([0, 0])];
171
+ tensor<int32, [1]> h_19_dilations_0 = const()[name = string("h_19_dilations_0"), val = tensor<int32, [1]>([1])];
172
+ tensor<fp16, [64, 1, 5]> m_convnext_3_dwconv__conv_weight_to_fp16 = const()[name = string("m_convnext_3_dwconv__conv_weight_to_fp16"), val = tensor<fp16, [64, 1, 5]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1268672)))];
173
+ tensor<fp16, [64]> m_convnext_3_dwconv__conv_bias_to_fp16 = const()[name = string("m_convnext_3_dwconv__conv_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1269376)))];
174
+ tensor<fp16, [1, 64, 129]> h_19_cast_fp16 = conv(bias = m_convnext_3_dwconv__conv_bias_to_fp16, dilations = h_19_dilations_0, groups = h_19_groups_0, pad = h_19_pad_0, pad_type = h_19_pad_type_0, strides = h_19_strides_0, weight = m_convnext_3_dwconv__conv_weight_to_fp16, x = input_33_cast_fp16)[name = string("h_19_cast_fp16")];
175
+ tensor<fp16, [1, 64, 129]> x_19_cast_fp16 = mul(x = h_19_cast_fp16, y = mask_cast_fp16)[name = string("x_19_cast_fp16")];
176
+ tensor<int32, [3]> input_35_perm_0 = const()[name = string("input_35_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
177
+ tensor<int32, [1]> var_211_axes_0 = const()[name = string("op_211_axes_0"), val = tensor<int32, [1]>([-1])];
178
+ tensor<fp16, [64]> m_convnext_3_norm_norm_weight_to_fp16 = const()[name = string("m_convnext_3_norm_norm_weight_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1269568)))];
179
+ tensor<fp16, [64]> m_convnext_3_norm_norm_bias_to_fp16 = const()[name = string("m_convnext_3_norm_norm_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1269760)))];
180
+ tensor<fp16, [1, 129, 64]> input_35_cast_fp16 = transpose(perm = input_35_perm_0, x = x_19_cast_fp16)[name = string("transpose_17")];
181
+ tensor<fp16, [1, 129, 64]> var_211_cast_fp16 = layer_norm(axes = var_211_axes_0, beta = m_convnext_3_norm_norm_bias_to_fp16, epsilon = var_20_to_fp16, gamma = m_convnext_3_norm_norm_weight_to_fp16, x = input_35_cast_fp16)[name = string("op_211_cast_fp16")];
182
+ tensor<int32, [3]> input_37_perm_0 = const()[name = string("input_37_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
183
+ string h_21_pad_type_0 = const()[name = string("h_21_pad_type_0"), val = string("valid")];
184
+ tensor<int32, [1]> h_21_strides_0 = const()[name = string("h_21_strides_0"), val = tensor<int32, [1]>([1])];
185
+ tensor<int32, [2]> h_21_pad_0 = const()[name = string("h_21_pad_0"), val = tensor<int32, [2]>([0, 0])];
186
+ tensor<int32, [1]> h_21_dilations_0 = const()[name = string("h_21_dilations_0"), val = tensor<int32, [1]>([1])];
187
+ int32 h_21_groups_0 = const()[name = string("h_21_groups_0"), val = int32(1)];
188
+ tensor<fp16, [256, 64, 1]> m_convnext_3_pwconv1_weight_to_fp16 = const()[name = string("m_convnext_3_pwconv1_weight_to_fp16"), val = tensor<fp16, [256, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1269952)))];
189
+ tensor<fp16, [256]> m_convnext_3_pwconv1_bias_to_fp16 = const()[name = string("m_convnext_3_pwconv1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1302784)))];
190
+ tensor<fp16, [1, 64, 129]> input_37_cast_fp16 = transpose(perm = input_37_perm_0, x = var_211_cast_fp16)[name = string("transpose_16")];
191
+ tensor<fp16, [1, 256, 129]> h_21_cast_fp16 = conv(bias = m_convnext_3_pwconv1_bias_to_fp16, dilations = h_21_dilations_0, groups = h_21_groups_0, pad = h_21_pad_0, pad_type = h_21_pad_type_0, strides = h_21_strides_0, weight = m_convnext_3_pwconv1_weight_to_fp16, x = input_37_cast_fp16)[name = string("h_21_cast_fp16")];
192
+ string input_39_mode_0 = const()[name = string("input_39_mode_0"), val = string("EXACT")];
193
+ tensor<fp16, [1, 256, 129]> input_39_cast_fp16 = gelu(mode = input_39_mode_0, x = h_21_cast_fp16)[name = string("input_39_cast_fp16")];
194
+ string h_23_pad_type_0 = const()[name = string("h_23_pad_type_0"), val = string("valid")];
195
+ tensor<int32, [1]> h_23_strides_0 = const()[name = string("h_23_strides_0"), val = tensor<int32, [1]>([1])];
196
+ tensor<int32, [2]> h_23_pad_0 = const()[name = string("h_23_pad_0"), val = tensor<int32, [2]>([0, 0])];
197
+ tensor<int32, [1]> h_23_dilations_0 = const()[name = string("h_23_dilations_0"), val = tensor<int32, [1]>([1])];
198
+ int32 h_23_groups_0 = const()[name = string("h_23_groups_0"), val = int32(1)];
199
+ tensor<fp16, [64, 256, 1]> var_228_weight_0_to_fp16 = const()[name = string("op_228_weight_0_to_fp16"), val = tensor<fp16, [64, 256, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1303360)))];
200
+ tensor<fp16, [64]> var_228_bias_0_to_fp16 = const()[name = string("op_228_bias_0_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1336192)))];
201
+ tensor<fp16, [1, 64, 129]> var_228_cast_fp16 = conv(bias = var_228_bias_0_to_fp16, dilations = h_23_dilations_0, groups = h_23_groups_0, pad = h_23_pad_0, pad_type = h_23_pad_type_0, strides = h_23_strides_0, weight = var_228_weight_0_to_fp16, x = input_39_cast_fp16)[name = string("op_228_cast_fp16")];
202
+ tensor<fp16, [1, 64, 129]> out_7_cast_fp16 = add(x = input_31_cast_fp16, y = var_228_cast_fp16)[name = string("out_7_cast_fp16")];
203
+ tensor<fp16, [1, 64, 129]> x_21_cast_fp16 = mul(x = out_7_cast_fp16, y = mask_cast_fp16)[name = string("x_21_cast_fp16")];
204
+ tensor<fp16, [1, 64, 129]> input_41_cast_fp16 = mul(x = x_21_cast_fp16, y = mask_cast_fp16)[name = string("input_41_cast_fp16")];
205
+ tensor<int32, [6]> input_43_pad_0 = const()[name = string("input_43_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 2, 2])];
206
+ string input_43_mode_0 = const()[name = string("input_43_mode_0"), val = string("replicate")];
207
+ fp16 const_5_to_fp16 = const()[name = string("const_5_to_fp16"), val = fp16(0x0p+0)];
208
+ tensor<fp16, [1, 64, 133]> input_43_cast_fp16 = pad(constant_val = const_5_to_fp16, mode = input_43_mode_0, pad = input_43_pad_0, x = input_41_cast_fp16)[name = string("input_43_cast_fp16")];
209
+ string h_25_pad_type_0 = const()[name = string("h_25_pad_type_0"), val = string("valid")];
210
+ int32 h_25_groups_0 = const()[name = string("h_25_groups_0"), val = int32(64)];
211
+ tensor<int32, [1]> h_25_strides_0 = const()[name = string("h_25_strides_0"), val = tensor<int32, [1]>([1])];
212
+ tensor<int32, [2]> h_25_pad_0 = const()[name = string("h_25_pad_0"), val = tensor<int32, [2]>([0, 0])];
213
+ tensor<int32, [1]> h_25_dilations_0 = const()[name = string("h_25_dilations_0"), val = tensor<int32, [1]>([1])];
214
+ tensor<fp16, [64, 1, 5]> m_convnext_4_dwconv__conv_weight_to_fp16 = const()[name = string("m_convnext_4_dwconv__conv_weight_to_fp16"), val = tensor<fp16, [64, 1, 5]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1336384)))];
215
+ tensor<fp16, [64]> m_convnext_4_dwconv__conv_bias_to_fp16 = const()[name = string("m_convnext_4_dwconv__conv_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1337088)))];
216
+ tensor<fp16, [1, 64, 129]> h_25_cast_fp16 = conv(bias = m_convnext_4_dwconv__conv_bias_to_fp16, dilations = h_25_dilations_0, groups = h_25_groups_0, pad = h_25_pad_0, pad_type = h_25_pad_type_0, strides = h_25_strides_0, weight = m_convnext_4_dwconv__conv_weight_to_fp16, x = input_43_cast_fp16)[name = string("h_25_cast_fp16")];
217
+ tensor<fp16, [1, 64, 129]> x_23_cast_fp16 = mul(x = h_25_cast_fp16, y = mask_cast_fp16)[name = string("x_23_cast_fp16")];
218
+ tensor<int32, [3]> input_45_perm_0 = const()[name = string("input_45_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
219
+ tensor<int32, [1]> var_253_axes_0 = const()[name = string("op_253_axes_0"), val = tensor<int32, [1]>([-1])];
220
+ tensor<fp16, [64]> m_convnext_4_norm_norm_weight_to_fp16 = const()[name = string("m_convnext_4_norm_norm_weight_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1337280)))];
221
+ tensor<fp16, [64]> m_convnext_4_norm_norm_bias_to_fp16 = const()[name = string("m_convnext_4_norm_norm_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1337472)))];
222
+ tensor<fp16, [1, 129, 64]> input_45_cast_fp16 = transpose(perm = input_45_perm_0, x = x_23_cast_fp16)[name = string("transpose_15")];
223
+ tensor<fp16, [1, 129, 64]> var_253_cast_fp16 = layer_norm(axes = var_253_axes_0, beta = m_convnext_4_norm_norm_bias_to_fp16, epsilon = var_20_to_fp16, gamma = m_convnext_4_norm_norm_weight_to_fp16, x = input_45_cast_fp16)[name = string("op_253_cast_fp16")];
224
+ tensor<int32, [3]> input_47_perm_0 = const()[name = string("input_47_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
225
+ string h_27_pad_type_0 = const()[name = string("h_27_pad_type_0"), val = string("valid")];
226
+ tensor<int32, [1]> h_27_strides_0 = const()[name = string("h_27_strides_0"), val = tensor<int32, [1]>([1])];
227
+ tensor<int32, [2]> h_27_pad_0 = const()[name = string("h_27_pad_0"), val = tensor<int32, [2]>([0, 0])];
228
+ tensor<int32, [1]> h_27_dilations_0 = const()[name = string("h_27_dilations_0"), val = tensor<int32, [1]>([1])];
229
+ int32 h_27_groups_0 = const()[name = string("h_27_groups_0"), val = int32(1)];
230
+ tensor<fp16, [256, 64, 1]> m_convnext_4_pwconv1_weight_to_fp16 = const()[name = string("m_convnext_4_pwconv1_weight_to_fp16"), val = tensor<fp16, [256, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1337664)))];
231
+ tensor<fp16, [256]> m_convnext_4_pwconv1_bias_to_fp16 = const()[name = string("m_convnext_4_pwconv1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1370496)))];
232
+ tensor<fp16, [1, 64, 129]> input_47_cast_fp16 = transpose(perm = input_47_perm_0, x = var_253_cast_fp16)[name = string("transpose_14")];
233
+ tensor<fp16, [1, 256, 129]> h_27_cast_fp16 = conv(bias = m_convnext_4_pwconv1_bias_to_fp16, dilations = h_27_dilations_0, groups = h_27_groups_0, pad = h_27_pad_0, pad_type = h_27_pad_type_0, strides = h_27_strides_0, weight = m_convnext_4_pwconv1_weight_to_fp16, x = input_47_cast_fp16)[name = string("h_27_cast_fp16")];
234
+ string input_49_mode_0 = const()[name = string("input_49_mode_0"), val = string("EXACT")];
235
+ tensor<fp16, [1, 256, 129]> input_49_cast_fp16 = gelu(mode = input_49_mode_0, x = h_27_cast_fp16)[name = string("input_49_cast_fp16")];
236
+ string h_29_pad_type_0 = const()[name = string("h_29_pad_type_0"), val = string("valid")];
237
+ tensor<int32, [1]> h_29_strides_0 = const()[name = string("h_29_strides_0"), val = tensor<int32, [1]>([1])];
238
+ tensor<int32, [2]> h_29_pad_0 = const()[name = string("h_29_pad_0"), val = tensor<int32, [2]>([0, 0])];
239
+ tensor<int32, [1]> h_29_dilations_0 = const()[name = string("h_29_dilations_0"), val = tensor<int32, [1]>([1])];
240
+ int32 h_29_groups_0 = const()[name = string("h_29_groups_0"), val = int32(1)];
241
+ tensor<fp16, [64, 256, 1]> var_270_weight_0_to_fp16 = const()[name = string("op_270_weight_0_to_fp16"), val = tensor<fp16, [64, 256, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1371072)))];
242
+ tensor<fp16, [64]> var_270_bias_0_to_fp16 = const()[name = string("op_270_bias_0_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1403904)))];
243
+ tensor<fp16, [1, 64, 129]> var_270_cast_fp16 = conv(bias = var_270_bias_0_to_fp16, dilations = h_29_dilations_0, groups = h_29_groups_0, pad = h_29_pad_0, pad_type = h_29_pad_type_0, strides = h_29_strides_0, weight = var_270_weight_0_to_fp16, x = input_49_cast_fp16)[name = string("op_270_cast_fp16")];
244
+ tensor<fp16, [1, 64, 129]> out_9_cast_fp16 = add(x = input_41_cast_fp16, y = var_270_cast_fp16)[name = string("out_9_cast_fp16")];
245
+ tensor<fp16, [1, 64, 129]> x_25_cast_fp16 = mul(x = out_9_cast_fp16, y = mask_cast_fp16)[name = string("x_25_cast_fp16")];
246
+ tensor<fp16, [1, 64, 129]> input_51_cast_fp16 = mul(x = x_25_cast_fp16, y = mask_cast_fp16)[name = string("input_51_cast_fp16")];
247
+ tensor<int32, [6]> input_53_pad_0 = const()[name = string("input_53_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 2, 2])];
248
+ string input_53_mode_0 = const()[name = string("input_53_mode_0"), val = string("replicate")];
249
+ fp16 const_6_to_fp16 = const()[name = string("const_6_to_fp16"), val = fp16(0x0p+0)];
250
+ tensor<fp16, [1, 64, 133]> input_53_cast_fp16 = pad(constant_val = const_6_to_fp16, mode = input_53_mode_0, pad = input_53_pad_0, x = input_51_cast_fp16)[name = string("input_53_cast_fp16")];
251
+ string h_31_pad_type_0 = const()[name = string("h_31_pad_type_0"), val = string("valid")];
252
+ int32 h_31_groups_0 = const()[name = string("h_31_groups_0"), val = int32(64)];
253
+ tensor<int32, [1]> h_31_strides_0 = const()[name = string("h_31_strides_0"), val = tensor<int32, [1]>([1])];
254
+ tensor<int32, [2]> h_31_pad_0 = const()[name = string("h_31_pad_0"), val = tensor<int32, [2]>([0, 0])];
255
+ tensor<int32, [1]> h_31_dilations_0 = const()[name = string("h_31_dilations_0"), val = tensor<int32, [1]>([1])];
256
+ tensor<fp16, [64, 1, 5]> m_convnext_5_dwconv__conv_weight_to_fp16 = const()[name = string("m_convnext_5_dwconv__conv_weight_to_fp16"), val = tensor<fp16, [64, 1, 5]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1404096)))];
257
+ tensor<fp16, [64]> m_convnext_5_dwconv__conv_bias_to_fp16 = const()[name = string("m_convnext_5_dwconv__conv_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1404800)))];
258
+ tensor<fp16, [1, 64, 129]> h_31_cast_fp16 = conv(bias = m_convnext_5_dwconv__conv_bias_to_fp16, dilations = h_31_dilations_0, groups = h_31_groups_0, pad = h_31_pad_0, pad_type = h_31_pad_type_0, strides = h_31_strides_0, weight = m_convnext_5_dwconv__conv_weight_to_fp16, x = input_53_cast_fp16)[name = string("h_31_cast_fp16")];
259
+ tensor<fp16, [1, 64, 129]> x_27_cast_fp16 = mul(x = h_31_cast_fp16, y = mask_cast_fp16)[name = string("x_27_cast_fp16")];
260
+ tensor<int32, [3]> input_55_perm_0 = const()[name = string("input_55_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
261
+ tensor<int32, [1]> var_295_axes_0 = const()[name = string("op_295_axes_0"), val = tensor<int32, [1]>([-1])];
262
+ tensor<fp16, [64]> m_convnext_5_norm_norm_weight_to_fp16 = const()[name = string("m_convnext_5_norm_norm_weight_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1404992)))];
263
+ tensor<fp16, [64]> m_convnext_5_norm_norm_bias_to_fp16 = const()[name = string("m_convnext_5_norm_norm_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1405184)))];
264
+ tensor<fp16, [1, 129, 64]> input_55_cast_fp16 = transpose(perm = input_55_perm_0, x = x_27_cast_fp16)[name = string("transpose_13")];
265
+ tensor<fp16, [1, 129, 64]> var_295_cast_fp16 = layer_norm(axes = var_295_axes_0, beta = m_convnext_5_norm_norm_bias_to_fp16, epsilon = var_20_to_fp16, gamma = m_convnext_5_norm_norm_weight_to_fp16, x = input_55_cast_fp16)[name = string("op_295_cast_fp16")];
266
+ tensor<int32, [3]> input_57_perm_0 = const()[name = string("input_57_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
267
+ string h_33_pad_type_0 = const()[name = string("h_33_pad_type_0"), val = string("valid")];
268
+ tensor<int32, [1]> h_33_strides_0 = const()[name = string("h_33_strides_0"), val = tensor<int32, [1]>([1])];
269
+ tensor<int32, [2]> h_33_pad_0 = const()[name = string("h_33_pad_0"), val = tensor<int32, [2]>([0, 0])];
270
+ tensor<int32, [1]> h_33_dilations_0 = const()[name = string("h_33_dilations_0"), val = tensor<int32, [1]>([1])];
271
+ int32 h_33_groups_0 = const()[name = string("h_33_groups_0"), val = int32(1)];
272
+ tensor<fp16, [256, 64, 1]> m_convnext_5_pwconv1_weight_to_fp16 = const()[name = string("m_convnext_5_pwconv1_weight_to_fp16"), val = tensor<fp16, [256, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1405376)))];
273
+ tensor<fp16, [256]> m_convnext_5_pwconv1_bias_to_fp16 = const()[name = string("m_convnext_5_pwconv1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1438208)))];
274
+ tensor<fp16, [1, 64, 129]> input_57_cast_fp16 = transpose(perm = input_57_perm_0, x = var_295_cast_fp16)[name = string("transpose_12")];
275
+ tensor<fp16, [1, 256, 129]> h_33_cast_fp16 = conv(bias = m_convnext_5_pwconv1_bias_to_fp16, dilations = h_33_dilations_0, groups = h_33_groups_0, pad = h_33_pad_0, pad_type = h_33_pad_type_0, strides = h_33_strides_0, weight = m_convnext_5_pwconv1_weight_to_fp16, x = input_57_cast_fp16)[name = string("h_33_cast_fp16")];
276
+ string input_59_mode_0 = const()[name = string("input_59_mode_0"), val = string("EXACT")];
277
+ tensor<fp16, [1, 256, 129]> input_59_cast_fp16 = gelu(mode = input_59_mode_0, x = h_33_cast_fp16)[name = string("input_59_cast_fp16")];
278
+ string h_35_pad_type_0 = const()[name = string("h_35_pad_type_0"), val = string("valid")];
279
+ tensor<int32, [1]> h_35_strides_0 = const()[name = string("h_35_strides_0"), val = tensor<int32, [1]>([1])];
280
+ tensor<int32, [2]> h_35_pad_0 = const()[name = string("h_35_pad_0"), val = tensor<int32, [2]>([0, 0])];
281
+ tensor<int32, [1]> h_35_dilations_0 = const()[name = string("h_35_dilations_0"), val = tensor<int32, [1]>([1])];
282
+ int32 h_35_groups_0 = const()[name = string("h_35_groups_0"), val = int32(1)];
283
+ tensor<fp16, [64, 256, 1]> var_312_weight_0_to_fp16 = const()[name = string("op_312_weight_0_to_fp16"), val = tensor<fp16, [64, 256, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1438784)))];
284
+ tensor<fp16, [64]> var_312_bias_0_to_fp16 = const()[name = string("op_312_bias_0_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1471616)))];
285
+ tensor<fp16, [1, 64, 129]> var_312_cast_fp16 = conv(bias = var_312_bias_0_to_fp16, dilations = h_35_dilations_0, groups = h_35_groups_0, pad = h_35_pad_0, pad_type = h_35_pad_type_0, strides = h_35_strides_0, weight = var_312_weight_0_to_fp16, x = input_59_cast_fp16)[name = string("op_312_cast_fp16")];
286
+ tensor<fp16, [1, 64, 129]> out_11_cast_fp16 = add(x = input_51_cast_fp16, y = var_312_cast_fp16)[name = string("out_11_cast_fp16")];
287
+ tensor<fp16, [1, 64, 129]> x_29_cast_fp16 = mul(x = out_11_cast_fp16, y = mask_cast_fp16)[name = string("x_29_cast_fp16")];
288
+ tensor<fp16, [1, 64, 129]> x_31_cast_fp16 = mul(x = x_29_cast_fp16, y = mask_cast_fp16)[name = string("x_31_cast_fp16")];
289
+ string var_335_pad_type_0 = const()[name = string("op_335_pad_type_0"), val = string("valid")];
290
+ tensor<int32, [1]> var_335_strides_0 = const()[name = string("op_335_strides_0"), val = tensor<int32, [1]>([1])];
291
+ tensor<int32, [2]> var_335_pad_0 = const()[name = string("op_335_pad_0"), val = tensor<int32, [2]>([0, 0])];
292
+ tensor<int32, [1]> var_335_dilations_0 = const()[name = string("op_335_dilations_0"), val = tensor<int32, [1]>([1])];
293
+ int32 var_335_groups_0 = const()[name = string("op_335_groups_0"), val = int32(1)];
294
+ tensor<fp16, [64, 64, 1]> m_attn_layers_0_attn_conv_q_weight_to_fp16 = const()[name = string("m_attn_layers_0_attn_conv_q_weight_to_fp16"), val = tensor<fp16, [64, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1471808)))];
295
+ tensor<fp16, [64]> m_attn_layers_0_attn_conv_q_bias_to_fp16 = const()[name = string("m_attn_layers_0_attn_conv_q_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1480064)))];
296
+ tensor<fp16, [1, 64, 129]> var_335_cast_fp16 = conv(bias = m_attn_layers_0_attn_conv_q_bias_to_fp16, dilations = var_335_dilations_0, groups = var_335_groups_0, pad = var_335_pad_0, pad_type = var_335_pad_type_0, strides = var_335_strides_0, weight = m_attn_layers_0_attn_conv_q_weight_to_fp16, x = x_31_cast_fp16)[name = string("op_335_cast_fp16")];
297
+ tensor<int32, [4]> var_336 = const()[name = string("op_336"), val = tensor<int32, [4]>([1, 2, 32, 129])];
298
+ tensor<fp16, [1, 2, 32, 129]> var_337_cast_fp16 = reshape(shape = var_336, x = var_335_cast_fp16)[name = string("op_337_cast_fp16")];
299
+ tensor<int32, [4]> q_1_perm_0 = const()[name = string("q_1_perm_0"), val = tensor<int32, [4]>([0, 1, 3, 2])];
300
+ string var_345_pad_type_0 = const()[name = string("op_345_pad_type_0"), val = string("valid")];
301
+ tensor<int32, [1]> var_345_strides_0 = const()[name = string("op_345_strides_0"), val = tensor<int32, [1]>([1])];
302
+ tensor<int32, [2]> var_345_pad_0 = const()[name = string("op_345_pad_0"), val = tensor<int32, [2]>([0, 0])];
303
+ tensor<int32, [1]> var_345_dilations_0 = const()[name = string("op_345_dilations_0"), val = tensor<int32, [1]>([1])];
304
+ int32 var_345_groups_0 = const()[name = string("op_345_groups_0"), val = int32(1)];
305
+ tensor<fp16, [64, 64, 1]> m_attn_layers_0_attn_conv_k_weight_to_fp16 = const()[name = string("m_attn_layers_0_attn_conv_k_weight_to_fp16"), val = tensor<fp16, [64, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1480256)))];
306
+ tensor<fp16, [64]> m_attn_layers_0_attn_conv_k_bias_to_fp16 = const()[name = string("m_attn_layers_0_attn_conv_k_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1488512)))];
307
+ tensor<fp16, [1, 64, 129]> var_345_cast_fp16 = conv(bias = m_attn_layers_0_attn_conv_k_bias_to_fp16, dilations = var_345_dilations_0, groups = var_345_groups_0, pad = var_345_pad_0, pad_type = var_345_pad_type_0, strides = var_345_strides_0, weight = m_attn_layers_0_attn_conv_k_weight_to_fp16, x = x_31_cast_fp16)[name = string("op_345_cast_fp16")];
308
+ tensor<int32, [4]> var_346 = const()[name = string("op_346"), val = tensor<int32, [4]>([1, 2, 32, 129])];
309
+ tensor<fp16, [1, 2, 32, 129]> var_347_cast_fp16 = reshape(shape = var_346, x = var_345_cast_fp16)[name = string("op_347_cast_fp16")];
310
+ string var_355_pad_type_0 = const()[name = string("op_355_pad_type_0"), val = string("valid")];
311
+ tensor<int32, [1]> var_355_strides_0 = const()[name = string("op_355_strides_0"), val = tensor<int32, [1]>([1])];
312
+ tensor<int32, [2]> var_355_pad_0 = const()[name = string("op_355_pad_0"), val = tensor<int32, [2]>([0, 0])];
313
+ tensor<int32, [1]> var_355_dilations_0 = const()[name = string("op_355_dilations_0"), val = tensor<int32, [1]>([1])];
314
+ int32 var_355_groups_0 = const()[name = string("op_355_groups_0"), val = int32(1)];
315
+ tensor<fp16, [64, 64, 1]> m_attn_layers_0_attn_conv_v_weight_to_fp16 = const()[name = string("m_attn_layers_0_attn_conv_v_weight_to_fp16"), val = tensor<fp16, [64, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1488704)))];
316
+ tensor<fp16, [64]> m_attn_layers_0_attn_conv_v_bias_to_fp16 = const()[name = string("m_attn_layers_0_attn_conv_v_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1496960)))];
317
+ tensor<fp16, [1, 64, 129]> var_355_cast_fp16 = conv(bias = m_attn_layers_0_attn_conv_v_bias_to_fp16, dilations = var_355_dilations_0, groups = var_355_groups_0, pad = var_355_pad_0, pad_type = var_355_pad_type_0, strides = var_355_strides_0, weight = m_attn_layers_0_attn_conv_v_weight_to_fp16, x = x_31_cast_fp16)[name = string("op_355_cast_fp16")];
318
+ tensor<int32, [4]> var_356 = const()[name = string("op_356"), val = tensor<int32, [4]>([1, 2, 32, 129])];
319
+ tensor<fp16, [1, 2, 32, 129]> var_357_cast_fp16 = reshape(shape = var_356, x = var_355_cast_fp16)[name = string("op_357_cast_fp16")];
320
+ fp16 var_359_to_fp16 = const()[name = string("op_359_to_fp16"), val = fp16(0x1.6ap-3)];
321
+ tensor<fp16, [1, 2, 129, 32]> q_1_cast_fp16 = transpose(perm = q_1_perm_0, x = var_337_cast_fp16)[name = string("transpose_11")];
322
+ tensor<fp16, [1, 2, 129, 32]> var_360_cast_fp16 = mul(x = q_1_cast_fp16, y = var_359_to_fp16)[name = string("op_360_cast_fp16")];
323
+ bool scores_1_transpose_x_0 = const()[name = string("scores_1_transpose_x_0"), val = bool(false)];
324
+ bool scores_1_transpose_y_0 = const()[name = string("scores_1_transpose_y_0"), val = bool(false)];
325
+ tensor<fp16, [1, 2, 129, 129]> scores_1_cast_fp16 = matmul(transpose_x = scores_1_transpose_x_0, transpose_y = scores_1_transpose_y_0, x = var_360_cast_fp16, y = var_347_cast_fp16)[name = string("scores_1_cast_fp16")];
326
+ bool x_33_transpose_x_0 = const()[name = string("x_33_transpose_x_0"), val = bool(false)];
327
+ bool x_33_transpose_y_0 = const()[name = string("x_33_transpose_y_0"), val = bool(false)];
328
+ tensor<fp16, [1, 1, 32, 257]> var_384_to_fp16 = const()[name = string("op_384_to_fp16"), val = tensor<fp16, [1, 1, 32, 257]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1497152)))];
329
+ tensor<fp16, [1, 2, 129, 257]> x_33_cast_fp16 = matmul(transpose_x = x_33_transpose_x_0, transpose_y = x_33_transpose_y_0, x = var_360_cast_fp16, y = var_384_to_fp16)[name = string("x_33_cast_fp16")];
330
+ tensor<int32, [8]> x_35_pad_0 = const()[name = string("x_35_pad_0"), val = tensor<int32, [8]>([0, 0, 0, 0, 0, 0, 0, 1])];
331
+ string x_35_mode_0 = const()[name = string("x_35_mode_0"), val = string("constant")];
332
+ fp16 const_13_to_fp16 = const()[name = string("const_13_to_fp16"), val = fp16(0x0p+0)];
333
+ tensor<fp16, [1, 2, 129, 258]> x_35_cast_fp16 = pad(constant_val = const_13_to_fp16, mode = x_35_mode_0, pad = x_35_pad_0, x = x_33_cast_fp16)[name = string("x_35_cast_fp16")];
334
+ tensor<int32, [3]> var_396 = const()[name = string("op_396"), val = tensor<int32, [3]>([1, 2, 33282])];
335
+ tensor<fp16, [1, 2, 33282]> input_63_cast_fp16 = reshape(shape = var_396, x = x_35_cast_fp16)[name = string("input_63_cast_fp16")];
336
+ tensor<int32, [6]> x_37_pad_0 = const()[name = string("x_37_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 0, 128])];
337
+ string x_37_mode_0 = const()[name = string("x_37_mode_0"), val = string("constant")];
338
+ fp16 const_14_to_fp16 = const()[name = string("const_14_to_fp16"), val = fp16(0x0p+0)];
339
+ tensor<fp16, [1, 2, 33410]> x_37_cast_fp16 = pad(constant_val = const_14_to_fp16, mode = x_37_mode_0, pad = x_37_pad_0, x = input_63_cast_fp16)[name = string("x_37_cast_fp16")];
340
+ tensor<int32, [4]> var_411 = const()[name = string("op_411"), val = tensor<int32, [4]>([1, 2, 130, 257])];
341
+ tensor<fp16, [1, 2, 130, 257]> x_39_cast_fp16 = reshape(shape = var_411, x = x_37_cast_fp16)[name = string("x_39_cast_fp16")];
342
+ tensor<int32, [4]> var_418_begin_0 = const()[name = string("op_418_begin_0"), val = tensor<int32, [4]>([0, 0, 0, 0])];
343
+ tensor<int32, [4]> var_418_end_0 = const()[name = string("op_418_end_0"), val = tensor<int32, [4]>([1, 2, 129, 257])];
344
+ tensor<bool, [4]> var_418_end_mask_0 = const()[name = string("op_418_end_mask_0"), val = tensor<bool, [4]>([true, true, false, true])];
345
+ tensor<fp16, [1, 2, 129, 257]> var_418_cast_fp16 = slice_by_index(begin = var_418_begin_0, end = var_418_end_0, end_mask = var_418_end_mask_0, x = x_39_cast_fp16)[name = string("op_418_cast_fp16")];
346
+ tensor<int32, [4]> var_419_begin_0 = const()[name = string("op_419_begin_0"), val = tensor<int32, [4]>([0, 0, 0, 128])];
347
+ tensor<int32, [4]> var_419_end_0 = const()[name = string("op_419_end_0"), val = tensor<int32, [4]>([1, 2, 129, 257])];
348
+ tensor<bool, [4]> var_419_end_mask_0 = const()[name = string("op_419_end_mask_0"), val = tensor<bool, [4]>([true, true, true, true])];
349
+ tensor<fp16, [1, 2, 129, 129]> var_419_cast_fp16 = slice_by_index(begin = var_419_begin_0, end = var_419_end_0, end_mask = var_419_end_mask_0, x = var_418_cast_fp16)[name = string("op_419_cast_fp16")];
350
+ tensor<fp16, [1, 2, 129, 129]> scores_3_cast_fp16 = add(x = scores_1_cast_fp16, y = var_419_cast_fp16)[name = string("scores_3_cast_fp16")];
351
+ tensor<int32, [1]> var_421_axes_0 = const()[name = string("op_421_axes_0"), val = tensor<int32, [1]>([2])];
352
+ tensor<fp16, [1, 1, 1, 129]> var_421_cast_fp16 = expand_dims(axes = var_421_axes_0, x = mask_cast_fp16)[name = string("op_421_cast_fp16")];
353
+ fp16 var_10_to_fp16 = const()[name = string("op_10_to_fp16"), val = fp16(0x1p+0)];
354
+ tensor<fp16, [1, 1, 1, 129]> var_422_cast_fp16 = sub(x = var_10_to_fp16, y = var_421_cast_fp16)[name = string("op_422_cast_fp16")];
355
+ fp16 var_423_to_fp16 = const()[name = string("op_423_to_fp16"), val = fp16(0x1.388p+13)];
356
+ tensor<fp16, [1, 1, 1, 129]> var_424_cast_fp16 = mul(x = var_422_cast_fp16, y = var_423_to_fp16)[name = string("op_424_cast_fp16")];
357
+ tensor<fp16, [1, 2, 129, 129]> input_65_cast_fp16 = sub(x = scores_3_cast_fp16, y = var_424_cast_fp16)[name = string("input_65_cast_fp16")];
358
+ tensor<fp16, [1, 2, 129, 129]> p_attn_1_cast_fp16 = softmax(axis = var_27, x = input_65_cast_fp16)[name = string("p_attn_1_cast_fp16")];
359
+ bool out_13_transpose_x_1 = const()[name = string("out_13_transpose_x_1"), val = bool(false)];
360
+ bool out_13_transpose_y_1 = const()[name = string("out_13_transpose_y_1"), val = bool(true)];
361
+ tensor<fp16, [1, 2, 129, 32]> out_13_cast_fp16 = matmul(transpose_x = out_13_transpose_x_1, transpose_y = out_13_transpose_y_1, x = p_attn_1_cast_fp16, y = var_357_cast_fp16)[name = string("out_13_cast_fp16")];
362
+ tensor<int32, [8]> x_41_pad_0 = const()[name = string("x_41_pad_0"), val = tensor<int32, [8]>([0, 0, 0, 0, 0, 0, 0, 128])];
363
+ string x_41_mode_0 = const()[name = string("x_41_mode_0"), val = string("constant")];
364
+ fp16 const_19_to_fp16 = const()[name = string("const_19_to_fp16"), val = fp16(0x0p+0)];
365
+ tensor<fp16, [1, 2, 129, 257]> x_41_cast_fp16 = pad(constant_val = const_19_to_fp16, mode = x_41_mode_0, pad = x_41_pad_0, x = p_attn_1_cast_fp16)[name = string("x_41_cast_fp16")];
366
+ tensor<int32, [3]> var_461 = const()[name = string("op_461"), val = tensor<int32, [3]>([1, 2, 33153])];
367
+ tensor<fp16, [1, 2, 33153]> input_69_cast_fp16 = reshape(shape = var_461, x = x_41_cast_fp16)[name = string("input_69_cast_fp16")];
368
+ tensor<int32, [6]> x_43_pad_0 = const()[name = string("x_43_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 129, 0])];
369
+ string x_43_mode_0 = const()[name = string("x_43_mode_0"), val = string("constant")];
370
+ fp16 const_20_to_fp16 = const()[name = string("const_20_to_fp16"), val = fp16(0x0p+0)];
371
+ tensor<fp16, [1, 2, 33282]> x_43_cast_fp16 = pad(constant_val = const_20_to_fp16, mode = x_43_mode_0, pad = x_43_pad_0, x = input_69_cast_fp16)[name = string("x_43_cast_fp16")];
372
+ tensor<int32, [4]> var_468 = const()[name = string("op_468"), val = tensor<int32, [4]>([1, 2, 129, 258])];
373
+ tensor<fp16, [1, 2, 129, 258]> x_45_cast_fp16 = reshape(shape = var_468, x = x_43_cast_fp16)[name = string("x_45_cast_fp16")];
374
+ tensor<int32, [4]> rel_weights_1_begin_0 = const()[name = string("rel_weights_1_begin_0"), val = tensor<int32, [4]>([0, 0, 0, 1])];
375
+ tensor<int32, [4]> rel_weights_1_end_0 = const()[name = string("rel_weights_1_end_0"), val = tensor<int32, [4]>([1, 2, 129, 258])];
376
+ tensor<bool, [4]> rel_weights_1_end_mask_0 = const()[name = string("rel_weights_1_end_mask_0"), val = tensor<bool, [4]>([true, true, true, true])];
377
+ tensor<fp16, [1, 2, 129, 257]> rel_weights_1_cast_fp16 = slice_by_index(begin = rel_weights_1_begin_0, end = rel_weights_1_end_0, end_mask = rel_weights_1_end_mask_0, x = x_45_cast_fp16)[name = string("rel_weights_1_cast_fp16")];
378
+ bool var_475_transpose_x_0 = const()[name = string("op_475_transpose_x_0"), val = bool(false)];
379
+ bool var_475_transpose_y_0 = const()[name = string("op_475_transpose_y_0"), val = bool(false)];
380
+ tensor<fp16, [1, 1, 257, 32]> var_474_to_fp16 = const()[name = string("op_474_to_fp16"), val = tensor<fp16, [1, 1, 257, 32]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1513664)))];
381
+ tensor<fp16, [1, 2, 129, 32]> var_475_cast_fp16 = matmul(transpose_x = var_475_transpose_x_0, transpose_y = var_475_transpose_y_0, x = rel_weights_1_cast_fp16, y = var_474_to_fp16)[name = string("op_475_cast_fp16")];
382
+ tensor<fp16, [1, 2, 129, 32]> out_15_cast_fp16 = add(x = out_13_cast_fp16, y = var_475_cast_fp16)[name = string("out_15_cast_fp16")];
383
+ tensor<int32, [4]> var_477_perm_0 = const()[name = string("op_477_perm_0"), val = tensor<int32, [4]>([0, 1, 3, 2])];
384
+ tensor<int32, [3]> var_478 = const()[name = string("op_478"), val = tensor<int32, [3]>([1, 64, 129])];
385
+ tensor<fp16, [1, 2, 32, 129]> var_477_cast_fp16 = transpose(perm = var_477_perm_0, x = out_15_cast_fp16)[name = string("transpose_10")];
386
+ tensor<fp16, [1, 64, 129]> input_71_cast_fp16 = reshape(shape = var_478, x = var_477_cast_fp16)[name = string("input_71_cast_fp16")];
387
+ string y_1_pad_type_0 = const()[name = string("y_1_pad_type_0"), val = string("valid")];
388
+ tensor<int32, [1]> y_1_strides_0 = const()[name = string("y_1_strides_0"), val = tensor<int32, [1]>([1])];
389
+ tensor<int32, [2]> y_1_pad_0 = const()[name = string("y_1_pad_0"), val = tensor<int32, [2]>([0, 0])];
390
+ tensor<int32, [1]> y_1_dilations_0 = const()[name = string("y_1_dilations_0"), val = tensor<int32, [1]>([1])];
391
+ int32 y_1_groups_0 = const()[name = string("y_1_groups_0"), val = int32(1)];
392
+ tensor<fp16, [64, 64, 1]> m_attn_layers_0_attn_conv_o_weight_to_fp16 = const()[name = string("m_attn_layers_0_attn_conv_o_weight_to_fp16"), val = tensor<fp16, [64, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1530176)))];
393
+ tensor<fp16, [64]> m_attn_layers_0_attn_conv_o_bias_to_fp16 = const()[name = string("m_attn_layers_0_attn_conv_o_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1538432)))];
394
+ tensor<fp16, [1, 64, 129]> y_1_cast_fp16 = conv(bias = m_attn_layers_0_attn_conv_o_bias_to_fp16, dilations = y_1_dilations_0, groups = y_1_groups_0, pad = y_1_pad_0, pad_type = y_1_pad_type_0, strides = y_1_strides_0, weight = m_attn_layers_0_attn_conv_o_weight_to_fp16, x = input_71_cast_fp16)[name = string("y_1_cast_fp16")];
395
+ tensor<fp16, [1, 64, 129]> x_47_cast_fp16 = add(x = x_31_cast_fp16, y = y_1_cast_fp16)[name = string("x_47_cast_fp16")];
396
+ tensor<int32, [3]> input_73_perm_0 = const()[name = string("input_73_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
397
+ tensor<int32, [1]> var_493_axes_0 = const()[name = string("op_493_axes_0"), val = tensor<int32, [1]>([-1])];
398
+ tensor<fp16, [64]> m_attn_layers_0_norm_1_norm_weight_to_fp16 = const()[name = string("m_attn_layers_0_norm_1_norm_weight_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1538624)))];
399
+ tensor<fp16, [64]> m_attn_layers_0_norm_1_norm_bias_to_fp16 = const()[name = string("m_attn_layers_0_norm_1_norm_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1538816)))];
400
+ tensor<fp16, [1, 129, 64]> input_73_cast_fp16 = transpose(perm = input_73_perm_0, x = x_47_cast_fp16)[name = string("transpose_9")];
401
+ tensor<fp16, [1, 129, 64]> var_493_cast_fp16 = layer_norm(axes = var_493_axes_0, beta = m_attn_layers_0_norm_1_norm_bias_to_fp16, epsilon = var_20_to_fp16, gamma = m_attn_layers_0_norm_1_norm_weight_to_fp16, x = input_73_cast_fp16)[name = string("op_493_cast_fp16")];
402
+ tensor<int32, [3]> x_49_perm_0 = const()[name = string("x_49_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
403
+ tensor<fp16, [1, 64, 129]> x_49_cast_fp16 = transpose(perm = x_49_perm_0, x = var_493_cast_fp16)[name = string("transpose_8")];
404
+ tensor<fp16, [1, 64, 129]> input_75_cast_fp16 = mul(x = x_49_cast_fp16, y = mask_cast_fp16)[name = string("input_75_cast_fp16")];
405
+ string input_77_pad_type_0 = const()[name = string("input_77_pad_type_0"), val = string("valid")];
406
+ tensor<int32, [1]> input_77_strides_0 = const()[name = string("input_77_strides_0"), val = tensor<int32, [1]>([1])];
407
+ tensor<int32, [2]> input_77_pad_0 = const()[name = string("input_77_pad_0"), val = tensor<int32, [2]>([0, 0])];
408
+ tensor<int32, [1]> input_77_dilations_0 = const()[name = string("input_77_dilations_0"), val = tensor<int32, [1]>([1])];
409
+ int32 input_77_groups_0 = const()[name = string("input_77_groups_0"), val = int32(1)];
410
+ tensor<fp16, [256, 64, 1]> m_attn_layers_0_ffn_conv_1_weight_to_fp16 = const()[name = string("m_attn_layers_0_ffn_conv_1_weight_to_fp16"), val = tensor<fp16, [256, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1539008)))];
411
+ tensor<fp16, [256]> m_attn_layers_0_ffn_conv_1_bias_to_fp16 = const()[name = string("m_attn_layers_0_ffn_conv_1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1571840)))];
412
+ tensor<fp16, [1, 256, 129]> input_77_cast_fp16 = conv(bias = m_attn_layers_0_ffn_conv_1_bias_to_fp16, dilations = input_77_dilations_0, groups = input_77_groups_0, pad = input_77_pad_0, pad_type = input_77_pad_type_0, strides = input_77_strides_0, weight = m_attn_layers_0_ffn_conv_1_weight_to_fp16, x = input_75_cast_fp16)[name = string("input_77_cast_fp16")];
413
+ tensor<fp16, [1, 256, 129]> h_37_cast_fp16 = relu(x = input_77_cast_fp16)[name = string("h_37_cast_fp16")];
414
+ tensor<fp16, [1, 256, 129]> input_79_cast_fp16 = mul(x = h_37_cast_fp16, y = mask_cast_fp16)[name = string("input_79_cast_fp16")];
415
+ string h_39_pad_type_0 = const()[name = string("h_39_pad_type_0"), val = string("valid")];
416
+ tensor<int32, [1]> h_39_strides_0 = const()[name = string("h_39_strides_0"), val = tensor<int32, [1]>([1])];
417
+ tensor<int32, [2]> h_39_pad_0 = const()[name = string("h_39_pad_0"), val = tensor<int32, [2]>([0, 0])];
418
+ tensor<int32, [1]> h_39_dilations_0 = const()[name = string("h_39_dilations_0"), val = tensor<int32, [1]>([1])];
419
+ int32 h_39_groups_0 = const()[name = string("h_39_groups_0"), val = int32(1)];
420
+ tensor<fp16, [64, 256, 1]> m_attn_layers_0_ffn_conv_2_weight_to_fp16 = const()[name = string("m_attn_layers_0_ffn_conv_2_weight_to_fp16"), val = tensor<fp16, [64, 256, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1572416)))];
421
+ tensor<fp16, [64]> m_attn_layers_0_ffn_conv_2_bias_to_fp16 = const()[name = string("m_attn_layers_0_ffn_conv_2_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1605248)))];
422
+ tensor<fp16, [1, 64, 129]> h_39_cast_fp16 = conv(bias = m_attn_layers_0_ffn_conv_2_bias_to_fp16, dilations = h_39_dilations_0, groups = h_39_groups_0, pad = h_39_pad_0, pad_type = h_39_pad_type_0, strides = h_39_strides_0, weight = m_attn_layers_0_ffn_conv_2_weight_to_fp16, x = input_79_cast_fp16)[name = string("h_39_cast_fp16")];
423
+ tensor<fp16, [1, 64, 129]> y_3_cast_fp16 = mul(x = h_39_cast_fp16, y = mask_cast_fp16)[name = string("y_3_cast_fp16")];
424
+ tensor<fp16, [1, 64, 129]> x_51_cast_fp16 = add(x = x_49_cast_fp16, y = y_3_cast_fp16)[name = string("x_51_cast_fp16")];
425
+ tensor<int32, [3]> input_81_perm_0 = const()[name = string("input_81_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
426
+ tensor<int32, [1]> var_521_axes_0 = const()[name = string("op_521_axes_0"), val = tensor<int32, [1]>([-1])];
427
+ tensor<fp16, [64]> m_attn_layers_0_norm_2_norm_weight_to_fp16 = const()[name = string("m_attn_layers_0_norm_2_norm_weight_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1605440)))];
428
+ tensor<fp16, [64]> m_attn_layers_0_norm_2_norm_bias_to_fp16 = const()[name = string("m_attn_layers_0_norm_2_norm_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1605632)))];
429
+ tensor<fp16, [1, 129, 64]> input_81_cast_fp16 = transpose(perm = input_81_perm_0, x = x_51_cast_fp16)[name = string("transpose_7")];
430
+ tensor<fp16, [1, 129, 64]> var_521_cast_fp16 = layer_norm(axes = var_521_axes_0, beta = m_attn_layers_0_norm_2_norm_bias_to_fp16, epsilon = var_20_to_fp16, gamma = m_attn_layers_0_norm_2_norm_weight_to_fp16, x = input_81_cast_fp16)[name = string("op_521_cast_fp16")];
431
+ tensor<int32, [3]> x_53_perm_0 = const()[name = string("x_53_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
432
+ tensor<fp16, [1, 64, 129]> x_53_cast_fp16 = transpose(perm = x_53_perm_0, x = var_521_cast_fp16)[name = string("transpose_6")];
433
+ tensor<fp16, [1, 64, 129]> x_55_cast_fp16 = mul(x = x_53_cast_fp16, y = mask_cast_fp16)[name = string("x_55_cast_fp16")];
434
+ string var_543_pad_type_0 = const()[name = string("op_543_pad_type_0"), val = string("valid")];
435
+ tensor<int32, [1]> var_543_strides_0 = const()[name = string("op_543_strides_0"), val = tensor<int32, [1]>([1])];
436
+ tensor<int32, [2]> var_543_pad_0 = const()[name = string("op_543_pad_0"), val = tensor<int32, [2]>([0, 0])];
437
+ tensor<int32, [1]> var_543_dilations_0 = const()[name = string("op_543_dilations_0"), val = tensor<int32, [1]>([1])];
438
+ int32 var_543_groups_0 = const()[name = string("op_543_groups_0"), val = int32(1)];
439
+ tensor<fp16, [64, 64, 1]> m_attn_layers_1_attn_conv_q_weight_to_fp16 = const()[name = string("m_attn_layers_1_attn_conv_q_weight_to_fp16"), val = tensor<fp16, [64, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1605824)))];
440
+ tensor<fp16, [64]> m_attn_layers_1_attn_conv_q_bias_to_fp16 = const()[name = string("m_attn_layers_1_attn_conv_q_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1614080)))];
441
+ tensor<fp16, [1, 64, 129]> var_543_cast_fp16 = conv(bias = m_attn_layers_1_attn_conv_q_bias_to_fp16, dilations = var_543_dilations_0, groups = var_543_groups_0, pad = var_543_pad_0, pad_type = var_543_pad_type_0, strides = var_543_strides_0, weight = m_attn_layers_1_attn_conv_q_weight_to_fp16, x = x_55_cast_fp16)[name = string("op_543_cast_fp16")];
442
+ tensor<int32, [4]> var_544 = const()[name = string("op_544"), val = tensor<int32, [4]>([1, 2, 32, 129])];
443
+ tensor<fp16, [1, 2, 32, 129]> var_545_cast_fp16 = reshape(shape = var_544, x = var_543_cast_fp16)[name = string("op_545_cast_fp16")];
444
+ tensor<int32, [4]> q_perm_0 = const()[name = string("q_perm_0"), val = tensor<int32, [4]>([0, 1, 3, 2])];
445
+ string var_553_pad_type_0 = const()[name = string("op_553_pad_type_0"), val = string("valid")];
446
+ tensor<int32, [1]> var_553_strides_0 = const()[name = string("op_553_strides_0"), val = tensor<int32, [1]>([1])];
447
+ tensor<int32, [2]> var_553_pad_0 = const()[name = string("op_553_pad_0"), val = tensor<int32, [2]>([0, 0])];
448
+ tensor<int32, [1]> var_553_dilations_0 = const()[name = string("op_553_dilations_0"), val = tensor<int32, [1]>([1])];
449
+ int32 var_553_groups_0 = const()[name = string("op_553_groups_0"), val = int32(1)];
450
+ tensor<fp16, [64, 64, 1]> m_attn_layers_1_attn_conv_k_weight_to_fp16 = const()[name = string("m_attn_layers_1_attn_conv_k_weight_to_fp16"), val = tensor<fp16, [64, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1614272)))];
451
+ tensor<fp16, [64]> m_attn_layers_1_attn_conv_k_bias_to_fp16 = const()[name = string("m_attn_layers_1_attn_conv_k_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1622528)))];
452
+ tensor<fp16, [1, 64, 129]> var_553_cast_fp16 = conv(bias = m_attn_layers_1_attn_conv_k_bias_to_fp16, dilations = var_553_dilations_0, groups = var_553_groups_0, pad = var_553_pad_0, pad_type = var_553_pad_type_0, strides = var_553_strides_0, weight = m_attn_layers_1_attn_conv_k_weight_to_fp16, x = x_55_cast_fp16)[name = string("op_553_cast_fp16")];
453
+ tensor<int32, [4]> var_554 = const()[name = string("op_554"), val = tensor<int32, [4]>([1, 2, 32, 129])];
454
+ tensor<fp16, [1, 2, 32, 129]> var_555_cast_fp16 = reshape(shape = var_554, x = var_553_cast_fp16)[name = string("op_555_cast_fp16")];
455
+ string var_563_pad_type_0 = const()[name = string("op_563_pad_type_0"), val = string("valid")];
456
+ tensor<int32, [1]> var_563_strides_0 = const()[name = string("op_563_strides_0"), val = tensor<int32, [1]>([1])];
457
+ tensor<int32, [2]> var_563_pad_0 = const()[name = string("op_563_pad_0"), val = tensor<int32, [2]>([0, 0])];
458
+ tensor<int32, [1]> var_563_dilations_0 = const()[name = string("op_563_dilations_0"), val = tensor<int32, [1]>([1])];
459
+ int32 var_563_groups_0 = const()[name = string("op_563_groups_0"), val = int32(1)];
460
+ tensor<fp16, [64, 64, 1]> m_attn_layers_1_attn_conv_v_weight_to_fp16 = const()[name = string("m_attn_layers_1_attn_conv_v_weight_to_fp16"), val = tensor<fp16, [64, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1622720)))];
461
+ tensor<fp16, [64]> m_attn_layers_1_attn_conv_v_bias_to_fp16 = const()[name = string("m_attn_layers_1_attn_conv_v_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1630976)))];
462
+ tensor<fp16, [1, 64, 129]> var_563_cast_fp16 = conv(bias = m_attn_layers_1_attn_conv_v_bias_to_fp16, dilations = var_563_dilations_0, groups = var_563_groups_0, pad = var_563_pad_0, pad_type = var_563_pad_type_0, strides = var_563_strides_0, weight = m_attn_layers_1_attn_conv_v_weight_to_fp16, x = x_55_cast_fp16)[name = string("op_563_cast_fp16")];
463
+ tensor<int32, [4]> var_564 = const()[name = string("op_564"), val = tensor<int32, [4]>([1, 2, 32, 129])];
464
+ tensor<fp16, [1, 2, 32, 129]> var_565_cast_fp16 = reshape(shape = var_564, x = var_563_cast_fp16)[name = string("op_565_cast_fp16")];
465
+ fp16 var_567_to_fp16 = const()[name = string("op_567_to_fp16"), val = fp16(0x1.6ap-3)];
466
+ tensor<fp16, [1, 2, 129, 32]> q_cast_fp16 = transpose(perm = q_perm_0, x = var_545_cast_fp16)[name = string("transpose_5")];
467
+ tensor<fp16, [1, 2, 129, 32]> var_568_cast_fp16 = mul(x = q_cast_fp16, y = var_567_to_fp16)[name = string("op_568_cast_fp16")];
468
+ bool scores_5_transpose_x_0 = const()[name = string("scores_5_transpose_x_0"), val = bool(false)];
469
+ bool scores_5_transpose_y_0 = const()[name = string("scores_5_transpose_y_0"), val = bool(false)];
470
+ tensor<fp16, [1, 2, 129, 129]> scores_5_cast_fp16 = matmul(transpose_x = scores_5_transpose_x_0, transpose_y = scores_5_transpose_y_0, x = var_568_cast_fp16, y = var_555_cast_fp16)[name = string("scores_5_cast_fp16")];
471
+ bool x_57_transpose_x_0 = const()[name = string("x_57_transpose_x_0"), val = bool(false)];
472
+ bool x_57_transpose_y_0 = const()[name = string("x_57_transpose_y_0"), val = bool(false)];
473
+ tensor<fp16, [1, 1, 32, 257]> var_592_to_fp16 = const()[name = string("op_592_to_fp16"), val = tensor<fp16, [1, 1, 32, 257]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1631168)))];
474
+ tensor<fp16, [1, 2, 129, 257]> x_57_cast_fp16 = matmul(transpose_x = x_57_transpose_x_0, transpose_y = x_57_transpose_y_0, x = var_568_cast_fp16, y = var_592_to_fp16)[name = string("x_57_cast_fp16")];
475
+ tensor<int32, [8]> x_59_pad_0 = const()[name = string("x_59_pad_0"), val = tensor<int32, [8]>([0, 0, 0, 0, 0, 0, 0, 1])];
476
+ string x_59_mode_0 = const()[name = string("x_59_mode_0"), val = string("constant")];
477
+ fp16 const_27_to_fp16 = const()[name = string("const_27_to_fp16"), val = fp16(0x0p+0)];
478
+ tensor<fp16, [1, 2, 129, 258]> x_59_cast_fp16 = pad(constant_val = const_27_to_fp16, mode = x_59_mode_0, pad = x_59_pad_0, x = x_57_cast_fp16)[name = string("x_59_cast_fp16")];
479
+ tensor<int32, [3]> var_604 = const()[name = string("op_604"), val = tensor<int32, [3]>([1, 2, 33282])];
480
+ tensor<fp16, [1, 2, 33282]> input_85_cast_fp16 = reshape(shape = var_604, x = x_59_cast_fp16)[name = string("input_85_cast_fp16")];
481
+ tensor<int32, [6]> x_61_pad_0 = const()[name = string("x_61_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 0, 128])];
482
+ string x_61_mode_0 = const()[name = string("x_61_mode_0"), val = string("constant")];
483
+ fp16 const_28_to_fp16 = const()[name = string("const_28_to_fp16"), val = fp16(0x0p+0)];
484
+ tensor<fp16, [1, 2, 33410]> x_61_cast_fp16 = pad(constant_val = const_28_to_fp16, mode = x_61_mode_0, pad = x_61_pad_0, x = input_85_cast_fp16)[name = string("x_61_cast_fp16")];
485
+ tensor<int32, [4]> var_619 = const()[name = string("op_619"), val = tensor<int32, [4]>([1, 2, 130, 257])];
486
+ tensor<fp16, [1, 2, 130, 257]> x_63_cast_fp16 = reshape(shape = var_619, x = x_61_cast_fp16)[name = string("x_63_cast_fp16")];
487
+ tensor<int32, [4]> var_626_begin_0 = const()[name = string("op_626_begin_0"), val = tensor<int32, [4]>([0, 0, 0, 0])];
488
+ tensor<int32, [4]> var_626_end_0 = const()[name = string("op_626_end_0"), val = tensor<int32, [4]>([1, 2, 129, 257])];
489
+ tensor<bool, [4]> var_626_end_mask_0 = const()[name = string("op_626_end_mask_0"), val = tensor<bool, [4]>([true, true, false, true])];
490
+ tensor<fp16, [1, 2, 129, 257]> var_626_cast_fp16 = slice_by_index(begin = var_626_begin_0, end = var_626_end_0, end_mask = var_626_end_mask_0, x = x_63_cast_fp16)[name = string("op_626_cast_fp16")];
491
+ tensor<int32, [4]> var_627_begin_0 = const()[name = string("op_627_begin_0"), val = tensor<int32, [4]>([0, 0, 0, 128])];
492
+ tensor<int32, [4]> var_627_end_0 = const()[name = string("op_627_end_0"), val = tensor<int32, [4]>([1, 2, 129, 257])];
493
+ tensor<bool, [4]> var_627_end_mask_0 = const()[name = string("op_627_end_mask_0"), val = tensor<bool, [4]>([true, true, true, true])];
494
+ tensor<fp16, [1, 2, 129, 129]> var_627_cast_fp16 = slice_by_index(begin = var_627_begin_0, end = var_627_end_0, end_mask = var_627_end_mask_0, x = var_626_cast_fp16)[name = string("op_627_cast_fp16")];
495
+ tensor<fp16, [1, 2, 129, 129]> scores_cast_fp16 = add(x = scores_5_cast_fp16, y = var_627_cast_fp16)[name = string("scores_cast_fp16")];
496
+ tensor<fp16, [1, 2, 129, 129]> input_87_cast_fp16 = sub(x = scores_cast_fp16, y = var_424_cast_fp16)[name = string("input_87_cast_fp16")];
497
+ tensor<fp16, [1, 2, 129, 129]> p_attn_cast_fp16 = softmax(axis = var_27, x = input_87_cast_fp16)[name = string("p_attn_cast_fp16")];
498
+ bool out_17_transpose_x_1 = const()[name = string("out_17_transpose_x_1"), val = bool(false)];
499
+ bool out_17_transpose_y_1 = const()[name = string("out_17_transpose_y_1"), val = bool(true)];
500
+ tensor<fp16, [1, 2, 129, 32]> out_17_cast_fp16 = matmul(transpose_x = out_17_transpose_x_1, transpose_y = out_17_transpose_y_1, x = p_attn_cast_fp16, y = var_565_cast_fp16)[name = string("out_17_cast_fp16")];
501
+ tensor<int32, [8]> x_65_pad_0 = const()[name = string("x_65_pad_0"), val = tensor<int32, [8]>([0, 0, 0, 0, 0, 0, 0, 128])];
502
+ string x_65_mode_0 = const()[name = string("x_65_mode_0"), val = string("constant")];
503
+ fp16 const_33_to_fp16 = const()[name = string("const_33_to_fp16"), val = fp16(0x0p+0)];
504
+ tensor<fp16, [1, 2, 129, 257]> x_65_cast_fp16 = pad(constant_val = const_33_to_fp16, mode = x_65_mode_0, pad = x_65_pad_0, x = p_attn_cast_fp16)[name = string("x_65_cast_fp16")];
505
+ tensor<int32, [3]> var_669 = const()[name = string("op_669"), val = tensor<int32, [3]>([1, 2, 33153])];
506
+ tensor<fp16, [1, 2, 33153]> input_91_cast_fp16 = reshape(shape = var_669, x = x_65_cast_fp16)[name = string("input_91_cast_fp16")];
507
+ tensor<int32, [6]> x_67_pad_0 = const()[name = string("x_67_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 129, 0])];
508
+ string x_67_mode_0 = const()[name = string("x_67_mode_0"), val = string("constant")];
509
+ fp16 const_34_to_fp16 = const()[name = string("const_34_to_fp16"), val = fp16(0x0p+0)];
510
+ tensor<fp16, [1, 2, 33282]> x_67_cast_fp16 = pad(constant_val = const_34_to_fp16, mode = x_67_mode_0, pad = x_67_pad_0, x = input_91_cast_fp16)[name = string("x_67_cast_fp16")];
511
+ tensor<int32, [4]> var_676 = const()[name = string("op_676"), val = tensor<int32, [4]>([1, 2, 129, 258])];
512
+ tensor<fp16, [1, 2, 129, 258]> x_69_cast_fp16 = reshape(shape = var_676, x = x_67_cast_fp16)[name = string("x_69_cast_fp16")];
513
+ tensor<int32, [4]> rel_weights_begin_0 = const()[name = string("rel_weights_begin_0"), val = tensor<int32, [4]>([0, 0, 0, 1])];
514
+ tensor<int32, [4]> rel_weights_end_0 = const()[name = string("rel_weights_end_0"), val = tensor<int32, [4]>([1, 2, 129, 258])];
515
+ tensor<bool, [4]> rel_weights_end_mask_0 = const()[name = string("rel_weights_end_mask_0"), val = tensor<bool, [4]>([true, true, true, true])];
516
+ tensor<fp16, [1, 2, 129, 257]> rel_weights_cast_fp16 = slice_by_index(begin = rel_weights_begin_0, end = rel_weights_end_0, end_mask = rel_weights_end_mask_0, x = x_69_cast_fp16)[name = string("rel_weights_cast_fp16")];
517
+ bool var_683_transpose_x_0 = const()[name = string("op_683_transpose_x_0"), val = bool(false)];
518
+ bool var_683_transpose_y_0 = const()[name = string("op_683_transpose_y_0"), val = bool(false)];
519
+ tensor<fp16, [1, 1, 257, 32]> var_682_to_fp16 = const()[name = string("op_682_to_fp16"), val = tensor<fp16, [1, 1, 257, 32]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1647680)))];
520
+ tensor<fp16, [1, 2, 129, 32]> var_683_cast_fp16 = matmul(transpose_x = var_683_transpose_x_0, transpose_y = var_683_transpose_y_0, x = rel_weights_cast_fp16, y = var_682_to_fp16)[name = string("op_683_cast_fp16")];
521
+ tensor<fp16, [1, 2, 129, 32]> out_cast_fp16 = add(x = out_17_cast_fp16, y = var_683_cast_fp16)[name = string("out_cast_fp16")];
522
+ tensor<int32, [4]> var_685_perm_0 = const()[name = string("op_685_perm_0"), val = tensor<int32, [4]>([0, 1, 3, 2])];
523
+ tensor<int32, [3]> var_686 = const()[name = string("op_686"), val = tensor<int32, [3]>([1, 64, 129])];
524
+ tensor<fp16, [1, 2, 32, 129]> var_685_cast_fp16 = transpose(perm = var_685_perm_0, x = out_cast_fp16)[name = string("transpose_4")];
525
+ tensor<fp16, [1, 64, 129]> input_93_cast_fp16 = reshape(shape = var_686, x = var_685_cast_fp16)[name = string("input_93_cast_fp16")];
526
+ string y_5_pad_type_0 = const()[name = string("y_5_pad_type_0"), val = string("valid")];
527
+ tensor<int32, [1]> y_5_strides_0 = const()[name = string("y_5_strides_0"), val = tensor<int32, [1]>([1])];
528
+ tensor<int32, [2]> y_5_pad_0 = const()[name = string("y_5_pad_0"), val = tensor<int32, [2]>([0, 0])];
529
+ tensor<int32, [1]> y_5_dilations_0 = const()[name = string("y_5_dilations_0"), val = tensor<int32, [1]>([1])];
530
+ int32 y_5_groups_0 = const()[name = string("y_5_groups_0"), val = int32(1)];
531
+ tensor<fp16, [64, 64, 1]> m_attn_layers_1_attn_conv_o_weight_to_fp16 = const()[name = string("m_attn_layers_1_attn_conv_o_weight_to_fp16"), val = tensor<fp16, [64, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1664192)))];
532
+ tensor<fp16, [64]> m_attn_layers_1_attn_conv_o_bias_to_fp16 = const()[name = string("m_attn_layers_1_attn_conv_o_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1672448)))];
533
+ tensor<fp16, [1, 64, 129]> y_5_cast_fp16 = conv(bias = m_attn_layers_1_attn_conv_o_bias_to_fp16, dilations = y_5_dilations_0, groups = y_5_groups_0, pad = y_5_pad_0, pad_type = y_5_pad_type_0, strides = y_5_strides_0, weight = m_attn_layers_1_attn_conv_o_weight_to_fp16, x = input_93_cast_fp16)[name = string("y_5_cast_fp16")];
534
+ tensor<fp16, [1, 64, 129]> x_71_cast_fp16 = add(x = x_55_cast_fp16, y = y_5_cast_fp16)[name = string("x_71_cast_fp16")];
535
+ tensor<int32, [3]> input_95_perm_0 = const()[name = string("input_95_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
536
+ tensor<int32, [1]> var_701_axes_0 = const()[name = string("op_701_axes_0"), val = tensor<int32, [1]>([-1])];
537
+ tensor<fp16, [64]> m_attn_layers_1_norm_1_norm_weight_to_fp16 = const()[name = string("m_attn_layers_1_norm_1_norm_weight_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1672640)))];
538
+ tensor<fp16, [64]> m_attn_layers_1_norm_1_norm_bias_to_fp16 = const()[name = string("m_attn_layers_1_norm_1_norm_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1672832)))];
539
+ tensor<fp16, [1, 129, 64]> input_95_cast_fp16 = transpose(perm = input_95_perm_0, x = x_71_cast_fp16)[name = string("transpose_3")];
540
+ tensor<fp16, [1, 129, 64]> var_701_cast_fp16 = layer_norm(axes = var_701_axes_0, beta = m_attn_layers_1_norm_1_norm_bias_to_fp16, epsilon = var_20_to_fp16, gamma = m_attn_layers_1_norm_1_norm_weight_to_fp16, x = input_95_cast_fp16)[name = string("op_701_cast_fp16")];
541
+ tensor<int32, [3]> x_73_perm_0 = const()[name = string("x_73_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
542
+ tensor<fp16, [1, 64, 129]> x_73_cast_fp16 = transpose(perm = x_73_perm_0, x = var_701_cast_fp16)[name = string("transpose_2")];
543
+ tensor<fp16, [1, 64, 129]> input_97_cast_fp16 = mul(x = x_73_cast_fp16, y = mask_cast_fp16)[name = string("input_97_cast_fp16")];
544
+ string input_99_pad_type_0 = const()[name = string("input_99_pad_type_0"), val = string("valid")];
545
+ tensor<int32, [1]> input_99_strides_0 = const()[name = string("input_99_strides_0"), val = tensor<int32, [1]>([1])];
546
+ tensor<int32, [2]> input_99_pad_0 = const()[name = string("input_99_pad_0"), val = tensor<int32, [2]>([0, 0])];
547
+ tensor<int32, [1]> input_99_dilations_0 = const()[name = string("input_99_dilations_0"), val = tensor<int32, [1]>([1])];
548
+ int32 input_99_groups_0 = const()[name = string("input_99_groups_0"), val = int32(1)];
549
+ tensor<fp16, [256, 64, 1]> m_attn_layers_1_ffn_conv_1_weight_to_fp16 = const()[name = string("m_attn_layers_1_ffn_conv_1_weight_to_fp16"), val = tensor<fp16, [256, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1673024)))];
550
+ tensor<fp16, [256]> m_attn_layers_1_ffn_conv_1_bias_to_fp16 = const()[name = string("m_attn_layers_1_ffn_conv_1_bias_to_fp16"), val = tensor<fp16, [256]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1705856)))];
551
+ tensor<fp16, [1, 256, 129]> input_99_cast_fp16 = conv(bias = m_attn_layers_1_ffn_conv_1_bias_to_fp16, dilations = input_99_dilations_0, groups = input_99_groups_0, pad = input_99_pad_0, pad_type = input_99_pad_type_0, strides = input_99_strides_0, weight = m_attn_layers_1_ffn_conv_1_weight_to_fp16, x = input_97_cast_fp16)[name = string("input_99_cast_fp16")];
552
+ tensor<fp16, [1, 256, 129]> h_41_cast_fp16 = relu(x = input_99_cast_fp16)[name = string("h_41_cast_fp16")];
553
+ tensor<fp16, [1, 256, 129]> input_101_cast_fp16 = mul(x = h_41_cast_fp16, y = mask_cast_fp16)[name = string("input_101_cast_fp16")];
554
+ string h_pad_type_0 = const()[name = string("h_pad_type_0"), val = string("valid")];
555
+ tensor<int32, [1]> h_strides_0 = const()[name = string("h_strides_0"), val = tensor<int32, [1]>([1])];
556
+ tensor<int32, [2]> h_pad_0 = const()[name = string("h_pad_0"), val = tensor<int32, [2]>([0, 0])];
557
+ tensor<int32, [1]> h_dilations_0 = const()[name = string("h_dilations_0"), val = tensor<int32, [1]>([1])];
558
+ int32 h_groups_0 = const()[name = string("h_groups_0"), val = int32(1)];
559
+ tensor<fp16, [64, 256, 1]> m_attn_layers_1_ffn_conv_2_weight_to_fp16 = const()[name = string("m_attn_layers_1_ffn_conv_2_weight_to_fp16"), val = tensor<fp16, [64, 256, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1706432)))];
560
+ tensor<fp16, [64]> m_attn_layers_1_ffn_conv_2_bias_to_fp16 = const()[name = string("m_attn_layers_1_ffn_conv_2_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1739264)))];
561
+ tensor<fp16, [1, 64, 129]> h_cast_fp16 = conv(bias = m_attn_layers_1_ffn_conv_2_bias_to_fp16, dilations = h_dilations_0, groups = h_groups_0, pad = h_pad_0, pad_type = h_pad_type_0, strides = h_strides_0, weight = m_attn_layers_1_ffn_conv_2_weight_to_fp16, x = input_101_cast_fp16)[name = string("h_cast_fp16")];
562
+ tensor<fp16, [1, 64, 129]> y_cast_fp16 = mul(x = h_cast_fp16, y = mask_cast_fp16)[name = string("y_cast_fp16")];
563
+ tensor<fp16, [1, 64, 129]> x_75_cast_fp16 = add(x = x_73_cast_fp16, y = y_cast_fp16)[name = string("x_75_cast_fp16")];
564
+ tensor<int32, [3]> input_103_perm_0 = const()[name = string("input_103_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
565
+ tensor<int32, [1]> var_729_axes_0 = const()[name = string("op_729_axes_0"), val = tensor<int32, [1]>([-1])];
566
+ tensor<fp16, [64]> m_attn_layers_1_norm_2_norm_weight_to_fp16 = const()[name = string("m_attn_layers_1_norm_2_norm_weight_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1739456)))];
567
+ tensor<fp16, [64]> m_attn_layers_1_norm_2_norm_bias_to_fp16 = const()[name = string("m_attn_layers_1_norm_2_norm_bias_to_fp16"), val = tensor<fp16, [64]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1739648)))];
568
+ tensor<fp16, [1, 129, 64]> input_103_cast_fp16 = transpose(perm = input_103_perm_0, x = x_75_cast_fp16)[name = string("transpose_1")];
569
+ tensor<fp16, [1, 129, 64]> var_729_cast_fp16 = layer_norm(axes = var_729_axes_0, beta = m_attn_layers_1_norm_2_norm_bias_to_fp16, epsilon = var_20_to_fp16, gamma = m_attn_layers_1_norm_2_norm_weight_to_fp16, x = input_103_cast_fp16)[name = string("op_729_cast_fp16")];
570
+ tensor<int32, [3]> x_77_perm_0 = const()[name = string("x_77_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
571
+ tensor<fp16, [1, 64, 129]> x_77_cast_fp16 = transpose(perm = x_77_perm_0, x = var_729_cast_fp16)[name = string("transpose_0")];
572
+ tensor<fp16, [1, 64, 129]> x_79_cast_fp16 = mul(x = x_77_cast_fp16, y = mask_cast_fp16)[name = string("x_79_cast_fp16")];
573
+ tensor<fp16, [1, 64, 129]> x_81_cast_fp16 = add(x = x_79_cast_fp16, y = x_29_cast_fp16)[name = string("x_81_cast_fp16")];
574
+ tensor<int32, [3]> input_105_begin_0 = const()[name = string("input_105_begin_0"), val = tensor<int32, [3]>([0, 0, 0])];
575
+ tensor<int32, [3]> input_105_end_0 = const()[name = string("input_105_end_0"), val = tensor<int32, [3]>([1, 64, 1])];
576
+ tensor<bool, [3]> input_105_end_mask_0 = const()[name = string("input_105_end_mask_0"), val = tensor<bool, [3]>([true, true, false])];
577
+ tensor<fp16, [1, 64, 1]> input_105_cast_fp16 = slice_by_index(begin = input_105_begin_0, end = input_105_end_0, end_mask = input_105_end_mask_0, x = x_81_cast_fp16)[name = string("input_105_cast_fp16")];
578
+ tensor<int32, [3]> sentence_mask_begin_0 = const()[name = string("sentence_mask_begin_0"), val = tensor<int32, [3]>([0, 0, 0])];
579
+ tensor<int32, [3]> sentence_mask_end_0 = const()[name = string("sentence_mask_end_0"), val = tensor<int32, [3]>([1, 1, 1])];
580
+ tensor<bool, [3]> sentence_mask_end_mask_0 = const()[name = string("sentence_mask_end_mask_0"), val = tensor<bool, [3]>([true, true, false])];
581
+ tensor<fp16, [1, 1, 1]> sentence_mask_cast_fp16 = slice_by_index(begin = sentence_mask_begin_0, end = sentence_mask_end_0, end_mask = sentence_mask_end_mask_0, x = mask_cast_fp16)[name = string("sentence_mask_cast_fp16")];
582
+ string var_744_pad_type_0 = const()[name = string("op_744_pad_type_0"), val = string("valid")];
583
+ tensor<int32, [1]> var_744_strides_0 = const()[name = string("op_744_strides_0"), val = tensor<int32, [1]>([1])];
584
+ tensor<int32, [2]> var_744_pad_0 = const()[name = string("op_744_pad_0"), val = tensor<int32, [2]>([0, 0])];
585
+ tensor<int32, [1]> var_744_dilations_0 = const()[name = string("op_744_dilations_0"), val = tensor<int32, [1]>([1])];
586
+ int32 var_744_groups_0 = const()[name = string("op_744_groups_0"), val = int32(1)];
587
+ tensor<fp16, [64, 64, 1]> m_proj_out_conv_weight_to_fp16 = const()[name = string("m_proj_out_conv_weight_to_fp16"), val = tensor<fp16, [64, 64, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1739840)))];
588
+ tensor<fp16, [1, 64, 1]> var_744_cast_fp16 = conv(dilations = var_744_dilations_0, groups = var_744_groups_0, pad = var_744_pad_0, pad_type = var_744_pad_type_0, strides = var_744_strides_0, weight = m_proj_out_conv_weight_to_fp16, x = input_105_cast_fp16)[name = string("op_744_cast_fp16")];
589
+ tensor<fp16, [1, 64, 1]> sentence_emb_cast_fp16 = mul(x = var_744_cast_fp16, y = sentence_mask_cast_fp16)[name = string("sentence_emb_cast_fp16")];
590
+ tensor<int32, [2]> var_752 = const()[name = string("op_752"), val = tensor<int32, [2]>([1, -1])];
591
+ tensor<fp16, [1, 64]> s_cast_fp16 = reshape(shape = var_752, x = sentence_emb_cast_fp16)[name = string("s_cast_fp16")];
592
+ tensor<int32, [2]> var_754 = const()[name = string("op_754"), val = tensor<int32, [2]>([1, -1])];
593
+ string style_dp_to_fp16_dtype_0 = const()[name = string("style_dp_to_fp16_dtype_0"), val = string("fp16")];
594
+ tensor<fp16, [1, 8, 16]> style_dp_to_fp16 = cast(dtype = style_dp_to_fp16_dtype_0, x = style_dp)[name = string("cast_21")];
595
+ tensor<fp16, [1, 128]> v_cast_fp16 = reshape(shape = var_754, x = style_dp_to_fp16)[name = string("v_cast_fp16")];
596
+ bool input_107_interleave_0 = const()[name = string("input_107_interleave_0"), val = bool(false)];
597
+ tensor<fp16, [1, 192]> input_107_cast_fp16 = concat(axis = var_26, interleave = input_107_interleave_0, values = (s_cast_fp16, v_cast_fp16))[name = string("input_107_cast_fp16")];
598
+ tensor<fp16, [128, 192]> m_predictor_layers_0_weight_to_fp16 = const()[name = string("m_predictor_layers_0_weight_to_fp16"), val = tensor<fp16, [128, 192]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1748096)))];
599
+ tensor<fp16, [128]> m_predictor_layers_0_bias_to_fp16 = const()[name = string("m_predictor_layers_0_bias_to_fp16"), val = tensor<fp16, [128]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1797312)))];
600
+ tensor<fp16, [1, 128]> linear_0_cast_fp16 = linear(bias = m_predictor_layers_0_bias_to_fp16, weight = m_predictor_layers_0_weight_to_fp16, x = input_107_cast_fp16)[name = string("linear_0_cast_fp16")];
601
+ tensor<int32, [1]> expand_dims_0_axes_0 = const()[name = string("expand_dims_0_axes_0"), val = tensor<int32, [1]>([2])];
602
+ tensor<fp16, [1, 128, 1]> expand_dims_0_cast_fp16 = expand_dims(axes = expand_dims_0_axes_0, x = linear_0_cast_fp16)[name = string("expand_dims_0_cast_fp16")];
603
+ fp32 prelu_0_alpha_1 = const()[name = string("prelu_0_alpha_1"), val = fp32(0x1.b19f5ap-3)];
604
+ tensor<fp16, [1, 128, 1]> prelu_0_cast_fp16 = leaky_relu(alpha = prelu_0_alpha_1, x = expand_dims_0_cast_fp16)[name = string("prelu_0_cast_fp16")];
605
+ tensor<int32, [1]> input_axes_0 = const()[name = string("input_axes_0"), val = tensor<int32, [1]>([2])];
606
+ tensor<fp16, [1, 128]> input_cast_fp16 = squeeze(axes = input_axes_0, x = prelu_0_cast_fp16)[name = string("input_cast_fp16")];
607
+ tensor<fp16, [1, 128]> m_predictor_layers_1_weight_to_fp16 = const()[name = string("m_predictor_layers_1_weight_to_fp16"), val = tensor<fp16, [1, 128]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1797632)))];
608
+ tensor<fp16, [1]> m_predictor_layers_1_bias_to_fp16 = const()[name = string("m_predictor_layers_1_bias_to_fp16"), val = tensor<fp16, [1]>([0x1.60cp-6])];
609
+ tensor<fp16, [1, 1]> linear_1_cast_fp16 = linear(bias = m_predictor_layers_1_bias_to_fp16, weight = m_predictor_layers_1_weight_to_fp16, x = input_cast_fp16)[name = string("linear_1_cast_fp16")];
610
+ tensor<fp16, [1, 1]> x_cast_fp16 = exp(x = linear_1_cast_fp16)[name = string("x_cast_fp16")];
611
+ tensor<int32, [1]> var_767_axes_0 = const()[name = string("op_767_axes_0"), val = tensor<int32, [1]>([-1])];
612
+ tensor<fp16, [1]> var_767_cast_fp16 = squeeze(axes = var_767_axes_0, x = x_cast_fp16)[name = string("op_767_cast_fp16")];
613
+ string var_767_cast_fp16_to_fp32_dtype_0 = const()[name = string("op_767_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
614
+ tensor<fp32, [1]> duration = cast(dtype = var_767_cast_fp16_to_fp32_dtype_0, x = var_767_cast_fp16)[name = string("cast_20")];
615
+ } -> (duration);
616
+ }
DurationPredictor.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f709c544087966d6b2538ac50b7a4014d6e1049e21d713cf3bc71c4c5b9307c
3
+ size 1797952
DurationPredictor.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ccb56ff515b7d41cf79d21d6be2f2d23a31a295810950dd86e39620f9629cad4
3
+ size 93494
DurationPredictor.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f709c544087966d6b2538ac50b7a4014d6e1049e21d713cf3bc71c4c5b9307c
3
+ size 1797952
DurationPredictor.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "51A57480-22F2-40DD-9B71-02466AEB3397": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Weights",
7
+ "name": "weights",
8
+ "path": "com.apple.CoreML/weights"
9
+ },
10
+ "6B8FFD11-C3E0-4F96-B464-7BC65C9BAE5A": {
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": "6B8FFD11-C3E0-4F96-B464-7BC65C9BAE5A"
18
+ }
README.md ADDED
@@ -0,0 +1,202 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: openrail++
3
+ track_downloads: true
4
+ language:
5
+ - en
6
+ - ko
7
+ - ja
8
+ - ar
9
+ - bg
10
+ - cs
11
+ - da
12
+ - de
13
+ - el
14
+ - es
15
+ - et
16
+ - fi
17
+ - fr
18
+ - hi
19
+ - hr
20
+ - hu
21
+ - id
22
+ - it
23
+ - lt
24
+ - lv
25
+ - nl
26
+ - pl
27
+ - pt
28
+ - ro
29
+ - ru
30
+ - sk
31
+ - sl
32
+ - sv
33
+ - tr
34
+ - uk
35
+ - vi
36
+ pipeline_tag: text-to-speech
37
+ library_name: coreml
38
+ datasets: []
39
+ thumbnail: null
40
+ tags:
41
+ - text-to-speech
42
+ - speech
43
+ - audio
44
+ - tts
45
+ - coreml
46
+ - ane
47
+ - apple-silicon
48
+ - flow-matching
49
+ - diffusion
50
+ - multilingual
51
+ - supertonic
52
+ base_model:
53
+ - Supertone/supertonic-3
54
+ ---
55
+
56
+ # **<span style="color:#5DAF8D"> 🧃 supertonic-3: Multilingual Text-to-Speech CoreML </span>**
57
+
58
+ <style>
59
+ img {
60
+ display: inline;
61
+ }
62
+ </style>
63
+
64
+ [![Model architecture](https://img.shields.io/badge/Model_Arch-Flow--Matching%20Diffusion-blue#model-badge)](#model-architecture)
65
+ | [![Sampling rate](https://img.shields.io/badge/Sample_Rate-44.1kHz-green#model-badge)](#model-details)
66
+ | [![Language](https://img.shields.io/badge/Languages-31-blue#model-badge)](#supported-languages)
67
+ | [![Discord](https://img.shields.io/badge/Discord-Join%20Chat-7289da.svg)](https://discord.gg/WNsvaCtmDe)
68
+ | [![GitHub Repo stars](https://img.shields.io/github/stars/FluidInference/FluidAudio?style=flat&logo=github)](https://github.com/FluidInference/FluidAudio)
69
+
70
+ On‑device multilingual TTS model converted to Core ML for Apple platforms.
71
+ This is a hand‑port of [Supertone Supertonic‑3 v1.7.3](https://huggingface.co/Supertone/supertonic-3)
72
+ from ONNX → PyTorch → Core ML, suitable for FluidAudio's TTS pipeline on
73
+ macOS/iOS. 31 languages, 44.1 kHz output, flow‑matching diffusion with
74
+ classifier‑free guidance (8 denoising steps).
75
+
76
+ The conversion script is here:
77
+ https://github.com/FluidInference/mobius/tree/main/models/tts/supertonic-3/coreml
78
+
79
+ And the FluidAudio integration is here:
80
+ https://github.com/FluidInference/FluidAudio/tree/main/Sources/FluidAudio/TTS/Supertonic3
81
+
82
+ ## Highlights
83
+
84
+ - **Core ML**: Runs on‑device (ANE + CPU) on Apple Silicon.
85
+ - **Multilingual**: 31 languages — see [Supported Languages](#supported-languages).
86
+ - **High quality**: 44.1 kHz output via flow‑matching diffusion + ConvNeXt vocoder.
87
+ - **Voice styling**: zero‑shot voice style embeddings (single JSON per voice).
88
+ - **Performance**: end‑to‑end RTFx ≈ 8.5× on M2 (CoreML), ≈ 17–19× on M2 with current ANE assignment (3 of 4 modules on ANE).
89
+ - **Privacy**: No network calls required once models are downloaded.
90
+
91
+ ## Intended Use
92
+
93
+ - **Batch TTS** for full text segments on macOS/iOS.
94
+ - **Local voice synthesis** for note‑taking, accessibility, and creative tools.
95
+ - **Embedded TTS** in production apps via the FluidAudio Swift framework.
96
+
97
+ ## Supported Platforms
98
+
99
+ - macOS 14+ (Apple Silicon recommended)
100
+ - iOS 17+
101
+
102
+ ## Model Details
103
+
104
+ - **Architecture**: Supertonic‑3 v1.7.3 — 4‑stage pipeline:
105
+ 1. `text_encoder` — token embeddings → contextual text features `[B, 256, T]`.
106
+ 2. `duration_predictor` — predicts utterance duration from text features.
107
+ 3. `vector_estimator` — flow‑matching diffusion in latent space
108
+ (8 steps, classifier‑free guidance via batch‑2 duplication, ConvNeXt + cross‑attention to text + style attention).
109
+ 4. `vocoder` — ConvNeXt decoder → 44.1 kHz waveform.
110
+ - **Output audio**: 44.1 kHz mono, Float32 PCM.
111
+ - **Languages**: 31 (see below).
112
+ - **Precision**: FP16 weights and activations (mlprogram, iOS 18+ minimum deployment target).
113
+ - **Granularity**: vocoder frame ≈ 11.6 ms; latent tick ≈ 69.7 ms.
114
+
115
+ ## Supported Languages
116
+
117
+ English, Korean, Japanese, Arabic, Bulgarian, Czech, Danish, German, Greek,
118
+ Spanish, Estonian, Finnish, French, Hindi, Croatian, Hungarian, Indonesian,
119
+ Italian, Lithuanian, Latvian, Dutch, Polish, Portuguese, Romanian, Russian,
120
+ Slovak, Slovenian, Swedish, Turkish, Ukrainian, Vietnamese.
121
+
122
+ ## Performance (Apple M2, macOS 26.5, FP16)
123
+
124
+ | Module | Size | Predict | Compute placement |
125
+ | ------------------ | ----- | ------- | ----------------- |
126
+ | duration_predictor | 1.8 MB| 0.82 ms | CPU (tiny) |
127
+ | text_encoder | 17 MB | 2.15 ms | 62 % ANE |
128
+ | vocoder | 48 MB | 1.17 ms | 100 % ANE |
129
+ | vector_estimator | 122 MB| 9.29 ms | CPU + GPU (see notes) |
130
+
131
+ End‑to‑end on M2: ≈ 0.74 s to synthesize 6.32 s of audio for a single English
132
+ sentence (RTFx ≈ 8.5×), 8 denoising steps. Output verified against
133
+ FluidAudio Parakeet TDT ASR.
134
+
135
+ **Note on `vector_estimator`**: 100 % of its ops are ANE‑eligible after
136
+ the float‑mask + precompute refactor, but Apple's ANECCompile currently
137
+ returns opaque error 11 on this graph and silently falls back to CPU/GPU.
138
+ See `coreml/trials.md` in the conversion repo for the full investigation.
139
+
140
+ ## Files
141
+
142
+ Both `.mlpackage` (Core ML source bundle, includes weights + spec) and the
143
+ precompiled `.mlmodelc` (ready for direct `MLModel(contentsOf:)` load) are
144
+ shipped — use `.mlmodelc` to skip the on‑device compile step on first load.
145
+
146
+ - `TextEncoder.mlpackage` / `TextEncoder.mlmodelc` — fixed `T=128` text input.
147
+ - `DurationPredictor.mlpackage` / `DurationPredictor.mlmodelc` — fixed `T=128` text input.
148
+ - `VectorEstimator.mlpackage` / `VectorEstimator.mlmodelc` — `latent.L` and `text.T` as RangeDim(17..512).
149
+ - `Vocoder.mlpackage` / `Vocoder.mlmodelc` — `latent.L_ttl` as RangeDim(4..512).
150
+ - `tts.json` — token / text frontend configuration.
151
+ - `unicode_indexer.json` — Unicode → token id mapping (multilingual frontend).
152
+ - `voice_styles/M1.json` — example voice style embedding (single male reference).
153
+ - `manifest.json` — file inventory (sha256 + sizes) for both `.mlpackage` and `.mlmodelc`.
154
+
155
+ ## Usage
156
+
157
+ For quickest integration, use the FluidAudio Swift framework which handles
158
+ model loading, text frontend, and the diffusion / vocoder loop.
159
+
160
+ ### Swift (FluidAudio)
161
+
162
+ ```swift
163
+ import AVFoundation
164
+ import FluidAudio
165
+
166
+ Task {
167
+ // Download and load Supertonic-3 models (first run only)
168
+ let models = try await Supertonic3Models.downloadAndLoad()
169
+
170
+ // Initialize the TTS manager
171
+ let tts = Supertonic3Manager(config: .default)
172
+ try await tts.initialize(models: models)
173
+
174
+ // Synthesize speech for some text with a voice style
175
+ let style = try VoiceStyle.load(path: "voice_styles/M1.json")
176
+ let audio = try await tts.synthesize(text: "Hello, world.", style: style)
177
+
178
+ // audio.samples is 44.1 kHz Float32 PCM in [-1, 1]
179
+ try AudioWriter.writeWav(audio.samples, sampleRate: 44_100, to: "hello.wav")
180
+
181
+ tts.cleanup()
182
+ }
183
+ ```
184
+
185
+ For more examples (including CLI usage and benchmarking), see the FluidAudio
186
+ repository: https://github.com/FluidInference/FluidAudio
187
+
188
+ ## Limitations
189
+
190
+ - 44.1 kHz output is high quality but heavier than 16/22.05 kHz TTS — plan
191
+ for the bandwidth and storage cost.
192
+ - `vector_estimator` currently runs on CPU + GPU instead of ANE due to an
193
+ Apple‑side ANE compiler limitation (see [Performance](#performance-apple-m2-macos-265-fp16)).
194
+ - Text frontend currently uses fixed `T=128` token windows; longer text
195
+ must be segmented by the caller.
196
+
197
+ ## License
198
+
199
+ OpenRAIL‑M (inherited from upstream [Supertone/supertonic-3](https://huggingface.co/Supertone/supertonic-3)).
200
+ The Core ML conversion tooling and FluidAudio integration are MIT‑licensed.
201
+ See the [FluidAudio repository](https://github.com/FluidInference/FluidAudio)
202
+ for details and usage guidance.
TextEncoder.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b39c208b46f180dcbf4cfcd3b4437eeb93371fb77d6d636b99ca3ddff5327253
3
+ size 243
TextEncoder.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4349aa84bb4a4731e3be09eb235e1816a3d0c88500b151479e08fe15662a446c
3
+ size 431
TextEncoder.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
TextEncoder.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c7afa5d02426a8a363c9cef0e27ce64a5c57a67d1a59f60aac0ac58ffb9877f
3
+ size 18022016
TextEncoder.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9674a311dd5145e27554c1d021f25531f90ed772a3758ab349b605967faec667
3
+ size 144168
TextEncoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c7afa5d02426a8a363c9cef0e27ce64a5c57a67d1a59f60aac0ac58ffb9877f
3
+ size 18022016
TextEncoder.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "5CB6F436-EBD5-483E-BB13-532FA7BAC495": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Specification",
7
+ "name": "model.mlmodel",
8
+ "path": "com.apple.CoreML/model.mlmodel"
9
+ },
10
+ "7BB55824-4E1A-445F-9CBA-D219959EE4C1": {
11
+ "author": "com.apple.CoreML",
12
+ "description": "CoreML Model Weights",
13
+ "name": "weights",
14
+ "path": "com.apple.CoreML/weights"
15
+ }
16
+ },
17
+ "rootModelIdentifier": "5CB6F436-EBD5-483E-BB13-532FA7BAC495"
18
+ }
VectorEstimator.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f43626d16b133a8b302f4f7a0e6d0f789422b22df135b7af11df25c385bebe0
3
+ size 243
VectorEstimator.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4ae2bd30a9130ac2d09996ad4f63ea2c5703f735d11d9cf7e494d14397fd495
3
+ size 633
VectorEstimator.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
VectorEstimator.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:101cebe7759b5f5e8f188ed3c2bf5f624bbf048aa657fe5bed9c3110c5ce0c26
3
+ size 127648384
VectorEstimator.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c93445cd09b7e6190fd26873cafb4a209fbb24d836b665428f7114ed6b0b41df
3
+ size 328305
VectorEstimator.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:101cebe7759b5f5e8f188ed3c2bf5f624bbf048aa657fe5bed9c3110c5ce0c26
3
+ size 127648384
VectorEstimator.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "BF47852D-32B0-4ACD-B523-68E04EDA15F4": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Specification",
7
+ "name": "model.mlmodel",
8
+ "path": "com.apple.CoreML/model.mlmodel"
9
+ },
10
+ "F0986E1C-EB8F-47CF-848C-CEB3D5B25667": {
11
+ "author": "com.apple.CoreML",
12
+ "description": "CoreML Model Weights",
13
+ "name": "weights",
14
+ "path": "com.apple.CoreML/weights"
15
+ }
16
+ },
17
+ "rootModelIdentifier": "BF47852D-32B0-4ACD-B523-68E04EDA15F4"
18
+ }
Vocoder.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4fe1b825137629a96dc58a1339bc4ece32041b755f99d638f21a153f2e7faed6
3
+ size 243
Vocoder.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbf54fc98879ee1c9a213452e16d09367f3c34ad489b175bcae7ef5f74f8d121
3
+ size 388
Vocoder.mlmodelc/model.mil ADDED
@@ -0,0 +1,465 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ program(1.3)
2
+ [buildInfo = dict<string, string>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.5.1"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
3
+ {
4
+ func main<ios18>(tensor<fp32, [1, 144, ?]> latent) [FlexibleShapeInformation = tuple<tuple<string, dict<string, tensor<int32, [?]>>>, tuple<string, dict<string, list<tensor<int32, [2]>, ?>>>>((("DefaultShapes", {{"latent", [1, 144, 4]}}), ("RangeDims", {{"latent", [[1, 1], [144, 144], [4, 512]]}})))] {
5
+ string latent_to_fp16_dtype_0 = const()[name = string("latent_to_fp16_dtype_0"), val = string("fp16")];
6
+ fp16 _inversed_x_1_y_0_to_fp16 = const()[name = string("_inversed_x_1_y_0_to_fp16"), val = fp16(0x1p+2)];
7
+ tensor<fp16, [1, 144, ?]> latent_to_fp16 = cast(dtype = latent_to_fp16_dtype_0, x = latent)[name = string("cast_6")];
8
+ tensor<fp16, [1, 144, ?]> _inversed_x_1_cast_fp16 = mul(x = latent_to_fp16, y = _inversed_x_1_y_0_to_fp16)[name = string("_inversed_x_1_cast_fp16")];
9
+ tensor<int32, [4]> var_33 = const()[name = string("op_33"), val = tensor<int32, [4]>([1, 24, 6, -1])];
10
+ tensor<fp16, [1, 24, 6, ?]> x_3_cast_fp16 = reshape(shape = var_33, x = _inversed_x_1_cast_fp16)[name = string("x_3_cast_fp16")];
11
+ tensor<int32, [4]> var_37_perm_0 = const()[name = string("op_37_perm_0"), val = tensor<int32, [4]>([0, 1, 3, 2])];
12
+ tensor<int32, [3]> var_43 = const()[name = string("op_43"), val = tensor<int32, [3]>([1, 24, -1])];
13
+ tensor<fp16, [1, 24, ?, 6]> var_37_cast_fp16 = transpose(perm = var_37_perm_0, x = x_3_cast_fp16)[name = string("transpose_21")];
14
+ tensor<fp16, [1, 24, ?]> x_7_cast_fp16 = reshape(shape = var_43, x = var_37_cast_fp16)[name = string("x_7_cast_fp16")];
15
+ tensor<fp16, [1, 24, 1]> latent_std_to_fp16 = const()[name = string("latent_std_to_fp16"), val = tensor<fp16, [1, 24, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
16
+ tensor<fp16, [1, 24, ?]> var_45_cast_fp16 = mul(x = x_7_cast_fp16, y = latent_std_to_fp16)[name = string("op_45_cast_fp16")];
17
+ tensor<fp16, [1, 24, 1]> latent_mean_to_fp16 = const()[name = string("latent_mean_to_fp16"), val = tensor<fp16, [1, 24, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(192)))];
18
+ tensor<fp16, [1, 24, ?]> input_1_cast_fp16 = add(x = var_45_cast_fp16, y = latent_mean_to_fp16)[name = string("input_1_cast_fp16")];
19
+ tensor<int32, [6]> input_3_pad_0 = const()[name = string("input_3_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 6, 0])];
20
+ string input_3_mode_0 = const()[name = string("input_3_mode_0"), val = string("replicate")];
21
+ fp16 const_0_to_fp16 = const()[name = string("const_0_to_fp16"), val = fp16(0x0p+0)];
22
+ tensor<fp16, [1, 24, ?]> input_3_cast_fp16 = pad(constant_val = const_0_to_fp16, mode = input_3_mode_0, pad = input_3_pad_0, x = input_1_cast_fp16)[name = string("input_3_cast_fp16")];
23
+ string input_5_pad_type_0 = const()[name = string("input_5_pad_type_0"), val = string("valid")];
24
+ tensor<int32, [1]> input_5_strides_0 = const()[name = string("input_5_strides_0"), val = tensor<int32, [1]>([1])];
25
+ tensor<int32, [2]> input_5_pad_0 = const()[name = string("input_5_pad_0"), val = tensor<int32, [2]>([0, 0])];
26
+ tensor<int32, [1]> input_5_dilations_0 = const()[name = string("input_5_dilations_0"), val = tensor<int32, [1]>([1])];
27
+ int32 input_5_groups_0 = const()[name = string("input_5_groups_0"), val = int32(1)];
28
+ tensor<fp16, [512, 24, 7]> embed_net_weight_to_fp16 = const()[name = string("embed_net_weight_to_fp16"), val = tensor<fp16, [512, 24, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(320)))];
29
+ tensor<fp16, [512]> embed_net_bias_to_fp16 = const()[name = string("embed_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(172416)))];
30
+ tensor<fp16, [1, 512, ?]> input_5_cast_fp16 = conv(bias = embed_net_bias_to_fp16, dilations = input_5_dilations_0, groups = input_5_groups_0, pad = input_5_pad_0, pad_type = input_5_pad_type_0, strides = input_5_strides_0, weight = embed_net_weight_to_fp16, x = input_3_cast_fp16)[name = string("input_5_cast_fp16")];
31
+ tensor<int32, [6]> input_7_pad_0 = const()[name = string("input_7_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 6, 0])];
32
+ string input_7_mode_0 = const()[name = string("input_7_mode_0"), val = string("replicate")];
33
+ fp16 const_1_to_fp16 = const()[name = string("const_1_to_fp16"), val = fp16(0x0p+0)];
34
+ tensor<fp16, [1, 512, ?]> input_7_cast_fp16 = pad(constant_val = const_1_to_fp16, mode = input_7_mode_0, pad = input_7_pad_0, x = input_5_cast_fp16)[name = string("input_7_cast_fp16")];
35
+ string x_9_pad_type_0 = const()[name = string("x_9_pad_type_0"), val = string("valid")];
36
+ int32 x_9_groups_0 = const()[name = string("x_9_groups_0"), val = int32(512)];
37
+ tensor<int32, [1]> x_9_strides_0 = const()[name = string("x_9_strides_0"), val = tensor<int32, [1]>([1])];
38
+ tensor<int32, [2]> x_9_pad_0 = const()[name = string("x_9_pad_0"), val = tensor<int32, [2]>([0, 0])];
39
+ tensor<int32, [1]> x_9_dilations_0 = const()[name = string("x_9_dilations_0"), val = tensor<int32, [1]>([1])];
40
+ tensor<fp16, [512, 1, 7]> convnext_0_dwconv_net_weight_to_fp16 = const()[name = string("convnext_0_dwconv_net_weight_to_fp16"), val = tensor<fp16, [512, 1, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(173504)))];
41
+ tensor<fp16, [512]> convnext_0_dwconv_net_bias_to_fp16 = const()[name = string("convnext_0_dwconv_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(180736)))];
42
+ tensor<fp16, [1, 512, ?]> x_9_cast_fp16 = conv(bias = convnext_0_dwconv_net_bias_to_fp16, dilations = x_9_dilations_0, groups = x_9_groups_0, pad = x_9_pad_0, pad_type = x_9_pad_type_0, strides = x_9_strides_0, weight = convnext_0_dwconv_net_weight_to_fp16, x = input_7_cast_fp16)[name = string("x_9_cast_fp16")];
43
+ tensor<int32, [3]> input_9_perm_0 = const()[name = string("input_9_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
44
+ tensor<int32, [1]> var_96_axes_0 = const()[name = string("op_96_axes_0"), val = tensor<int32, [1]>([-1])];
45
+ tensor<fp16, [512]> convnext_0_norm_norm_weight_to_fp16 = const()[name = string("convnext_0_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(181824)))];
46
+ tensor<fp16, [512]> convnext_0_norm_norm_bias_to_fp16 = const()[name = string("convnext_0_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(182912)))];
47
+ fp16 var_67_to_fp16 = const()[name = string("op_67_to_fp16"), val = fp16(0x1.5p-17)];
48
+ tensor<fp16, [1, ?, 512]> input_9_cast_fp16 = transpose(perm = input_9_perm_0, x = x_9_cast_fp16)[name = string("transpose_20")];
49
+ tensor<fp16, [1, ?, 512]> var_96_cast_fp16 = layer_norm(axes = var_96_axes_0, beta = convnext_0_norm_norm_bias_to_fp16, epsilon = var_67_to_fp16, gamma = convnext_0_norm_norm_weight_to_fp16, x = input_9_cast_fp16)[name = string("op_96_cast_fp16")];
50
+ tensor<int32, [3]> input_11_perm_0 = const()[name = string("input_11_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
51
+ string h_1_pad_type_0 = const()[name = string("h_1_pad_type_0"), val = string("valid")];
52
+ tensor<int32, [1]> h_1_strides_0 = const()[name = string("h_1_strides_0"), val = tensor<int32, [1]>([1])];
53
+ tensor<int32, [2]> h_1_pad_0 = const()[name = string("h_1_pad_0"), val = tensor<int32, [2]>([0, 0])];
54
+ tensor<int32, [1]> h_1_dilations_0 = const()[name = string("h_1_dilations_0"), val = tensor<int32, [1]>([1])];
55
+ int32 h_1_groups_0 = const()[name = string("h_1_groups_0"), val = int32(1)];
56
+ tensor<fp16, [2048, 512, 1]> convnext_0_pwconv1_weight_to_fp16 = const()[name = string("convnext_0_pwconv1_weight_to_fp16"), val = tensor<fp16, [2048, 512, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(184000)))];
57
+ tensor<fp16, [2048]> convnext_0_pwconv1_bias_to_fp16 = const()[name = string("convnext_0_pwconv1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2281216)))];
58
+ tensor<fp16, [1, 512, ?]> input_11_cast_fp16 = transpose(perm = input_11_perm_0, x = var_96_cast_fp16)[name = string("transpose_19")];
59
+ tensor<fp16, [1, 2048, ?]> h_1_cast_fp16 = conv(bias = convnext_0_pwconv1_bias_to_fp16, dilations = h_1_dilations_0, groups = h_1_groups_0, pad = h_1_pad_0, pad_type = h_1_pad_type_0, strides = h_1_strides_0, weight = convnext_0_pwconv1_weight_to_fp16, x = input_11_cast_fp16)[name = string("h_1_cast_fp16")];
60
+ string input_13_mode_0 = const()[name = string("input_13_mode_0"), val = string("EXACT")];
61
+ tensor<fp16, [1, 2048, ?]> input_13_cast_fp16 = gelu(mode = input_13_mode_0, x = h_1_cast_fp16)[name = string("input_13_cast_fp16")];
62
+ string h_3_pad_type_0 = const()[name = string("h_3_pad_type_0"), val = string("valid")];
63
+ tensor<int32, [1]> h_3_strides_0 = const()[name = string("h_3_strides_0"), val = tensor<int32, [1]>([1])];
64
+ tensor<int32, [2]> h_3_pad_0 = const()[name = string("h_3_pad_0"), val = tensor<int32, [2]>([0, 0])];
65
+ tensor<int32, [1]> h_3_dilations_0 = const()[name = string("h_3_dilations_0"), val = tensor<int32, [1]>([1])];
66
+ int32 h_3_groups_0 = const()[name = string("h_3_groups_0"), val = int32(1)];
67
+ tensor<fp16, [512, 2048, 1]> var_113_weight_0_to_fp16 = const()[name = string("op_113_weight_0_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2285376)))];
68
+ tensor<fp16, [512]> var_113_bias_0_to_fp16 = const()[name = string("op_113_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4382592)))];
69
+ tensor<fp16, [1, 512, ?]> var_113_cast_fp16 = conv(bias = var_113_bias_0_to_fp16, dilations = h_3_dilations_0, groups = h_3_groups_0, pad = h_3_pad_0, pad_type = h_3_pad_type_0, strides = h_3_strides_0, weight = var_113_weight_0_to_fp16, x = input_13_cast_fp16)[name = string("op_113_cast_fp16")];
70
+ tensor<fp16, [1, 512, ?]> input_15_cast_fp16 = add(x = input_5_cast_fp16, y = var_113_cast_fp16)[name = string("input_15_cast_fp16")];
71
+ tensor<int32, [6]> input_17_pad_0 = const()[name = string("input_17_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 12, 0])];
72
+ string input_17_mode_0 = const()[name = string("input_17_mode_0"), val = string("replicate")];
73
+ fp16 const_2_to_fp16 = const()[name = string("const_2_to_fp16"), val = fp16(0x0p+0)];
74
+ tensor<fp16, [1, 512, ?]> input_17_cast_fp16 = pad(constant_val = const_2_to_fp16, mode = input_17_mode_0, pad = input_17_pad_0, x = input_15_cast_fp16)[name = string("input_17_cast_fp16")];
75
+ string x_11_pad_type_0 = const()[name = string("x_11_pad_type_0"), val = string("valid")];
76
+ tensor<int32, [1]> x_11_dilations_0 = const()[name = string("x_11_dilations_0"), val = tensor<int32, [1]>([2])];
77
+ int32 x_11_groups_0 = const()[name = string("x_11_groups_0"), val = int32(512)];
78
+ tensor<int32, [1]> x_11_strides_0 = const()[name = string("x_11_strides_0"), val = tensor<int32, [1]>([1])];
79
+ tensor<int32, [2]> x_11_pad_0 = const()[name = string("x_11_pad_0"), val = tensor<int32, [2]>([0, 0])];
80
+ tensor<fp16, [512, 1, 7]> convnext_1_dwconv_net_weight_to_fp16 = const()[name = string("convnext_1_dwconv_net_weight_to_fp16"), val = tensor<fp16, [512, 1, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4383680)))];
81
+ tensor<fp16, [512]> convnext_1_dwconv_net_bias_to_fp16 = const()[name = string("convnext_1_dwconv_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4390912)))];
82
+ tensor<fp16, [1, 512, ?]> x_11_cast_fp16 = conv(bias = convnext_1_dwconv_net_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 = convnext_1_dwconv_net_weight_to_fp16, x = input_17_cast_fp16)[name = string("x_11_cast_fp16")];
83
+ tensor<int32, [3]> input_19_perm_0 = const()[name = string("input_19_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
84
+ tensor<int32, [1]> var_146_axes_0 = const()[name = string("op_146_axes_0"), val = tensor<int32, [1]>([-1])];
85
+ tensor<fp16, [512]> convnext_1_norm_norm_weight_to_fp16 = const()[name = string("convnext_1_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4392000)))];
86
+ tensor<fp16, [512]> convnext_1_norm_norm_bias_to_fp16 = const()[name = string("convnext_1_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4393088)))];
87
+ fp16 var_116_to_fp16 = const()[name = string("op_116_to_fp16"), val = fp16(0x1.5p-17)];
88
+ tensor<fp16, [1, ?, 512]> input_19_cast_fp16 = transpose(perm = input_19_perm_0, x = x_11_cast_fp16)[name = string("transpose_18")];
89
+ tensor<fp16, [1, ?, 512]> var_146_cast_fp16 = layer_norm(axes = var_146_axes_0, beta = convnext_1_norm_norm_bias_to_fp16, epsilon = var_116_to_fp16, gamma = convnext_1_norm_norm_weight_to_fp16, x = input_19_cast_fp16)[name = string("op_146_cast_fp16")];
90
+ tensor<int32, [3]> input_21_perm_0 = const()[name = string("input_21_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
91
+ string h_5_pad_type_0 = const()[name = string("h_5_pad_type_0"), val = string("valid")];
92
+ tensor<int32, [1]> h_5_strides_0 = const()[name = string("h_5_strides_0"), val = tensor<int32, [1]>([1])];
93
+ tensor<int32, [2]> h_5_pad_0 = const()[name = string("h_5_pad_0"), val = tensor<int32, [2]>([0, 0])];
94
+ tensor<int32, [1]> h_5_dilations_0 = const()[name = string("h_5_dilations_0"), val = tensor<int32, [1]>([1])];
95
+ int32 h_5_groups_0 = const()[name = string("h_5_groups_0"), val = int32(1)];
96
+ tensor<fp16, [2048, 512, 1]> convnext_1_pwconv1_weight_to_fp16 = const()[name = string("convnext_1_pwconv1_weight_to_fp16"), val = tensor<fp16, [2048, 512, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4394176)))];
97
+ tensor<fp16, [2048]> convnext_1_pwconv1_bias_to_fp16 = const()[name = string("convnext_1_pwconv1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6491392)))];
98
+ tensor<fp16, [1, 512, ?]> input_21_cast_fp16 = transpose(perm = input_21_perm_0, x = var_146_cast_fp16)[name = string("transpose_17")];
99
+ tensor<fp16, [1, 2048, ?]> h_5_cast_fp16 = conv(bias = convnext_1_pwconv1_bias_to_fp16, dilations = h_5_dilations_0, groups = h_5_groups_0, pad = h_5_pad_0, pad_type = h_5_pad_type_0, strides = h_5_strides_0, weight = convnext_1_pwconv1_weight_to_fp16, x = input_21_cast_fp16)[name = string("h_5_cast_fp16")];
100
+ string input_23_mode_0 = const()[name = string("input_23_mode_0"), val = string("EXACT")];
101
+ tensor<fp16, [1, 2048, ?]> input_23_cast_fp16 = gelu(mode = input_23_mode_0, x = h_5_cast_fp16)[name = string("input_23_cast_fp16")];
102
+ string h_7_pad_type_0 = const()[name = string("h_7_pad_type_0"), val = string("valid")];
103
+ tensor<int32, [1]> h_7_strides_0 = const()[name = string("h_7_strides_0"), val = tensor<int32, [1]>([1])];
104
+ tensor<int32, [2]> h_7_pad_0 = const()[name = string("h_7_pad_0"), val = tensor<int32, [2]>([0, 0])];
105
+ tensor<int32, [1]> h_7_dilations_0 = const()[name = string("h_7_dilations_0"), val = tensor<int32, [1]>([1])];
106
+ int32 h_7_groups_0 = const()[name = string("h_7_groups_0"), val = int32(1)];
107
+ tensor<fp16, [512, 2048, 1]> var_163_weight_0_to_fp16 = const()[name = string("op_163_weight_0_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6495552)))];
108
+ tensor<fp16, [512]> var_163_bias_0_to_fp16 = const()[name = string("op_163_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8592768)))];
109
+ tensor<fp16, [1, 512, ?]> var_163_cast_fp16 = conv(bias = var_163_bias_0_to_fp16, dilations = h_7_dilations_0, groups = h_7_groups_0, pad = h_7_pad_0, pad_type = h_7_pad_type_0, strides = h_7_strides_0, weight = var_163_weight_0_to_fp16, x = input_23_cast_fp16)[name = string("op_163_cast_fp16")];
110
+ tensor<fp16, [1, 512, ?]> input_25_cast_fp16 = add(x = input_15_cast_fp16, y = var_163_cast_fp16)[name = string("input_25_cast_fp16")];
111
+ tensor<int32, [6]> input_27_pad_0 = const()[name = string("input_27_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 24, 0])];
112
+ string input_27_mode_0 = const()[name = string("input_27_mode_0"), val = string("replicate")];
113
+ fp16 const_3_to_fp16 = const()[name = string("const_3_to_fp16"), val = fp16(0x0p+0)];
114
+ tensor<fp16, [1, 512, ?]> input_27_cast_fp16 = pad(constant_val = const_3_to_fp16, mode = input_27_mode_0, pad = input_27_pad_0, x = input_25_cast_fp16)[name = string("input_27_cast_fp16")];
115
+ string x_13_pad_type_0 = const()[name = string("x_13_pad_type_0"), val = string("valid")];
116
+ tensor<int32, [1]> x_13_dilations_0 = const()[name = string("x_13_dilations_0"), val = tensor<int32, [1]>([4])];
117
+ int32 x_13_groups_0 = const()[name = string("x_13_groups_0"), val = int32(512)];
118
+ tensor<int32, [1]> x_13_strides_0 = const()[name = string("x_13_strides_0"), val = tensor<int32, [1]>([1])];
119
+ tensor<int32, [2]> x_13_pad_0 = const()[name = string("x_13_pad_0"), val = tensor<int32, [2]>([0, 0])];
120
+ tensor<fp16, [512, 1, 7]> convnext_2_dwconv_net_weight_to_fp16 = const()[name = string("convnext_2_dwconv_net_weight_to_fp16"), val = tensor<fp16, [512, 1, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8593856)))];
121
+ tensor<fp16, [512]> convnext_2_dwconv_net_bias_to_fp16 = const()[name = string("convnext_2_dwconv_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8601088)))];
122
+ tensor<fp16, [1, 512, ?]> x_13_cast_fp16 = conv(bias = convnext_2_dwconv_net_bias_to_fp16, dilations = x_13_dilations_0, groups = x_13_groups_0, pad = x_13_pad_0, pad_type = x_13_pad_type_0, strides = x_13_strides_0, weight = convnext_2_dwconv_net_weight_to_fp16, x = input_27_cast_fp16)[name = string("x_13_cast_fp16")];
123
+ tensor<int32, [3]> input_29_perm_0 = const()[name = string("input_29_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
124
+ tensor<int32, [1]> var_197_axes_0 = const()[name = string("op_197_axes_0"), val = tensor<int32, [1]>([-1])];
125
+ tensor<fp16, [512]> convnext_2_norm_norm_weight_to_fp16 = const()[name = string("convnext_2_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8602176)))];
126
+ tensor<fp16, [512]> convnext_2_norm_norm_bias_to_fp16 = const()[name = string("convnext_2_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8603264)))];
127
+ fp16 var_167_to_fp16 = const()[name = string("op_167_to_fp16"), val = fp16(0x1.5p-17)];
128
+ tensor<fp16, [1, ?, 512]> input_29_cast_fp16 = transpose(perm = input_29_perm_0, x = x_13_cast_fp16)[name = string("transpose_16")];
129
+ tensor<fp16, [1, ?, 512]> var_197_cast_fp16 = layer_norm(axes = var_197_axes_0, beta = convnext_2_norm_norm_bias_to_fp16, epsilon = var_167_to_fp16, gamma = convnext_2_norm_norm_weight_to_fp16, x = input_29_cast_fp16)[name = string("op_197_cast_fp16")];
130
+ tensor<int32, [3]> input_31_perm_0 = const()[name = string("input_31_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
131
+ string h_9_pad_type_0 = const()[name = string("h_9_pad_type_0"), val = string("valid")];
132
+ tensor<int32, [1]> h_9_strides_0 = const()[name = string("h_9_strides_0"), val = tensor<int32, [1]>([1])];
133
+ tensor<int32, [2]> h_9_pad_0 = const()[name = string("h_9_pad_0"), val = tensor<int32, [2]>([0, 0])];
134
+ tensor<int32, [1]> h_9_dilations_0 = const()[name = string("h_9_dilations_0"), val = tensor<int32, [1]>([1])];
135
+ int32 h_9_groups_0 = const()[name = string("h_9_groups_0"), val = int32(1)];
136
+ tensor<fp16, [2048, 512, 1]> convnext_2_pwconv1_weight_to_fp16 = const()[name = string("convnext_2_pwconv1_weight_to_fp16"), val = tensor<fp16, [2048, 512, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8604352)))];
137
+ tensor<fp16, [2048]> convnext_2_pwconv1_bias_to_fp16 = const()[name = string("convnext_2_pwconv1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10701568)))];
138
+ tensor<fp16, [1, 512, ?]> input_31_cast_fp16 = transpose(perm = input_31_perm_0, x = var_197_cast_fp16)[name = string("transpose_15")];
139
+ tensor<fp16, [1, 2048, ?]> h_9_cast_fp16 = conv(bias = convnext_2_pwconv1_bias_to_fp16, dilations = h_9_dilations_0, groups = h_9_groups_0, pad = h_9_pad_0, pad_type = h_9_pad_type_0, strides = h_9_strides_0, weight = convnext_2_pwconv1_weight_to_fp16, x = input_31_cast_fp16)[name = string("h_9_cast_fp16")];
140
+ string input_33_mode_0 = const()[name = string("input_33_mode_0"), val = string("EXACT")];
141
+ tensor<fp16, [1, 2048, ?]> input_33_cast_fp16 = gelu(mode = input_33_mode_0, x = h_9_cast_fp16)[name = string("input_33_cast_fp16")];
142
+ string h_11_pad_type_0 = const()[name = string("h_11_pad_type_0"), val = string("valid")];
143
+ tensor<int32, [1]> h_11_strides_0 = const()[name = string("h_11_strides_0"), val = tensor<int32, [1]>([1])];
144
+ tensor<int32, [2]> h_11_pad_0 = const()[name = string("h_11_pad_0"), val = tensor<int32, [2]>([0, 0])];
145
+ tensor<int32, [1]> h_11_dilations_0 = const()[name = string("h_11_dilations_0"), val = tensor<int32, [1]>([1])];
146
+ int32 h_11_groups_0 = const()[name = string("h_11_groups_0"), val = int32(1)];
147
+ tensor<fp16, [512, 2048, 1]> var_214_weight_0_to_fp16 = const()[name = string("op_214_weight_0_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10705728)))];
148
+ tensor<fp16, [512]> var_214_bias_0_to_fp16 = const()[name = string("op_214_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(12802944)))];
149
+ tensor<fp16, [1, 512, ?]> var_214_cast_fp16 = conv(bias = var_214_bias_0_to_fp16, dilations = h_11_dilations_0, groups = h_11_groups_0, pad = h_11_pad_0, pad_type = h_11_pad_type_0, strides = h_11_strides_0, weight = var_214_weight_0_to_fp16, x = input_33_cast_fp16)[name = string("op_214_cast_fp16")];
150
+ tensor<fp16, [1, 512, ?]> input_35_cast_fp16 = add(x = input_25_cast_fp16, y = var_214_cast_fp16)[name = string("input_35_cast_fp16")];
151
+ tensor<int32, [6]> input_37_pad_0 = const()[name = string("input_37_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 6, 0])];
152
+ string input_37_mode_0 = const()[name = string("input_37_mode_0"), val = string("replicate")];
153
+ fp16 const_4_to_fp16 = const()[name = string("const_4_to_fp16"), val = fp16(0x0p+0)];
154
+ tensor<fp16, [1, 512, ?]> input_37_cast_fp16 = pad(constant_val = const_4_to_fp16, mode = input_37_mode_0, pad = input_37_pad_0, x = input_35_cast_fp16)[name = string("input_37_cast_fp16")];
155
+ string x_15_pad_type_0 = const()[name = string("x_15_pad_type_0"), val = string("valid")];
156
+ int32 x_15_groups_0 = const()[name = string("x_15_groups_0"), val = int32(512)];
157
+ tensor<int32, [1]> x_15_strides_0 = const()[name = string("x_15_strides_0"), val = tensor<int32, [1]>([1])];
158
+ tensor<int32, [2]> x_15_pad_0 = const()[name = string("x_15_pad_0"), val = tensor<int32, [2]>([0, 0])];
159
+ tensor<int32, [1]> x_15_dilations_0 = const()[name = string("x_15_dilations_0"), val = tensor<int32, [1]>([1])];
160
+ tensor<fp16, [512, 1, 7]> convnext_3_dwconv_net_weight_to_fp16 = const()[name = string("convnext_3_dwconv_net_weight_to_fp16"), val = tensor<fp16, [512, 1, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(12804032)))];
161
+ tensor<fp16, [512]> convnext_3_dwconv_net_bias_to_fp16 = const()[name = string("convnext_3_dwconv_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(12811264)))];
162
+ tensor<fp16, [1, 512, ?]> x_15_cast_fp16 = conv(bias = convnext_3_dwconv_net_bias_to_fp16, dilations = x_15_dilations_0, groups = x_15_groups_0, pad = x_15_pad_0, pad_type = x_15_pad_type_0, strides = x_15_strides_0, weight = convnext_3_dwconv_net_weight_to_fp16, x = input_37_cast_fp16)[name = string("x_15_cast_fp16")];
163
+ tensor<int32, [3]> input_39_perm_0 = const()[name = string("input_39_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
164
+ tensor<int32, [1]> var_247_axes_0 = const()[name = string("op_247_axes_0"), val = tensor<int32, [1]>([-1])];
165
+ tensor<fp16, [512]> convnext_3_norm_norm_weight_to_fp16 = const()[name = string("convnext_3_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(12812352)))];
166
+ tensor<fp16, [512]> convnext_3_norm_norm_bias_to_fp16 = const()[name = string("convnext_3_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(12813440)))];
167
+ fp16 var_218_to_fp16 = const()[name = string("op_218_to_fp16"), val = fp16(0x1.5p-17)];
168
+ tensor<fp16, [1, ?, 512]> input_39_cast_fp16 = transpose(perm = input_39_perm_0, x = x_15_cast_fp16)[name = string("transpose_14")];
169
+ tensor<fp16, [1, ?, 512]> var_247_cast_fp16 = layer_norm(axes = var_247_axes_0, beta = convnext_3_norm_norm_bias_to_fp16, epsilon = var_218_to_fp16, gamma = convnext_3_norm_norm_weight_to_fp16, x = input_39_cast_fp16)[name = string("op_247_cast_fp16")];
170
+ tensor<int32, [3]> input_41_perm_0 = const()[name = string("input_41_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
171
+ string h_13_pad_type_0 = const()[name = string("h_13_pad_type_0"), val = string("valid")];
172
+ tensor<int32, [1]> h_13_strides_0 = const()[name = string("h_13_strides_0"), val = tensor<int32, [1]>([1])];
173
+ tensor<int32, [2]> h_13_pad_0 = const()[name = string("h_13_pad_0"), val = tensor<int32, [2]>([0, 0])];
174
+ tensor<int32, [1]> h_13_dilations_0 = const()[name = string("h_13_dilations_0"), val = tensor<int32, [1]>([1])];
175
+ int32 h_13_groups_0 = const()[name = string("h_13_groups_0"), val = int32(1)];
176
+ tensor<fp16, [2048, 512, 1]> convnext_3_pwconv1_weight_to_fp16 = const()[name = string("convnext_3_pwconv1_weight_to_fp16"), val = tensor<fp16, [2048, 512, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(12814528)))];
177
+ tensor<fp16, [2048]> convnext_3_pwconv1_bias_to_fp16 = const()[name = string("convnext_3_pwconv1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14911744)))];
178
+ tensor<fp16, [1, 512, ?]> input_41_cast_fp16 = transpose(perm = input_41_perm_0, x = var_247_cast_fp16)[name = string("transpose_13")];
179
+ tensor<fp16, [1, 2048, ?]> h_13_cast_fp16 = conv(bias = convnext_3_pwconv1_bias_to_fp16, dilations = h_13_dilations_0, groups = h_13_groups_0, pad = h_13_pad_0, pad_type = h_13_pad_type_0, strides = h_13_strides_0, weight = convnext_3_pwconv1_weight_to_fp16, x = input_41_cast_fp16)[name = string("h_13_cast_fp16")];
180
+ string input_43_mode_0 = const()[name = string("input_43_mode_0"), val = string("EXACT")];
181
+ tensor<fp16, [1, 2048, ?]> input_43_cast_fp16 = gelu(mode = input_43_mode_0, x = h_13_cast_fp16)[name = string("input_43_cast_fp16")];
182
+ string h_15_pad_type_0 = const()[name = string("h_15_pad_type_0"), val = string("valid")];
183
+ tensor<int32, [1]> h_15_strides_0 = const()[name = string("h_15_strides_0"), val = tensor<int32, [1]>([1])];
184
+ tensor<int32, [2]> h_15_pad_0 = const()[name = string("h_15_pad_0"), val = tensor<int32, [2]>([0, 0])];
185
+ tensor<int32, [1]> h_15_dilations_0 = const()[name = string("h_15_dilations_0"), val = tensor<int32, [1]>([1])];
186
+ int32 h_15_groups_0 = const()[name = string("h_15_groups_0"), val = int32(1)];
187
+ tensor<fp16, [512, 2048, 1]> var_264_weight_0_to_fp16 = const()[name = string("op_264_weight_0_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14915904)))];
188
+ tensor<fp16, [512]> var_264_bias_0_to_fp16 = const()[name = string("op_264_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(17013120)))];
189
+ tensor<fp16, [1, 512, ?]> var_264_cast_fp16 = conv(bias = var_264_bias_0_to_fp16, dilations = h_15_dilations_0, groups = h_15_groups_0, pad = h_15_pad_0, pad_type = h_15_pad_type_0, strides = h_15_strides_0, weight = var_264_weight_0_to_fp16, x = input_43_cast_fp16)[name = string("op_264_cast_fp16")];
190
+ tensor<fp16, [1, 512, ?]> input_45_cast_fp16 = add(x = input_35_cast_fp16, y = var_264_cast_fp16)[name = string("input_45_cast_fp16")];
191
+ tensor<int32, [6]> input_47_pad_0 = const()[name = string("input_47_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 12, 0])];
192
+ string input_47_mode_0 = const()[name = string("input_47_mode_0"), val = string("replicate")];
193
+ fp16 const_5_to_fp16 = const()[name = string("const_5_to_fp16"), val = fp16(0x0p+0)];
194
+ tensor<fp16, [1, 512, ?]> input_47_cast_fp16 = pad(constant_val = const_5_to_fp16, mode = input_47_mode_0, pad = input_47_pad_0, x = input_45_cast_fp16)[name = string("input_47_cast_fp16")];
195
+ string x_17_pad_type_0 = const()[name = string("x_17_pad_type_0"), val = string("valid")];
196
+ tensor<int32, [1]> x_17_dilations_0 = const()[name = string("x_17_dilations_0"), val = tensor<int32, [1]>([2])];
197
+ int32 x_17_groups_0 = const()[name = string("x_17_groups_0"), val = int32(512)];
198
+ tensor<int32, [1]> x_17_strides_0 = const()[name = string("x_17_strides_0"), val = tensor<int32, [1]>([1])];
199
+ tensor<int32, [2]> x_17_pad_0 = const()[name = string("x_17_pad_0"), val = tensor<int32, [2]>([0, 0])];
200
+ tensor<fp16, [512, 1, 7]> convnext_4_dwconv_net_weight_to_fp16 = const()[name = string("convnext_4_dwconv_net_weight_to_fp16"), val = tensor<fp16, [512, 1, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(17014208)))];
201
+ tensor<fp16, [512]> convnext_4_dwconv_net_bias_to_fp16 = const()[name = string("convnext_4_dwconv_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(17021440)))];
202
+ tensor<fp16, [1, 512, ?]> x_17_cast_fp16 = conv(bias = convnext_4_dwconv_net_bias_to_fp16, dilations = x_17_dilations_0, groups = x_17_groups_0, pad = x_17_pad_0, pad_type = x_17_pad_type_0, strides = x_17_strides_0, weight = convnext_4_dwconv_net_weight_to_fp16, x = input_47_cast_fp16)[name = string("x_17_cast_fp16")];
203
+ tensor<int32, [3]> input_49_perm_0 = const()[name = string("input_49_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
204
+ tensor<int32, [1]> var_297_axes_0 = const()[name = string("op_297_axes_0"), val = tensor<int32, [1]>([-1])];
205
+ tensor<fp16, [512]> convnext_4_norm_norm_weight_to_fp16 = const()[name = string("convnext_4_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(17022528)))];
206
+ tensor<fp16, [512]> convnext_4_norm_norm_bias_to_fp16 = const()[name = string("convnext_4_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(17023616)))];
207
+ fp16 var_267_to_fp16 = const()[name = string("op_267_to_fp16"), val = fp16(0x1.5p-17)];
208
+ tensor<fp16, [1, ?, 512]> input_49_cast_fp16 = transpose(perm = input_49_perm_0, x = x_17_cast_fp16)[name = string("transpose_12")];
209
+ tensor<fp16, [1, ?, 512]> var_297_cast_fp16 = layer_norm(axes = var_297_axes_0, beta = convnext_4_norm_norm_bias_to_fp16, epsilon = var_267_to_fp16, gamma = convnext_4_norm_norm_weight_to_fp16, x = input_49_cast_fp16)[name = string("op_297_cast_fp16")];
210
+ tensor<int32, [3]> input_51_perm_0 = const()[name = string("input_51_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
211
+ string h_17_pad_type_0 = const()[name = string("h_17_pad_type_0"), val = string("valid")];
212
+ tensor<int32, [1]> h_17_strides_0 = const()[name = string("h_17_strides_0"), val = tensor<int32, [1]>([1])];
213
+ tensor<int32, [2]> h_17_pad_0 = const()[name = string("h_17_pad_0"), val = tensor<int32, [2]>([0, 0])];
214
+ tensor<int32, [1]> h_17_dilations_0 = const()[name = string("h_17_dilations_0"), val = tensor<int32, [1]>([1])];
215
+ int32 h_17_groups_0 = const()[name = string("h_17_groups_0"), val = int32(1)];
216
+ tensor<fp16, [2048, 512, 1]> convnext_4_pwconv1_weight_to_fp16 = const()[name = string("convnext_4_pwconv1_weight_to_fp16"), val = tensor<fp16, [2048, 512, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(17024704)))];
217
+ tensor<fp16, [2048]> convnext_4_pwconv1_bias_to_fp16 = const()[name = string("convnext_4_pwconv1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(19121920)))];
218
+ tensor<fp16, [1, 512, ?]> input_51_cast_fp16 = transpose(perm = input_51_perm_0, x = var_297_cast_fp16)[name = string("transpose_11")];
219
+ tensor<fp16, [1, 2048, ?]> h_17_cast_fp16 = conv(bias = convnext_4_pwconv1_bias_to_fp16, dilations = h_17_dilations_0, groups = h_17_groups_0, pad = h_17_pad_0, pad_type = h_17_pad_type_0, strides = h_17_strides_0, weight = convnext_4_pwconv1_weight_to_fp16, x = input_51_cast_fp16)[name = string("h_17_cast_fp16")];
220
+ string input_53_mode_0 = const()[name = string("input_53_mode_0"), val = string("EXACT")];
221
+ tensor<fp16, [1, 2048, ?]> input_53_cast_fp16 = gelu(mode = input_53_mode_0, x = h_17_cast_fp16)[name = string("input_53_cast_fp16")];
222
+ string h_19_pad_type_0 = const()[name = string("h_19_pad_type_0"), val = string("valid")];
223
+ tensor<int32, [1]> h_19_strides_0 = const()[name = string("h_19_strides_0"), val = tensor<int32, [1]>([1])];
224
+ tensor<int32, [2]> h_19_pad_0 = const()[name = string("h_19_pad_0"), val = tensor<int32, [2]>([0, 0])];
225
+ tensor<int32, [1]> h_19_dilations_0 = const()[name = string("h_19_dilations_0"), val = tensor<int32, [1]>([1])];
226
+ int32 h_19_groups_0 = const()[name = string("h_19_groups_0"), val = int32(1)];
227
+ tensor<fp16, [512, 2048, 1]> var_314_weight_0_to_fp16 = const()[name = string("op_314_weight_0_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(19126080)))];
228
+ tensor<fp16, [512]> var_314_bias_0_to_fp16 = const()[name = string("op_314_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(21223296)))];
229
+ tensor<fp16, [1, 512, ?]> var_314_cast_fp16 = conv(bias = var_314_bias_0_to_fp16, dilations = h_19_dilations_0, groups = h_19_groups_0, pad = h_19_pad_0, pad_type = h_19_pad_type_0, strides = h_19_strides_0, weight = var_314_weight_0_to_fp16, x = input_53_cast_fp16)[name = string("op_314_cast_fp16")];
230
+ tensor<fp16, [1, 512, ?]> input_55_cast_fp16 = add(x = input_45_cast_fp16, y = var_314_cast_fp16)[name = string("input_55_cast_fp16")];
231
+ tensor<int32, [6]> input_57_pad_0 = const()[name = string("input_57_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 24, 0])];
232
+ string input_57_mode_0 = const()[name = string("input_57_mode_0"), val = string("replicate")];
233
+ fp16 const_6_to_fp16 = const()[name = string("const_6_to_fp16"), val = fp16(0x0p+0)];
234
+ tensor<fp16, [1, 512, ?]> input_57_cast_fp16 = pad(constant_val = const_6_to_fp16, mode = input_57_mode_0, pad = input_57_pad_0, x = input_55_cast_fp16)[name = string("input_57_cast_fp16")];
235
+ string x_19_pad_type_0 = const()[name = string("x_19_pad_type_0"), val = string("valid")];
236
+ tensor<int32, [1]> x_19_dilations_0 = const()[name = string("x_19_dilations_0"), val = tensor<int32, [1]>([4])];
237
+ int32 x_19_groups_0 = const()[name = string("x_19_groups_0"), val = int32(512)];
238
+ tensor<int32, [1]> x_19_strides_0 = const()[name = string("x_19_strides_0"), val = tensor<int32, [1]>([1])];
239
+ tensor<int32, [2]> x_19_pad_0 = const()[name = string("x_19_pad_0"), val = tensor<int32, [2]>([0, 0])];
240
+ tensor<fp16, [512, 1, 7]> convnext_5_dwconv_net_weight_to_fp16 = const()[name = string("convnext_5_dwconv_net_weight_to_fp16"), val = tensor<fp16, [512, 1, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(21224384)))];
241
+ tensor<fp16, [512]> convnext_5_dwconv_net_bias_to_fp16 = const()[name = string("convnext_5_dwconv_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(21231616)))];
242
+ tensor<fp16, [1, 512, ?]> x_19_cast_fp16 = conv(bias = convnext_5_dwconv_net_bias_to_fp16, dilations = x_19_dilations_0, groups = x_19_groups_0, pad = x_19_pad_0, pad_type = x_19_pad_type_0, strides = x_19_strides_0, weight = convnext_5_dwconv_net_weight_to_fp16, x = input_57_cast_fp16)[name = string("x_19_cast_fp16")];
243
+ tensor<int32, [3]> input_59_perm_0 = const()[name = string("input_59_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
244
+ tensor<int32, [1]> var_348_axes_0 = const()[name = string("op_348_axes_0"), val = tensor<int32, [1]>([-1])];
245
+ tensor<fp16, [512]> convnext_5_norm_norm_weight_to_fp16 = const()[name = string("convnext_5_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(21232704)))];
246
+ tensor<fp16, [512]> convnext_5_norm_norm_bias_to_fp16 = const()[name = string("convnext_5_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(21233792)))];
247
+ fp16 var_318_to_fp16 = const()[name = string("op_318_to_fp16"), val = fp16(0x1.5p-17)];
248
+ tensor<fp16, [1, ?, 512]> input_59_cast_fp16 = transpose(perm = input_59_perm_0, x = x_19_cast_fp16)[name = string("transpose_10")];
249
+ tensor<fp16, [1, ?, 512]> var_348_cast_fp16 = layer_norm(axes = var_348_axes_0, beta = convnext_5_norm_norm_bias_to_fp16, epsilon = var_318_to_fp16, gamma = convnext_5_norm_norm_weight_to_fp16, x = input_59_cast_fp16)[name = string("op_348_cast_fp16")];
250
+ tensor<int32, [3]> input_61_perm_0 = const()[name = string("input_61_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
251
+ string h_21_pad_type_0 = const()[name = string("h_21_pad_type_0"), val = string("valid")];
252
+ tensor<int32, [1]> h_21_strides_0 = const()[name = string("h_21_strides_0"), val = tensor<int32, [1]>([1])];
253
+ tensor<int32, [2]> h_21_pad_0 = const()[name = string("h_21_pad_0"), val = tensor<int32, [2]>([0, 0])];
254
+ tensor<int32, [1]> h_21_dilations_0 = const()[name = string("h_21_dilations_0"), val = tensor<int32, [1]>([1])];
255
+ int32 h_21_groups_0 = const()[name = string("h_21_groups_0"), val = int32(1)];
256
+ tensor<fp16, [2048, 512, 1]> convnext_5_pwconv1_weight_to_fp16 = const()[name = string("convnext_5_pwconv1_weight_to_fp16"), val = tensor<fp16, [2048, 512, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(21234880)))];
257
+ tensor<fp16, [2048]> convnext_5_pwconv1_bias_to_fp16 = const()[name = string("convnext_5_pwconv1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(23332096)))];
258
+ tensor<fp16, [1, 512, ?]> input_61_cast_fp16 = transpose(perm = input_61_perm_0, x = var_348_cast_fp16)[name = string("transpose_9")];
259
+ tensor<fp16, [1, 2048, ?]> h_21_cast_fp16 = conv(bias = convnext_5_pwconv1_bias_to_fp16, dilations = h_21_dilations_0, groups = h_21_groups_0, pad = h_21_pad_0, pad_type = h_21_pad_type_0, strides = h_21_strides_0, weight = convnext_5_pwconv1_weight_to_fp16, x = input_61_cast_fp16)[name = string("h_21_cast_fp16")];
260
+ string input_63_mode_0 = const()[name = string("input_63_mode_0"), val = string("EXACT")];
261
+ tensor<fp16, [1, 2048, ?]> input_63_cast_fp16 = gelu(mode = input_63_mode_0, x = h_21_cast_fp16)[name = string("input_63_cast_fp16")];
262
+ string h_23_pad_type_0 = const()[name = string("h_23_pad_type_0"), val = string("valid")];
263
+ tensor<int32, [1]> h_23_strides_0 = const()[name = string("h_23_strides_0"), val = tensor<int32, [1]>([1])];
264
+ tensor<int32, [2]> h_23_pad_0 = const()[name = string("h_23_pad_0"), val = tensor<int32, [2]>([0, 0])];
265
+ tensor<int32, [1]> h_23_dilations_0 = const()[name = string("h_23_dilations_0"), val = tensor<int32, [1]>([1])];
266
+ int32 h_23_groups_0 = const()[name = string("h_23_groups_0"), val = int32(1)];
267
+ tensor<fp16, [512, 2048, 1]> var_365_weight_0_to_fp16 = const()[name = string("op_365_weight_0_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(23336256)))];
268
+ tensor<fp16, [512]> var_365_bias_0_to_fp16 = const()[name = string("op_365_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(25433472)))];
269
+ tensor<fp16, [1, 512, ?]> var_365_cast_fp16 = conv(bias = var_365_bias_0_to_fp16, dilations = h_23_dilations_0, groups = h_23_groups_0, pad = h_23_pad_0, pad_type = h_23_pad_type_0, strides = h_23_strides_0, weight = var_365_weight_0_to_fp16, x = input_63_cast_fp16)[name = string("op_365_cast_fp16")];
270
+ tensor<fp16, [1, 512, ?]> input_65_cast_fp16 = add(x = input_55_cast_fp16, y = var_365_cast_fp16)[name = string("input_65_cast_fp16")];
271
+ tensor<int32, [6]> input_67_pad_0 = const()[name = string("input_67_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 6, 0])];
272
+ string input_67_mode_0 = const()[name = string("input_67_mode_0"), val = string("replicate")];
273
+ fp16 const_7_to_fp16 = const()[name = string("const_7_to_fp16"), val = fp16(0x0p+0)];
274
+ tensor<fp16, [1, 512, ?]> input_67_cast_fp16 = pad(constant_val = const_7_to_fp16, mode = input_67_mode_0, pad = input_67_pad_0, x = input_65_cast_fp16)[name = string("input_67_cast_fp16")];
275
+ string x_21_pad_type_0 = const()[name = string("x_21_pad_type_0"), val = string("valid")];
276
+ int32 x_21_groups_0 = const()[name = string("x_21_groups_0"), val = int32(512)];
277
+ tensor<int32, [1]> x_21_strides_0 = const()[name = string("x_21_strides_0"), val = tensor<int32, [1]>([1])];
278
+ tensor<int32, [2]> x_21_pad_0 = const()[name = string("x_21_pad_0"), val = tensor<int32, [2]>([0, 0])];
279
+ tensor<int32, [1]> x_21_dilations_0 = const()[name = string("x_21_dilations_0"), val = tensor<int32, [1]>([1])];
280
+ tensor<fp16, [512, 1, 7]> convnext_6_dwconv_net_weight_to_fp16 = const()[name = string("convnext_6_dwconv_net_weight_to_fp16"), val = tensor<fp16, [512, 1, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(25434560)))];
281
+ tensor<fp16, [512]> convnext_6_dwconv_net_bias_to_fp16 = const()[name = string("convnext_6_dwconv_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(25441792)))];
282
+ tensor<fp16, [1, 512, ?]> x_21_cast_fp16 = conv(bias = convnext_6_dwconv_net_bias_to_fp16, dilations = x_21_dilations_0, groups = x_21_groups_0, pad = x_21_pad_0, pad_type = x_21_pad_type_0, strides = x_21_strides_0, weight = convnext_6_dwconv_net_weight_to_fp16, x = input_67_cast_fp16)[name = string("x_21_cast_fp16")];
283
+ tensor<int32, [3]> input_69_perm_0 = const()[name = string("input_69_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
284
+ tensor<int32, [1]> var_398_axes_0 = const()[name = string("op_398_axes_0"), val = tensor<int32, [1]>([-1])];
285
+ tensor<fp16, [512]> convnext_6_norm_norm_weight_to_fp16 = const()[name = string("convnext_6_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(25442880)))];
286
+ tensor<fp16, [512]> convnext_6_norm_norm_bias_to_fp16 = const()[name = string("convnext_6_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(25443968)))];
287
+ fp16 var_369_to_fp16 = const()[name = string("op_369_to_fp16"), val = fp16(0x1.5p-17)];
288
+ tensor<fp16, [1, ?, 512]> input_69_cast_fp16 = transpose(perm = input_69_perm_0, x = x_21_cast_fp16)[name = string("transpose_8")];
289
+ tensor<fp16, [1, ?, 512]> var_398_cast_fp16 = layer_norm(axes = var_398_axes_0, beta = convnext_6_norm_norm_bias_to_fp16, epsilon = var_369_to_fp16, gamma = convnext_6_norm_norm_weight_to_fp16, x = input_69_cast_fp16)[name = string("op_398_cast_fp16")];
290
+ tensor<int32, [3]> input_71_perm_0 = const()[name = string("input_71_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
291
+ string h_25_pad_type_0 = const()[name = string("h_25_pad_type_0"), val = string("valid")];
292
+ tensor<int32, [1]> h_25_strides_0 = const()[name = string("h_25_strides_0"), val = tensor<int32, [1]>([1])];
293
+ tensor<int32, [2]> h_25_pad_0 = const()[name = string("h_25_pad_0"), val = tensor<int32, [2]>([0, 0])];
294
+ tensor<int32, [1]> h_25_dilations_0 = const()[name = string("h_25_dilations_0"), val = tensor<int32, [1]>([1])];
295
+ int32 h_25_groups_0 = const()[name = string("h_25_groups_0"), val = int32(1)];
296
+ tensor<fp16, [2048, 512, 1]> convnext_6_pwconv1_weight_to_fp16 = const()[name = string("convnext_6_pwconv1_weight_to_fp16"), val = tensor<fp16, [2048, 512, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(25445056)))];
297
+ tensor<fp16, [2048]> convnext_6_pwconv1_bias_to_fp16 = const()[name = string("convnext_6_pwconv1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(27542272)))];
298
+ tensor<fp16, [1, 512, ?]> input_71_cast_fp16 = transpose(perm = input_71_perm_0, x = var_398_cast_fp16)[name = string("transpose_7")];
299
+ tensor<fp16, [1, 2048, ?]> h_25_cast_fp16 = conv(bias = convnext_6_pwconv1_bias_to_fp16, dilations = h_25_dilations_0, groups = h_25_groups_0, pad = h_25_pad_0, pad_type = h_25_pad_type_0, strides = h_25_strides_0, weight = convnext_6_pwconv1_weight_to_fp16, x = input_71_cast_fp16)[name = string("h_25_cast_fp16")];
300
+ string input_73_mode_0 = const()[name = string("input_73_mode_0"), val = string("EXACT")];
301
+ tensor<fp16, [1, 2048, ?]> input_73_cast_fp16 = gelu(mode = input_73_mode_0, x = h_25_cast_fp16)[name = string("input_73_cast_fp16")];
302
+ string h_27_pad_type_0 = const()[name = string("h_27_pad_type_0"), val = string("valid")];
303
+ tensor<int32, [1]> h_27_strides_0 = const()[name = string("h_27_strides_0"), val = tensor<int32, [1]>([1])];
304
+ tensor<int32, [2]> h_27_pad_0 = const()[name = string("h_27_pad_0"), val = tensor<int32, [2]>([0, 0])];
305
+ tensor<int32, [1]> h_27_dilations_0 = const()[name = string("h_27_dilations_0"), val = tensor<int32, [1]>([1])];
306
+ int32 h_27_groups_0 = const()[name = string("h_27_groups_0"), val = int32(1)];
307
+ tensor<fp16, [512, 2048, 1]> var_415_weight_0_to_fp16 = const()[name = string("op_415_weight_0_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(27546432)))];
308
+ tensor<fp16, [512]> var_415_bias_0_to_fp16 = const()[name = string("op_415_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(29643648)))];
309
+ tensor<fp16, [1, 512, ?]> var_415_cast_fp16 = conv(bias = var_415_bias_0_to_fp16, dilations = h_27_dilations_0, groups = h_27_groups_0, pad = h_27_pad_0, pad_type = h_27_pad_type_0, strides = h_27_strides_0, weight = var_415_weight_0_to_fp16, x = input_73_cast_fp16)[name = string("op_415_cast_fp16")];
310
+ tensor<fp16, [1, 512, ?]> input_75_cast_fp16 = add(x = input_65_cast_fp16, y = var_415_cast_fp16)[name = string("input_75_cast_fp16")];
311
+ tensor<int32, [6]> input_77_pad_0 = const()[name = string("input_77_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 6, 0])];
312
+ string input_77_mode_0 = const()[name = string("input_77_mode_0"), val = string("replicate")];
313
+ fp16 const_8_to_fp16 = const()[name = string("const_8_to_fp16"), val = fp16(0x0p+0)];
314
+ tensor<fp16, [1, 512, ?]> input_77_cast_fp16 = pad(constant_val = const_8_to_fp16, mode = input_77_mode_0, pad = input_77_pad_0, x = input_75_cast_fp16)[name = string("input_77_cast_fp16")];
315
+ string x_23_pad_type_0 = const()[name = string("x_23_pad_type_0"), val = string("valid")];
316
+ int32 x_23_groups_0 = const()[name = string("x_23_groups_0"), val = int32(512)];
317
+ tensor<int32, [1]> x_23_strides_0 = const()[name = string("x_23_strides_0"), val = tensor<int32, [1]>([1])];
318
+ tensor<int32, [2]> x_23_pad_0 = const()[name = string("x_23_pad_0"), val = tensor<int32, [2]>([0, 0])];
319
+ tensor<int32, [1]> x_23_dilations_0 = const()[name = string("x_23_dilations_0"), val = tensor<int32, [1]>([1])];
320
+ tensor<fp16, [512, 1, 7]> convnext_7_dwconv_net_weight_to_fp16 = const()[name = string("convnext_7_dwconv_net_weight_to_fp16"), val = tensor<fp16, [512, 1, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(29644736)))];
321
+ tensor<fp16, [512]> convnext_7_dwconv_net_bias_to_fp16 = const()[name = string("convnext_7_dwconv_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(29651968)))];
322
+ tensor<fp16, [1, 512, ?]> x_23_cast_fp16 = conv(bias = convnext_7_dwconv_net_bias_to_fp16, dilations = x_23_dilations_0, groups = x_23_groups_0, pad = x_23_pad_0, pad_type = x_23_pad_type_0, strides = x_23_strides_0, weight = convnext_7_dwconv_net_weight_to_fp16, x = input_77_cast_fp16)[name = string("x_23_cast_fp16")];
323
+ tensor<int32, [3]> input_79_perm_0 = const()[name = string("input_79_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
324
+ tensor<int32, [1]> var_448_axes_0 = const()[name = string("op_448_axes_0"), val = tensor<int32, [1]>([-1])];
325
+ tensor<fp16, [512]> convnext_7_norm_norm_weight_to_fp16 = const()[name = string("convnext_7_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(29653056)))];
326
+ tensor<fp16, [512]> convnext_7_norm_norm_bias_to_fp16 = const()[name = string("convnext_7_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(29654144)))];
327
+ fp16 var_419_to_fp16 = const()[name = string("op_419_to_fp16"), val = fp16(0x1.5p-17)];
328
+ tensor<fp16, [1, ?, 512]> input_79_cast_fp16 = transpose(perm = input_79_perm_0, x = x_23_cast_fp16)[name = string("transpose_6")];
329
+ tensor<fp16, [1, ?, 512]> var_448_cast_fp16 = layer_norm(axes = var_448_axes_0, beta = convnext_7_norm_norm_bias_to_fp16, epsilon = var_419_to_fp16, gamma = convnext_7_norm_norm_weight_to_fp16, x = input_79_cast_fp16)[name = string("op_448_cast_fp16")];
330
+ tensor<int32, [3]> input_81_perm_0 = const()[name = string("input_81_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
331
+ string h_29_pad_type_0 = const()[name = string("h_29_pad_type_0"), val = string("valid")];
332
+ tensor<int32, [1]> h_29_strides_0 = const()[name = string("h_29_strides_0"), val = tensor<int32, [1]>([1])];
333
+ tensor<int32, [2]> h_29_pad_0 = const()[name = string("h_29_pad_0"), val = tensor<int32, [2]>([0, 0])];
334
+ tensor<int32, [1]> h_29_dilations_0 = const()[name = string("h_29_dilations_0"), val = tensor<int32, [1]>([1])];
335
+ int32 h_29_groups_0 = const()[name = string("h_29_groups_0"), val = int32(1)];
336
+ tensor<fp16, [2048, 512, 1]> convnext_7_pwconv1_weight_to_fp16 = const()[name = string("convnext_7_pwconv1_weight_to_fp16"), val = tensor<fp16, [2048, 512, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(29655232)))];
337
+ tensor<fp16, [2048]> convnext_7_pwconv1_bias_to_fp16 = const()[name = string("convnext_7_pwconv1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(31752448)))];
338
+ tensor<fp16, [1, 512, ?]> input_81_cast_fp16 = transpose(perm = input_81_perm_0, x = var_448_cast_fp16)[name = string("transpose_5")];
339
+ tensor<fp16, [1, 2048, ?]> h_29_cast_fp16 = conv(bias = convnext_7_pwconv1_bias_to_fp16, dilations = h_29_dilations_0, groups = h_29_groups_0, pad = h_29_pad_0, pad_type = h_29_pad_type_0, strides = h_29_strides_0, weight = convnext_7_pwconv1_weight_to_fp16, x = input_81_cast_fp16)[name = string("h_29_cast_fp16")];
340
+ string input_83_mode_0 = const()[name = string("input_83_mode_0"), val = string("EXACT")];
341
+ tensor<fp16, [1, 2048, ?]> input_83_cast_fp16 = gelu(mode = input_83_mode_0, x = h_29_cast_fp16)[name = string("input_83_cast_fp16")];
342
+ string h_31_pad_type_0 = const()[name = string("h_31_pad_type_0"), val = string("valid")];
343
+ tensor<int32, [1]> h_31_strides_0 = const()[name = string("h_31_strides_0"), val = tensor<int32, [1]>([1])];
344
+ tensor<int32, [2]> h_31_pad_0 = const()[name = string("h_31_pad_0"), val = tensor<int32, [2]>([0, 0])];
345
+ tensor<int32, [1]> h_31_dilations_0 = const()[name = string("h_31_dilations_0"), val = tensor<int32, [1]>([1])];
346
+ int32 h_31_groups_0 = const()[name = string("h_31_groups_0"), val = int32(1)];
347
+ tensor<fp16, [512, 2048, 1]> var_465_weight_0_to_fp16 = const()[name = string("op_465_weight_0_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(31756608)))];
348
+ tensor<fp16, [512]> var_465_bias_0_to_fp16 = const()[name = string("op_465_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(33853824)))];
349
+ tensor<fp16, [1, 512, ?]> var_465_cast_fp16 = conv(bias = var_465_bias_0_to_fp16, dilations = h_31_dilations_0, groups = h_31_groups_0, pad = h_31_pad_0, pad_type = h_31_pad_type_0, strides = h_31_strides_0, weight = var_465_weight_0_to_fp16, x = input_83_cast_fp16)[name = string("op_465_cast_fp16")];
350
+ tensor<fp16, [1, 512, ?]> input_85_cast_fp16 = add(x = input_75_cast_fp16, y = var_465_cast_fp16)[name = string("input_85_cast_fp16")];
351
+ tensor<int32, [6]> input_87_pad_0 = const()[name = string("input_87_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 6, 0])];
352
+ string input_87_mode_0 = const()[name = string("input_87_mode_0"), val = string("replicate")];
353
+ fp16 const_9_to_fp16 = const()[name = string("const_9_to_fp16"), val = fp16(0x0p+0)];
354
+ tensor<fp16, [1, 512, ?]> input_87_cast_fp16 = pad(constant_val = const_9_to_fp16, mode = input_87_mode_0, pad = input_87_pad_0, x = input_85_cast_fp16)[name = string("input_87_cast_fp16")];
355
+ string x_25_pad_type_0 = const()[name = string("x_25_pad_type_0"), val = string("valid")];
356
+ int32 x_25_groups_0 = const()[name = string("x_25_groups_0"), val = int32(512)];
357
+ tensor<int32, [1]> x_25_strides_0 = const()[name = string("x_25_strides_0"), val = tensor<int32, [1]>([1])];
358
+ tensor<int32, [2]> x_25_pad_0 = const()[name = string("x_25_pad_0"), val = tensor<int32, [2]>([0, 0])];
359
+ tensor<int32, [1]> x_25_dilations_0 = const()[name = string("x_25_dilations_0"), val = tensor<int32, [1]>([1])];
360
+ tensor<fp16, [512, 1, 7]> convnext_8_dwconv_net_weight_to_fp16 = const()[name = string("convnext_8_dwconv_net_weight_to_fp16"), val = tensor<fp16, [512, 1, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(33854912)))];
361
+ tensor<fp16, [512]> convnext_8_dwconv_net_bias_to_fp16 = const()[name = string("convnext_8_dwconv_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(33862144)))];
362
+ tensor<fp16, [1, 512, ?]> x_25_cast_fp16 = conv(bias = convnext_8_dwconv_net_bias_to_fp16, dilations = x_25_dilations_0, groups = x_25_groups_0, pad = x_25_pad_0, pad_type = x_25_pad_type_0, strides = x_25_strides_0, weight = convnext_8_dwconv_net_weight_to_fp16, x = input_87_cast_fp16)[name = string("x_25_cast_fp16")];
363
+ tensor<int32, [3]> input_89_perm_0 = const()[name = string("input_89_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
364
+ tensor<int32, [1]> var_498_axes_0 = const()[name = string("op_498_axes_0"), val = tensor<int32, [1]>([-1])];
365
+ tensor<fp16, [512]> convnext_8_norm_norm_weight_to_fp16 = const()[name = string("convnext_8_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(33863232)))];
366
+ tensor<fp16, [512]> convnext_8_norm_norm_bias_to_fp16 = const()[name = string("convnext_8_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(33864320)))];
367
+ fp16 var_469_to_fp16 = const()[name = string("op_469_to_fp16"), val = fp16(0x1.5p-17)];
368
+ tensor<fp16, [1, ?, 512]> input_89_cast_fp16 = transpose(perm = input_89_perm_0, x = x_25_cast_fp16)[name = string("transpose_4")];
369
+ tensor<fp16, [1, ?, 512]> var_498_cast_fp16 = layer_norm(axes = var_498_axes_0, beta = convnext_8_norm_norm_bias_to_fp16, epsilon = var_469_to_fp16, gamma = convnext_8_norm_norm_weight_to_fp16, x = input_89_cast_fp16)[name = string("op_498_cast_fp16")];
370
+ tensor<int32, [3]> input_91_perm_0 = const()[name = string("input_91_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
371
+ string h_33_pad_type_0 = const()[name = string("h_33_pad_type_0"), val = string("valid")];
372
+ tensor<int32, [1]> h_33_strides_0 = const()[name = string("h_33_strides_0"), val = tensor<int32, [1]>([1])];
373
+ tensor<int32, [2]> h_33_pad_0 = const()[name = string("h_33_pad_0"), val = tensor<int32, [2]>([0, 0])];
374
+ tensor<int32, [1]> h_33_dilations_0 = const()[name = string("h_33_dilations_0"), val = tensor<int32, [1]>([1])];
375
+ int32 h_33_groups_0 = const()[name = string("h_33_groups_0"), val = int32(1)];
376
+ tensor<fp16, [2048, 512, 1]> convnext_8_pwconv1_weight_to_fp16 = const()[name = string("convnext_8_pwconv1_weight_to_fp16"), val = tensor<fp16, [2048, 512, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(33865408)))];
377
+ tensor<fp16, [2048]> convnext_8_pwconv1_bias_to_fp16 = const()[name = string("convnext_8_pwconv1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(35962624)))];
378
+ tensor<fp16, [1, 512, ?]> input_91_cast_fp16 = transpose(perm = input_91_perm_0, x = var_498_cast_fp16)[name = string("transpose_3")];
379
+ tensor<fp16, [1, 2048, ?]> h_33_cast_fp16 = conv(bias = convnext_8_pwconv1_bias_to_fp16, dilations = h_33_dilations_0, groups = h_33_groups_0, pad = h_33_pad_0, pad_type = h_33_pad_type_0, strides = h_33_strides_0, weight = convnext_8_pwconv1_weight_to_fp16, x = input_91_cast_fp16)[name = string("h_33_cast_fp16")];
380
+ string input_93_mode_0 = const()[name = string("input_93_mode_0"), val = string("EXACT")];
381
+ tensor<fp16, [1, 2048, ?]> input_93_cast_fp16 = gelu(mode = input_93_mode_0, x = h_33_cast_fp16)[name = string("input_93_cast_fp16")];
382
+ string h_35_pad_type_0 = const()[name = string("h_35_pad_type_0"), val = string("valid")];
383
+ tensor<int32, [1]> h_35_strides_0 = const()[name = string("h_35_strides_0"), val = tensor<int32, [1]>([1])];
384
+ tensor<int32, [2]> h_35_pad_0 = const()[name = string("h_35_pad_0"), val = tensor<int32, [2]>([0, 0])];
385
+ tensor<int32, [1]> h_35_dilations_0 = const()[name = string("h_35_dilations_0"), val = tensor<int32, [1]>([1])];
386
+ int32 h_35_groups_0 = const()[name = string("h_35_groups_0"), val = int32(1)];
387
+ tensor<fp16, [512, 2048, 1]> var_515_weight_0_to_fp16 = const()[name = string("op_515_weight_0_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(35966784)))];
388
+ tensor<fp16, [512]> var_515_bias_0_to_fp16 = const()[name = string("op_515_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(38064000)))];
389
+ tensor<fp16, [1, 512, ?]> var_515_cast_fp16 = conv(bias = var_515_bias_0_to_fp16, dilations = h_35_dilations_0, groups = h_35_groups_0, pad = h_35_pad_0, pad_type = h_35_pad_type_0, strides = h_35_strides_0, weight = var_515_weight_0_to_fp16, x = input_93_cast_fp16)[name = string("op_515_cast_fp16")];
390
+ tensor<fp16, [1, 512, ?]> input_95_cast_fp16 = add(x = input_85_cast_fp16, y = var_515_cast_fp16)[name = string("input_95_cast_fp16")];
391
+ tensor<int32, [6]> input_97_pad_0 = const()[name = string("input_97_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 6, 0])];
392
+ string input_97_mode_0 = const()[name = string("input_97_mode_0"), val = string("replicate")];
393
+ fp16 const_10_to_fp16 = const()[name = string("const_10_to_fp16"), val = fp16(0x0p+0)];
394
+ tensor<fp16, [1, 512, ?]> input_97_cast_fp16 = pad(constant_val = const_10_to_fp16, mode = input_97_mode_0, pad = input_97_pad_0, x = input_95_cast_fp16)[name = string("input_97_cast_fp16")];
395
+ string x_27_pad_type_0 = const()[name = string("x_27_pad_type_0"), val = string("valid")];
396
+ int32 x_27_groups_0 = const()[name = string("x_27_groups_0"), val = int32(512)];
397
+ tensor<int32, [1]> x_27_strides_0 = const()[name = string("x_27_strides_0"), val = tensor<int32, [1]>([1])];
398
+ tensor<int32, [2]> x_27_pad_0 = const()[name = string("x_27_pad_0"), val = tensor<int32, [2]>([0, 0])];
399
+ tensor<int32, [1]> x_27_dilations_0 = const()[name = string("x_27_dilations_0"), val = tensor<int32, [1]>([1])];
400
+ tensor<fp16, [512, 1, 7]> convnext_9_dwconv_net_weight_to_fp16 = const()[name = string("convnext_9_dwconv_net_weight_to_fp16"), val = tensor<fp16, [512, 1, 7]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(38065088)))];
401
+ tensor<fp16, [512]> convnext_9_dwconv_net_bias_to_fp16 = const()[name = string("convnext_9_dwconv_net_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(38072320)))];
402
+ tensor<fp16, [1, 512, ?]> x_27_cast_fp16 = conv(bias = convnext_9_dwconv_net_bias_to_fp16, dilations = x_27_dilations_0, groups = x_27_groups_0, pad = x_27_pad_0, pad_type = x_27_pad_type_0, strides = x_27_strides_0, weight = convnext_9_dwconv_net_weight_to_fp16, x = input_97_cast_fp16)[name = string("x_27_cast_fp16")];
403
+ tensor<int32, [3]> input_99_perm_0 = const()[name = string("input_99_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
404
+ tensor<int32, [1]> var_548_axes_0 = const()[name = string("op_548_axes_0"), val = tensor<int32, [1]>([-1])];
405
+ tensor<fp16, [512]> convnext_9_norm_norm_weight_to_fp16 = const()[name = string("convnext_9_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(38073408)))];
406
+ tensor<fp16, [512]> convnext_9_norm_norm_bias_to_fp16 = const()[name = string("convnext_9_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(38074496)))];
407
+ fp16 var_519_to_fp16 = const()[name = string("op_519_to_fp16"), val = fp16(0x1.5p-17)];
408
+ tensor<fp16, [1, ?, 512]> input_99_cast_fp16 = transpose(perm = input_99_perm_0, x = x_27_cast_fp16)[name = string("transpose_2")];
409
+ tensor<fp16, [1, ?, 512]> var_548_cast_fp16 = layer_norm(axes = var_548_axes_0, beta = convnext_9_norm_norm_bias_to_fp16, epsilon = var_519_to_fp16, gamma = convnext_9_norm_norm_weight_to_fp16, x = input_99_cast_fp16)[name = string("op_548_cast_fp16")];
410
+ tensor<int32, [3]> input_101_perm_0 = const()[name = string("input_101_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
411
+ string h_37_pad_type_0 = const()[name = string("h_37_pad_type_0"), val = string("valid")];
412
+ tensor<int32, [1]> h_37_strides_0 = const()[name = string("h_37_strides_0"), val = tensor<int32, [1]>([1])];
413
+ tensor<int32, [2]> h_37_pad_0 = const()[name = string("h_37_pad_0"), val = tensor<int32, [2]>([0, 0])];
414
+ tensor<int32, [1]> h_37_dilations_0 = const()[name = string("h_37_dilations_0"), val = tensor<int32, [1]>([1])];
415
+ int32 h_37_groups_0 = const()[name = string("h_37_groups_0"), val = int32(1)];
416
+ tensor<fp16, [2048, 512, 1]> convnext_9_pwconv1_weight_to_fp16 = const()[name = string("convnext_9_pwconv1_weight_to_fp16"), val = tensor<fp16, [2048, 512, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(38075584)))];
417
+ tensor<fp16, [2048]> convnext_9_pwconv1_bias_to_fp16 = const()[name = string("convnext_9_pwconv1_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(40172800)))];
418
+ tensor<fp16, [1, 512, ?]> input_101_cast_fp16 = transpose(perm = input_101_perm_0, x = var_548_cast_fp16)[name = string("transpose_1")];
419
+ tensor<fp16, [1, 2048, ?]> h_37_cast_fp16 = conv(bias = convnext_9_pwconv1_bias_to_fp16, dilations = h_37_dilations_0, groups = h_37_groups_0, pad = h_37_pad_0, pad_type = h_37_pad_type_0, strides = h_37_strides_0, weight = convnext_9_pwconv1_weight_to_fp16, x = input_101_cast_fp16)[name = string("h_37_cast_fp16")];
420
+ string input_103_mode_0 = const()[name = string("input_103_mode_0"), val = string("EXACT")];
421
+ tensor<fp16, [1, 2048, ?]> input_103_cast_fp16 = gelu(mode = input_103_mode_0, x = h_37_cast_fp16)[name = string("input_103_cast_fp16")];
422
+ string h_pad_type_0 = const()[name = string("h_pad_type_0"), val = string("valid")];
423
+ tensor<int32, [1]> h_strides_0 = const()[name = string("h_strides_0"), val = tensor<int32, [1]>([1])];
424
+ tensor<int32, [2]> h_pad_0 = const()[name = string("h_pad_0"), val = tensor<int32, [2]>([0, 0])];
425
+ tensor<int32, [1]> h_dilations_0 = const()[name = string("h_dilations_0"), val = tensor<int32, [1]>([1])];
426
+ int32 h_groups_0 = const()[name = string("h_groups_0"), val = int32(1)];
427
+ tensor<fp16, [512, 2048, 1]> var_565_weight_0_to_fp16 = const()[name = string("op_565_weight_0_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(40176960)))];
428
+ tensor<fp16, [512]> var_565_bias_0_to_fp16 = const()[name = string("op_565_bias_0_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(42274176)))];
429
+ tensor<fp16, [1, 512, ?]> var_565_cast_fp16 = conv(bias = var_565_bias_0_to_fp16, dilations = h_dilations_0, groups = h_groups_0, pad = h_pad_0, pad_type = h_pad_type_0, strides = h_strides_0, weight = var_565_weight_0_to_fp16, x = input_103_cast_fp16)[name = string("op_565_cast_fp16")];
430
+ tensor<fp16, [1, 512, ?]> input_105_cast_fp16 = add(x = input_95_cast_fp16, y = var_565_cast_fp16)[name = string("input_105_cast_fp16")];
431
+ tensor<fp16, [512]> final_norm_norm_running_mean_to_fp16 = const()[name = string("final_norm_norm_running_mean_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(42275264)))];
432
+ tensor<fp16, [512]> final_norm_norm_running_var_to_fp16 = const()[name = string("final_norm_norm_running_var_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(42276352)))];
433
+ tensor<fp16, [512]> final_norm_norm_weight_to_fp16 = const()[name = string("final_norm_norm_weight_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(42277440)))];
434
+ tensor<fp16, [512]> final_norm_norm_bias_to_fp16 = const()[name = string("final_norm_norm_bias_to_fp16"), val = tensor<fp16, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(42278528)))];
435
+ fp16 var_569_to_fp16 = const()[name = string("op_569_to_fp16"), val = fp16(0x1.5p-17)];
436
+ tensor<fp16, [1, 512, ?]> input_107_cast_fp16 = batch_norm(beta = final_norm_norm_bias_to_fp16, epsilon = var_569_to_fp16, gamma = final_norm_norm_weight_to_fp16, mean = final_norm_norm_running_mean_to_fp16, variance = final_norm_norm_running_var_to_fp16, x = input_105_cast_fp16)[name = string("input_107_cast_fp16")];
437
+ tensor<int32, [6]> input_109_pad_0 = const()[name = string("input_109_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 2, 0])];
438
+ string input_109_mode_0 = const()[name = string("input_109_mode_0"), val = string("replicate")];
439
+ fp16 const_11_to_fp16 = const()[name = string("const_11_to_fp16"), val = fp16(0x0p+0)];
440
+ tensor<fp16, [1, 512, ?]> input_109_cast_fp16 = pad(constant_val = const_11_to_fp16, mode = input_109_mode_0, pad = input_109_pad_0, x = input_107_cast_fp16)[name = string("input_109_cast_fp16")];
441
+ string input_111_pad_type_0 = const()[name = string("input_111_pad_type_0"), val = string("valid")];
442
+ tensor<int32, [1]> input_111_strides_0 = const()[name = string("input_111_strides_0"), val = tensor<int32, [1]>([1])];
443
+ tensor<int32, [2]> input_111_pad_0 = const()[name = string("input_111_pad_0"), val = tensor<int32, [2]>([0, 0])];
444
+ tensor<int32, [1]> input_111_dilations_0 = const()[name = string("input_111_dilations_0"), val = tensor<int32, [1]>([1])];
445
+ int32 input_111_groups_0 = const()[name = string("input_111_groups_0"), val = int32(1)];
446
+ tensor<fp16, [2048, 512, 3]> head_layer1_net_weight_to_fp16 = const()[name = string("head_layer1_net_weight_to_fp16"), val = tensor<fp16, [2048, 512, 3]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(42279616)))];
447
+ tensor<fp16, [2048]> head_layer1_net_bias_to_fp16 = const()[name = string("head_layer1_net_bias_to_fp16"), val = tensor<fp16, [2048]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(48571136)))];
448
+ tensor<fp16, [1, 2048, ?]> input_111_cast_fp16 = conv(bias = head_layer1_net_bias_to_fp16, dilations = input_111_dilations_0, groups = input_111_groups_0, pad = input_111_pad_0, pad_type = input_111_pad_type_0, strides = input_111_strides_0, weight = head_layer1_net_weight_to_fp16, x = input_109_cast_fp16)[name = string("input_111_cast_fp16")];
449
+ fp32 input_alpha_1 = const()[name = string("input_alpha_1"), val = fp32(0x1.9814fap-13)];
450
+ tensor<fp16, [1, 2048, ?]> input_cast_fp16 = leaky_relu(alpha = input_alpha_1, x = input_111_cast_fp16)[name = string("input_cast_fp16")];
451
+ string x_29_pad_type_0 = const()[name = string("x_29_pad_type_0"), val = string("valid")];
452
+ tensor<int32, [1]> x_29_strides_0 = const()[name = string("x_29_strides_0"), val = tensor<int32, [1]>([1])];
453
+ tensor<int32, [2]> x_29_pad_0 = const()[name = string("x_29_pad_0"), val = tensor<int32, [2]>([0, 0])];
454
+ tensor<int32, [1]> x_29_dilations_0 = const()[name = string("x_29_dilations_0"), val = tensor<int32, [1]>([1])];
455
+ int32 x_29_groups_0 = const()[name = string("x_29_groups_0"), val = int32(1)];
456
+ tensor<fp16, [512, 2048, 1]> head_layer2_weight_to_fp16 = const()[name = string("head_layer2_weight_to_fp16"), val = tensor<fp16, [512, 2048, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(48575296)))];
457
+ tensor<fp16, [1, 512, ?]> x_29_cast_fp16 = conv(dilations = x_29_dilations_0, groups = x_29_groups_0, pad = x_29_pad_0, pad_type = x_29_pad_type_0, strides = x_29_strides_0, weight = head_layer2_weight_to_fp16, x = input_cast_fp16)[name = string("x_29_cast_fp16")];
458
+ tensor<int32, [3]> var_607_perm_0 = const()[name = string("op_607_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
459
+ tensor<int32, [2]> var_612 = const()[name = string("op_612"), val = tensor<int32, [2]>([1, -1])];
460
+ tensor<fp16, [1, ?, 512]> var_607_cast_fp16 = transpose(perm = var_607_perm_0, x = x_29_cast_fp16)[name = string("transpose_0")];
461
+ tensor<fp16, [1, ?]> var_613_cast_fp16 = reshape(shape = var_612, x = var_607_cast_fp16)[name = string("op_613_cast_fp16")];
462
+ string var_613_cast_fp16_to_fp32_dtype_0 = const()[name = string("op_613_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
463
+ tensor<fp32, [1, ?]> wav = cast(dtype = var_613_cast_fp16_to_fp32_dtype_0, x = var_613_cast_fp16)[name = string("cast_5")];
464
+ } -> (wav);
465
+ }
Vocoder.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b45e4c17a45cc4de30dcedae9f662548aa35f8f9763632c342b8bbef3089fcee
3
+ size 50672512
Vocoder.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0aefc35639793ac4dfcb74cddfca17744e7bb302f7af34a330d83d4b59e2831
3
+ size 70695
Vocoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b45e4c17a45cc4de30dcedae9f662548aa35f8f9763632c342b8bbef3089fcee
3
+ size 50672512
Vocoder.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "25CD729D-5964-4AA7-8FCA-CFCB279429B1": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Weights",
7
+ "name": "weights",
8
+ "path": "com.apple.CoreML/weights"
9
+ },
10
+ "53EE82CD-FCB9-4071-BACB-71F6BA6D3B60": {
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": "53EE82CD-FCB9-4071-BACB-71F6BA6D3B60"
18
+ }
manifest.json ADDED
@@ -0,0 +1,250 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "supertonic-3-coreml",
3
+ "version": "1.7.3",
4
+ "upstream": "https://huggingface.co/Supertone/supertonic-3",
5
+ "conversion_repo": "https://github.com/FluidInference/mobius/tree/main/models/tts/supertonic-3/coreml",
6
+ "integration": "https://github.com/FluidInference/FluidAudio/tree/main/Sources/FluidAudio/TTS/Supertonic3",
7
+ "precision": "float16",
8
+ "core_ml_target": "iOS18",
9
+ "sample_rate": 44100,
10
+ "languages": [
11
+ "en",
12
+ "ko",
13
+ "ja",
14
+ "ar",
15
+ "bg",
16
+ "cs",
17
+ "da",
18
+ "de",
19
+ "el",
20
+ "es",
21
+ "et",
22
+ "fi",
23
+ "fr",
24
+ "hi",
25
+ "hr",
26
+ "hu",
27
+ "id",
28
+ "it",
29
+ "lt",
30
+ "lv",
31
+ "nl",
32
+ "pl",
33
+ "pt",
34
+ "ro",
35
+ "ru",
36
+ "sk",
37
+ "sl",
38
+ "sv",
39
+ "tr",
40
+ "uk",
41
+ "vi"
42
+ ],
43
+ "modules": [
44
+ {
45
+ "name": "TextEncoder",
46
+ "package": "TextEncoder.mlpackage",
47
+ "compiled": "TextEncoder.mlmodelc",
48
+ "shape_policy": "fixed T=128",
49
+ "package_size_bytes": 18166801,
50
+ "compiled_size_bytes": 18173642,
51
+ "package_files": [
52
+ {
53
+ "path": "TextEncoder.mlpackage/Data/com.apple.CoreML/model.mlmodel",
54
+ "sha256": "9674a311dd5145e27554c1d021f25531f90ed772a3758ab349b605967faec667",
55
+ "size_bytes": 144168
56
+ },
57
+ {
58
+ "path": "TextEncoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin",
59
+ "sha256": "7c7afa5d02426a8a363c9cef0e27ce64a5c57a67d1a59f60aac0ac58ffb9877f",
60
+ "size_bytes": 18022016
61
+ },
62
+ {
63
+ "path": "TextEncoder.mlpackage/Manifest.json",
64
+ "sha256": "3d1496b0ff38b32feb32f36468e5920909176c6dc4dbe3bf70db97b03beb9e85",
65
+ "size_bytes": 617
66
+ }
67
+ ],
68
+ "compiled_files": [
69
+ {
70
+ "path": "TextEncoder.mlmodelc/analytics/coremldata.bin",
71
+ "sha256": "b39c208b46f180dcbf4cfcd3b4437eeb93371fb77d6d636b99ca3ddff5327253",
72
+ "size_bytes": 243
73
+ },
74
+ {
75
+ "path": "TextEncoder.mlmodelc/coremldata.bin",
76
+ "sha256": "4349aa84bb4a4731e3be09eb235e1816a3d0c88500b151479e08fe15662a446c",
77
+ "size_bytes": 431
78
+ },
79
+ {
80
+ "path": "TextEncoder.mlmodelc/model.mil",
81
+ "sha256": "6ef23a27a343f540bee78f5b2e85bbd83443c960c151d0e3bd57ffe5c6082ed1",
82
+ "size_bytes": 150952
83
+ },
84
+ {
85
+ "path": "TextEncoder.mlmodelc/weights/weight.bin",
86
+ "sha256": "7c7afa5d02426a8a363c9cef0e27ce64a5c57a67d1a59f60aac0ac58ffb9877f",
87
+ "size_bytes": 18022016
88
+ }
89
+ ]
90
+ },
91
+ {
92
+ "name": "DurationPredictor",
93
+ "package": "DurationPredictor.mlpackage",
94
+ "compiled": "DurationPredictor.mlmodelc",
95
+ "shape_policy": "fixed T=128",
96
+ "package_size_bytes": 1892063,
97
+ "compiled_size_bytes": 1896415,
98
+ "package_files": [
99
+ {
100
+ "path": "DurationPredictor.mlpackage/Data/com.apple.CoreML/model.mlmodel",
101
+ "sha256": "ccb56ff515b7d41cf79d21d6be2f2d23a31a295810950dd86e39620f9629cad4",
102
+ "size_bytes": 93494
103
+ },
104
+ {
105
+ "path": "DurationPredictor.mlpackage/Data/com.apple.CoreML/weights/weight.bin",
106
+ "sha256": "7f709c544087966d6b2538ac50b7a4014d6e1049e21d713cf3bc71c4c5b9307c",
107
+ "size_bytes": 1797952
108
+ },
109
+ {
110
+ "path": "DurationPredictor.mlpackage/Manifest.json",
111
+ "sha256": "36d830eb7bfbe233cdfca3ac155eb0a8894a2a52fa96ca81b38d00d5c1267f8a",
112
+ "size_bytes": 617
113
+ }
114
+ ],
115
+ "compiled_files": [
116
+ {
117
+ "path": "DurationPredictor.mlmodelc/analytics/coremldata.bin",
118
+ "sha256": "b836b72b388c2bc009a13bcc495cd08278a1742b1a866cdaf7faab7d80786bcb",
119
+ "size_bytes": 243
120
+ },
121
+ {
122
+ "path": "DurationPredictor.mlmodelc/coremldata.bin",
123
+ "sha256": "3ee71b864377bd54479f6b07d141ed37c9fc49718aba6dba21c56cc5cb975be0",
124
+ "size_bytes": 425
125
+ },
126
+ {
127
+ "path": "DurationPredictor.mlmodelc/model.mil",
128
+ "sha256": "e87f25d43a28b967628e6c0089535390687203e54520e7e3223c771b26fbc0bd",
129
+ "size_bytes": 97795
130
+ },
131
+ {
132
+ "path": "DurationPredictor.mlmodelc/weights/weight.bin",
133
+ "sha256": "7f709c544087966d6b2538ac50b7a4014d6e1049e21d713cf3bc71c4c5b9307c",
134
+ "size_bytes": 1797952
135
+ }
136
+ ]
137
+ },
138
+ {
139
+ "name": "VectorEstimator",
140
+ "package": "VectorEstimator.mlpackage",
141
+ "compiled": "VectorEstimator.mlmodelc",
142
+ "shape_policy": "RangeDim L,T 17..512",
143
+ "package_size_bytes": 127977306,
144
+ "compiled_size_bytes": 127993906,
145
+ "package_files": [
146
+ {
147
+ "path": "VectorEstimator.mlpackage/Data/com.apple.CoreML/model.mlmodel",
148
+ "sha256": "c93445cd09b7e6190fd26873cafb4a209fbb24d836b665428f7114ed6b0b41df",
149
+ "size_bytes": 328305
150
+ },
151
+ {
152
+ "path": "VectorEstimator.mlpackage/Data/com.apple.CoreML/weights/weight.bin",
153
+ "sha256": "101cebe7759b5f5e8f188ed3c2bf5f624bbf048aa657fe5bed9c3110c5ce0c26",
154
+ "size_bytes": 127648384
155
+ },
156
+ {
157
+ "path": "VectorEstimator.mlpackage/Manifest.json",
158
+ "sha256": "825048428f8073ddcc4f22dfeb1d6a649f5799ef8e37fdb7d47ecaac8ec48590",
159
+ "size_bytes": 617
160
+ }
161
+ ],
162
+ "compiled_files": [
163
+ {
164
+ "path": "VectorEstimator.mlmodelc/analytics/coremldata.bin",
165
+ "sha256": "1f43626d16b133a8b302f4f7a0e6d0f789422b22df135b7af11df25c385bebe0",
166
+ "size_bytes": 243
167
+ },
168
+ {
169
+ "path": "VectorEstimator.mlmodelc/coremldata.bin",
170
+ "sha256": "e4ae2bd30a9130ac2d09996ad4f63ea2c5703f735d11d9cf7e494d14397fd495",
171
+ "size_bytes": 633
172
+ },
173
+ {
174
+ "path": "VectorEstimator.mlmodelc/model.mil",
175
+ "sha256": "56c549b708a496f94e2b7c3ff7671414fb4d8a676df0b026ebab71c334a76ccf",
176
+ "size_bytes": 344646
177
+ },
178
+ {
179
+ "path": "VectorEstimator.mlmodelc/weights/weight.bin",
180
+ "sha256": "101cebe7759b5f5e8f188ed3c2bf5f624bbf048aa657fe5bed9c3110c5ce0c26",
181
+ "size_bytes": 127648384
182
+ }
183
+ ]
184
+ },
185
+ {
186
+ "name": "Vocoder",
187
+ "package": "Vocoder.mlpackage",
188
+ "compiled": "Vocoder.mlmodelc",
189
+ "shape_policy": "RangeDim L_ttl 4..512",
190
+ "package_size_bytes": 50743824,
191
+ "compiled_size_bytes": 50748769,
192
+ "package_files": [
193
+ {
194
+ "path": "Vocoder.mlpackage/Data/com.apple.CoreML/model.mlmodel",
195
+ "sha256": "e0aefc35639793ac4dfcb74cddfca17744e7bb302f7af34a330d83d4b59e2831",
196
+ "size_bytes": 70695
197
+ },
198
+ {
199
+ "path": "Vocoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin",
200
+ "sha256": "b45e4c17a45cc4de30dcedae9f662548aa35f8f9763632c342b8bbef3089fcee",
201
+ "size_bytes": 50672512
202
+ },
203
+ {
204
+ "path": "Vocoder.mlpackage/Manifest.json",
205
+ "sha256": "5f675492dc6d4d345386d7c974c246ce3228c457acaed27f7aff00754a7bce45",
206
+ "size_bytes": 617
207
+ }
208
+ ],
209
+ "compiled_files": [
210
+ {
211
+ "path": "Vocoder.mlmodelc/analytics/coremldata.bin",
212
+ "sha256": "4fe1b825137629a96dc58a1339bc4ece32041b755f99d638f21a153f2e7faed6",
213
+ "size_bytes": 243
214
+ },
215
+ {
216
+ "path": "Vocoder.mlmodelc/coremldata.bin",
217
+ "sha256": "dbf54fc98879ee1c9a213452e16d09367f3c34ad489b175bcae7ef5f74f8d121",
218
+ "size_bytes": 388
219
+ },
220
+ {
221
+ "path": "Vocoder.mlmodelc/model.mil",
222
+ "sha256": "5f57d6b16a4ee7237e046a423c265aa2c487857eb0003bba1c89fa66e577353f",
223
+ "size_bytes": 75626
224
+ },
225
+ {
226
+ "path": "Vocoder.mlmodelc/weights/weight.bin",
227
+ "sha256": "b45e4c17a45cc4de30dcedae9f662548aa35f8f9763632c342b8bbef3089fcee",
228
+ "size_bytes": 50672512
229
+ }
230
+ ]
231
+ }
232
+ ],
233
+ "assets": [
234
+ {
235
+ "path": "tts.json",
236
+ "sha256": "42078d3aef1cd43ab43021f3c54f47d2d75ceb4e75f627f118890128b06a0d09",
237
+ "size_bytes": 8253
238
+ },
239
+ {
240
+ "path": "unicode_indexer.json",
241
+ "sha256": "9bf7346e43883a81f8645c81224f786d43c5b57f3641f6e7671a7d6c493cb24f",
242
+ "size_bytes": 277676
243
+ },
244
+ {
245
+ "path": "voice_styles/M1.json",
246
+ "sha256": "e35604687f5d23694b8e91593a93eec0e4eca6c0b02bb8ed69139ab2ea6b0a5b",
247
+ "size_bytes": 291748
248
+ }
249
+ ]
250
+ }
tts.json ADDED
@@ -0,0 +1,311 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tts_version": "v1.7.3",
3
+ "split": "opensource-multilingual",
4
+ "ttl": {
5
+ "latent_dim": 24,
6
+ "chunk_compress_factor": 6,
7
+ "batch_expander": {
8
+ "n_batch_expand": 6
9
+ },
10
+ "normalizer": {
11
+ "scale": 0.25
12
+ },
13
+ "text_encoder": {
14
+ "n_langs": 0,
15
+ "lang_emb_dim": 0,
16
+ "text_embedder": {
17
+ "char_emb_dim": 256
18
+ },
19
+ "convnext": {
20
+ "idim": 256,
21
+ "ksz": 5,
22
+ "intermediate_dim": 1024,
23
+ "num_layers": 6,
24
+ "dilation_lst": [
25
+ 1,
26
+ 1,
27
+ 2,
28
+ 2,
29
+ 4,
30
+ 4
31
+ ]
32
+ },
33
+ "attn_encoder": {
34
+ "hidden_channels": 256,
35
+ "filter_channels": 1024,
36
+ "n_heads": 4,
37
+ "n_layers": 4,
38
+ "p_dropout": 0.0
39
+ },
40
+ "proj_out": {
41
+ "idim": 256,
42
+ "odim": 256
43
+ }
44
+ },
45
+ "flow_matching": {
46
+ "sig_min": 1e-08
47
+ },
48
+ "style_encoder": {
49
+ "proj_in": {
50
+ "ldim": 24,
51
+ "chunk_compress_factor": 6,
52
+ "odim": 256
53
+ },
54
+ "convnext": {
55
+ "idim": 256,
56
+ "ksz": 5,
57
+ "intermediate_dim": 1024,
58
+ "num_layers": 6,
59
+ "dilation_lst": [
60
+ 1,
61
+ 1,
62
+ 1,
63
+ 1,
64
+ 1,
65
+ 1
66
+ ]
67
+ },
68
+ "style_token_layer": {
69
+ "input_dim": 256,
70
+ "n_style": 50,
71
+ "style_key_dim": 256,
72
+ "style_value_dim": 256,
73
+ "prototype_dim": 256,
74
+ "n_units": 256,
75
+ "n_heads": 2
76
+ }
77
+ },
78
+ "speech_prompted_text_encoder": {
79
+ "text_dim": 256,
80
+ "style_dim": 256,
81
+ "n_units": 256,
82
+ "n_heads": 2
83
+ },
84
+ "uncond_masker": {
85
+ "prob_both_uncond": 0.04,
86
+ "prob_text_uncond": 0.01,
87
+ "std": 0.1,
88
+ "text_dim": 256,
89
+ "n_style": 50,
90
+ "style_key_dim": 256,
91
+ "style_value_dim": 256
92
+ },
93
+ "vector_field": {
94
+ "n_langs": 0,
95
+ "lang_emb_dim": 0,
96
+ "proj_in": {
97
+ "ldim": 24,
98
+ "chunk_compress_factor": 6,
99
+ "odim": 512
100
+ },
101
+ "time_encoder": {
102
+ "time_dim": 64,
103
+ "hdim": 256
104
+ },
105
+ "main_blocks": {
106
+ "n_blocks": 4,
107
+ "time_cond_layer": {
108
+ "idim": 512,
109
+ "time_dim": 64
110
+ },
111
+ "style_cond_layer": {
112
+ "idim": 512,
113
+ "style_dim": 256
114
+ },
115
+ "text_cond_layer": {
116
+ "idim": 512,
117
+ "text_dim": 256,
118
+ "n_heads": 8,
119
+ "n_units": 512,
120
+ "use_residual": true,
121
+ "rotary_base": 10000,
122
+ "rotary_scale": 10
123
+ },
124
+ "convnext_0": {
125
+ "idim": 512,
126
+ "ksz": 5,
127
+ "intermediate_dim": 2048,
128
+ "num_layers": 4,
129
+ "dilation_lst": [
130
+ 1,
131
+ 2,
132
+ 4,
133
+ 8
134
+ ]
135
+ },
136
+ "convnext_1": {
137
+ "idim": 512,
138
+ "ksz": 5,
139
+ "intermediate_dim": 2048,
140
+ "num_layers": 1,
141
+ "dilation_lst": [
142
+ 1
143
+ ]
144
+ },
145
+ "convnext_2": {
146
+ "idim": 512,
147
+ "ksz": 5,
148
+ "intermediate_dim": 2048,
149
+ "num_layers": 1,
150
+ "dilation_lst": [
151
+ 1
152
+ ]
153
+ }
154
+ },
155
+ "last_convnext": {
156
+ "idim": 512,
157
+ "ksz": 5,
158
+ "intermediate_dim": 2048,
159
+ "num_layers": 4,
160
+ "dilation_lst": [
161
+ 1,
162
+ 1,
163
+ 1,
164
+ 1
165
+ ]
166
+ },
167
+ "proj_out": {
168
+ "idim": 512,
169
+ "chunk_compress_factor": 6,
170
+ "ldim": 24
171
+ }
172
+ }
173
+ },
174
+ "ae": {
175
+ "sample_rate": 44100,
176
+ "n_delay": 0,
177
+ "base_chunk_size": 512,
178
+ "chunk_compress_factor": 1,
179
+ "ldim": 24,
180
+ "encoder": {
181
+ "spec_processor": {
182
+ "n_fft": 2048,
183
+ "win_length": 2048,
184
+ "hop_length": 512,
185
+ "n_mels": 228,
186
+ "sample_rate": 44100,
187
+ "eps": 1e-05,
188
+ "norm_mean": 0.0,
189
+ "norm_std": 1.0
190
+ },
191
+ "ksz_init": 7,
192
+ "ksz": 7,
193
+ "num_layers": 10,
194
+ "dilation_lst": [
195
+ 1,
196
+ 1,
197
+ 1,
198
+ 1,
199
+ 1,
200
+ 1,
201
+ 1,
202
+ 1,
203
+ 1,
204
+ 1
205
+ ],
206
+ "intermediate_dim": 2048,
207
+ "idim": 1253,
208
+ "hdim": 512,
209
+ "odim": 24
210
+ },
211
+ "decoder": {
212
+ "ksz_init": 7,
213
+ "ksz": 7,
214
+ "num_layers": 10,
215
+ "dilation_lst": [
216
+ 1,
217
+ 2,
218
+ 4,
219
+ 1,
220
+ 2,
221
+ 4,
222
+ 1,
223
+ 1,
224
+ 1,
225
+ 1
226
+ ],
227
+ "intermediate_dim": 2048,
228
+ "idim": 24,
229
+ "hdim": 512,
230
+ "head": {
231
+ "idim": 512,
232
+ "hdim": 2048,
233
+ "odim": 512,
234
+ "ksz": 3
235
+ }
236
+ }
237
+ },
238
+ "dp": {
239
+ "latent_dim": 24,
240
+ "chunk_compress_factor": 6,
241
+ "normalizer": {
242
+ "scale": 1.0
243
+ },
244
+ "sentence_encoder": {
245
+ "char_emb_dim": 64,
246
+ "text_embedder": {
247
+ "char_emb_dim": 64
248
+ },
249
+ "convnext": {
250
+ "idim": 64,
251
+ "ksz": 5,
252
+ "intermediate_dim": 256,
253
+ "num_layers": 6,
254
+ "dilation_lst": [
255
+ 1,
256
+ 1,
257
+ 1,
258
+ 1,
259
+ 1,
260
+ 1
261
+ ]
262
+ },
263
+ "attn_encoder": {
264
+ "hidden_channels": 64,
265
+ "filter_channels": 256,
266
+ "n_heads": 2,
267
+ "n_layers": 2,
268
+ "p_dropout": 0.0
269
+ },
270
+ "proj_out": {
271
+ "idim": 64,
272
+ "odim": 64
273
+ }
274
+ },
275
+ "style_encoder": {
276
+ "proj_in": {
277
+ "ldim": 24,
278
+ "chunk_compress_factor": 6,
279
+ "odim": 64
280
+ },
281
+ "convnext": {
282
+ "idim": 64,
283
+ "ksz": 5,
284
+ "intermediate_dim": 256,
285
+ "num_layers": 4,
286
+ "dilation_lst": [
287
+ 1,
288
+ 1,
289
+ 1,
290
+ 1
291
+ ]
292
+ },
293
+ "style_token_layer": {
294
+ "input_dim": 64,
295
+ "n_style": 8,
296
+ "style_key_dim": 0,
297
+ "style_value_dim": 16,
298
+ "prototype_dim": 64,
299
+ "n_units": 64,
300
+ "n_heads": 2
301
+ }
302
+ },
303
+ "predictor": {
304
+ "sentence_dim": 64,
305
+ "n_style": 8,
306
+ "style_dim": 16,
307
+ "hdim": 128,
308
+ "n_layer": 2
309
+ }
310
+ }
311
+ }
unicode_indexer.json ADDED
The diff for this file is too large to render. See raw diff
 
voice_styles/M1.json ADDED
The diff for this file is too large to render. See raw diff