Text-to-Speech
Core ML
Supertonic
speech
audio
tts
ane
apple-silicon
flow-matching
diffusion
multilingual
Instructions to use FluidInference/supertonic-3-coreml with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Supertonic
How to use FluidInference/supertonic-3-coreml with Supertonic:
from supertonic import TTS tts = TTS(auto_download=True) style = tts.get_voice_style(voice_name="M1") text = "The train delay was announced at 4:45 PM on Wed, Apr 3, 2024 due to track maintenance." wav, duration = tts.synthesize(text, voice_style=style) tts.save_audio(wav, "output.wav")
- Notebooks
- Google Colab
- Kaggle
Upload 33 files
Browse files- DurationPredictor.mlmodelc/analytics/coremldata.bin +3 -0
- DurationPredictor.mlmodelc/coremldata.bin +3 -0
- DurationPredictor.mlmodelc/model.mil +616 -0
- DurationPredictor.mlmodelc/weights/weight.bin +3 -0
- DurationPredictor.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- DurationPredictor.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- DurationPredictor.mlpackage/Manifest.json +18 -0
- README.md +202 -0
- TextEncoder.mlmodelc/analytics/coremldata.bin +3 -0
- TextEncoder.mlmodelc/coremldata.bin +3 -0
- TextEncoder.mlmodelc/model.mil +0 -0
- TextEncoder.mlmodelc/weights/weight.bin +3 -0
- TextEncoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- TextEncoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- TextEncoder.mlpackage/Manifest.json +18 -0
- VectorEstimator.mlmodelc/analytics/coremldata.bin +3 -0
- VectorEstimator.mlmodelc/coremldata.bin +3 -0
- VectorEstimator.mlmodelc/model.mil +0 -0
- VectorEstimator.mlmodelc/weights/weight.bin +3 -0
- VectorEstimator.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- VectorEstimator.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- VectorEstimator.mlpackage/Manifest.json +18 -0
- Vocoder.mlmodelc/analytics/coremldata.bin +3 -0
- Vocoder.mlmodelc/coremldata.bin +3 -0
- Vocoder.mlmodelc/model.mil +465 -0
- Vocoder.mlmodelc/weights/weight.bin +3 -0
- Vocoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- Vocoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- Vocoder.mlpackage/Manifest.json +18 -0
- manifest.json +250 -0
- tts.json +311 -0
- unicode_indexer.json +0 -0
- 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)
|
| 65 |
+
| [](#model-details)
|
| 66 |
+
| [](#supported-languages)
|
| 67 |
+
| [](https://discord.gg/WNsvaCtmDe)
|
| 68 |
+
| [](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
|
|
|