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