program(1.0) [buildInfo = dict, tensor>({{"coremlc-component-MIL", "3510.2.1"}, {"coremlc-version", "3500.32.1"}, {"coremltools-component-torch", "2.13.0+cu130"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})] { func main(tensor boundary_kind, tensor language_id, tensor mel, tensor text_bytes) { tensor var_12 = const()[name = tensor("op_12"), val = tensor(1)]; tensor input_1_pad_type_0 = const()[name = tensor("input_1_pad_type_0"), val = tensor("custom")]; tensor input_1_pad_0 = const()[name = tensor("input_1_pad_0"), val = tensor([2, 2, 2, 2])]; tensor input_1_strides_0 = const()[name = tensor("input_1_strides_0"), val = tensor([1, 1])]; tensor input_1_dilations_0 = const()[name = tensor("input_1_dilations_0"), val = tensor([1, 1])]; tensor input_1_groups_0 = const()[name = tensor("input_1_groups_0"), val = tensor(1)]; tensor const_3_to_fp16 = const()[name = tensor("const_3_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(64)))]; tensor const_4_to_fp16 = const()[name = tensor("const_4_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(960)))]; tensor input_3_cast_fp16 = conv(bias = const_4_to_fp16, dilations = input_1_dilations_0, groups = input_1_groups_0, pad = input_1_pad_0, pad_type = input_1_pad_type_0, strides = input_1_strides_0, weight = const_3_to_fp16, x = mel)[name = tensor("input_3_cast_fp16")]; tensor input_5_cast_fp16 = silu(x = input_3_cast_fp16)[name = tensor("input_5_cast_fp16")]; tensor var_45 = const()[name = tensor("op_45"), val = tensor([2, 1])]; tensor var_46 = const()[name = tensor("op_46"), val = tensor([2, 1])]; tensor input_7_pad_type_0 = const()[name = tensor("input_7_pad_type_0"), val = tensor("custom")]; tensor input_7_pad_0 = const()[name = tensor("input_7_pad_0"), val = tensor([0, 0, 0, 0])]; tensor input_7_ceil_mode_0 = const()[name = tensor("input_7_ceil_mode_0"), val = tensor(false)]; tensor input_7_cast_fp16 = max_pool(ceil_mode = input_7_ceil_mode_0, kernel_sizes = var_45, pad = input_7_pad_0, pad_type = input_7_pad_type_0, strides = var_46, x = input_5_cast_fp16)[name = tensor("input_7_cast_fp16")]; tensor input_9_pad_type_0 = const()[name = tensor("input_9_pad_type_0"), val = tensor("custom")]; tensor input_9_pad_0 = const()[name = tensor("input_9_pad_0"), val = tensor([1, 1, 1, 1])]; tensor input_9_strides_0 = const()[name = tensor("input_9_strides_0"), val = tensor([1, 1])]; tensor input_9_dilations_0 = const()[name = tensor("input_9_dilations_0"), val = tensor([1, 1])]; tensor input_9_groups_0 = const()[name = tensor("input_9_groups_0"), val = tensor(1)]; tensor const_5_to_fp16 = const()[name = tensor("const_5_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1088)))]; tensor const_6_to_fp16 = const()[name = tensor("const_6_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10368)))]; tensor input_11_cast_fp16 = conv(bias = const_6_to_fp16, dilations = input_9_dilations_0, groups = input_9_groups_0, pad = input_9_pad_0, pad_type = input_9_pad_type_0, strides = input_9_strides_0, weight = const_5_to_fp16, x = input_7_cast_fp16)[name = tensor("input_11_cast_fp16")]; tensor input_13_cast_fp16 = silu(x = input_11_cast_fp16)[name = tensor("input_13_cast_fp16")]; tensor var_63 = const()[name = tensor("op_63"), val = tensor([2, 1])]; tensor var_64 = const()[name = tensor("op_64"), val = tensor([2, 1])]; tensor input_15_pad_type_0 = const()[name = tensor("input_15_pad_type_0"), val = tensor("custom")]; tensor input_15_pad_0 = const()[name = tensor("input_15_pad_0"), val = tensor([0, 0, 0, 0])]; tensor input_15_ceil_mode_0 = const()[name = tensor("input_15_ceil_mode_0"), val = tensor(false)]; tensor input_15_cast_fp16 = max_pool(ceil_mode = input_15_ceil_mode_0, kernel_sizes = var_63, pad = input_15_pad_0, pad_type = input_15_pad_type_0, strides = var_64, x = input_13_cast_fp16)[name = tensor("input_15_cast_fp16")]; tensor input_17_pad_type_0 = const()[name = tensor("input_17_pad_type_0"), val = tensor("custom")]; tensor input_17_pad_0 = const()[name = tensor("input_17_pad_0"), val = tensor([1, 1, 1, 1])]; tensor input_17_strides_0 = const()[name = tensor("input_17_strides_0"), val = tensor([1, 1])]; tensor input_17_dilations_0 = const()[name = tensor("input_17_dilations_0"), val = tensor([1, 1])]; tensor input_17_groups_0 = const()[name = tensor("input_17_groups_0"), val = tensor(1)]; tensor const_7_to_fp16 = const()[name = tensor("const_7_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10496)))]; tensor const_8_to_fp16 = const()[name = tensor("const_8_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(38208)))]; tensor input_19_cast_fp16 = conv(bias = const_8_to_fp16, dilations = input_17_dilations_0, groups = input_17_groups_0, pad = input_17_pad_0, pad_type = input_17_pad_type_0, strides = input_17_strides_0, weight = const_7_to_fp16, x = input_15_cast_fp16)[name = tensor("input_19_cast_fp16")]; tensor input_21_cast_fp16 = silu(x = input_19_cast_fp16)[name = tensor("input_21_cast_fp16")]; tensor var_81 = const()[name = tensor("op_81"), val = tensor([2, 1])]; tensor var_82 = const()[name = tensor("op_82"), val = tensor([2, 1])]; tensor x_pad_type_0 = const()[name = tensor("x_pad_type_0"), val = tensor("custom")]; tensor x_pad_0 = const()[name = tensor("x_pad_0"), val = tensor([0, 0, 0, 0])]; tensor x_ceil_mode_0 = const()[name = tensor("x_ceil_mode_0"), val = tensor(false)]; tensor x_cast_fp16 = max_pool(ceil_mode = x_ceil_mode_0, kernel_sizes = var_81, pad = x_pad_0, pad_type = x_pad_type_0, strides = var_82, x = input_21_cast_fp16)[name = tensor("x_cast_fp16")]; tensor audio_1_axes_0 = const()[name = tensor("audio_1_axes_0"), val = tensor([2])]; tensor audio_1_keep_dims_0 = const()[name = tensor("audio_1_keep_dims_0"), val = tensor(false)]; tensor audio_1_cast_fp16 = reduce_mean(axes = audio_1_axes_0, keep_dims = audio_1_keep_dims_0, x = x_cast_fp16)[name = tensor("audio_1_cast_fp16")]; tensor e_1_batch_dims_0 = const()[name = tensor("e_1_batch_dims_0"), val = tensor(0)]; tensor e_1_validate_indices_0 = const()[name = tensor("e_1_validate_indices_0"), val = tensor(false)]; tensor m_lex_byte_emb_weight_to_fp16 = const()[name = tensor("m_lex_byte_emb_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(38400)))]; tensor text_bytes_to_int16_dtype_0 = const()[name = tensor("text_bytes_to_int16_dtype_0"), val = tensor("int16")]; tensor cast_5_dtype_0 = const()[name = tensor("cast_5_dtype_0"), val = tensor("int32")]; tensor greater_equal_0_y_0 = const()[name = tensor("greater_equal_0_y_0"), val = tensor(0)]; tensor text_bytes_to_int16 = cast(dtype = text_bytes_to_int16_dtype_0, x = text_bytes)[name = tensor("cast_10")]; tensor cast_5 = cast(dtype = cast_5_dtype_0, x = text_bytes_to_int16)[name = tensor("cast_9")]; tensor greater_equal_0 = greater_equal(x = cast_5, y = greater_equal_0_y_0)[name = tensor("greater_equal_0")]; tensor slice_by_index_0 = const()[name = tensor("slice_by_index_0"), val = tensor(257)]; tensor add_0 = add(x = cast_5, y = slice_by_index_0)[name = tensor("add_0")]; tensor select_0 = select(a = cast_5, b = add_0, cond = greater_equal_0)[name = tensor("select_0")]; tensor e_1_cast_fp16_cast_uint16_axis_0 = const()[name = tensor("e_1_cast_fp16_cast_uint16_axis_0"), val = tensor(0)]; tensor select_0_to_int16_dtype_0 = const()[name = tensor("select_0_to_int16_dtype_0"), val = tensor("int16")]; tensor select_0_to_int16 = cast(dtype = select_0_to_int16_dtype_0, x = select_0)[name = tensor("cast_8")]; tensor e_1_cast_fp16_cast_uint16_cast_uint16 = gather(axis = e_1_cast_fp16_cast_uint16_axis_0, batch_dims = e_1_batch_dims_0, indices = select_0_to_int16, validate_indices = e_1_validate_indices_0, x = m_lex_byte_emb_weight_to_fp16)[name = tensor("e_1_cast_fp16_cast_uint16_cast_uint16")]; tensor var_94 = const()[name = tensor("op_94"), val = tensor([16, -1])]; tensor e_cast_fp16 = reshape(shape = var_94, x = e_1_cast_fp16_cast_uint16_cast_uint16)[name = tensor("e_cast_fp16")]; tensor lang_axis_0 = const()[name = tensor("lang_axis_0"), val = tensor(0)]; tensor lang_batch_dims_0 = const()[name = tensor("lang_batch_dims_0"), val = tensor(0)]; tensor lang_validate_indices_0 = const()[name = tensor("lang_validate_indices_0"), val = tensor(false)]; tensor m_lex_lang_emb_weight_to_fp16 = const()[name = tensor("m_lex_lang_emb_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(44672)))]; tensor language_id_to_uint16_dtype_0 = const()[name = tensor("language_id_to_uint16_dtype_0"), val = tensor("uint16")]; tensor language_id_to_uint16 = cast(dtype = language_id_to_uint16_dtype_0, x = language_id)[name = tensor("cast_7")]; tensor lang_cast_fp16_cast_uint16 = gather(axis = lang_axis_0, batch_dims = lang_batch_dims_0, indices = language_id_to_uint16, validate_indices = lang_validate_indices_0, x = m_lex_lang_emb_weight_to_fp16)[name = tensor("lang_cast_fp16_cast_uint16")]; tensor var_98_one_hot_vector_size_0 = const()[name = tensor("op_98_one_hot_vector_size_0"), val = tensor(2)]; tensor var_98_axis_0 = const()[name = tensor("op_98_axis_0"), val = tensor(-1)]; tensor var_98_on_value_0 = const()[name = tensor("op_98_on_value_0"), val = tensor(1)]; tensor var_98_off_value_0 = const()[name = tensor("op_98_off_value_0"), val = tensor(0)]; tensor var_98 = one_hot(axis = var_98_axis_0, indices = boundary_kind, off_value = var_98_off_value_0, on_value = var_98_on_value_0, one_hot_vector_size = var_98_one_hot_vector_size_0)[name = tensor("op_98")]; tensor input_23_interleave_0 = const()[name = tensor("input_23_interleave_0"), val = tensor(false)]; tensor kind_to_fp16_dtype_0 = const()[name = tensor("kind_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_98_to_fp16 = cast(dtype = kind_to_fp16_dtype_0, x = var_98)[name = tensor("cast_6")]; tensor input_23_cast_fp16 = concat(axis = var_12, interleave = input_23_interleave_0, values = (e_cast_fp16, lang_cast_fp16_cast_uint16, var_98_to_fp16))[name = tensor("input_23_cast_fp16")]; tensor m_lex_head_0_weight_to_fp16 = const()[name = tensor("m_lex_head_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(44928)))]; tensor m_lex_head_0_bias_to_fp16 = const()[name = tensor("m_lex_head_0_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(95424)))]; tensor linear_0_cast_fp16 = linear(bias = m_lex_head_0_bias_to_fp16, weight = m_lex_head_0_weight_to_fp16, x = input_23_cast_fp16)[name = tensor("linear_0_cast_fp16")]; tensor input_27_cast_fp16 = silu(x = linear_0_cast_fp16)[name = tensor("input_27_cast_fp16")]; tensor m_lex_head_3_weight_to_fp16 = const()[name = tensor("m_lex_head_3_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(95616)))]; tensor m_lex_head_3_bias_to_fp16 = const()[name = tensor("m_lex_head_3_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(101824)))]; tensor linear_1_cast_fp16 = linear(bias = m_lex_head_3_bias_to_fp16, weight = m_lex_head_3_weight_to_fp16, x = input_27_cast_fp16)[name = tensor("linear_1_cast_fp16")]; tensor var_112_cast_fp16 = silu(x = linear_1_cast_fp16)[name = tensor("op_112_cast_fp16")]; tensor var_113_axes_0 = const()[name = tensor("op_113_axes_0"), val = tensor([-1])]; tensor var_113_cast_fp16 = expand_dims(axes = var_113_axes_0, x = var_112_cast_fp16)[name = tensor("op_113_cast_fp16")]; tensor lexv_reps_0 = const()[name = tensor("lexv_reps_0"), val = tensor([1, 1, 81])]; tensor lexv_cast_fp16 = tile(reps = lexv_reps_0, x = var_113_cast_fp16)[name = tensor("lexv_cast_fp16")]; tensor input_33_interleave_0 = const()[name = tensor("input_33_interleave_0"), val = tensor(false)]; tensor pos_1_to_fp16 = const()[name = tensor("pos_1_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(102016)))]; tensor input_33_cast_fp16 = concat(axis = var_12, interleave = input_33_interleave_0, values = (audio_1_cast_fp16, lexv_cast_fp16, pos_1_to_fp16))[name = tensor("input_33_cast_fp16")]; tensor input_35_pad_type_0 = const()[name = tensor("input_35_pad_type_0"), val = tensor("custom")]; tensor input_35_pad_0 = const()[name = tensor("input_35_pad_0"), val = tensor([2, 2])]; tensor input_35_strides_0 = const()[name = tensor("input_35_strides_0"), val = tensor([1])]; tensor input_35_dilations_0 = const()[name = tensor("input_35_dilations_0"), val = tensor([1])]; tensor input_35_groups_0 = const()[name = tensor("input_35_groups_0"), val = tensor(1)]; tensor const_9_to_fp16 = const()[name = tensor("const_9_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(107264)))]; tensor const_10_to_fp16 = const()[name = tensor("const_10_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(201408)))]; tensor input_37_cast_fp16 = conv(bias = const_10_to_fp16, dilations = input_35_dilations_0, groups = input_35_groups_0, pad = input_35_pad_0, pad_type = input_35_pad_type_0, strides = input_35_strides_0, weight = const_9_to_fp16, x = input_33_cast_fp16)[name = tensor("input_37_cast_fp16")]; tensor input_39_cast_fp16 = silu(x = input_37_cast_fp16)[name = tensor("input_39_cast_fp16")]; tensor input_43_pad_type_0 = const()[name = tensor("input_43_pad_type_0"), val = tensor("custom")]; tensor input_43_pad_0 = const()[name = tensor("input_43_pad_0"), val = tensor([2, 2])]; tensor input_43_strides_0 = const()[name = tensor("input_43_strides_0"), val = tensor([1])]; tensor input_43_dilations_0 = const()[name = tensor("input_43_dilations_0"), val = tensor([1])]; tensor input_43_groups_0 = const()[name = tensor("input_43_groups_0"), val = tensor(1)]; tensor m_temporal_net_4_weight_to_fp16 = const()[name = tensor("m_temporal_net_4_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(201664)))]; tensor m_temporal_net_4_bias_to_fp16 = const()[name = tensor("m_temporal_net_4_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(247808)))]; tensor input_43_cast_fp16 = conv(bias = m_temporal_net_4_bias_to_fp16, dilations = input_43_dilations_0, groups = input_43_groups_0, pad = input_43_pad_0, pad_type = input_43_pad_type_0, strides = input_43_strides_0, weight = m_temporal_net_4_weight_to_fp16, x = input_39_cast_fp16)[name = tensor("input_43_cast_fp16")]; tensor input_cast_fp16 = silu(x = input_43_cast_fp16)[name = tensor("input_cast_fp16")]; tensor var_154_pad_type_0 = const()[name = tensor("op_154_pad_type_0"), val = tensor("valid")]; tensor var_154_strides_0 = const()[name = tensor("op_154_strides_0"), val = tensor([1])]; tensor var_154_pad_0 = const()[name = tensor("op_154_pad_0"), val = tensor([0, 0])]; tensor var_154_dilations_0 = const()[name = tensor("op_154_dilations_0"), val = tensor([1])]; tensor var_154_groups_0 = const()[name = tensor("op_154_groups_0"), val = tensor(1)]; tensor m_temporal_net_6_weight_to_fp16 = const()[name = tensor("m_temporal_net_6_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(248000)))]; tensor m_temporal_net_6_bias_to_fp16 = const()[name = tensor("m_temporal_net_6_bias_to_fp16"), val = tensor([0x1.69cp+3])]; tensor var_154_cast_fp16 = conv(bias = m_temporal_net_6_bias_to_fp16, dilations = var_154_dilations_0, groups = var_154_groups_0, pad = var_154_pad_0, pad_type = var_154_pad_type_0, strides = var_154_strides_0, weight = m_temporal_net_6_weight_to_fp16, x = input_cast_fp16)[name = tensor("op_154_cast_fp16")]; tensor var_155_axes_0 = const()[name = tensor("op_155_axes_0"), val = tensor([1])]; tensor var_155 = squeeze(axes = var_155_axes_0, x = var_154_cast_fp16)[name = tensor("op_155_cast_fp16")]; } -> (var_155); }