program(1.0) [buildInfo = dict, tensor>({{"coremlc-component-MIL", "3510.2.1"}, {"coremlc-version", "3500.32.1"}, {"coremltools-component-torch", "2.10.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})] { func main(tensor erb_features, tensor spec_features) { tensor encoder_erb_conv1_0_weight = const()[name = tensor("encoder_erb_conv1_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(64)))]; tensor encoder_erb_conv2_0_weight = const()[name = tensor("encoder_erb_conv2_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(896)))]; tensor encoder_erb_conv3_0_weight = const()[name = tensor("encoder_erb_conv3_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1728)))]; tensor encoder_df_conv0_1_weight = const()[name = tensor("encoder_df_conv0_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2560)))]; tensor encoder_df_conv1_0_weight = const()[name = tensor("encoder_df_conv1_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4928)))]; tensor encoder_df_fc_emb_0_weight = const()[name = tensor("encoder_df_fc_emb_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(5760)))]; tensor encoder_emb_gru_linear_in_0_weight = const()[name = tensor("encoder_emb_gru_linear_in_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(202432)))]; tensor encoder_emb_gru_linear_out_0_weight = const()[name = tensor("encoder_emb_gru_linear_out_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(235264)))]; tensor encoder_lsnr_fc_0_bias = const()[name = tensor("encoder_lsnr_fc_0_bias"), val = tensor([-0x1.331464p-2])]; tensor encoder_lsnr_fc_0_weight = const()[name = tensor("encoder_lsnr_fc_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(268096)))]; tensor const_0 = const()[name = tensor("const_0"), val = tensor(0x0p+0)]; tensor input_1_pad_0 = const()[name = tensor("input_1_pad_0"), val = tensor([0, 0, 0, 0, 2, 0, 0, 0])]; tensor input_1_mode_0 = const()[name = tensor("input_1_mode_0"), val = tensor("constant")]; tensor input_1 = pad(constant_val = const_0, mode = input_1_mode_0, pad = input_1_pad_0, x = erb_features)[name = tensor("input_1")]; tensor input_3_pad_type_0 = const()[name = tensor("input_3_pad_type_0"), val = tensor("custom")]; tensor input_3_pad_0 = const()[name = tensor("input_3_pad_0"), val = tensor([0, 0, 1, 1])]; tensor input_3_strides_0 = const()[name = tensor("input_3_strides_0"), val = tensor([1, 1])]; tensor input_3_dilations_0 = const()[name = tensor("input_3_dilations_0"), val = tensor([1, 1])]; tensor input_3_groups_0 = const()[name = tensor("input_3_groups_0"), val = tensor(1)]; tensor const_9 = const()[name = tensor("const_9"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(270208)))]; tensor const_10 = const()[name = tensor("const_10"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(272576)))]; tensor input_5 = conv(bias = const_10, dilations = input_3_dilations_0, groups = input_3_groups_0, pad = input_3_pad_0, pad_type = input_3_pad_type_0, strides = input_3_strides_0, weight = const_9, x = input_1)[name = tensor("input_5")]; tensor e0 = relu(x = input_5)[name = tensor("input_7")]; 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([0, 0, 1, 1])]; tensor input_9_strides_0 = const()[name = tensor("input_9_strides_0"), val = tensor([1, 2])]; tensor input_9_groups_0 = const()[name = tensor("input_9_groups_0"), val = tensor(64)]; tensor input_9_dilations_0 = const()[name = tensor("input_9_dilations_0"), val = tensor([1, 1])]; tensor input_9 = conv(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 = encoder_erb_conv1_0_weight, x = e0)[name = tensor("input_9")]; tensor input_11_pad_type_0 = const()[name = tensor("input_11_pad_type_0"), val = tensor("valid")]; tensor input_11_strides_0 = const()[name = tensor("input_11_strides_0"), val = tensor([1, 1])]; tensor input_11_pad_0 = const()[name = tensor("input_11_pad_0"), val = tensor([0, 0, 0, 0])]; tensor input_11_dilations_0 = const()[name = tensor("input_11_dilations_0"), val = tensor([1, 1])]; tensor input_11_groups_0 = const()[name = tensor("input_11_groups_0"), val = tensor(1)]; tensor const_11 = const()[name = tensor("const_11"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(272896)))]; tensor const_12 = const()[name = tensor("const_12"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(289344)))]; tensor input_13 = conv(bias = const_12, dilations = input_11_dilations_0, groups = input_11_groups_0, pad = input_11_pad_0, pad_type = input_11_pad_type_0, strides = input_11_strides_0, weight = const_11, x = input_9)[name = tensor("input_13")]; tensor e1 = relu(x = input_13)[name = tensor("input_15")]; 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([0, 0, 1, 1])]; tensor input_17_strides_0 = const()[name = tensor("input_17_strides_0"), val = tensor([1, 2])]; tensor input_17_groups_0 = const()[name = tensor("input_17_groups_0"), val = tensor(64)]; tensor input_17_dilations_0 = const()[name = tensor("input_17_dilations_0"), val = tensor([1, 1])]; tensor input_17 = conv(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 = encoder_erb_conv2_0_weight, x = e1)[name = tensor("input_17")]; tensor input_19_pad_type_0 = const()[name = tensor("input_19_pad_type_0"), val = tensor("valid")]; tensor input_19_strides_0 = const()[name = tensor("input_19_strides_0"), val = tensor([1, 1])]; tensor input_19_pad_0 = const()[name = tensor("input_19_pad_0"), val = tensor([0, 0, 0, 0])]; tensor input_19_dilations_0 = const()[name = tensor("input_19_dilations_0"), val = tensor([1, 1])]; tensor input_19_groups_0 = const()[name = tensor("input_19_groups_0"), val = tensor(1)]; tensor const_13 = const()[name = tensor("const_13"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(289664)))]; tensor const_14 = const()[name = tensor("const_14"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(306112)))]; tensor input_21 = conv(bias = const_14, dilations = input_19_dilations_0, groups = input_19_groups_0, pad = input_19_pad_0, pad_type = input_19_pad_type_0, strides = input_19_strides_0, weight = const_13, x = input_17)[name = tensor("input_21")]; tensor e2 = relu(x = input_21)[name = tensor("input_23")]; tensor input_25_pad_type_0 = const()[name = tensor("input_25_pad_type_0"), val = tensor("custom")]; tensor input_25_pad_0 = const()[name = tensor("input_25_pad_0"), val = tensor([0, 0, 1, 1])]; tensor input_25_groups_0 = const()[name = tensor("input_25_groups_0"), val = tensor(64)]; tensor input_25_strides_0 = const()[name = tensor("input_25_strides_0"), val = tensor([1, 1])]; tensor input_25_dilations_0 = const()[name = tensor("input_25_dilations_0"), val = tensor([1, 1])]; tensor input_25 = conv(dilations = input_25_dilations_0, groups = input_25_groups_0, pad = input_25_pad_0, pad_type = input_25_pad_type_0, strides = input_25_strides_0, weight = encoder_erb_conv3_0_weight, x = e2)[name = tensor("input_25")]; tensor input_27_pad_type_0 = const()[name = tensor("input_27_pad_type_0"), val = tensor("valid")]; tensor input_27_strides_0 = const()[name = tensor("input_27_strides_0"), val = tensor([1, 1])]; tensor input_27_pad_0 = const()[name = tensor("input_27_pad_0"), val = tensor([0, 0, 0, 0])]; tensor input_27_dilations_0 = const()[name = tensor("input_27_dilations_0"), val = tensor([1, 1])]; tensor input_27_groups_0 = const()[name = tensor("input_27_groups_0"), val = tensor(1)]; tensor const_15 = const()[name = tensor("const_15"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(306432)))]; tensor const_16 = const()[name = tensor("const_16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(322880)))]; tensor input_29 = conv(bias = const_16, dilations = input_27_dilations_0, groups = input_27_groups_0, pad = input_27_pad_0, pad_type = input_27_pad_type_0, strides = input_27_strides_0, weight = const_15, x = input_25)[name = tensor("input_29")]; tensor e3 = relu(x = input_29)[name = tensor("e3")]; tensor const_1 = const()[name = tensor("const_1"), val = tensor(0x0p+0)]; tensor input_31_pad_0 = const()[name = tensor("input_31_pad_0"), val = tensor([0, 0, 0, 0, 2, 0, 0, 0])]; tensor input_31_mode_0 = const()[name = tensor("input_31_mode_0"), val = tensor("constant")]; tensor input_31 = pad(constant_val = const_1, mode = input_31_mode_0, pad = input_31_pad_0, x = spec_features)[name = tensor("input_31")]; tensor input_33_pad_type_0 = const()[name = tensor("input_33_pad_type_0"), val = tensor("custom")]; tensor input_33_pad_0 = const()[name = tensor("input_33_pad_0"), val = tensor([0, 0, 1, 1])]; tensor input_33_groups_0 = const()[name = tensor("input_33_groups_0"), val = tensor(2)]; tensor input_33_strides_0 = const()[name = tensor("input_33_strides_0"), val = tensor([1, 1])]; tensor input_33_dilations_0 = const()[name = tensor("input_33_dilations_0"), val = tensor([1, 1])]; tensor input_33 = conv(dilations = input_33_dilations_0, groups = input_33_groups_0, pad = input_33_pad_0, pad_type = input_33_pad_type_0, strides = input_33_strides_0, weight = encoder_df_conv0_1_weight, x = input_31)[name = tensor("input_33")]; tensor input_35_pad_type_0 = const()[name = tensor("input_35_pad_type_0"), val = tensor("valid")]; tensor input_35_strides_0 = const()[name = tensor("input_35_strides_0"), val = tensor([1, 1])]; tensor input_35_pad_0 = const()[name = tensor("input_35_pad_0"), val = tensor([0, 0, 0, 0])]; tensor input_35_dilations_0 = const()[name = tensor("input_35_dilations_0"), val = tensor([1, 1])]; tensor input_35_groups_0 = const()[name = tensor("input_35_groups_0"), val = tensor(1)]; tensor const_17 = const()[name = tensor("const_17"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(323200)))]; tensor const_18 = const()[name = tensor("const_18"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(339648)))]; tensor input_37 = conv(bias = const_18, 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_17, x = input_33)[name = tensor("input_37")]; tensor c0 = relu(x = input_37)[name = tensor("input_39")]; tensor input_41_pad_type_0 = const()[name = tensor("input_41_pad_type_0"), val = tensor("custom")]; tensor input_41_pad_0 = const()[name = tensor("input_41_pad_0"), val = tensor([0, 0, 1, 1])]; tensor input_41_strides_0 = const()[name = tensor("input_41_strides_0"), val = tensor([1, 2])]; tensor input_41_groups_0 = const()[name = tensor("input_41_groups_0"), val = tensor(64)]; tensor input_41_dilations_0 = const()[name = tensor("input_41_dilations_0"), val = tensor([1, 1])]; tensor input_41 = conv(dilations = input_41_dilations_0, groups = input_41_groups_0, pad = input_41_pad_0, pad_type = input_41_pad_type_0, strides = input_41_strides_0, weight = encoder_df_conv1_0_weight, x = c0)[name = tensor("input_41")]; tensor input_43_pad_type_0 = const()[name = tensor("input_43_pad_type_0"), val = tensor("valid")]; tensor input_43_strides_0 = const()[name = tensor("input_43_strides_0"), val = tensor([1, 1])]; tensor input_43_pad_0 = const()[name = tensor("input_43_pad_0"), val = tensor([0, 0, 0, 0])]; tensor input_43_dilations_0 = const()[name = tensor("input_43_dilations_0"), val = tensor([1, 1])]; tensor input_43_groups_0 = const()[name = tensor("input_43_groups_0"), val = tensor(1)]; tensor const_19 = const()[name = tensor("const_19"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(339968)))]; tensor const_20 = const()[name = tensor("const_20"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(356416)))]; tensor input_45 = conv(bias = const_20, 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 = const_19, x = input_41)[name = tensor("input_45")]; tensor c1 = relu(x = input_45)[name = tensor("c1")]; tensor var_156 = const()[name = tensor("op_156"), val = tensor([0, 2, 3, 1])]; tensor var_163 = const()[name = tensor("op_163"), val = tensor([1, 100, 32, 96])]; tensor var_157 = transpose(perm = var_156, x = c1)[name = tensor("transpose_16")]; tensor var_164 = reshape(shape = var_163, x = var_157)[name = tensor("op_164")]; tensor transpose_0_perm_0 = const()[name = tensor("transpose_0_perm_0"), val = tensor([2, 0, 1, 3])]; tensor concat_5 = const()[name = tensor("concat_5"), val = tensor([32, 100, 96])]; tensor transpose_0 = transpose(perm = transpose_0_perm_0, x = var_164)[name = tensor("transpose_15")]; tensor reshape_0 = reshape(shape = concat_5, x = transpose_0)[name = tensor("reshape_0")]; tensor matmul_0_transpose_x_0 = const()[name = tensor("matmul_0_transpose_x_0"), val = tensor(false)]; tensor matmul_0_transpose_y_0 = const()[name = tensor("matmul_0_transpose_y_0"), val = tensor(false)]; tensor matmul_0 = matmul(transpose_x = matmul_0_transpose_x_0, transpose_y = matmul_0_transpose_y_0, x = reshape_0, y = encoder_df_fc_emb_0_weight)[name = tensor("matmul_0")]; tensor concat_10 = const()[name = tensor("concat_10"), val = tensor([32, 1, 100, 16])]; tensor reshape_2 = reshape(shape = concat_10, x = matmul_0)[name = tensor("reshape_2")]; tensor x_3_perm_0 = const()[name = tensor("x_3_perm_0"), val = tensor([1, 2, 0, 3])]; tensor concat_11 = const()[name = tensor("concat_11"), val = tensor([1, 100, 512])]; tensor x_3 = transpose(perm = x_3_perm_0, x = reshape_2)[name = tensor("transpose_14")]; tensor input_47 = reshape(shape = concat_11, x = x_3)[name = tensor("input_47")]; tensor b_3 = relu(x = input_47)[name = tensor("b_3")]; tensor var_169 = const()[name = tensor("op_169"), val = tensor([0, 2, 3, 1])]; tensor concat_12 = const()[name = tensor("concat_12"), val = tensor([1, 100, 512])]; tensor var_170 = transpose(perm = var_169, x = e3)[name = tensor("transpose_13")]; tensor a = reshape(shape = concat_12, x = var_170)[name = tensor("a")]; tensor x_5 = add(x = a, y = b_3)[name = tensor("x_5")]; tensor var_180 = const()[name = tensor("op_180"), val = tensor([1, 100, 16, 32])]; tensor var_181 = reshape(shape = var_180, x = x_5)[name = tensor("op_181")]; tensor transpose_2_perm_0 = const()[name = tensor("transpose_2_perm_0"), val = tensor([2, 0, 1, 3])]; tensor concat_17 = const()[name = tensor("concat_17"), val = tensor([16, 100, 32])]; tensor transpose_2 = transpose(perm = transpose_2_perm_0, x = var_181)[name = tensor("transpose_12")]; tensor reshape_3 = reshape(shape = concat_17, x = transpose_2)[name = tensor("reshape_3")]; tensor matmul_1_transpose_x_0 = const()[name = tensor("matmul_1_transpose_x_0"), val = tensor(false)]; tensor matmul_1_transpose_y_0 = const()[name = tensor("matmul_1_transpose_y_0"), val = tensor(false)]; tensor matmul_1 = matmul(transpose_x = matmul_1_transpose_x_0, transpose_y = matmul_1_transpose_y_0, x = reshape_3, y = encoder_emb_gru_linear_in_0_weight)[name = tensor("matmul_1")]; tensor concat_22 = const()[name = tensor("concat_22"), val = tensor([16, 1, 100, 16])]; tensor reshape_5 = reshape(shape = concat_22, x = matmul_1)[name = tensor("reshape_5")]; tensor x_7_perm_0 = const()[name = tensor("x_7_perm_0"), val = tensor([1, 2, 0, 3])]; tensor concat_23 = const()[name = tensor("concat_23"), val = tensor([1, 100, 256])]; tensor x_7 = transpose(perm = x_7_perm_0, x = reshape_5)[name = tensor("transpose_11")]; tensor input_49 = reshape(shape = concat_23, x = x_7)[name = tensor("input_49")]; tensor input_51 = relu(x = input_49)[name = tensor("input_51")]; tensor transpose_4_perm_0 = const()[name = tensor("transpose_4_perm_0"), val = tensor([1, 0, 2])]; tensor slice_by_index_23 = const()[name = tensor("slice_by_index_23"), val = tensor([100])]; tensor concat_25 = const()[name = tensor("concat_25"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(356736)))]; tensor while_loop_0_loop_vars0_0 = const()[name = tensor("while_loop_0_loop_vars0_0"), val = tensor([0])]; tensor transpose_4 = transpose(perm = transpose_4_perm_0, x = input_51)[name = tensor("transpose_10")]; tensor while_loop_0_0, tensor while_loop_0_1 = while_loop(loop_vars = (while_loop_0_loop_vars0_0, concat_25))[name = tensor("while_loop_0")] (tensor while_loop_0_loop_vars0_0_x0_1_1_1_0, tensor concat_25_x0_1_1_1_0) { tensor less_1 = less(x = while_loop_0_loop_vars0_0_x0_1_1_1_0, y = slice_by_index_23)[name = tensor("less_1")]; } -> (less_1) (tensor while_loop_0_loop_vars0_0_x0_1_1_1_1, tensor concat_25_x0_1_1_1_1) { tensor gather_2_batch_dims_0 = const()[name = tensor("gather_2_batch_dims_0"), val = tensor(0)]; tensor gather_2_validate_indices_0 = const()[name = tensor("gather_2_validate_indices_0"), val = tensor(false)]; tensor greater_equal_0_y_0 = const()[name = tensor("greater_equal_0_y_0"), val = tensor(0)]; tensor greater_equal_0 = greater_equal(x = while_loop_0_loop_vars0_0_x0_1_1_1_1, y = greater_equal_0_y_0)[name = tensor("greater_equal_0")]; tensor slice_by_index_34 = const()[name = tensor("slice_by_index_34"), val = tensor(100)]; tensor add_10 = add(x = while_loop_0_loop_vars0_0_x0_1_1_1_1, y = slice_by_index_34)[name = tensor("add_10")]; tensor select_0 = select(a = while_loop_0_loop_vars0_0_x0_1_1_1_1, b = add_10, cond = greater_equal_0)[name = tensor("select_0")]; tensor gather_2_axis_1 = const()[name = tensor("gather_2_axis_1"), val = tensor(0)]; tensor gather_2 = gather(axis = gather_2_axis_1, batch_dims = gather_2_batch_dims_0, indices = select_0, validate_indices = gather_2_validate_indices_0, x = transpose_4)[name = tensor("gather_2")]; tensor gather_3_batch_dims_0 = const()[name = tensor("gather_3_batch_dims_0"), val = tensor(0)]; tensor gather_3_validate_indices_0 = const()[name = tensor("gather_3_validate_indices_0"), val = tensor(false)]; tensor slice_by_index_35 = const()[name = tensor("slice_by_index_35"), val = tensor(101)]; tensor add_11 = add(x = while_loop_0_loop_vars0_0_x0_1_1_1_1, y = slice_by_index_35)[name = tensor("add_11")]; tensor select_1 = select(a = while_loop_0_loop_vars0_0_x0_1_1_1_1, b = add_11, cond = greater_equal_0)[name = tensor("select_1")]; tensor gather_3_axis_1 = const()[name = tensor("gather_3_axis_1"), val = tensor(0)]; tensor gather_3 = gather(axis = gather_3_axis_1, batch_dims = gather_3_batch_dims_0, indices = select_1, validate_indices = gather_3_validate_indices_0, x = concat_25_x0_1_1_1_1)[name = tensor("gather_3")]; tensor squeeze_2_axes_0 = const()[name = tensor("squeeze_2_axes_0"), val = tensor([0])]; tensor squeeze_2 = squeeze(axes = squeeze_2_axes_0, x = gather_2)[name = tensor("squeeze_2")]; tensor squeeze_3_axes_0 = const()[name = tensor("squeeze_3_axes_0"), val = tensor([0])]; tensor squeeze_3 = squeeze(axes = squeeze_3_axes_0, x = gather_3)[name = tensor("squeeze_3")]; tensor linear_6_weight_0 = const()[name = tensor("linear_6_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(460224)))]; tensor linear_6_bias_0 = const()[name = tensor("linear_6_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(722432)))]; tensor linear_6 = linear(bias = linear_6_bias_0, weight = linear_6_weight_0, x = squeeze_2)[name = tensor("linear_6")]; tensor linear_7_weight_0 = const()[name = tensor("linear_7_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(723520)))]; tensor linear_7_bias_0 = const()[name = tensor("linear_7_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(985728)))]; tensor linear_7 = linear(bias = linear_7_bias_0, weight = linear_7_weight_0, x = squeeze_3)[name = tensor("linear_7")]; tensor add_5 = add(x = linear_6, y = linear_7)[name = tensor("add_5")]; tensor sigmoid_2 = sigmoid(x = add_5)[name = tensor("sigmoid_2")]; tensor linear_8_weight_0 = const()[name = tensor("linear_8_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(986816)))]; tensor linear_8_bias_0 = const()[name = tensor("linear_8_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1249024)))]; tensor linear_8 = linear(bias = linear_8_bias_0, weight = linear_8_weight_0, x = squeeze_2)[name = tensor("linear_8")]; tensor linear_9_weight_0 = const()[name = tensor("linear_9_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1250112)))]; tensor linear_9_bias_0 = const()[name = tensor("linear_9_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1512320)))]; tensor linear_9 = linear(bias = linear_9_bias_0, weight = linear_9_weight_0, x = squeeze_3)[name = tensor("linear_9")]; tensor add_6 = add(x = linear_8, y = linear_9)[name = tensor("add_6")]; tensor sigmoid_3 = sigmoid(x = add_6)[name = tensor("sigmoid_3")]; tensor linear_10_weight_0 = const()[name = tensor("linear_10_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1513408)))]; tensor linear_10_bias_0 = const()[name = tensor("linear_10_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1775616)))]; tensor linear_10 = linear(bias = linear_10_bias_0, weight = linear_10_weight_0, x = squeeze_2)[name = tensor("linear_10")]; tensor linear_11_weight_0 = const()[name = tensor("linear_11_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1776704)))]; tensor linear_11_bias_0 = const()[name = tensor("linear_11_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2038912)))]; tensor linear_11 = linear(bias = linear_11_bias_0, weight = linear_11_weight_0, x = squeeze_3)[name = tensor("linear_11")]; tensor mul_3 = mul(x = sigmoid_2, y = linear_11)[name = tensor("mul_3")]; tensor add_7 = add(x = linear_10, y = mul_3)[name = tensor("add_7")]; tensor tanh_1 = tanh(x = add_7)[name = tensor("tanh_1")]; tensor sub_1_x_0 = const()[name = tensor("sub_1_x_0"), val = tensor(0x1p+0)]; tensor sub_1 = sub(x = sub_1_x_0, y = sigmoid_3)[name = tensor("sub_1")]; tensor mul_4 = mul(x = sub_1, y = tanh_1)[name = tensor("mul_4")]; tensor mul_5 = mul(x = sigmoid_3, y = squeeze_3)[name = tensor("mul_5")]; tensor add_8 = add(x = mul_4, y = mul_5)[name = tensor("add_8")]; tensor add_9_y_0 = const()[name = tensor("add_9_y_0"), val = tensor(1)]; tensor add_9 = add(x = while_loop_0_loop_vars0_0_x0_1_1_1_1, y = add_9_y_0)[name = tensor("add_9")]; tensor expand_dims_1_axes_0 = const()[name = tensor("expand_dims_1_axes_0"), val = tensor([0])]; tensor expand_dims_1 = expand_dims(axes = expand_dims_1_axes_0, x = add_8)[name = tensor("expand_dims_1")]; tensor scatter_1_axis_0 = const()[name = tensor("scatter_1_axis_0"), val = tensor(0)]; tensor scatter_1_mode_0 = const()[name = tensor("scatter_1_mode_0"), val = tensor("add")]; tensor scatter_1_validate_indices_0 = const()[name = tensor("scatter_1_validate_indices_0"), val = tensor(false)]; tensor scatter_1 = scatter(axis = scatter_1_axis_0, data = concat_25_x0_1_1_1_1, indices = add_9, mode = scatter_1_mode_0, updates = expand_dims_1, validate_indices = scatter_1_validate_indices_0)[name = tensor("scatter_1")]; } -> (add_9, scatter_1); tensor x_9_tmp_begin_0 = const()[name = tensor("x_9_tmp_begin_0"), val = tensor([1, 0, 0])]; tensor x_9_tmp_end_0 = const()[name = tensor("x_9_tmp_end_0"), val = tensor([0, 0, 0])]; tensor x_9_tmp_begin_mask_0 = const()[name = tensor("x_9_tmp_begin_mask_0"), val = tensor([false, true, true])]; tensor x_9_tmp_end_mask_0 = const()[name = tensor("x_9_tmp_end_mask_0"), val = tensor([true, true, true])]; tensor x_9_tmp = slice_by_index(begin = x_9_tmp_begin_0, begin_mask = x_9_tmp_begin_mask_0, end = x_9_tmp_end_0, end_mask = x_9_tmp_end_mask_0, x = while_loop_0_1)[name = tensor("x_9_tmp")]; tensor x_9_perm_0 = const()[name = tensor("x_9_perm_0"), val = tensor([1, 0, 2])]; tensor var_200 = const()[name = tensor("op_200"), val = tensor([1, 100, 16, 16])]; tensor x_9 = transpose(perm = x_9_perm_0, x = x_9_tmp)[name = tensor("transpose_9")]; tensor var_201 = reshape(shape = var_200, x = x_9)[name = tensor("op_201")]; tensor transpose_5_perm_0 = const()[name = tensor("transpose_5_perm_0"), val = tensor([2, 0, 1, 3])]; tensor concat_30 = const()[name = tensor("concat_30"), val = tensor([16, 100, 16])]; tensor transpose_5 = transpose(perm = transpose_5_perm_0, x = var_201)[name = tensor("transpose_8")]; tensor reshape_6 = reshape(shape = concat_30, x = transpose_5)[name = tensor("reshape_6")]; tensor matmul_2_transpose_x_0 = const()[name = tensor("matmul_2_transpose_x_0"), val = tensor(false)]; tensor matmul_2_transpose_y_0 = const()[name = tensor("matmul_2_transpose_y_0"), val = tensor(false)]; tensor matmul_2 = matmul(transpose_x = matmul_2_transpose_x_0, transpose_y = matmul_2_transpose_y_0, x = reshape_6, y = encoder_emb_gru_linear_out_0_weight)[name = tensor("matmul_2")]; tensor concat_35 = const()[name = tensor("concat_35"), val = tensor([16, 1, 100, 32])]; tensor reshape_8 = reshape(shape = concat_35, x = matmul_2)[name = tensor("reshape_8")]; tensor x_perm_0 = const()[name = tensor("x_perm_0"), val = tensor([1, 2, 0, 3])]; tensor concat_36 = const()[name = tensor("concat_36"), val = tensor([1, 100, 512])]; tensor x = transpose(perm = x_perm_0, x = reshape_8)[name = tensor("transpose_7")]; tensor input_53 = reshape(shape = concat_36, x = x)[name = tensor("input_53")]; tensor emb = relu(x = input_53)[name = tensor("input_55")]; tensor input = linear(bias = encoder_lsnr_fc_0_bias, weight = encoder_lsnr_fc_0_weight, x = emb)[name = tensor("linear_12")]; tensor var_210 = sigmoid(x = input)[name = tensor("op_210")]; tensor var_211_promoted = const()[name = tensor("op_211_promoted"), val = tensor(0x1.9p+5)]; tensor var_212 = mul(x = var_210, y = var_211_promoted)[name = tensor("op_212")]; tensor var_213_promoted = const()[name = tensor("op_213_promoted"), val = tensor(-0x1.ep+3)]; tensor lsnr = add(x = var_212, y = var_213_promoted)[name = tensor("op_214")]; } -> (e0, e1, e2, e3, emb, c0, lsnr); }