alexwengg commited on
Commit
e14200d
·
verified ·
1 Parent(s): f3f2c19

Upload 352 files

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. it/1120ms/decoder.mlmodelc/analytics/coremldata.bin +3 -0
  2. it/1120ms/decoder.mlmodelc/coremldata.bin +3 -0
  3. it/1120ms/decoder.mlmodelc/model.mil +64 -0
  4. it/1120ms/decoder.mlmodelc/weights/weight.bin +3 -0
  5. it/1120ms/decoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  6. it/1120ms/decoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  7. it/1120ms/decoder.mlpackage/Manifest.json +18 -0
  8. it/1120ms/decoder_joint.mlmodelc/analytics/coremldata.bin +3 -0
  9. it/1120ms/decoder_joint.mlmodelc/coremldata.bin +3 -0
  10. it/1120ms/decoder_joint.mlmodelc/model.mil +83 -0
  11. it/1120ms/decoder_joint.mlmodelc/weights/weight.bin +3 -0
  12. it/1120ms/decoder_joint.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  13. it/1120ms/decoder_joint.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  14. it/1120ms/decoder_joint.mlpackage/Manifest.json +18 -0
  15. it/1120ms/decoder_joint_noencproj.mlmodelc/analytics/coremldata.bin +3 -0
  16. it/1120ms/decoder_joint_noencproj.mlmodelc/coremldata.bin +3 -0
  17. it/1120ms/decoder_joint_noencproj.mlmodelc/model.mil +91 -0
  18. it/1120ms/decoder_joint_noencproj.mlmodelc/weights/weight.bin +3 -0
  19. it/1120ms/decoder_joint_noencproj.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  20. it/1120ms/decoder_joint_noencproj.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  21. it/1120ms/decoder_joint_noencproj.mlpackage/Manifest.json +18 -0
  22. it/1120ms/encoder.mlmodelc/analytics/coremldata.bin +3 -0
  23. it/1120ms/encoder.mlmodelc/coremldata.bin +3 -0
  24. it/1120ms/encoder.mlmodelc/model.mil +0 -0
  25. it/1120ms/encoder.mlmodelc/weights/weight.bin +3 -0
  26. it/1120ms/encoder.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  27. it/1120ms/encoder.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  28. it/1120ms/encoder.mlpackage/Manifest.json +18 -0
  29. it/1120ms/joint.mlmodelc/analytics/coremldata.bin +3 -0
  30. it/1120ms/joint.mlmodelc/coremldata.bin +3 -0
  31. it/1120ms/joint.mlmodelc/model.mil +31 -0
  32. it/1120ms/joint.mlmodelc/weights/weight.bin +3 -0
  33. it/1120ms/joint.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  34. it/1120ms/joint.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  35. it/1120ms/joint.mlpackage/Manifest.json +18 -0
  36. it/1120ms/joint_noencproj_batched.mlmodelc/analytics/coremldata.bin +3 -0
  37. it/1120ms/joint_noencproj_batched.mlmodelc/coremldata.bin +3 -0
  38. it/1120ms/joint_noencproj_batched.mlmodelc/model.mil +26 -0
  39. it/1120ms/joint_noencproj_batched.mlmodelc/weights/weight.bin +3 -0
  40. it/1120ms/joint_noencproj_batched.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  41. it/1120ms/joint_noencproj_batched.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  42. it/1120ms/joint_noencproj_batched.mlpackage/Manifest.json +18 -0
  43. it/1120ms/metadata.json +198 -0
  44. it/1120ms/preprocessor.mlmodelc/analytics/coremldata.bin +3 -0
  45. it/1120ms/preprocessor.mlmodelc/coremldata.bin +3 -0
  46. it/1120ms/preprocessor.mlmodelc/model.mil +122 -0
  47. it/1120ms/preprocessor.mlmodelc/weights/weight.bin +3 -0
  48. it/1120ms/preprocessor.mlpackage/Data/com.apple.CoreML/model.mlmodel +3 -0
  49. it/1120ms/preprocessor.mlpackage/Data/com.apple.CoreML/weights/weight.bin +3 -0
  50. 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
+ }