Upload 352 files
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- it/1120ms/decoder.mlmodelc/analytics/coremldata.bin +3 -0
- it/1120ms/decoder.mlmodelc/coremldata.bin +3 -0
- it/1120ms/decoder.mlmodelc/model.mil +64 -0
- it/1120ms/decoder.mlmodelc/weights/weight.bin +3 -0
- it/1120ms/decoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- it/1120ms/decoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- it/1120ms/decoder.mlpackage/Manifest.json +18 -0
- it/1120ms/decoder_joint.mlmodelc/analytics/coremldata.bin +3 -0
- it/1120ms/decoder_joint.mlmodelc/coremldata.bin +3 -0
- it/1120ms/decoder_joint.mlmodelc/model.mil +83 -0
- it/1120ms/decoder_joint.mlmodelc/weights/weight.bin +3 -0
- it/1120ms/decoder_joint.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- it/1120ms/decoder_joint.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- it/1120ms/decoder_joint.mlpackage/Manifest.json +18 -0
- it/1120ms/decoder_joint_noencproj.mlmodelc/analytics/coremldata.bin +3 -0
- it/1120ms/decoder_joint_noencproj.mlmodelc/coremldata.bin +3 -0
- it/1120ms/decoder_joint_noencproj.mlmodelc/model.mil +91 -0
- it/1120ms/decoder_joint_noencproj.mlmodelc/weights/weight.bin +3 -0
- it/1120ms/decoder_joint_noencproj.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- it/1120ms/decoder_joint_noencproj.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- it/1120ms/decoder_joint_noencproj.mlpackage/Manifest.json +18 -0
- it/1120ms/encoder.mlmodelc/analytics/coremldata.bin +3 -0
- it/1120ms/encoder.mlmodelc/coremldata.bin +3 -0
- it/1120ms/encoder.mlmodelc/model.mil +0 -0
- it/1120ms/encoder.mlmodelc/weights/weight.bin +3 -0
- it/1120ms/encoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- it/1120ms/encoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- it/1120ms/encoder.mlpackage/Manifest.json +18 -0
- it/1120ms/joint.mlmodelc/analytics/coremldata.bin +3 -0
- it/1120ms/joint.mlmodelc/coremldata.bin +3 -0
- it/1120ms/joint.mlmodelc/model.mil +31 -0
- it/1120ms/joint.mlmodelc/weights/weight.bin +3 -0
- it/1120ms/joint.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- it/1120ms/joint.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- it/1120ms/joint.mlpackage/Manifest.json +18 -0
- it/1120ms/joint_noencproj_batched.mlmodelc/analytics/coremldata.bin +3 -0
- it/1120ms/joint_noencproj_batched.mlmodelc/coremldata.bin +3 -0
- it/1120ms/joint_noencproj_batched.mlmodelc/model.mil +26 -0
- it/1120ms/joint_noencproj_batched.mlmodelc/weights/weight.bin +3 -0
- it/1120ms/joint_noencproj_batched.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- it/1120ms/joint_noencproj_batched.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- it/1120ms/joint_noencproj_batched.mlpackage/Manifest.json +18 -0
- it/1120ms/metadata.json +198 -0
- it/1120ms/preprocessor.mlmodelc/analytics/coremldata.bin +3 -0
- it/1120ms/preprocessor.mlmodelc/coremldata.bin +3 -0
- it/1120ms/preprocessor.mlmodelc/model.mil +122 -0
- it/1120ms/preprocessor.mlmodelc/weights/weight.bin +3 -0
- it/1120ms/preprocessor.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
- it/1120ms/preprocessor.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
- it/1120ms/preprocessor.mlpackage/Manifest.json +18 -0
it/1120ms/decoder.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4cdca6bf678463f31354072f526088e5bdf5115ae94c04e387bb35b2c7a607d6
|
| 3 |
+
size 243
|
it/1120ms/decoder.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3c993f8b96ce22027cd2ed42d99b7e61f93a01197bb17cadada8eb989e946dec
|
| 3 |
+
size 433
|
it/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, [806, 640]> module_prediction_embed_weight_to_fp16 = const()[name = string("module_prediction_embed_weight_to_fp16"), val = tensor<fp16, [806, 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(1031808)))];
|
| 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(4308672)))];
|
| 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(7585536)))];
|
| 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(7590720)))];
|
| 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(10867584)))];
|
| 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(14144448)))];
|
| 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 |
+
}
|
it/1120ms/decoder.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1360d93c68c3e9c54bda4adaec860753949f3b0dc93bc98f4edc9d6f8dd5595c
|
| 3 |
+
size 14149632
|
it/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:73bb3afa62698bc822b6d32b3731d0bc40521e03737e3139e10a768542fca1fe
|
| 3 |
+
size 10359
|
it/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:1360d93c68c3e9c54bda4adaec860753949f3b0dc93bc98f4edc9d6f8dd5595c
|
| 3 |
+
size 14149632
|
it/1120ms/decoder.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"7CBCED8D-FA6A-45B0-BF60-30DB0A653074": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"AFD197FC-BECC-451A-961C-C0CA05D58065": {
|
| 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": "AFD197FC-BECC-451A-961C-C0CA05D58065"
|
| 18 |
+
}
|
it/1120ms/decoder_joint.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:12f4bcf5114baa2b3a37b8ebeab6c519109bd857e50ec345c458b7a6c4deb20e
|
| 3 |
+
size 243
|
it/1120ms/decoder_joint.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:53754e2eaa7e0f7435220b47b621c5f3d8c5f2da83edd46efa5950fa723ef1d9
|
| 3 |
+
size 454
|
it/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, [806, 640]> decoder_module_prediction_embed_weight_to_fp16 = const()[name = string("decoder_module_prediction_embed_weight_to_fp16"), val = tensor<fp16, [806, 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(1031808)))];
|
| 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(4308672)))];
|
| 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(7585536)))];
|
| 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(7590720)))];
|
| 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(10867584)))];
|
| 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(14144448)))];
|
| 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(14149632)))];
|
| 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(15460416)))];
|
| 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(15461760)))];
|
| 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(16281024)))];
|
| 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, [806, 640]> joint_module_joint_net_2_weight_to_fp16 = const()[name = string("joint_module_joint_net_2_weight_to_fp16"), val = tensor<fp16, [806, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(16282368)))];
|
| 75 |
+
tensor<fp16, [806]> joint_module_joint_net_2_bias_to_fp16 = const()[name = string("joint_module_joint_net_2_bias_to_fp16"), val = tensor<fp16, [806]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(17314112)))];
|
| 76 |
+
tensor<fp16, [1, 1, 1, 806]> 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, 806]> 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 |
+
}
|
it/1120ms/decoder_joint.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e507a69196a04e30adfafc302b6a5f5f527e45c1965c65dd81d63a621cae2064
|
| 3 |
+
size 17315788
|
it/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:5fa6a6d89c1c07ae16f162f5a3b6809b12aafe57a663f5cdb270be3dec7b1427
|
| 3 |
+
size 13745
|
it/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:e507a69196a04e30adfafc302b6a5f5f527e45c1965c65dd81d63a621cae2064
|
| 3 |
+
size 17315788
|
it/1120ms/decoder_joint.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"2B19A50C-1D16-4D97-BE3C-D9BCF35884CF": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Specification",
|
| 7 |
+
"name": "model.mlmodel",
|
| 8 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 9 |
+
},
|
| 10 |
+
"9CA734BC-CFD2-4F39-B068-BE69ABCAAD1F": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Weights",
|
| 13 |
+
"name": "weights",
|
| 14 |
+
"path": "com.apple.CoreML/weights"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "2B19A50C-1D16-4D97-BE3C-D9BCF35884CF"
|
| 18 |
+
}
|
it/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:40ec657603479e7dbf8cdb3d6368349eb8b766a52439a26a735d1fadf1b4281d
|
| 3 |
+
size 243
|
it/1120ms/decoder_joint_noencproj.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c682ba0e028fb7ab6557f8ac1006febc8ec8dd81e4ef8d3a2c05d876e2dbcc8e
|
| 3 |
+
size 519
|
it/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, [806, 640]> decoder_module_prediction_embed_weight_to_fp16 = const()[name = string("decoder_module_prediction_embed_weight_to_fp16"), val = tensor<fp16, [806, 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(806)];
|
| 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(1031808)))];
|
| 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(4308672)))];
|
| 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(7585536)))];
|
| 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(7590720)))];
|
| 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(10867584)))];
|
| 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(14144448)))];
|
| 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(14149632)))];
|
| 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(14968896)))];
|
| 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, [806, 640]> joint_module_joint_net_2_weight_to_fp16 = const()[name = string("joint_module_joint_net_2_weight_to_fp16"), val = tensor<fp16, [806, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14970240)))];
|
| 79 |
+
tensor<fp16, [806]> joint_module_joint_net_2_bias_to_fp16 = const()[name = string("joint_module_joint_net_2_bias_to_fp16"), val = tensor<fp16, [806]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(16001984)))];
|
| 80 |
+
tensor<fp16, [1, 1, 1, 806]> 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, 806]> 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, 806]> 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, 806]> 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 |
+
}
|
it/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:5ca6a467ebf44612a8032c6c4ddf323e35a9ffaa15c01822528fb97144ec9439
|
| 3 |
+
size 16003660
|
it/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:1a46ef6436e8c1e5af6f379658d871f2a40dff3a13c946d4f8df736b19cdc45e
|
| 3 |
+
size 14630
|
it/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:5ca6a467ebf44612a8032c6c4ddf323e35a9ffaa15c01822528fb97144ec9439
|
| 3 |
+
size 16003660
|
it/1120ms/decoder_joint_noencproj.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"E92AE661-EE59-447C-B9BE-52D50BC4FC37": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"F7138F3F-8633-4C54-A3F4-D0F6E8220E39": {
|
| 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": "F7138F3F-8633-4C54-A3F4-D0F6E8220E39"
|
| 18 |
+
}
|
it/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
|
it/1120ms/encoder.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0f76d62779c43fa5272f8977cfc6d3e368abd5cc38841834d05834021633dd2f
|
| 3 |
+
size 662
|
it/1120ms/encoder.mlmodelc/model.mil
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
it/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
|
it/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
|
it/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
|
it/1120ms/encoder.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"2A6DB86E-A930-4F02-B70E-CA03D7F3AD84": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Specification",
|
| 7 |
+
"name": "model.mlmodel",
|
| 8 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 9 |
+
},
|
| 10 |
+
"712F619E-4FA1-4165-A290-6C1344B0A816": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Weights",
|
| 13 |
+
"name": "weights",
|
| 14 |
+
"path": "com.apple.CoreML/weights"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "2A6DB86E-A930-4F02-B70E-CA03D7F3AD84"
|
| 18 |
+
}
|
it/1120ms/joint.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cae47ef3b8cb372625305f4cdd31159b5ce56be1470c94613ff946be342f6d7a
|
| 3 |
+
size 243
|
it/1120ms/joint.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:215d8dd2e33da0c37f08e6b0c0a1e997a3e056f2cb9113fdcf17f8027a61216d
|
| 3 |
+
size 341
|
it/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, [806, 640]> module_joint_net_2_weight_to_fp16 = const()[name = string("module_joint_net_2_weight_to_fp16"), val = tensor<fp16, [806, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2132800)))];
|
| 26 |
+
tensor<fp16, [806]> module_joint_net_2_bias_to_fp16 = const()[name = string("module_joint_net_2_bias_to_fp16"), val = tensor<fp16, [806]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(3164544)))];
|
| 27 |
+
tensor<fp16, [1, 1, 1, 806]> 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, 806]> logits = cast(dtype = linear_2_cast_fp16_to_fp32_dtype_0, x = linear_2_cast_fp16)[name = string("cast_0")];
|
| 30 |
+
} -> (logits);
|
| 31 |
+
}
|
it/1120ms/joint.mlmodelc/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1740c99cbe34ebeaa0163c8421135b4586df09960ef07fe02abb2a94b5693411
|
| 3 |
+
size 3166220
|
it/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:d902ae83b8a93f0f4240a4a6939466dbd1a6b2291f1615d81d7ac26d9115bc23
|
| 3 |
+
size 4486
|
it/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:1740c99cbe34ebeaa0163c8421135b4586df09960ef07fe02abb2a94b5693411
|
| 3 |
+
size 3166220
|
it/1120ms/joint.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"6D4EAD3B-A17D-4807-8572-3E0A513C0C7E": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Specification",
|
| 7 |
+
"name": "model.mlmodel",
|
| 8 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 9 |
+
},
|
| 10 |
+
"8242AF76-3606-4A98-8B3D-E0BB730201A4": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Weights",
|
| 13 |
+
"name": "weights",
|
| 14 |
+
"path": "com.apple.CoreML/weights"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "6D4EAD3B-A17D-4807-8572-3E0A513C0C7E"
|
| 18 |
+
}
|
it/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:24510a0ca90eea355a0e446ba54d63004a3f2c51d30eb3e415e4af938ac1acba
|
| 3 |
+
size 243
|
it/1120ms/joint_noencproj_batched.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dc93bb8f241754f1359837f597d600b029d687ce0c57a4280fa586f8c8386337
|
| 3 |
+
size 406
|
it/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, [806, 640]> joint_module_joint_net_2_weight_to_fp16 = const()[name = string("joint_module_joint_net_2_weight_to_fp16"), val = tensor<fp16, [806, 640]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(820672)))];
|
| 21 |
+
tensor<fp16, [806]> joint_module_joint_net_2_bias_to_fp16 = const()[name = string("joint_module_joint_net_2_bias_to_fp16"), val = tensor<fp16, [806]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1852416)))];
|
| 22 |
+
tensor<fp16, [1, 4, 1, 806]> 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, 806]> logits = cast(dtype = linear_1_cast_fp16_to_fp32_dtype_0, x = linear_1_cast_fp16)[name = string("cast_0")];
|
| 25 |
+
} -> (logits);
|
| 26 |
+
}
|
it/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:cd83e82dcfec315f28c8a8872b0d7f22e668a2c485821de86a0379ae2b3864ad
|
| 3 |
+
size 1854092
|
it/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:26a19f2e269811070e57eda978e8226ca9dcca6177b46b95f916538b5f815716
|
| 3 |
+
size 3839
|
it/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:cd83e82dcfec315f28c8a8872b0d7f22e668a2c485821de86a0379ae2b3864ad
|
| 3 |
+
size 1854092
|
it/1120ms/joint_noencproj_batched.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"4233CE8E-FB95-4FF9-BCD8-2A834D55C580": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"96E0F26C-90DC-49EE-B510-D0FB3FC812CC": {
|
| 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": "96E0F26C-90DC-49EE-B510-D0FB3FC812CC"
|
| 18 |
+
}
|
it/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": 805,
|
| 14 |
+
"blank_idx": 805,
|
| 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 |
+
63,
|
| 160 |
+
115,
|
| 161 |
+
167,
|
| 162 |
+
226,
|
| 163 |
+
227,
|
| 164 |
+
259,
|
| 165 |
+
276,
|
| 166 |
+
328,
|
| 167 |
+
353,
|
| 168 |
+
368,
|
| 169 |
+
462,
|
| 170 |
+
481,
|
| 171 |
+
499,
|
| 172 |
+
518,
|
| 173 |
+
542,
|
| 174 |
+
571,
|
| 175 |
+
602,
|
| 176 |
+
603,
|
| 177 |
+
612,
|
| 178 |
+
624,
|
| 179 |
+
646,
|
| 180 |
+
647,
|
| 181 |
+
667,
|
| 182 |
+
689,
|
| 183 |
+
699,
|
| 184 |
+
720,
|
| 185 |
+
727,
|
| 186 |
+
747,
|
| 187 |
+
748,
|
| 188 |
+
750,
|
| 189 |
+
751,
|
| 190 |
+
752,
|
| 191 |
+
756,
|
| 192 |
+
774,
|
| 193 |
+
787,
|
| 194 |
+
788,
|
| 195 |
+
801,
|
| 196 |
+
802
|
| 197 |
+
]
|
| 198 |
+
}
|
it/1120ms/preprocessor.mlmodelc/analytics/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:97be07beee7a6a28b7db85d47087d5b018ebcd1fc0b1565707141d574244bdc9
|
| 3 |
+
size 243
|
it/1120ms/preprocessor.mlmodelc/coremldata.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f7b11e08aba46d1845d8ad3f247717e0f6fae35b21d71d52e44a69ea73587bfe
|
| 3 |
+
size 371
|
it/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 |
+
}
|
it/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
|
it/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:f1df1cad6f43ce4fa090e0f9a33bb5ddf25a0aaeca2be136339878b5b9de45c0
|
| 3 |
+
size 15878
|
it/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
|
it/1120ms/preprocessor.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"162F4F34-AB45-4D09-8F9D-2F76AAE79E0F": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"B5F24360-A5E1-4F41-9872-3D77094622F0": {
|
| 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": "B5F24360-A5E1-4F41-9872-3D77094622F0"
|
| 18 |
+
}
|