Upload 204 files
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- fr/1120ms/decoder.mlmodelc/analytics/coremldata.bin +3 -0
- fr/1120ms/decoder.mlmodelc/coremldata.bin +3 -0
- fr/1120ms/decoder.mlmodelc/model.mil +64 -0
- fr/1120ms/decoder.mlmodelc/weights/weight.bin +3 -0
- fr/1120ms/decoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- fr/1120ms/decoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- fr/1120ms/decoder.mlpackage/Manifest.json +18 -0
- fr/1120ms/decoder_joint.mlmodelc/analytics/coremldata.bin +3 -0
- fr/1120ms/decoder_joint.mlmodelc/coremldata.bin +3 -0
- fr/1120ms/decoder_joint.mlmodelc/model.mil +83 -0
- fr/1120ms/decoder_joint.mlmodelc/weights/weight.bin +3 -0
- fr/1120ms/decoder_joint.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- fr/1120ms/decoder_joint.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- fr/1120ms/decoder_joint.mlpackage/Manifest.json +18 -0
- fr/1120ms/decoder_joint_noencproj.mlmodelc/analytics/coremldata.bin +3 -0
- fr/1120ms/decoder_joint_noencproj.mlmodelc/coremldata.bin +3 -0
- fr/1120ms/decoder_joint_noencproj.mlmodelc/model.mil +91 -0
- fr/1120ms/decoder_joint_noencproj.mlmodelc/weights/weight.bin +3 -0
- fr/1120ms/decoder_joint_noencproj.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- fr/1120ms/decoder_joint_noencproj.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- fr/1120ms/decoder_joint_noencproj.mlpackage/Manifest.json +18 -0
- fr/1120ms/encoder.mlmodelc/analytics/coremldata.bin +3 -0
- fr/1120ms/encoder.mlmodelc/coremldata.bin +3 -0
- fr/1120ms/encoder.mlmodelc/model.mil +0 -0
- fr/1120ms/encoder.mlmodelc/weights/weight.bin +3 -0
- fr/1120ms/encoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- fr/1120ms/encoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- fr/1120ms/encoder.mlpackage/Manifest.json +18 -0
- fr/1120ms/joint.mlmodelc/analytics/coremldata.bin +3 -0
- fr/1120ms/joint.mlmodelc/coremldata.bin +3 -0
- fr/1120ms/joint.mlmodelc/model.mil +31 -0
- fr/1120ms/joint.mlmodelc/weights/weight.bin +3 -0
- fr/1120ms/joint.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- fr/1120ms/joint.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- fr/1120ms/joint.mlpackage/Manifest.json +18 -0
- fr/1120ms/joint_noencproj_batched.mlmodelc/analytics/coremldata.bin +3 -0
- fr/1120ms/joint_noencproj_batched.mlmodelc/coremldata.bin +3 -0
- fr/1120ms/joint_noencproj_batched.mlmodelc/model.mil +26 -0
- fr/1120ms/joint_noencproj_batched.mlmodelc/weights/weight.bin +3 -0
- fr/1120ms/joint_noencproj_batched.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- fr/1120ms/joint_noencproj_batched.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- fr/1120ms/joint_noencproj_batched.mlpackage/Manifest.json +18 -0
- fr/1120ms/metadata.json +198 -0
- fr/1120ms/preprocessor.mlmodelc/analytics/coremldata.bin +3 -0
- fr/1120ms/preprocessor.mlmodelc/coremldata.bin +3 -0
- fr/1120ms/preprocessor.mlmodelc/model.mil +122 -0
- fr/1120ms/preprocessor.mlmodelc/weights/weight.bin +3 -0
- fr/1120ms/preprocessor.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- fr/1120ms/preprocessor.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- fr/1120ms/preprocessor.mlpackage/Manifest.json +18 -0
fr/1120ms/decoder.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3d3becf0abd7a7efd30577f33f4144cfcb8c08155fdd62903eb1a45d7b02b87c
|
| 3 |
+
size 243
|
fr/1120ms/decoder.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d6b7630abf0c1b447913bf9d1ed5935d625bcb70e978c607820b2ff35ee8f07f
|
| 3 |
+
size 433
|
fr/1120ms/decoder.mlmodelc/model.mil
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
program(1.3)
|
| 2 |
+
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.5.1"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "8.3.0"}})]
|
| 3 |
+
{
|
| 4 |
+
func main<ios18>(tensor<fp32, [2, 1, 640]> c_in, tensor<fp32, [2, 1, 640]> h_in, tensor<int32, [1, 1]> token, tensor<int32, [1]> token_length) {
|
| 5 |
+
int32 y_axis_0 = const()[name = string("y_axis_0"), val = int32(0)];
|
| 6 |
+
int32 y_batch_dims_0 = const()[name = string("y_batch_dims_0"), val = int32(0)];
|
| 7 |
+
bool y_validate_indices_0 = const()[name = string("y_validate_indices_0"), val = bool(false)];
|
| 8 |
+
tensor<fp16, [849, 640]> module_prediction_embed_weight_to_fp16 = const()[name = string("module_prediction_embed_weight_to_fp16"), val = tensor<fp16, [849, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
| 9 |
+
string token_to_int16_dtype_0 = const()[name = string("token_to_int16_dtype_0"), val = string("int16")];
|
| 10 |
+
tensor<int16, [1, 1]> token_to_int16 = cast(dtype = token_to_int16_dtype_0, x = token)[name = string("cast_8")];
|
| 11 |
+
tensor<fp16, [1, 1, 640]> y_cast_fp16_cast_uint16 = gather(axis = y_axis_0, batch_dims = y_batch_dims_0, indices = token_to_int16, validate_indices = y_validate_indices_0, x = module_prediction_embed_weight_to_fp16)[name = string("y_cast_fp16_cast_uint16")];
|
| 12 |
+
tensor<int32, [3]> input_3_perm_0 = const()[name = string("input_3_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
|
| 13 |
+
int32 split_0_num_splits_0 = const()[name = string("split_0_num_splits_0"), val = int32(2)];
|
| 14 |
+
int32 split_0_axis_0 = const()[name = string("split_0_axis_0"), val = int32(0)];
|
| 15 |
+
string h_in_to_fp16_dtype_0 = const()[name = string("h_in_to_fp16_dtype_0"), val = string("fp16")];
|
| 16 |
+
tensor<fp16, [2, 1, 640]> h_in_to_fp16 = cast(dtype = h_in_to_fp16_dtype_0, x = h_in)[name = string("cast_7")];
|
| 17 |
+
tensor<fp16, [1, 1, 640]> split_0_cast_fp16_0, tensor<fp16, [1, 1, 640]> split_0_cast_fp16_1 = split(axis = split_0_axis_0, num_splits = split_0_num_splits_0, x = h_in_to_fp16)[name = string("split_0_cast_fp16")];
|
| 18 |
+
int32 split_1_num_splits_0 = const()[name = string("split_1_num_splits_0"), val = int32(2)];
|
| 19 |
+
int32 split_1_axis_0 = const()[name = string("split_1_axis_0"), val = int32(0)];
|
| 20 |
+
string c_in_to_fp16_dtype_0 = const()[name = string("c_in_to_fp16_dtype_0"), val = string("fp16")];
|
| 21 |
+
tensor<fp16, [2, 1, 640]> c_in_to_fp16 = cast(dtype = c_in_to_fp16_dtype_0, x = c_in)[name = string("cast_6")];
|
| 22 |
+
tensor<fp16, [1, 1, 640]> split_1_cast_fp16_0, tensor<fp16, [1, 1, 640]> split_1_cast_fp16_1 = split(axis = split_1_axis_0, num_splits = split_1_num_splits_0, x = c_in_to_fp16)[name = string("split_1_cast_fp16")];
|
| 23 |
+
tensor<int32, [1]> input_lstm_layer_0_lstm_h0_squeeze_axes_0 = const()[name = string("input_lstm_layer_0_lstm_h0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 24 |
+
tensor<fp16, [1, 640]> input_lstm_layer_0_lstm_h0_squeeze_cast_fp16 = squeeze(axes = input_lstm_layer_0_lstm_h0_squeeze_axes_0, x = split_0_cast_fp16_0)[name = string("input_lstm_layer_0_lstm_h0_squeeze_cast_fp16")];
|
| 25 |
+
tensor<int32, [1]> input_lstm_layer_0_lstm_c0_squeeze_axes_0 = const()[name = string("input_lstm_layer_0_lstm_c0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 26 |
+
tensor<fp16, [1, 640]> input_lstm_layer_0_lstm_c0_squeeze_cast_fp16 = squeeze(axes = input_lstm_layer_0_lstm_c0_squeeze_axes_0, x = split_1_cast_fp16_0)[name = string("input_lstm_layer_0_lstm_c0_squeeze_cast_fp16")];
|
| 27 |
+
string input_lstm_layer_0_direction_0 = const()[name = string("input_lstm_layer_0_direction_0"), val = string("forward")];
|
| 28 |
+
bool input_lstm_layer_0_output_sequence_0 = const()[name = string("input_lstm_layer_0_output_sequence_0"), val = bool(true)];
|
| 29 |
+
string input_lstm_layer_0_recurrent_activation_0 = const()[name = string("input_lstm_layer_0_recurrent_activation_0"), val = string("sigmoid")];
|
| 30 |
+
string input_lstm_layer_0_cell_activation_0 = const()[name = string("input_lstm_layer_0_cell_activation_0"), val = string("tanh")];
|
| 31 |
+
string input_lstm_layer_0_activation_0 = const()[name = string("input_lstm_layer_0_activation_0"), val = string("tanh")];
|
| 32 |
+
tensor<fp16, [2560, 640]> concat_1_to_fp16 = const()[name = string("concat_1_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1086848)))];
|
| 33 |
+
tensor<fp16, [2560, 640]> concat_2_to_fp16 = const()[name = string("concat_2_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4363712)))];
|
| 34 |
+
tensor<fp16, [2560]> concat_0_to_fp16 = const()[name = string("concat_0_to_fp16"), val = tensor<fp16, [2560]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(7640576)))];
|
| 35 |
+
tensor<fp16, [1, 1, 640]> input_3_cast_fp16 = transpose(perm = input_3_perm_0, x = y_cast_fp16_cast_uint16)[name = string("transpose_2")];
|
| 36 |
+
tensor<fp16, [1, 1, 640]> input_lstm_layer_0_cast_fp16_0, tensor<fp16, [1, 640]> input_lstm_layer_0_cast_fp16_1, tensor<fp16, [1, 640]> input_lstm_layer_0_cast_fp16_2 = lstm(activation = input_lstm_layer_0_activation_0, bias = concat_0_to_fp16, cell_activation = input_lstm_layer_0_cell_activation_0, direction = input_lstm_layer_0_direction_0, initial_c = input_lstm_layer_0_lstm_c0_squeeze_cast_fp16, initial_h = input_lstm_layer_0_lstm_h0_squeeze_cast_fp16, output_sequence = input_lstm_layer_0_output_sequence_0, recurrent_activation = input_lstm_layer_0_recurrent_activation_0, weight_hh = concat_2_to_fp16, weight_ih = concat_1_to_fp16, x = input_3_cast_fp16)[name = string("input_lstm_layer_0_cast_fp16")];
|
| 37 |
+
tensor<int32, [1]> input_lstm_h0_squeeze_axes_0 = const()[name = string("input_lstm_h0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 38 |
+
tensor<fp16, [1, 640]> input_lstm_h0_squeeze_cast_fp16 = squeeze(axes = input_lstm_h0_squeeze_axes_0, x = split_0_cast_fp16_1)[name = string("input_lstm_h0_squeeze_cast_fp16")];
|
| 39 |
+
tensor<int32, [1]> input_lstm_c0_squeeze_axes_0 = const()[name = string("input_lstm_c0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 40 |
+
tensor<fp16, [1, 640]> input_lstm_c0_squeeze_cast_fp16 = squeeze(axes = input_lstm_c0_squeeze_axes_0, x = split_1_cast_fp16_1)[name = string("input_lstm_c0_squeeze_cast_fp16")];
|
| 41 |
+
string input_direction_0 = const()[name = string("input_direction_0"), val = string("forward")];
|
| 42 |
+
bool input_output_sequence_0 = const()[name = string("input_output_sequence_0"), val = bool(true)];
|
| 43 |
+
string input_recurrent_activation_0 = const()[name = string("input_recurrent_activation_0"), val = string("sigmoid")];
|
| 44 |
+
string input_cell_activation_0 = const()[name = string("input_cell_activation_0"), val = string("tanh")];
|
| 45 |
+
string input_activation_0 = const()[name = string("input_activation_0"), val = string("tanh")];
|
| 46 |
+
tensor<fp16, [2560, 640]> concat_4_to_fp16 = const()[name = string("concat_4_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(7645760)))];
|
| 47 |
+
tensor<fp16, [2560, 640]> concat_5_to_fp16 = const()[name = string("concat_5_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10922624)))];
|
| 48 |
+
tensor<fp16, [2560]> concat_3_to_fp16 = const()[name = string("concat_3_to_fp16"), val = tensor<fp16, [2560]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14199488)))];
|
| 49 |
+
tensor<fp16, [1, 1, 640]> input_cast_fp16_0, tensor<fp16, [1, 640]> input_cast_fp16_1, tensor<fp16, [1, 640]> input_cast_fp16_2 = lstm(activation = input_activation_0, bias = concat_3_to_fp16, cell_activation = input_cell_activation_0, direction = input_direction_0, initial_c = input_lstm_c0_squeeze_cast_fp16, initial_h = input_lstm_h0_squeeze_cast_fp16, output_sequence = input_output_sequence_0, recurrent_activation = input_recurrent_activation_0, weight_hh = concat_5_to_fp16, weight_ih = concat_4_to_fp16, x = input_lstm_layer_0_cast_fp16_0)[name = string("input_cast_fp16")];
|
| 50 |
+
int32 obj_3_axis_0 = const()[name = string("obj_3_axis_0"), val = int32(0)];
|
| 51 |
+
tensor<fp16, [2, 1, 640]> obj_3_cast_fp16 = stack(axis = obj_3_axis_0, values = (input_lstm_layer_0_cast_fp16_1, input_cast_fp16_1))[name = string("obj_3_cast_fp16")];
|
| 52 |
+
string obj_3_cast_fp16_to_fp32_dtype_0 = const()[name = string("obj_3_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 53 |
+
int32 obj_axis_0 = const()[name = string("obj_axis_0"), val = int32(0)];
|
| 54 |
+
tensor<fp16, [2, 1, 640]> obj_cast_fp16 = stack(axis = obj_axis_0, values = (input_lstm_layer_0_cast_fp16_2, input_cast_fp16_2))[name = string("obj_cast_fp16")];
|
| 55 |
+
string obj_cast_fp16_to_fp32_dtype_0 = const()[name = string("obj_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 56 |
+
tensor<int32, [3]> transpose_0_perm_0 = const()[name = string("transpose_0_perm_0"), val = tensor<int32, [3]>([1, 2, 0])];
|
| 57 |
+
string transpose_0_cast_fp16_to_fp32_dtype_0 = const()[name = string("transpose_0_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 58 |
+
tensor<fp16, [1, 640, 1]> transpose_0_cast_fp16 = transpose(perm = transpose_0_perm_0, x = input_cast_fp16_0)[name = string("transpose_1")];
|
| 59 |
+
tensor<fp32, [1, 640, 1]> decoder_out = cast(dtype = transpose_0_cast_fp16_to_fp32_dtype_0, x = transpose_0_cast_fp16)[name = string("cast_3")];
|
| 60 |
+
tensor<fp32, [2, 1, 640]> c_out = cast(dtype = obj_cast_fp16_to_fp32_dtype_0, x = obj_cast_fp16)[name = string("cast_4")];
|
| 61 |
+
tensor<fp32, [2, 1, 640]> h_out = cast(dtype = obj_3_cast_fp16_to_fp32_dtype_0, x = obj_3_cast_fp16)[name = string("cast_5")];
|
| 62 |
+
tensor<int32, [1]> token_length_tmp = identity(x = token_length)[name = string("token_length_tmp")];
|
| 63 |
+
} -> (decoder_out, h_out, c_out);
|
| 64 |
+
}
|
fr/1120ms/decoder.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8ef2aeef85ed664eb60850330a014c5c299325ba1047d690654dc7ffa870e95a
|
| 3 |
+
size 14204672
|
fr/1120ms/decoder.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c82c0fbdef3af45d82f9b1fe84c0c822154020a5b3cd0e5070d33e9ef0beeca6
|
| 3 |
+
size 10359
|
fr/1120ms/decoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8ef2aeef85ed664eb60850330a014c5c299325ba1047d690654dc7ffa870e95a
|
| 3 |
+
size 14204672
|
fr/1120ms/decoder.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"08617906-EB26-428B-BCF5-BC6013603E2C": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"89EF9907-E7AF-4633-B6D7-C8D472B636BC": {
|
| 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": "89EF9907-E7AF-4633-B6D7-C8D472B636BC"
|
| 18 |
+
}
|
fr/1120ms/decoder_joint.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7f8367de6fb06c7b8053f5ecbbc8e4c7e5100f85f1e1c9ba6e9d9b3e31ecca3d
|
| 3 |
+
size 243
|
fr/1120ms/decoder_joint.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f0566656fc89cd81234bb22159c3ccebfb8c0277e3b740d12a8fb53288c1efcd
|
| 3 |
+
size 454
|
fr/1120ms/decoder_joint.mlmodelc/model.mil
ADDED
|
@@ -0,0 +1,83 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
program(1.3)
|
| 2 |
+
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.5.1"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "8.3.0"}})]
|
| 3 |
+
{
|
| 4 |
+
func main<ios18>(tensor<fp32, [2, 1, 640]> c_in, tensor<fp32, [1, 1024, 1]> encoder, tensor<fp32, [2, 1, 640]> h_in, tensor<int32, [1, 1]> token, tensor<int32, [1]> token_length) {
|
| 5 |
+
int32 y_axis_0 = const()[name = string("y_axis_0"), val = int32(0)];
|
| 6 |
+
int32 y_batch_dims_0 = const()[name = string("y_batch_dims_0"), val = int32(0)];
|
| 7 |
+
bool y_validate_indices_0 = const()[name = string("y_validate_indices_0"), val = bool(false)];
|
| 8 |
+
tensor<fp16, [849, 640]> decoder_module_prediction_embed_weight_to_fp16 = const()[name = string("decoder_module_prediction_embed_weight_to_fp16"), val = tensor<fp16, [849, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
| 9 |
+
string token_to_int16_dtype_0 = const()[name = string("token_to_int16_dtype_0"), val = string("int16")];
|
| 10 |
+
tensor<int16, [1, 1]> token_to_int16 = cast(dtype = token_to_int16_dtype_0, x = token)[name = string("cast_9")];
|
| 11 |
+
tensor<fp16, [1, 1, 640]> y_cast_fp16_cast_uint16 = gather(axis = y_axis_0, batch_dims = y_batch_dims_0, indices = token_to_int16, validate_indices = y_validate_indices_0, x = decoder_module_prediction_embed_weight_to_fp16)[name = string("y_cast_fp16_cast_uint16")];
|
| 12 |
+
tensor<int32, [3]> input_3_perm_0 = const()[name = string("input_3_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
|
| 13 |
+
int32 split_0_num_splits_0 = const()[name = string("split_0_num_splits_0"), val = int32(2)];
|
| 14 |
+
int32 split_0_axis_0 = const()[name = string("split_0_axis_0"), val = int32(0)];
|
| 15 |
+
string h_in_to_fp16_dtype_0 = const()[name = string("h_in_to_fp16_dtype_0"), val = string("fp16")];
|
| 16 |
+
tensor<fp16, [2, 1, 640]> h_in_to_fp16 = cast(dtype = h_in_to_fp16_dtype_0, x = h_in)[name = string("cast_8")];
|
| 17 |
+
tensor<fp16, [1, 1, 640]> split_0_cast_fp16_0, tensor<fp16, [1, 1, 640]> split_0_cast_fp16_1 = split(axis = split_0_axis_0, num_splits = split_0_num_splits_0, x = h_in_to_fp16)[name = string("split_0_cast_fp16")];
|
| 18 |
+
int32 split_1_num_splits_0 = const()[name = string("split_1_num_splits_0"), val = int32(2)];
|
| 19 |
+
int32 split_1_axis_0 = const()[name = string("split_1_axis_0"), val = int32(0)];
|
| 20 |
+
string c_in_to_fp16_dtype_0 = const()[name = string("c_in_to_fp16_dtype_0"), val = string("fp16")];
|
| 21 |
+
tensor<fp16, [2, 1, 640]> c_in_to_fp16 = cast(dtype = c_in_to_fp16_dtype_0, x = c_in)[name = string("cast_7")];
|
| 22 |
+
tensor<fp16, [1, 1, 640]> split_1_cast_fp16_0, tensor<fp16, [1, 1, 640]> split_1_cast_fp16_1 = split(axis = split_1_axis_0, num_splits = split_1_num_splits_0, x = c_in_to_fp16)[name = string("split_1_cast_fp16")];
|
| 23 |
+
tensor<int32, [1]> input_5_lstm_layer_0_lstm_h0_squeeze_axes_0 = const()[name = string("input_5_lstm_layer_0_lstm_h0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 24 |
+
tensor<fp16, [1, 640]> input_5_lstm_layer_0_lstm_h0_squeeze_cast_fp16 = squeeze(axes = input_5_lstm_layer_0_lstm_h0_squeeze_axes_0, x = split_0_cast_fp16_0)[name = string("input_5_lstm_layer_0_lstm_h0_squeeze_cast_fp16")];
|
| 25 |
+
tensor<int32, [1]> input_5_lstm_layer_0_lstm_c0_squeeze_axes_0 = const()[name = string("input_5_lstm_layer_0_lstm_c0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 26 |
+
tensor<fp16, [1, 640]> input_5_lstm_layer_0_lstm_c0_squeeze_cast_fp16 = squeeze(axes = input_5_lstm_layer_0_lstm_c0_squeeze_axes_0, x = split_1_cast_fp16_0)[name = string("input_5_lstm_layer_0_lstm_c0_squeeze_cast_fp16")];
|
| 27 |
+
string input_5_lstm_layer_0_direction_0 = const()[name = string("input_5_lstm_layer_0_direction_0"), val = string("forward")];
|
| 28 |
+
bool input_5_lstm_layer_0_output_sequence_0 = const()[name = string("input_5_lstm_layer_0_output_sequence_0"), val = bool(true)];
|
| 29 |
+
string input_5_lstm_layer_0_recurrent_activation_0 = const()[name = string("input_5_lstm_layer_0_recurrent_activation_0"), val = string("sigmoid")];
|
| 30 |
+
string input_5_lstm_layer_0_cell_activation_0 = const()[name = string("input_5_lstm_layer_0_cell_activation_0"), val = string("tanh")];
|
| 31 |
+
string input_5_lstm_layer_0_activation_0 = const()[name = string("input_5_lstm_layer_0_activation_0"), val = string("tanh")];
|
| 32 |
+
tensor<fp16, [2560, 640]> concat_1_to_fp16 = const()[name = string("concat_1_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1086848)))];
|
| 33 |
+
tensor<fp16, [2560, 640]> concat_2_to_fp16 = const()[name = string("concat_2_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4363712)))];
|
| 34 |
+
tensor<fp16, [2560]> concat_0_to_fp16 = const()[name = string("concat_0_to_fp16"), val = tensor<fp16, [2560]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(7640576)))];
|
| 35 |
+
tensor<fp16, [1, 1, 640]> input_3_cast_fp16 = transpose(perm = input_3_perm_0, x = y_cast_fp16_cast_uint16)[name = string("transpose_4")];
|
| 36 |
+
tensor<fp16, [1, 1, 640]> input_5_lstm_layer_0_cast_fp16_0, tensor<fp16, [1, 640]> input_5_lstm_layer_0_cast_fp16_1, tensor<fp16, [1, 640]> input_5_lstm_layer_0_cast_fp16_2 = lstm(activation = input_5_lstm_layer_0_activation_0, bias = concat_0_to_fp16, cell_activation = input_5_lstm_layer_0_cell_activation_0, direction = input_5_lstm_layer_0_direction_0, initial_c = input_5_lstm_layer_0_lstm_c0_squeeze_cast_fp16, initial_h = input_5_lstm_layer_0_lstm_h0_squeeze_cast_fp16, output_sequence = input_5_lstm_layer_0_output_sequence_0, recurrent_activation = input_5_lstm_layer_0_recurrent_activation_0, weight_hh = concat_2_to_fp16, weight_ih = concat_1_to_fp16, x = input_3_cast_fp16)[name = string("input_5_lstm_layer_0_cast_fp16")];
|
| 37 |
+
tensor<int32, [1]> input_5_lstm_h0_squeeze_axes_0 = const()[name = string("input_5_lstm_h0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 38 |
+
tensor<fp16, [1, 640]> input_5_lstm_h0_squeeze_cast_fp16 = squeeze(axes = input_5_lstm_h0_squeeze_axes_0, x = split_0_cast_fp16_1)[name = string("input_5_lstm_h0_squeeze_cast_fp16")];
|
| 39 |
+
tensor<int32, [1]> input_5_lstm_c0_squeeze_axes_0 = const()[name = string("input_5_lstm_c0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 40 |
+
tensor<fp16, [1, 640]> input_5_lstm_c0_squeeze_cast_fp16 = squeeze(axes = input_5_lstm_c0_squeeze_axes_0, x = split_1_cast_fp16_1)[name = string("input_5_lstm_c0_squeeze_cast_fp16")];
|
| 41 |
+
string input_5_direction_0 = const()[name = string("input_5_direction_0"), val = string("forward")];
|
| 42 |
+
bool input_5_output_sequence_0 = const()[name = string("input_5_output_sequence_0"), val = bool(true)];
|
| 43 |
+
string input_5_recurrent_activation_0 = const()[name = string("input_5_recurrent_activation_0"), val = string("sigmoid")];
|
| 44 |
+
string input_5_cell_activation_0 = const()[name = string("input_5_cell_activation_0"), val = string("tanh")];
|
| 45 |
+
string input_5_activation_0 = const()[name = string("input_5_activation_0"), val = string("tanh")];
|
| 46 |
+
tensor<fp16, [2560, 640]> concat_4_to_fp16 = const()[name = string("concat_4_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(7645760)))];
|
| 47 |
+
tensor<fp16, [2560, 640]> concat_5_to_fp16 = const()[name = string("concat_5_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10922624)))];
|
| 48 |
+
tensor<fp16, [2560]> concat_3_to_fp16 = const()[name = string("concat_3_to_fp16"), val = tensor<fp16, [2560]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14199488)))];
|
| 49 |
+
tensor<fp16, [1, 1, 640]> input_5_cast_fp16_0, tensor<fp16, [1, 640]> input_5_cast_fp16_1, tensor<fp16, [1, 640]> input_5_cast_fp16_2 = lstm(activation = input_5_activation_0, bias = concat_3_to_fp16, cell_activation = input_5_cell_activation_0, direction = input_5_direction_0, initial_c = input_5_lstm_c0_squeeze_cast_fp16, initial_h = input_5_lstm_h0_squeeze_cast_fp16, output_sequence = input_5_output_sequence_0, recurrent_activation = input_5_recurrent_activation_0, weight_hh = concat_5_to_fp16, weight_ih = concat_4_to_fp16, x = input_5_lstm_layer_0_cast_fp16_0)[name = string("input_5_cast_fp16")];
|
| 50 |
+
int32 obj_3_axis_0 = const()[name = string("obj_3_axis_0"), val = int32(0)];
|
| 51 |
+
tensor<fp16, [2, 1, 640]> obj_3_cast_fp16 = stack(axis = obj_3_axis_0, values = (input_5_lstm_layer_0_cast_fp16_1, input_5_cast_fp16_1))[name = string("obj_3_cast_fp16")];
|
| 52 |
+
string obj_3_cast_fp16_to_fp32_dtype_0 = const()[name = string("obj_3_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 53 |
+
int32 obj_axis_0 = const()[name = string("obj_axis_0"), val = int32(0)];
|
| 54 |
+
tensor<fp16, [2, 1, 640]> obj_cast_fp16 = stack(axis = obj_axis_0, values = (input_5_lstm_layer_0_cast_fp16_2, input_5_cast_fp16_2))[name = string("obj_cast_fp16")];
|
| 55 |
+
string obj_cast_fp16_to_fp32_dtype_0 = const()[name = string("obj_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 56 |
+
tensor<int32, [3]> transpose_1_perm_0 = const()[name = string("transpose_1_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
|
| 57 |
+
tensor<int32, [3]> input_7_perm_0 = const()[name = string("input_7_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
|
| 58 |
+
string encoder_to_fp16_dtype_0 = const()[name = string("encoder_to_fp16_dtype_0"), val = string("fp16")];
|
| 59 |
+
tensor<fp16, [640, 1024]> joint_module_enc_weight_to_fp16 = const()[name = string("joint_module_enc_weight_to_fp16"), val = tensor<fp16, [640, 1024]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14204672)))];
|
| 60 |
+
tensor<fp16, [640]> joint_module_enc_bias_to_fp16 = const()[name = string("joint_module_enc_bias_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(15515456)))];
|
| 61 |
+
tensor<fp16, [1, 1024, 1]> encoder_to_fp16 = cast(dtype = encoder_to_fp16_dtype_0, x = encoder)[name = string("cast_4")];
|
| 62 |
+
tensor<fp16, [1, 1, 1024]> input_7_cast_fp16 = transpose(perm = input_7_perm_0, x = encoder_to_fp16)[name = string("transpose_2")];
|
| 63 |
+
tensor<fp16, [1, 1, 640]> linear_0_cast_fp16 = linear(bias = joint_module_enc_bias_to_fp16, weight = joint_module_enc_weight_to_fp16, x = input_7_cast_fp16)[name = string("linear_0_cast_fp16")];
|
| 64 |
+
tensor<fp16, [640, 640]> joint_module_pred_weight_to_fp16 = const()[name = string("joint_module_pred_weight_to_fp16"), val = tensor<fp16, [640, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(15516800)))];
|
| 65 |
+
tensor<fp16, [640]> joint_module_pred_bias_to_fp16 = const()[name = string("joint_module_pred_bias_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(16336064)))];
|
| 66 |
+
tensor<fp16, [1, 1, 640]> transpose_1_cast_fp16 = transpose(perm = transpose_1_perm_0, x = input_5_cast_fp16_0)[name = string("transpose_3")];
|
| 67 |
+
tensor<fp16, [1, 1, 640]> linear_1_cast_fp16 = linear(bias = joint_module_pred_bias_to_fp16, weight = joint_module_pred_weight_to_fp16, x = transpose_1_cast_fp16)[name = string("linear_1_cast_fp16")];
|
| 68 |
+
tensor<int32, [1]> var_79_axes_0 = const()[name = string("op_79_axes_0"), val = tensor<int32, [1]>([2])];
|
| 69 |
+
tensor<fp16, [1, 1, 1, 640]> var_79_cast_fp16 = expand_dims(axes = var_79_axes_0, x = linear_0_cast_fp16)[name = string("op_79_cast_fp16")];
|
| 70 |
+
tensor<int32, [1]> var_80_axes_0 = const()[name = string("op_80_axes_0"), val = tensor<int32, [1]>([1])];
|
| 71 |
+
tensor<fp16, [1, 1, 1, 640]> var_80_cast_fp16 = expand_dims(axes = var_80_axes_0, x = linear_1_cast_fp16)[name = string("op_80_cast_fp16")];
|
| 72 |
+
tensor<fp16, [1, 1, 1, 640]> input_11_cast_fp16 = add(x = var_79_cast_fp16, y = var_80_cast_fp16)[name = string("input_11_cast_fp16")];
|
| 73 |
+
tensor<fp16, [1, 1, 1, 640]> input_13_cast_fp16 = relu(x = input_11_cast_fp16)[name = string("input_13_cast_fp16")];
|
| 74 |
+
tensor<fp16, [849, 640]> joint_module_joint_net_2_weight_to_fp16 = const()[name = string("joint_module_joint_net_2_weight_to_fp16"), val = tensor<fp16, [849, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(16337408)))];
|
| 75 |
+
tensor<fp16, [849]> joint_module_joint_net_2_bias_to_fp16 = const()[name = string("joint_module_joint_net_2_bias_to_fp16"), val = tensor<fp16, [849]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(17424192)))];
|
| 76 |
+
tensor<fp16, [1, 1, 1, 849]> linear_2_cast_fp16 = linear(bias = joint_module_joint_net_2_bias_to_fp16, weight = joint_module_joint_net_2_weight_to_fp16, x = input_13_cast_fp16)[name = string("linear_2_cast_fp16")];
|
| 77 |
+
string linear_2_cast_fp16_to_fp32_dtype_0 = const()[name = string("linear_2_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 78 |
+
tensor<fp32, [1, 1, 1, 849]> logits = cast(dtype = linear_2_cast_fp16_to_fp32_dtype_0, x = linear_2_cast_fp16)[name = string("cast_3")];
|
| 79 |
+
tensor<fp32, [2, 1, 640]> c_out = cast(dtype = obj_cast_fp16_to_fp32_dtype_0, x = obj_cast_fp16)[name = string("cast_5")];
|
| 80 |
+
tensor<fp32, [2, 1, 640]> h_out = cast(dtype = obj_3_cast_fp16_to_fp32_dtype_0, x = obj_3_cast_fp16)[name = string("cast_6")];
|
| 81 |
+
tensor<int32, [1]> token_length_tmp = identity(x = token_length)[name = string("token_length_tmp")];
|
| 82 |
+
} -> (logits, h_out, c_out);
|
| 83 |
+
}
|
fr/1120ms/decoder_joint.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:39ce7eecdfb6a432f391c931412d9a5cd15fb0ef85562ef0c207e4a18fe6b50e
|
| 3 |
+
size 17425954
|
fr/1120ms/decoder_joint.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:72ce6705a73b8b0488f31dd35a385908473aff8bef0ffe290f56bb51068a1d26
|
| 3 |
+
size 13745
|
fr/1120ms/decoder_joint.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:39ce7eecdfb6a432f391c931412d9a5cd15fb0ef85562ef0c207e4a18fe6b50e
|
| 3 |
+
size 17425954
|
fr/1120ms/decoder_joint.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"8FCEBC80-AF4D-42E1-B70D-96BB47913BC7": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Specification",
|
| 7 |
+
"name": "model.mlmodel",
|
| 8 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 9 |
+
},
|
| 10 |
+
"A77788EC-CA36-4DF3-BB1B-E6182E0ABF28": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Weights",
|
| 13 |
+
"name": "weights",
|
| 14 |
+
"path": "com.apple.CoreML/weights"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "8FCEBC80-AF4D-42E1-B70D-96BB47913BC7"
|
| 18 |
+
}
|
fr/1120ms/decoder_joint_noencproj.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:878825967d0c03e7b016222a7b824badb35d1d7e134ad3a973abc00dc5fe353a
|
| 3 |
+
size 243
|
fr/1120ms/decoder_joint_noencproj.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ebb7868572e2691055cf8d926fe61e2049bdc0a3e6349e82227f0b82dd849814
|
| 3 |
+
size 519
|
fr/1120ms/decoder_joint_noencproj.mlmodelc/model.mil
ADDED
|
@@ -0,0 +1,91 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
program(1.3)
|
| 2 |
+
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.10.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
|
| 3 |
+
{
|
| 4 |
+
func main<ios18>(tensor<fp32, [2, 1, 640]> c_in, tensor<fp32, [1, 1, 640]> encoder_proj, tensor<fp32, [2, 1, 640]> h_in, tensor<int32, [1, 1]> token, tensor<int32, [1]> token_length) {
|
| 5 |
+
int32 y_batch_dims_0 = const()[name = string("y_batch_dims_0"), val = int32(0)];
|
| 6 |
+
bool y_validate_indices_0 = const()[name = string("y_validate_indices_0"), val = bool(false)];
|
| 7 |
+
tensor<fp16, [849, 640]> decoder_module_prediction_embed_weight_to_fp16 = const()[name = string("decoder_module_prediction_embed_weight_to_fp16"), val = tensor<fp16, [849, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
| 8 |
+
string token_to_int16_dtype_0 = const()[name = string("token_to_int16_dtype_0"), val = string("int16")];
|
| 9 |
+
string cast_1_dtype_0 = const()[name = string("cast_1_dtype_0"), val = string("int32")];
|
| 10 |
+
int32 greater_equal_0_y_0 = const()[name = string("greater_equal_0_y_0"), val = int32(0)];
|
| 11 |
+
tensor<int16, [1, 1]> token_to_int16 = cast(dtype = token_to_int16_dtype_0, x = token)[name = string("cast_10")];
|
| 12 |
+
tensor<int32, [1, 1]> cast_1 = cast(dtype = cast_1_dtype_0, x = token_to_int16)[name = string("cast_9")];
|
| 13 |
+
tensor<bool, [1, 1]> greater_equal_0 = greater_equal(x = cast_1, y = greater_equal_0_y_0)[name = string("greater_equal_0")];
|
| 14 |
+
int32 slice_by_index_0 = const()[name = string("slice_by_index_0"), val = int32(849)];
|
| 15 |
+
tensor<int32, [1, 1]> add_2 = add(x = cast_1, y = slice_by_index_0)[name = string("add_2")];
|
| 16 |
+
tensor<int32, [1, 1]> select_0 = select(a = cast_1, b = add_2, cond = greater_equal_0)[name = string("select_0")];
|
| 17 |
+
int32 y_cast_fp16_cast_uint16_axis_0 = const()[name = string("y_cast_fp16_cast_uint16_axis_0"), val = int32(0)];
|
| 18 |
+
string select_0_to_int16_dtype_0 = const()[name = string("select_0_to_int16_dtype_0"), val = string("int16")];
|
| 19 |
+
tensor<int16, [1, 1]> select_0_to_int16 = cast(dtype = select_0_to_int16_dtype_0, x = select_0)[name = string("cast_8")];
|
| 20 |
+
tensor<fp16, [1, 1, 640]> y_cast_fp16_cast_uint16_cast_uint16 = gather(axis = y_cast_fp16_cast_uint16_axis_0, batch_dims = y_batch_dims_0, indices = select_0_to_int16, validate_indices = y_validate_indices_0, x = decoder_module_prediction_embed_weight_to_fp16)[name = string("y_cast_fp16_cast_uint16_cast_uint16")];
|
| 21 |
+
tensor<int32, [3]> input_3_perm_0 = const()[name = string("input_3_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
|
| 22 |
+
int32 split_0_num_splits_0 = const()[name = string("split_0_num_splits_0"), val = int32(2)];
|
| 23 |
+
int32 split_0_axis_0 = const()[name = string("split_0_axis_0"), val = int32(0)];
|
| 24 |
+
string h_in_to_fp16_dtype_0 = const()[name = string("h_in_to_fp16_dtype_0"), val = string("fp16")];
|
| 25 |
+
tensor<fp16, [2, 1, 640]> h_in_to_fp16 = cast(dtype = h_in_to_fp16_dtype_0, x = h_in)[name = string("cast_7")];
|
| 26 |
+
tensor<fp16, [1, 1, 640]> split_0_cast_fp16_0, tensor<fp16, [1, 1, 640]> split_0_cast_fp16_1 = split(axis = split_0_axis_0, num_splits = split_0_num_splits_0, x = h_in_to_fp16)[name = string("split_0_cast_fp16")];
|
| 27 |
+
int32 split_1_num_splits_0 = const()[name = string("split_1_num_splits_0"), val = int32(2)];
|
| 28 |
+
int32 split_1_axis_0 = const()[name = string("split_1_axis_0"), val = int32(0)];
|
| 29 |
+
string c_in_to_fp16_dtype_0 = const()[name = string("c_in_to_fp16_dtype_0"), val = string("fp16")];
|
| 30 |
+
tensor<fp16, [2, 1, 640]> c_in_to_fp16 = cast(dtype = c_in_to_fp16_dtype_0, x = c_in)[name = string("cast_6")];
|
| 31 |
+
tensor<fp16, [1, 1, 640]> split_1_cast_fp16_0, tensor<fp16, [1, 1, 640]> split_1_cast_fp16_1 = split(axis = split_1_axis_0, num_splits = split_1_num_splits_0, x = c_in_to_fp16)[name = string("split_1_cast_fp16")];
|
| 32 |
+
tensor<int32, [1]> input_5_lstm_layer_0_lstm_h0_squeeze_axes_0 = const()[name = string("input_5_lstm_layer_0_lstm_h0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 33 |
+
tensor<fp16, [1, 640]> input_5_lstm_layer_0_lstm_h0_squeeze_cast_fp16 = squeeze(axes = input_5_lstm_layer_0_lstm_h0_squeeze_axes_0, x = split_0_cast_fp16_0)[name = string("input_5_lstm_layer_0_lstm_h0_squeeze_cast_fp16")];
|
| 34 |
+
tensor<int32, [1]> input_5_lstm_layer_0_lstm_c0_squeeze_axes_0 = const()[name = string("input_5_lstm_layer_0_lstm_c0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 35 |
+
tensor<fp16, [1, 640]> input_5_lstm_layer_0_lstm_c0_squeeze_cast_fp16 = squeeze(axes = input_5_lstm_layer_0_lstm_c0_squeeze_axes_0, x = split_1_cast_fp16_0)[name = string("input_5_lstm_layer_0_lstm_c0_squeeze_cast_fp16")];
|
| 36 |
+
string input_5_lstm_layer_0_direction_0 = const()[name = string("input_5_lstm_layer_0_direction_0"), val = string("forward")];
|
| 37 |
+
bool input_5_lstm_layer_0_output_sequence_0 = const()[name = string("input_5_lstm_layer_0_output_sequence_0"), val = bool(true)];
|
| 38 |
+
string input_5_lstm_layer_0_recurrent_activation_0 = const()[name = string("input_5_lstm_layer_0_recurrent_activation_0"), val = string("sigmoid")];
|
| 39 |
+
string input_5_lstm_layer_0_cell_activation_0 = const()[name = string("input_5_lstm_layer_0_cell_activation_0"), val = string("tanh")];
|
| 40 |
+
string input_5_lstm_layer_0_activation_0 = const()[name = string("input_5_lstm_layer_0_activation_0"), val = string("tanh")];
|
| 41 |
+
tensor<fp16, [2560, 640]> concat_1_to_fp16 = const()[name = string("concat_1_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1086848)))];
|
| 42 |
+
tensor<fp16, [2560, 640]> concat_2_to_fp16 = const()[name = string("concat_2_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4363712)))];
|
| 43 |
+
tensor<fp16, [2560]> concat_0_to_fp16 = const()[name = string("concat_0_to_fp16"), val = tensor<fp16, [2560]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(7640576)))];
|
| 44 |
+
tensor<fp16, [1, 1, 640]> input_3_cast_fp16 = transpose(perm = input_3_perm_0, x = y_cast_fp16_cast_uint16_cast_uint16)[name = string("transpose_3")];
|
| 45 |
+
tensor<fp16, [1, 1, 640]> input_5_lstm_layer_0_cast_fp16_0, tensor<fp16, [1, 640]> input_5_lstm_layer_0_cast_fp16_1, tensor<fp16, [1, 640]> input_5_lstm_layer_0_cast_fp16_2 = lstm(activation = input_5_lstm_layer_0_activation_0, bias = concat_0_to_fp16, cell_activation = input_5_lstm_layer_0_cell_activation_0, direction = input_5_lstm_layer_0_direction_0, initial_c = input_5_lstm_layer_0_lstm_c0_squeeze_cast_fp16, initial_h = input_5_lstm_layer_0_lstm_h0_squeeze_cast_fp16, output_sequence = input_5_lstm_layer_0_output_sequence_0, recurrent_activation = input_5_lstm_layer_0_recurrent_activation_0, weight_hh = concat_2_to_fp16, weight_ih = concat_1_to_fp16, x = input_3_cast_fp16)[name = string("input_5_lstm_layer_0_cast_fp16")];
|
| 46 |
+
tensor<int32, [1]> input_5_lstm_h0_squeeze_axes_0 = const()[name = string("input_5_lstm_h0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 47 |
+
tensor<fp16, [1, 640]> input_5_lstm_h0_squeeze_cast_fp16 = squeeze(axes = input_5_lstm_h0_squeeze_axes_0, x = split_0_cast_fp16_1)[name = string("input_5_lstm_h0_squeeze_cast_fp16")];
|
| 48 |
+
tensor<int32, [1]> input_5_lstm_c0_squeeze_axes_0 = const()[name = string("input_5_lstm_c0_squeeze_axes_0"), val = tensor<int32, [1]>([0])];
|
| 49 |
+
tensor<fp16, [1, 640]> input_5_lstm_c0_squeeze_cast_fp16 = squeeze(axes = input_5_lstm_c0_squeeze_axes_0, x = split_1_cast_fp16_1)[name = string("input_5_lstm_c0_squeeze_cast_fp16")];
|
| 50 |
+
string input_5_direction_0 = const()[name = string("input_5_direction_0"), val = string("forward")];
|
| 51 |
+
bool input_5_output_sequence_0 = const()[name = string("input_5_output_sequence_0"), val = bool(true)];
|
| 52 |
+
string input_5_recurrent_activation_0 = const()[name = string("input_5_recurrent_activation_0"), val = string("sigmoid")];
|
| 53 |
+
string input_5_cell_activation_0 = const()[name = string("input_5_cell_activation_0"), val = string("tanh")];
|
| 54 |
+
string input_5_activation_0 = const()[name = string("input_5_activation_0"), val = string("tanh")];
|
| 55 |
+
tensor<fp16, [2560, 640]> concat_4_to_fp16 = const()[name = string("concat_4_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(7645760)))];
|
| 56 |
+
tensor<fp16, [2560, 640]> concat_5_to_fp16 = const()[name = string("concat_5_to_fp16"), val = tensor<fp16, [2560, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10922624)))];
|
| 57 |
+
tensor<fp16, [2560]> concat_3_to_fp16 = const()[name = string("concat_3_to_fp16"), val = tensor<fp16, [2560]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14199488)))];
|
| 58 |
+
tensor<fp16, [1, 1, 640]> input_5_cast_fp16_0, tensor<fp16, [1, 640]> input_5_cast_fp16_1, tensor<fp16, [1, 640]> input_5_cast_fp16_2 = lstm(activation = input_5_activation_0, bias = concat_3_to_fp16, cell_activation = input_5_cell_activation_0, direction = input_5_direction_0, initial_c = input_5_lstm_c0_squeeze_cast_fp16, initial_h = input_5_lstm_h0_squeeze_cast_fp16, output_sequence = input_5_output_sequence_0, recurrent_activation = input_5_recurrent_activation_0, weight_hh = concat_5_to_fp16, weight_ih = concat_4_to_fp16, x = input_5_lstm_layer_0_cast_fp16_0)[name = string("input_5_cast_fp16")];
|
| 59 |
+
int32 obj_3_axis_0 = const()[name = string("obj_3_axis_0"), val = int32(0)];
|
| 60 |
+
tensor<fp16, [2, 1, 640]> obj_3_cast_fp16 = stack(axis = obj_3_axis_0, values = (input_5_lstm_layer_0_cast_fp16_1, input_5_cast_fp16_1))[name = string("obj_3_cast_fp16")];
|
| 61 |
+
string obj_3_cast_fp16_to_fp32_dtype_0 = const()[name = string("obj_3_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 62 |
+
int32 obj_axis_0 = const()[name = string("obj_axis_0"), val = int32(0)];
|
| 63 |
+
tensor<fp16, [2, 1, 640]> obj_cast_fp16 = stack(axis = obj_axis_0, values = (input_5_lstm_layer_0_cast_fp16_2, input_5_cast_fp16_2))[name = string("obj_cast_fp16")];
|
| 64 |
+
string obj_cast_fp16_to_fp32_dtype_0 = const()[name = string("obj_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 65 |
+
tensor<int32, [3]> transpose_1_perm_0 = const()[name = string("transpose_1_perm_0"), val = tensor<int32, [3]>([1, 0, 2])];
|
| 66 |
+
tensor<fp16, [640, 640]> joint_module_pred_weight_to_fp16 = const()[name = string("joint_module_pred_weight_to_fp16"), val = tensor<fp16, [640, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14204672)))];
|
| 67 |
+
tensor<fp16, [640]> joint_module_pred_bias_to_fp16 = const()[name = string("joint_module_pred_bias_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(15023936)))];
|
| 68 |
+
tensor<fp16, [1, 1, 640]> transpose_1_cast_fp16 = transpose(perm = transpose_1_perm_0, x = input_5_cast_fp16_0)[name = string("transpose_2")];
|
| 69 |
+
tensor<fp16, [1, 1, 640]> linear_0_cast_fp16 = linear(bias = joint_module_pred_bias_to_fp16, weight = joint_module_pred_weight_to_fp16, x = transpose_1_cast_fp16)[name = string("linear_0_cast_fp16")];
|
| 70 |
+
tensor<int32, [1]> f_axes_0 = const()[name = string("f_axes_0"), val = tensor<int32, [1]>([2])];
|
| 71 |
+
string encoder_proj_to_fp16_dtype_0 = const()[name = string("encoder_proj_to_fp16_dtype_0"), val = string("fp16")];
|
| 72 |
+
tensor<fp16, [1, 1, 640]> encoder_proj_to_fp16 = cast(dtype = encoder_proj_to_fp16_dtype_0, x = encoder_proj)[name = string("cast_3")];
|
| 73 |
+
tensor<fp16, [1, 1, 1, 640]> f_cast_fp16 = expand_dims(axes = f_axes_0, x = encoder_proj_to_fp16)[name = string("f_cast_fp16")];
|
| 74 |
+
tensor<int32, [1]> g_axes_0 = const()[name = string("g_axes_0"), val = tensor<int32, [1]>([1])];
|
| 75 |
+
tensor<fp16, [1, 1, 1, 640]> g_cast_fp16 = expand_dims(axes = g_axes_0, x = linear_0_cast_fp16)[name = string("g_cast_fp16")];
|
| 76 |
+
tensor<fp16, [1, 1, 1, 640]> input_9_cast_fp16 = add(x = f_cast_fp16, y = g_cast_fp16)[name = string("input_9_cast_fp16")];
|
| 77 |
+
tensor<fp16, [1, 1, 1, 640]> input_11_cast_fp16 = relu(x = input_9_cast_fp16)[name = string("input_11_cast_fp16")];
|
| 78 |
+
tensor<fp16, [849, 640]> joint_module_joint_net_2_weight_to_fp16 = const()[name = string("joint_module_joint_net_2_weight_to_fp16"), val = tensor<fp16, [849, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(15025280)))];
|
| 79 |
+
tensor<fp16, [849]> joint_module_joint_net_2_bias_to_fp16 = const()[name = string("joint_module_joint_net_2_bias_to_fp16"), val = tensor<fp16, [849]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(16112064)))];
|
| 80 |
+
tensor<fp16, [1, 1, 1, 849]> linear_1_cast_fp16 = linear(bias = joint_module_joint_net_2_bias_to_fp16, weight = joint_module_joint_net_2_weight_to_fp16, x = input_11_cast_fp16)[name = string("linear_1_cast_fp16")];
|
| 81 |
+
int32 var_83 = const()[name = string("op_83"), val = int32(-1)];
|
| 82 |
+
tensor<fp16, [1, 1, 1, 849]> var_85_softmax_cast_fp16 = softmax(axis = var_83, x = linear_1_cast_fp16)[name = string("op_85_softmax_cast_fp16")];
|
| 83 |
+
fp32 var_85_epsilon_0 = const()[name = string("op_85_epsilon_0"), val = fp32(0x1p-149)];
|
| 84 |
+
tensor<fp16, [1, 1, 1, 849]> var_85_cast_fp16 = log(epsilon = var_85_epsilon_0, x = var_85_softmax_cast_fp16)[name = string("op_85_cast_fp16")];
|
| 85 |
+
string var_85_cast_fp16_to_fp32_dtype_0 = const()[name = string("op_85_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 86 |
+
tensor<fp32, [1, 1, 1, 849]> logits = cast(dtype = var_85_cast_fp16_to_fp32_dtype_0, x = var_85_cast_fp16)[name = string("cast_2")];
|
| 87 |
+
tensor<fp32, [2, 1, 640]> c_out = cast(dtype = obj_cast_fp16_to_fp32_dtype_0, x = obj_cast_fp16)[name = string("cast_4")];
|
| 88 |
+
tensor<fp32, [2, 1, 640]> h_out = cast(dtype = obj_3_cast_fp16_to_fp32_dtype_0, x = obj_3_cast_fp16)[name = string("cast_5")];
|
| 89 |
+
tensor<int32, [1]> token_length_tmp = identity(x = token_length)[name = string("token_length_tmp")];
|
| 90 |
+
} -> (logits, h_out, c_out);
|
| 91 |
+
}
|
fr/1120ms/decoder_joint_noencproj.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c691d3004f2ac6a956208569a29d899a7f8e5030087d53fbd488785fa0e8efad
|
| 3 |
+
size 16113826
|
fr/1120ms/decoder_joint_noencproj.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4196a821e716db75328fdd2f99f83214ecd9e6e6481a96a88907fe1c4232f143
|
| 3 |
+
size 14630
|
fr/1120ms/decoder_joint_noencproj.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c691d3004f2ac6a956208569a29d899a7f8e5030087d53fbd488785fa0e8efad
|
| 3 |
+
size 16113826
|
fr/1120ms/decoder_joint_noencproj.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"1D4938C1-F035-4D52-9376-C0FB72335508": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"BF28BA0A-268B-4C46-AB97-C0173D8B55F0": {
|
| 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": "BF28BA0A-268B-4C46-AB97-C0173D8B55F0"
|
| 18 |
+
}
|
fr/1120ms/encoder.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0638780288ed25026170632a153dbed5d798954deac843fa58ddaae3190914d4
|
| 3 |
+
size 243
|
fr/1120ms/encoder.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5df783ab0c78ea6386cb181e85e1458a9cc1a8f5e06a49833b7b7a4b017105ed
|
| 3 |
+
size 662
|
fr/1120ms/encoder.mlmodelc/model.mil
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
fr/1120ms/encoder.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:011a5342cc1e2901633a8f0468be561f50dfc289ebe51851cddb4831f9a6a23f
|
| 3 |
+
size 565952640
|
fr/1120ms/encoder.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:14655ea0dbccc666356247f05afc87bbea77ff469344bfb084526e00c0ce6d15
|
| 3 |
+
size 804512
|
fr/1120ms/encoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:011a5342cc1e2901633a8f0468be561f50dfc289ebe51851cddb4831f9a6a23f
|
| 3 |
+
size 565952640
|
fr/1120ms/encoder.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"6BA56D30-2135-407A-9AD7-D9996C9B4325": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Specification",
|
| 7 |
+
"name": "model.mlmodel",
|
| 8 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 9 |
+
},
|
| 10 |
+
"D3843811-369C-4145-9609-E516DF19D3E5": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Weights",
|
| 13 |
+
"name": "weights",
|
| 14 |
+
"path": "com.apple.CoreML/weights"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "6BA56D30-2135-407A-9AD7-D9996C9B4325"
|
| 18 |
+
}
|
fr/1120ms/joint.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ad1f6f5571ee9e7ad8a533b4679ed7416879f4d8e5fc10f40835c1d014b822a4
|
| 3 |
+
size 243
|
fr/1120ms/joint.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2e630dfefcd0f652f2815f76cfbacb89b51f2c94618fbc517e41779426099a1f
|
| 3 |
+
size 341
|
fr/1120ms/joint.mlmodelc/model.mil
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
program(1.3)
|
| 2 |
+
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.5.1"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "8.3.0"}})]
|
| 3 |
+
{
|
| 4 |
+
func main<ios18>(tensor<fp32, [1, 640, 1]> decoder, tensor<fp32, [1, 1024, 1]> encoder) {
|
| 5 |
+
tensor<int32, [3]> input_1_perm_0 = const()[name = string("input_1_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
|
| 6 |
+
string encoder_to_fp16_dtype_0 = const()[name = string("encoder_to_fp16_dtype_0"), val = string("fp16")];
|
| 7 |
+
tensor<int32, [3]> input_3_perm_0 = const()[name = string("input_3_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
|
| 8 |
+
string decoder_to_fp16_dtype_0 = const()[name = string("decoder_to_fp16_dtype_0"), val = string("fp16")];
|
| 9 |
+
tensor<fp16, [640, 1024]> module_enc_weight_to_fp16 = const()[name = string("module_enc_weight_to_fp16"), val = tensor<fp16, [640, 1024]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
| 10 |
+
tensor<fp16, [640]> module_enc_bias_to_fp16 = const()[name = string("module_enc_bias_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1310848)))];
|
| 11 |
+
tensor<fp16, [1, 1024, 1]> encoder_to_fp16 = cast(dtype = encoder_to_fp16_dtype_0, x = encoder)[name = string("cast_2")];
|
| 12 |
+
tensor<fp16, [1, 1, 1024]> input_1_cast_fp16 = transpose(perm = input_1_perm_0, x = encoder_to_fp16)[name = string("transpose_1")];
|
| 13 |
+
tensor<fp16, [1, 1, 640]> linear_0_cast_fp16 = linear(bias = module_enc_bias_to_fp16, weight = module_enc_weight_to_fp16, x = input_1_cast_fp16)[name = string("linear_0_cast_fp16")];
|
| 14 |
+
tensor<fp16, [640, 640]> module_pred_weight_to_fp16 = const()[name = string("module_pred_weight_to_fp16"), val = tensor<fp16, [640, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1312192)))];
|
| 15 |
+
tensor<fp16, [640]> module_pred_bias_to_fp16 = const()[name = string("module_pred_bias_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2131456)))];
|
| 16 |
+
tensor<fp16, [1, 640, 1]> decoder_to_fp16 = cast(dtype = decoder_to_fp16_dtype_0, x = decoder)[name = string("cast_1")];
|
| 17 |
+
tensor<fp16, [1, 1, 640]> input_3_cast_fp16 = transpose(perm = input_3_perm_0, x = decoder_to_fp16)[name = string("transpose_0")];
|
| 18 |
+
tensor<fp16, [1, 1, 640]> linear_1_cast_fp16 = linear(bias = module_pred_bias_to_fp16, weight = module_pred_weight_to_fp16, x = input_3_cast_fp16)[name = string("linear_1_cast_fp16")];
|
| 19 |
+
tensor<int32, [1]> var_23_axes_0 = const()[name = string("op_23_axes_0"), val = tensor<int32, [1]>([2])];
|
| 20 |
+
tensor<fp16, [1, 1, 1, 640]> var_23_cast_fp16 = expand_dims(axes = var_23_axes_0, x = linear_0_cast_fp16)[name = string("op_23_cast_fp16")];
|
| 21 |
+
tensor<int32, [1]> var_25_axes_0 = const()[name = string("op_25_axes_0"), val = tensor<int32, [1]>([1])];
|
| 22 |
+
tensor<fp16, [1, 1, 1, 640]> var_25_cast_fp16 = expand_dims(axes = var_25_axes_0, x = linear_1_cast_fp16)[name = string("op_25_cast_fp16")];
|
| 23 |
+
tensor<fp16, [1, 1, 1, 640]> input_5_cast_fp16 = add(x = var_23_cast_fp16, y = var_25_cast_fp16)[name = string("input_5_cast_fp16")];
|
| 24 |
+
tensor<fp16, [1, 1, 1, 640]> input_7_cast_fp16 = relu(x = input_5_cast_fp16)[name = string("input_7_cast_fp16")];
|
| 25 |
+
tensor<fp16, [849, 640]> module_joint_net_2_weight_to_fp16 = const()[name = string("module_joint_net_2_weight_to_fp16"), val = tensor<fp16, [849, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2132800)))];
|
| 26 |
+
tensor<fp16, [849]> module_joint_net_2_bias_to_fp16 = const()[name = string("module_joint_net_2_bias_to_fp16"), val = tensor<fp16, [849]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(3219584)))];
|
| 27 |
+
tensor<fp16, [1, 1, 1, 849]> linear_2_cast_fp16 = linear(bias = module_joint_net_2_bias_to_fp16, weight = module_joint_net_2_weight_to_fp16, x = input_7_cast_fp16)[name = string("linear_2_cast_fp16")];
|
| 28 |
+
string linear_2_cast_fp16_to_fp32_dtype_0 = const()[name = string("linear_2_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 29 |
+
tensor<fp32, [1, 1, 1, 849]> logits = cast(dtype = linear_2_cast_fp16_to_fp32_dtype_0, x = linear_2_cast_fp16)[name = string("cast_0")];
|
| 30 |
+
} -> (logits);
|
| 31 |
+
}
|
fr/1120ms/joint.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9fa5fb0d14d87484d9ffd1f4d30996ea456c14ea3e5ab9d910486d8e86d3a540
|
| 3 |
+
size 3221346
|
fr/1120ms/joint.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4bac39a8a21b3e7e8cdeba1d03b92e8f743bf87cb2de3b1d85c505a580a0b218
|
| 3 |
+
size 4486
|
fr/1120ms/joint.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9fa5fb0d14d87484d9ffd1f4d30996ea456c14ea3e5ab9d910486d8e86d3a540
|
| 3 |
+
size 3221346
|
fr/1120ms/joint.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"0DC18C6B-B3DE-4D3A-80EE-28CAA06F6708": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"88105E4A-1C49-40F4-A835-C9B725CEEC59": {
|
| 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": "88105E4A-1C49-40F4-A835-C9B725CEEC59"
|
| 18 |
+
}
|
fr/1120ms/joint_noencproj_batched.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:28b5c13f8ce2d8154edb31968c174d322ee4ae76db42f41ecf7ac5e0fd87f57d
|
| 3 |
+
size 243
|
fr/1120ms/joint_noencproj_batched.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d5dc74778713432d34163ae80901d3294eef3341dd020a29dc05d67eba8f6f0a
|
| 3 |
+
size 406
|
fr/1120ms/joint_noencproj_batched.mlmodelc/model.mil
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
program(1.3)
|
| 2 |
+
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.10.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
|
| 3 |
+
{
|
| 4 |
+
func main<ios18>(tensor<fp32, [1, 640, 1]> decoder, tensor<fp32, [1, 4, 640]> encoder_proj) {
|
| 5 |
+
tensor<int32, [3]> input_1_perm_0 = const()[name = string("input_1_perm_0"), val = tensor<int32, [3]>([0, 2, 1])];
|
| 6 |
+
string decoder_to_fp16_dtype_0 = const()[name = string("decoder_to_fp16_dtype_0"), val = string("fp16")];
|
| 7 |
+
tensor<fp16, [640, 640]> joint_module_pred_weight_to_fp16 = const()[name = string("joint_module_pred_weight_to_fp16"), val = tensor<fp16, [640, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
| 8 |
+
tensor<fp16, [640]> joint_module_pred_bias_to_fp16 = const()[name = string("joint_module_pred_bias_to_fp16"), val = tensor<fp16, [640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(819328)))];
|
| 9 |
+
tensor<fp16, [1, 640, 1]> decoder_to_fp16 = cast(dtype = decoder_to_fp16_dtype_0, x = decoder)[name = string("cast_2")];
|
| 10 |
+
tensor<fp16, [1, 1, 640]> input_1_cast_fp16 = transpose(perm = input_1_perm_0, x = decoder_to_fp16)[name = string("transpose_0")];
|
| 11 |
+
tensor<fp16, [1, 1, 640]> linear_0_cast_fp16 = linear(bias = joint_module_pred_bias_to_fp16, weight = joint_module_pred_weight_to_fp16, x = input_1_cast_fp16)[name = string("linear_0_cast_fp16")];
|
| 12 |
+
tensor<int32, [1]> var_15_axes_0 = const()[name = string("op_15_axes_0"), val = tensor<int32, [1]>([2])];
|
| 13 |
+
string encoder_proj_to_fp16_dtype_0 = const()[name = string("encoder_proj_to_fp16_dtype_0"), val = string("fp16")];
|
| 14 |
+
tensor<fp16, [1, 4, 640]> encoder_proj_to_fp16 = cast(dtype = encoder_proj_to_fp16_dtype_0, x = encoder_proj)[name = string("cast_1")];
|
| 15 |
+
tensor<fp16, [1, 4, 1, 640]> var_15_cast_fp16 = expand_dims(axes = var_15_axes_0, x = encoder_proj_to_fp16)[name = string("op_15_cast_fp16")];
|
| 16 |
+
tensor<int32, [1]> var_17_axes_0 = const()[name = string("op_17_axes_0"), val = tensor<int32, [1]>([1])];
|
| 17 |
+
tensor<fp16, [1, 1, 1, 640]> var_17_cast_fp16 = expand_dims(axes = var_17_axes_0, x = linear_0_cast_fp16)[name = string("op_17_cast_fp16")];
|
| 18 |
+
tensor<fp16, [1, 4, 1, 640]> input_3_cast_fp16 = add(x = var_15_cast_fp16, y = var_17_cast_fp16)[name = string("input_3_cast_fp16")];
|
| 19 |
+
tensor<fp16, [1, 4, 1, 640]> input_5_cast_fp16 = relu(x = input_3_cast_fp16)[name = string("input_5_cast_fp16")];
|
| 20 |
+
tensor<fp16, [849, 640]> joint_module_joint_net_2_weight_to_fp16 = const()[name = string("joint_module_joint_net_2_weight_to_fp16"), val = tensor<fp16, [849, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(820672)))];
|
| 21 |
+
tensor<fp16, [849]> joint_module_joint_net_2_bias_to_fp16 = const()[name = string("joint_module_joint_net_2_bias_to_fp16"), val = tensor<fp16, [849]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1907456)))];
|
| 22 |
+
tensor<fp16, [1, 4, 1, 849]> linear_1_cast_fp16 = linear(bias = joint_module_joint_net_2_bias_to_fp16, weight = joint_module_joint_net_2_weight_to_fp16, x = input_5_cast_fp16)[name = string("linear_1_cast_fp16")];
|
| 23 |
+
string linear_1_cast_fp16_to_fp32_dtype_0 = const()[name = string("linear_1_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 24 |
+
tensor<fp32, [1, 4, 1, 849]> logits = cast(dtype = linear_1_cast_fp16_to_fp32_dtype_0, x = linear_1_cast_fp16)[name = string("cast_0")];
|
| 25 |
+
} -> (logits);
|
| 26 |
+
}
|
fr/1120ms/joint_noencproj_batched.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f9a9dd24e6b63b97540b0db0e990418bc6fddc9c2bf85b82eb226e6df1ea02a4
|
| 3 |
+
size 1909218
|
fr/1120ms/joint_noencproj_batched.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d13f08c4961187e47001773d7d4d5d5937ba7ff68475eb7c1bac41d7014c64a1
|
| 3 |
+
size 3839
|
fr/1120ms/joint_noencproj_batched.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f9a9dd24e6b63b97540b0db0e990418bc6fddc9c2bf85b82eb226e6df1ea02a4
|
| 3 |
+
size 1909218
|
fr/1120ms/joint_noencproj_batched.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"EFDDA9F1-A088-4CE2-8A64-5928EB8F429C": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"F28FF7B3-AC0A-4ACA-93F8-DF201DFFE75B": {
|
| 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": "F28FF7B3-AC0A-4ACA-93F8-DF201DFFE75B"
|
| 18 |
+
}
|
fr/1120ms/metadata.json
ADDED
|
@@ -0,0 +1,198 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model": "nvidia/nemotron-asr-streaming-multilingual-0.6b",
|
| 3 |
+
"model_class": "nemo.collections.asr.models.rnnt_bpe_models_prompt.EncDecRNNTBPEModelWithPrompt",
|
| 4 |
+
"sample_rate": 16000,
|
| 5 |
+
"mel_features": 128,
|
| 6 |
+
"chunk_mel_frames": 112,
|
| 7 |
+
"pre_encode_cache": 9,
|
| 8 |
+
"total_mel_frames": 121,
|
| 9 |
+
"att_context_size": [
|
| 10 |
+
42,
|
| 11 |
+
13
|
| 12 |
+
],
|
| 13 |
+
"vocab_size": 848,
|
| 14 |
+
"blank_idx": 848,
|
| 15 |
+
"vocab_pruned": true,
|
| 16 |
+
"vocab_pruned_original_size": 13087,
|
| 17 |
+
"cache_channel_shape": [
|
| 18 |
+
1,
|
| 19 |
+
24,
|
| 20 |
+
42,
|
| 21 |
+
1024
|
| 22 |
+
],
|
| 23 |
+
"cache_time_shape": [
|
| 24 |
+
1,
|
| 25 |
+
24,
|
| 26 |
+
1024,
|
| 27 |
+
8
|
| 28 |
+
],
|
| 29 |
+
"decoder_hidden": 640,
|
| 30 |
+
"decoder_layers": 2,
|
| 31 |
+
"encoder_dim": 1024,
|
| 32 |
+
"num_prompts": 128,
|
| 33 |
+
"prompt_dictionary": {
|
| 34 |
+
"af-ZA": 54,
|
| 35 |
+
"am-ET": 49,
|
| 36 |
+
"ar": 7,
|
| 37 |
+
"ar-AR": 7,
|
| 38 |
+
"auto": 101,
|
| 39 |
+
"ay-BO": 81,
|
| 40 |
+
"az-AZ": 66,
|
| 41 |
+
"bg": 30,
|
| 42 |
+
"bg-BG": 30,
|
| 43 |
+
"bn-IN": 36,
|
| 44 |
+
"cs": 22,
|
| 45 |
+
"cs-CZ": 22,
|
| 46 |
+
"da": 25,
|
| 47 |
+
"da-DK": 25,
|
| 48 |
+
"de": 9,
|
| 49 |
+
"de-DE": 9,
|
| 50 |
+
"el": 21,
|
| 51 |
+
"el-GR": 21,
|
| 52 |
+
"en": 0,
|
| 53 |
+
"en-GB": 1,
|
| 54 |
+
"en-US": 0,
|
| 55 |
+
"enGB": 1,
|
| 56 |
+
"es": 3,
|
| 57 |
+
"es-ES": 2,
|
| 58 |
+
"es-US": 3,
|
| 59 |
+
"esES": 2,
|
| 60 |
+
"et": 60,
|
| 61 |
+
"et-EE": 60,
|
| 62 |
+
"fa-IR": 38,
|
| 63 |
+
"fi": 26,
|
| 64 |
+
"fi-FI": 26,
|
| 65 |
+
"fr": 8,
|
| 66 |
+
"fr-CA": 100,
|
| 67 |
+
"fr-FR": 8,
|
| 68 |
+
"gn-PY": 82,
|
| 69 |
+
"gu-IN": 42,
|
| 70 |
+
"ha-NG": 50,
|
| 71 |
+
"haw-US": 97,
|
| 72 |
+
"he-IL": 64,
|
| 73 |
+
"hi": 6,
|
| 74 |
+
"hi-HI": 6,
|
| 75 |
+
"hi-IN": 6,
|
| 76 |
+
"hr": 29,
|
| 77 |
+
"hr-HR": 29,
|
| 78 |
+
"hu": 23,
|
| 79 |
+
"hu-HU": 23,
|
| 80 |
+
"hy-AM": 68,
|
| 81 |
+
"id-ID": 34,
|
| 82 |
+
"ig-NG": 53,
|
| 83 |
+
"it": 15,
|
| 84 |
+
"it-IT": 15,
|
| 85 |
+
"ja-JA": 10,
|
| 86 |
+
"ja-JP": 10,
|
| 87 |
+
"ka-GE": 67,
|
| 88 |
+
"km-KH": 47,
|
| 89 |
+
"kn-IN": 43,
|
| 90 |
+
"ko": 14,
|
| 91 |
+
"ko-KO": 14,
|
| 92 |
+
"ko-KR": 14,
|
| 93 |
+
"ku-TR": 65,
|
| 94 |
+
"ky-KG": 71,
|
| 95 |
+
"ln-CD": 58,
|
| 96 |
+
"lt": 31,
|
| 97 |
+
"lt-LT": 31,
|
| 98 |
+
"lv": 61,
|
| 99 |
+
"lv-LV": 61,
|
| 100 |
+
"mi-NZ": 96,
|
| 101 |
+
"ml-IN": 44,
|
| 102 |
+
"mr-IN": 41,
|
| 103 |
+
"ms-MY": 35,
|
| 104 |
+
"mt-MT": 102,
|
| 105 |
+
"nah-MX": 83,
|
| 106 |
+
"nb": 103,
|
| 107 |
+
"nb-NO": 103,
|
| 108 |
+
"ne-NP": 46,
|
| 109 |
+
"nl": 16,
|
| 110 |
+
"nl-NL": 16,
|
| 111 |
+
"nn": 104,
|
| 112 |
+
"nn-NO": 104,
|
| 113 |
+
"no": 27,
|
| 114 |
+
"no-NO": 27,
|
| 115 |
+
"ny-MW": 57,
|
| 116 |
+
"or-KE": 59,
|
| 117 |
+
"pl": 17,
|
| 118 |
+
"pl-PL": 17,
|
| 119 |
+
"pt": 13,
|
| 120 |
+
"pt-BR": 12,
|
| 121 |
+
"pt-PT": 13,
|
| 122 |
+
"qu-PE": 80,
|
| 123 |
+
"ro": 20,
|
| 124 |
+
"ro-RO": 20,
|
| 125 |
+
"ru": 11,
|
| 126 |
+
"ru-RU": 11,
|
| 127 |
+
"rw-RW": 55,
|
| 128 |
+
"si-LK": 45,
|
| 129 |
+
"sk": 28,
|
| 130 |
+
"sk-SK": 28,
|
| 131 |
+
"sl": 62,
|
| 132 |
+
"sl-SI": 62,
|
| 133 |
+
"sm-WS": 98,
|
| 134 |
+
"so-SO": 56,
|
| 135 |
+
"sv": 24,
|
| 136 |
+
"sv-SE": 24,
|
| 137 |
+
"sw-KE": 48,
|
| 138 |
+
"ta-IN": 39,
|
| 139 |
+
"te-IN": 40,
|
| 140 |
+
"tg-TJ": 70,
|
| 141 |
+
"th-TH": 32,
|
| 142 |
+
"to-TO": 99,
|
| 143 |
+
"tr": 18,
|
| 144 |
+
"tr-TR": 18,
|
| 145 |
+
"uk": 19,
|
| 146 |
+
"uk-UA": 19,
|
| 147 |
+
"ur-PK": 37,
|
| 148 |
+
"uz-UZ": 69,
|
| 149 |
+
"vi-VN": 33,
|
| 150 |
+
"yo-NG": 52,
|
| 151 |
+
"zh-CN": 4,
|
| 152 |
+
"zh-TW": 5,
|
| 153 |
+
"zh-ZH": 4,
|
| 154 |
+
"zu-ZA": 51
|
| 155 |
+
},
|
| 156 |
+
"default_prompt_id": 101,
|
| 157 |
+
"lang_tag_token_ids": [
|
| 158 |
+
1,
|
| 159 |
+
72,
|
| 160 |
+
122,
|
| 161 |
+
170,
|
| 162 |
+
230,
|
| 163 |
+
231,
|
| 164 |
+
263,
|
| 165 |
+
282,
|
| 166 |
+
390,
|
| 167 |
+
415,
|
| 168 |
+
430,
|
| 169 |
+
483,
|
| 170 |
+
501,
|
| 171 |
+
519,
|
| 172 |
+
542,
|
| 173 |
+
565,
|
| 174 |
+
594,
|
| 175 |
+
619,
|
| 176 |
+
620,
|
| 177 |
+
629,
|
| 178 |
+
640,
|
| 179 |
+
659,
|
| 180 |
+
660,
|
| 181 |
+
677,
|
| 182 |
+
704,
|
| 183 |
+
717,
|
| 184 |
+
735,
|
| 185 |
+
741,
|
| 186 |
+
790,
|
| 187 |
+
791,
|
| 188 |
+
793,
|
| 189 |
+
794,
|
| 190 |
+
795,
|
| 191 |
+
800,
|
| 192 |
+
819,
|
| 193 |
+
829,
|
| 194 |
+
830,
|
| 195 |
+
844,
|
| 196 |
+
845
|
| 197 |
+
]
|
| 198 |
+
}
|
fr/1120ms/preprocessor.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:12264cf4116cb366a0a4cc606ffba954ad2f9494d25776e76112c294b847ffb7
|
| 3 |
+
size 243
|
fr/1120ms/preprocessor.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:66115dc0006e9aac8b4b2340806691cb4083df62b5757a8e5861b3e30b6065c6
|
| 3 |
+
size 371
|
fr/1120ms/preprocessor.mlmodelc/model.mil
ADDED
|
@@ -0,0 +1,122 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
program(1.3)
|
| 2 |
+
[buildInfo = dict<string, string>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.5.1"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "8.3.0"}})]
|
| 3 |
+
{
|
| 4 |
+
func main<ios18>(tensor<fp32, [1, ?]> audio, tensor<int32, [1]> audio_length) [FlexibleShapeInformation = tuple<tuple<string, dict<string, tensor<int32, [?]>>>, tuple<string, dict<string, list<tensor<int32, [2]>, ?>>>>((("DefaultShapes", {{"audio", [1, 1]}}), ("RangeDims", {{"audio", [[1, 1], [1, 480000]]}})))] {
|
| 5 |
+
int32 var_9 = const()[name = string("op_9"), val = int32(1)];
|
| 6 |
+
int32 var_10 = const()[name = string("op_10"), val = int32(160)];
|
| 7 |
+
int32 var_12 = const()[name = string("op_12"), val = int32(0)];
|
| 8 |
+
int32 var_33 = const()[name = string("op_33"), val = int32(512)];
|
| 9 |
+
tensor<int32, [1]> var_34 = add(x = audio_length, y = var_33)[name = string("op_34")];
|
| 10 |
+
int32 var_35 = const()[name = string("op_35"), val = int32(512)];
|
| 11 |
+
tensor<int32, [1]> var_36 = sub(x = var_34, y = var_35)[name = string("op_36")];
|
| 12 |
+
tensor<int32, [1]> floor_div_0 = floor_div(x = var_36, y = var_10)[name = string("floor_div_0")];
|
| 13 |
+
tensor<bool, [1]> var_39 = equal(x = audio_length, y = var_12)[name = string("op_39")];
|
| 14 |
+
tensor<int32, [1]> var_40 = const()[name = string("op_40"), val = tensor<int32, [1]>([0])];
|
| 15 |
+
tensor<int32, [1]> mel_length = select(a = var_40, b = floor_div_0, cond = var_39)[name = string("seq_len")];
|
| 16 |
+
string audio_to_fp16_dtype_0 = const()[name = string("audio_to_fp16_dtype_0"), val = string("fp16")];
|
| 17 |
+
tensor<fp16, [1, ?]> audio_to_fp16 = cast(dtype = audio_to_fp16_dtype_0, x = audio)[name = string("cast_14")];
|
| 18 |
+
tensor<int32, [2]> var_42_shape_cast_fp16 = shape(x = audio_to_fp16)[name = string("op_42_shape_cast_fp16")];
|
| 19 |
+
int32 gather_0_axis_0 = const()[name = string("gather_0_axis_0"), val = int32(0)];
|
| 20 |
+
int32 gather_0_batch_dims_0 = const()[name = string("gather_0_batch_dims_0"), val = int32(0)];
|
| 21 |
+
bool gather_0_validate_indices_0 = const()[name = string("gather_0_validate_indices_0"), val = bool(false)];
|
| 22 |
+
string var_42_shape_cast_fp16_to_int16_dtype_0 = const()[name = string("op_42_shape_cast_fp16_to_int16_dtype_0"), val = string("int16")];
|
| 23 |
+
uint16 select_0_to_uint16 = const()[name = string("select_0_to_uint16"), val = uint16(1)];
|
| 24 |
+
tensor<int16, [2]> var_42_shape_cast_fp16_to_int16 = cast(dtype = var_42_shape_cast_fp16_to_int16_dtype_0, x = var_42_shape_cast_fp16)[name = string("cast_13")];
|
| 25 |
+
int16 gather_0_cast_uint16 = gather(axis = gather_0_axis_0, batch_dims = gather_0_batch_dims_0, indices = select_0_to_uint16, validate_indices = gather_0_validate_indices_0, x = var_42_shape_cast_fp16_to_int16)[name = string("gather_0_cast_uint16")];
|
| 26 |
+
string gather_0_cast_uint16_to_int32_dtype_0 = const()[name = string("gather_0_cast_uint16_to_int32_dtype_0"), val = string("int32")];
|
| 27 |
+
int32 const_0 = const()[name = string("const_0"), val = int32(0)];
|
| 28 |
+
int32 const_1 = const()[name = string("const_1"), val = int32(1)];
|
| 29 |
+
int32 gather_0_cast_uint16_to_int32 = cast(dtype = gather_0_cast_uint16_to_int32_dtype_0, x = gather_0_cast_uint16)[name = string("cast_12")];
|
| 30 |
+
tensor<int32, [?]> var_43 = range_1d(end = gather_0_cast_uint16_to_int32, start = const_0, step = const_1)[name = string("op_43")];
|
| 31 |
+
tensor<int32, [1]> var_44_axes_0 = const()[name = string("op_44_axes_0"), val = tensor<int32, [1]>([0])];
|
| 32 |
+
tensor<int32, [1, ?]> var_44 = expand_dims(axes = var_44_axes_0, x = var_43)[name = string("op_44")];
|
| 33 |
+
tensor<int32, [1]> var_45_axes_0 = const()[name = string("op_45_axes_0"), val = tensor<int32, [1]>([1])];
|
| 34 |
+
tensor<int32, [1, 1]> var_45 = expand_dims(axes = var_45_axes_0, x = audio_length)[name = string("op_45")];
|
| 35 |
+
tensor<bool, [1, ?]> timemask = less(x = var_44, y = var_45)[name = string("timemask")];
|
| 36 |
+
tensor<int32, [2]> var_48_begin_0 = const()[name = string("op_48_begin_0"), val = tensor<int32, [2]>([0, 0])];
|
| 37 |
+
tensor<int32, [2]> var_48_end_0 = const()[name = string("op_48_end_0"), val = tensor<int32, [2]>([1, 1])];
|
| 38 |
+
tensor<bool, [2]> var_48_end_mask_0 = const()[name = string("op_48_end_mask_0"), val = tensor<bool, [2]>([true, false])];
|
| 39 |
+
tensor<bool, [2]> var_48_squeeze_mask_0 = const()[name = string("op_48_squeeze_mask_0"), val = tensor<bool, [2]>([false, true])];
|
| 40 |
+
tensor<fp16, [1]> var_48_cast_fp16 = slice_by_index(begin = var_48_begin_0, end = var_48_end_0, end_mask = var_48_end_mask_0, squeeze_mask = var_48_squeeze_mask_0, x = audio_to_fp16)[name = string("op_48_cast_fp16")];
|
| 41 |
+
tensor<int32, [1]> var_49_axes_0 = const()[name = string("op_49_axes_0"), val = tensor<int32, [1]>([1])];
|
| 42 |
+
tensor<fp16, [1, 1]> var_49_cast_fp16 = expand_dims(axes = var_49_axes_0, x = var_48_cast_fp16)[name = string("op_49_cast_fp16")];
|
| 43 |
+
tensor<int32, [2]> var_51_begin_0 = const()[name = string("op_51_begin_0"), val = tensor<int32, [2]>([0, 1])];
|
| 44 |
+
tensor<int32, [2]> var_51_end_0 = const()[name = string("op_51_end_0"), val = tensor<int32, [2]>([1, 0])];
|
| 45 |
+
tensor<bool, [2]> var_51_end_mask_0 = const()[name = string("op_51_end_mask_0"), val = tensor<bool, [2]>([true, true])];
|
| 46 |
+
tensor<fp16, [1, ?]> var_51_cast_fp16 = slice_by_index(begin = var_51_begin_0, end = var_51_end_0, end_mask = var_51_end_mask_0, x = audio_to_fp16)[name = string("op_51_cast_fp16")];
|
| 47 |
+
tensor<int32, [2]> var_53_begin_0 = const()[name = string("op_53_begin_0"), val = tensor<int32, [2]>([0, 0])];
|
| 48 |
+
tensor<int32, [2]> var_53_end_0 = const()[name = string("op_53_end_0"), val = tensor<int32, [2]>([1, -1])];
|
| 49 |
+
tensor<bool, [2]> var_53_end_mask_0 = const()[name = string("op_53_end_mask_0"), val = tensor<bool, [2]>([true, false])];
|
| 50 |
+
tensor<fp16, [1, ?]> var_53_cast_fp16 = slice_by_index(begin = var_53_begin_0, end = var_53_end_0, end_mask = var_53_end_mask_0, x = audio_to_fp16)[name = string("op_53_cast_fp16")];
|
| 51 |
+
fp16 var_54_to_fp16 = const()[name = string("op_54_to_fp16"), val = fp16(0x1.f0cp-1)];
|
| 52 |
+
tensor<fp16, [1, ?]> var_55_cast_fp16 = mul(x = var_53_cast_fp16, y = var_54_to_fp16)[name = string("op_55_cast_fp16")];
|
| 53 |
+
tensor<fp16, [1, ?]> var_56_cast_fp16 = sub(x = var_51_cast_fp16, y = var_55_cast_fp16)[name = string("op_56_cast_fp16")];
|
| 54 |
+
bool x_3_interleave_0 = const()[name = string("x_3_interleave_0"), val = bool(false)];
|
| 55 |
+
tensor<fp16, [1, ?]> x_3_cast_fp16 = concat(axis = var_9, interleave = x_3_interleave_0, values = (var_49_cast_fp16, var_56_cast_fp16))[name = string("x_3_cast_fp16")];
|
| 56 |
+
tensor<bool, [1, ?]> var_59 = logical_not(x = timemask)[name = string("op_59")];
|
| 57 |
+
fp16 var_16_to_fp16 = const()[name = string("op_16_to_fp16"), val = fp16(0x0p+0)];
|
| 58 |
+
tensor<fp16, [1, ?]> input_1_cast_fp16 = select(a = var_16_to_fp16, b = x_3_cast_fp16, cond = var_59)[name = string("input_1_cast_fp16")];
|
| 59 |
+
tensor<int32, [3]> concat_1x = const()[name = string("concat_1x"), val = tensor<int32, [3]>([1, 1, -1])];
|
| 60 |
+
tensor<fp16, [1, 1, ?]> input_3_cast_fp16 = reshape(shape = concat_1x, x = input_1_cast_fp16)[name = string("input_3_cast_fp16")];
|
| 61 |
+
tensor<int32, [6]> input_5_pad_0 = const()[name = string("input_5_pad_0"), val = tensor<int32, [6]>([0, 0, 0, 0, 256, 256])];
|
| 62 |
+
string input_5_mode_0 = const()[name = string("input_5_mode_0"), val = string("constant")];
|
| 63 |
+
fp16 const_3_to_fp16 = const()[name = string("const_3_to_fp16"), val = fp16(0x0p+0)];
|
| 64 |
+
tensor<fp16, [1, 1, ?]> input_5_cast_fp16 = pad(constant_val = const_3_to_fp16, mode = input_5_mode_0, pad = input_5_pad_0, x = input_3_cast_fp16)[name = string("input_5_cast_fp16")];
|
| 65 |
+
tensor<int32, [2]> concat_2x = const()[name = string("concat_2x"), val = tensor<int32, [2]>([1, -1])];
|
| 66 |
+
tensor<fp16, [1, ?]> input_cast_fp16 = reshape(shape = concat_2x, x = input_5_cast_fp16)[name = string("input_cast_fp16")];
|
| 67 |
+
tensor<int32, [1]> expand_dims_3 = const()[name = string("expand_dims_3"), val = tensor<int32, [1]>([160])];
|
| 68 |
+
tensor<int32, [1]> expand_dims_4_axes_0 = const()[name = string("expand_dims_4_axes_0"), val = tensor<int32, [1]>([1])];
|
| 69 |
+
tensor<fp16, [1, 1, ?]> expand_dims_4_cast_fp16 = expand_dims(axes = expand_dims_4_axes_0, x = input_cast_fp16)[name = string("expand_dims_4_cast_fp16")];
|
| 70 |
+
string conv_0_pad_type_0 = const()[name = string("conv_0_pad_type_0"), val = string("valid")];
|
| 71 |
+
tensor<int32, [2]> conv_0_pad_0 = const()[name = string("conv_0_pad_0"), val = tensor<int32, [2]>([0, 0])];
|
| 72 |
+
tensor<int32, [1]> conv_0_dilations_0 = const()[name = string("conv_0_dilations_0"), val = tensor<int32, [1]>([1])];
|
| 73 |
+
int32 conv_0_groups_0 = const()[name = string("conv_0_groups_0"), val = int32(1)];
|
| 74 |
+
tensor<fp16, [257, 1, 512]> expand_dims_1_to_fp16 = const()[name = string("expand_dims_1_to_fp16"), val = tensor<fp16, [257, 1, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
|
| 75 |
+
tensor<fp16, [1, 257, ?]> conv_0_cast_fp16 = conv(dilations = conv_0_dilations_0, groups = conv_0_groups_0, pad = conv_0_pad_0, pad_type = conv_0_pad_type_0, strides = expand_dims_3, weight = expand_dims_1_to_fp16, x = expand_dims_4_cast_fp16)[name = string("conv_0_cast_fp16")];
|
| 76 |
+
string conv_1_pad_type_0 = const()[name = string("conv_1_pad_type_0"), val = string("valid")];
|
| 77 |
+
tensor<int32, [2]> conv_1_pad_0 = const()[name = string("conv_1_pad_0"), val = tensor<int32, [2]>([0, 0])];
|
| 78 |
+
tensor<int32, [1]> conv_1_dilations_0 = const()[name = string("conv_1_dilations_0"), val = tensor<int32, [1]>([1])];
|
| 79 |
+
int32 conv_1_groups_0 = const()[name = string("conv_1_groups_0"), val = int32(1)];
|
| 80 |
+
tensor<fp16, [257, 1, 512]> expand_dims_2_to_fp16 = const()[name = string("expand_dims_2_to_fp16"), val = tensor<fp16, [257, 1, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(263296)))];
|
| 81 |
+
tensor<fp16, [1, 257, ?]> conv_1_cast_fp16 = conv(dilations = conv_1_dilations_0, groups = conv_1_groups_0, pad = conv_1_pad_0, pad_type = conv_1_pad_type_0, strides = expand_dims_3, weight = expand_dims_2_to_fp16, x = expand_dims_4_cast_fp16)[name = string("conv_1_cast_fp16")];
|
| 82 |
+
int32 stack_0_axis_0 = const()[name = string("stack_0_axis_0"), val = int32(-1)];
|
| 83 |
+
tensor<fp16, [1, 257, ?, 2]> stack_0_cast_fp16 = stack(axis = stack_0_axis_0, values = (conv_0_cast_fp16, conv_1_cast_fp16))[name = string("stack_0_cast_fp16")];
|
| 84 |
+
fp16 var_19_promoted_to_fp16 = const()[name = string("op_19_promoted_to_fp16"), val = fp16(0x1p+1)];
|
| 85 |
+
tensor<fp16, [1, 257, ?, 2]> var_74_cast_fp16 = pow(x = stack_0_cast_fp16, y = var_19_promoted_to_fp16)[name = string("op_74_cast_fp16")];
|
| 86 |
+
tensor<int32, [1]> var_76_axes_0 = const()[name = string("op_76_axes_0"), val = tensor<int32, [1]>([-1])];
|
| 87 |
+
bool var_76_keep_dims_0 = const()[name = string("op_76_keep_dims_0"), val = bool(false)];
|
| 88 |
+
tensor<fp16, [1, 257, ?]> var_76_cast_fp16 = reduce_sum(axes = var_76_axes_0, keep_dims = var_76_keep_dims_0, x = var_74_cast_fp16)[name = string("op_76_cast_fp16")];
|
| 89 |
+
tensor<fp16, [1, 257, ?]> x_11_cast_fp16 = identity(x = var_76_cast_fp16)[name = string("x_11_cast_fp16")];
|
| 90 |
+
bool x_13_transpose_x_0 = const()[name = string("x_13_transpose_x_0"), val = bool(false)];
|
| 91 |
+
bool x_13_transpose_y_0 = const()[name = string("x_13_transpose_y_0"), val = bool(false)];
|
| 92 |
+
tensor<fp16, [1, 128, 257]> const_4_to_fp16 = const()[name = string("const_4_to_fp16"), val = tensor<fp16, [1, 128, 257]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(526528)))];
|
| 93 |
+
tensor<fp16, [1, 128, ?]> x_13_cast_fp16 = matmul(transpose_x = x_13_transpose_x_0, transpose_y = x_13_transpose_y_0, x = const_4_to_fp16, y = x_11_cast_fp16)[name = string("x_13_cast_fp16")];
|
| 94 |
+
fp16 var_83_to_fp16 = const()[name = string("op_83_to_fp16"), val = fp16(0x1p-24)];
|
| 95 |
+
tensor<fp16, [1, 128, ?]> var_84_cast_fp16 = add(x = x_13_cast_fp16, y = var_83_to_fp16)[name = string("op_84_cast_fp16")];
|
| 96 |
+
fp32 x_epsilon_0 = const()[name = string("x_epsilon_0"), val = fp32(0x1p-149)];
|
| 97 |
+
tensor<fp16, [1, 128, ?]> x_cast_fp16 = log(epsilon = x_epsilon_0, x = var_84_cast_fp16)[name = string("x_cast_fp16")];
|
| 98 |
+
tensor<int32, [3]> var_86_shape_cast_fp16 = shape(x = x_cast_fp16)[name = string("op_86_shape_cast_fp16")];
|
| 99 |
+
int32 gather_5_axis_0 = const()[name = string("gather_5_axis_0"), val = int32(0)];
|
| 100 |
+
int32 gather_5_batch_dims_0 = const()[name = string("gather_5_batch_dims_0"), val = int32(0)];
|
| 101 |
+
bool gather_5_validate_indices_0 = const()[name = string("gather_5_validate_indices_0"), val = bool(false)];
|
| 102 |
+
string var_86_shape_cast_fp16_to_uint16_dtype_0 = const()[name = string("op_86_shape_cast_fp16_to_uint16_dtype_0"), val = string("uint16")];
|
| 103 |
+
uint16 select_5_to_uint16 = const()[name = string("select_5_to_uint16"), val = uint16(2)];
|
| 104 |
+
tensor<uint16, [3]> var_86_shape_cast_fp16_to_uint16 = cast(dtype = var_86_shape_cast_fp16_to_uint16_dtype_0, x = var_86_shape_cast_fp16)[name = string("cast_11")];
|
| 105 |
+
uint16 gather_5_cast_uint16 = gather(axis = gather_5_axis_0, batch_dims = gather_5_batch_dims_0, indices = select_5_to_uint16, validate_indices = gather_5_validate_indices_0, x = var_86_shape_cast_fp16_to_uint16)[name = string("gather_5_cast_uint16")];
|
| 106 |
+
string gather_5_cast_uint16_to_int32_dtype_0 = const()[name = string("gather_5_cast_uint16_to_int32_dtype_0"), val = string("int32")];
|
| 107 |
+
int32 const_5 = const()[name = string("const_5"), val = int32(0)];
|
| 108 |
+
int32 const_6 = const()[name = string("const_6"), val = int32(1)];
|
| 109 |
+
int32 gather_5_cast_uint16_to_int32 = cast(dtype = gather_5_cast_uint16_to_int32_dtype_0, x = gather_5_cast_uint16)[name = string("cast_10")];
|
| 110 |
+
tensor<int32, [?]> mask_1 = range_1d(end = gather_5_cast_uint16_to_int32, start = const_5, step = const_6)[name = string("mask_1")];
|
| 111 |
+
tensor<int32, [1]> expand_dims_0_axes_0 = const()[name = string("expand_dims_0_axes_0"), val = tensor<int32, [1]>([0])];
|
| 112 |
+
tensor<int32, [1, ?]> expand_dims_0 = expand_dims(axes = expand_dims_0_axes_0, x = mask_1)[name = string("expand_dims_0")];
|
| 113 |
+
tensor<int32, [1]> var_91_axes_0 = const()[name = string("op_91_axes_0"), val = tensor<int32, [1]>([1])];
|
| 114 |
+
tensor<int32, [1, 1]> var_91 = expand_dims(axes = var_91_axes_0, x = mel_length)[name = string("op_91")];
|
| 115 |
+
tensor<bool, [1, ?]> mask = greater_equal(x = expand_dims_0, y = var_91)[name = string("mask")];
|
| 116 |
+
tensor<int32, [1]> var_93_axes_0 = const()[name = string("op_93_axes_0"), val = tensor<int32, [1]>([1])];
|
| 117 |
+
tensor<bool, [1, 1, ?]> var_93 = expand_dims(axes = var_93_axes_0, x = mask)[name = string("op_93")];
|
| 118 |
+
tensor<fp16, [1, 128, ?]> processed_signal_cast_fp16 = select(a = var_16_to_fp16, b = x_cast_fp16, cond = var_93)[name = string("processed_signal_cast_fp16")];
|
| 119 |
+
string processed_signal_cast_fp16_to_fp32_dtype_0 = const()[name = string("processed_signal_cast_fp16_to_fp32_dtype_0"), val = string("fp32")];
|
| 120 |
+
tensor<fp32, [1, 128, ?]> mel = cast(dtype = processed_signal_cast_fp16_to_fp32_dtype_0, x = processed_signal_cast_fp16)[name = string("cast_9")];
|
| 121 |
+
} -> (mel, mel_length);
|
| 122 |
+
}
|
fr/1120ms/preprocessor.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:297514e2b211d14b0e53cb97193d679bb89ead98d28e578f3f1d049ddbcc36b3
|
| 3 |
+
size 592384
|
fr/1120ms/preprocessor.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:daad3e5dc1a13a1dc66997f9c1c1280b68646d7ccd7b6359fa8c361d63b5c09b
|
| 3 |
+
size 15878
|
fr/1120ms/preprocessor.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:297514e2b211d14b0e53cb97193d679bb89ead98d28e578f3f1d049ddbcc36b3
|
| 3 |
+
size 592384
|
fr/1120ms/preprocessor.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"11EFF503-1D75-4BC3-9480-44E97E36229D": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"A4EAB1E5-DDC2-4445-9D00-C29BCDE18F72": {
|
| 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": "A4EAB1E5-DDC2-4445-9D00-C29BCDE18F72"
|
| 18 |
+
}
|