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 c0, tensor emb) { tensor decoder_df_fc_a_0_bias = const()[name = tensor("decoder_df_fc_a_0_bias"), val = tensor([-0x1.ee457p-6])]; tensor decoder_df_fc_a_0_weight = const()[name = tensor("decoder_df_fc_a_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(64)))]; tensor decoder_df_gru_linear_in_0_weight = const()[name = tensor("decoder_df_gru_linear_in_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1152)))]; tensor decoder_df_skip_weight = const()[name = tensor("decoder_df_skip_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(66752)))]; tensor decoder_df_convp_1_weight = const()[name = tensor("decoder_df_convp_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(99584)))]; tensor decoder_df_out_0_weight = const()[name = tensor("decoder_df_out_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(106048)))]; tensor var_39 = const()[name = tensor("op_39"), val = tensor([1, 100, 8, 64])]; tensor var_40 = reshape(shape = var_39, x = emb)[name = tensor("op_40")]; tensor transpose_0_perm_0 = const()[name = tensor("transpose_0_perm_0"), val = tensor([2, 0, 1, 3])]; tensor concat_4 = const()[name = tensor("concat_4"), val = tensor([8, 100, 64])]; tensor transpose_0 = transpose(perm = transpose_0_perm_0, x = var_40)[name = tensor("transpose_15")]; tensor reshape_0 = reshape(shape = concat_4, 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 = decoder_df_gru_linear_in_0_weight)[name = tensor("matmul_0")]; tensor concat_9 = const()[name = tensor("concat_9"), val = tensor([8, 1, 100, 32])]; tensor reshape_2 = reshape(shape = concat_9, x = matmul_0)[name = tensor("reshape_2")]; tensor x_1_perm_0 = const()[name = tensor("x_1_perm_0"), val = tensor([1, 2, 0, 3])]; tensor concat_10 = const()[name = tensor("concat_10"), val = tensor([1, 100, 256])]; tensor x_1 = transpose(perm = x_1_perm_0, x = reshape_2)[name = tensor("transpose_14")]; tensor input_1 = reshape(shape = concat_10, x = x_1)[name = tensor("input_1")]; tensor input_3 = relu(x = input_1)[name = tensor("input_3")]; tensor transpose_2_perm_0 = const()[name = tensor("transpose_2_perm_0"), val = tensor([1, 0, 2])]; tensor slice_by_index_10 = const()[name = tensor("slice_by_index_10"), val = tensor([100])]; tensor concat_12 = const()[name = tensor("concat_12"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(167552)))]; tensor while_loop_0_loop_vars0_0 = const()[name = tensor("while_loop_0_loop_vars0_0"), val = tensor([0])]; tensor transpose_2 = transpose(perm = transpose_2_perm_0, x = input_3)[name = tensor("transpose_13")]; tensor while_loop_0_0, tensor while_loop_0_1 = while_loop(loop_vars = (while_loop_0_loop_vars0_0, concat_12))[name = tensor("while_loop_0")] (tensor while_loop_0_loop_vars0_0_x0_1_1_1_0, tensor concat_12_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_10)[name = tensor("less_1")]; } -> (less_1) (tensor while_loop_0_loop_vars0_0_x0_1_1_1_1, tensor concat_12_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_35 = const()[name = tensor("slice_by_index_35"), val = tensor(100)]; tensor add_20 = add(x = while_loop_0_loop_vars0_0_x0_1_1_1_1, y = slice_by_index_35)[name = tensor("add_20")]; tensor select_0 = select(a = while_loop_0_loop_vars0_0_x0_1_1_1_1, b = add_20, 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_2)[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_36 = const()[name = tensor("slice_by_index_36"), val = tensor(101)]; tensor add_21 = add(x = while_loop_0_loop_vars0_0_x0_1_1_1_1, y = slice_by_index_36)[name = tensor("add_21")]; tensor select_1 = select(a = while_loop_0_loop_vars0_0_x0_1_1_1_1, b = add_21, 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_12_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(271040)))]; tensor linear_6_bias_0 = const()[name = tensor("linear_6_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(533248)))]; 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(534336)))]; tensor linear_7_bias_0 = const()[name = tensor("linear_7_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(796544)))]; 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(797632)))]; tensor linear_8_bias_0 = const()[name = tensor("linear_8_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1059840)))]; 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(1060928)))]; tensor linear_9_bias_0 = const()[name = tensor("linear_9_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1323136)))]; 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(1324224)))]; tensor linear_10_bias_0 = const()[name = tensor("linear_10_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1586432)))]; 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(1587520)))]; tensor linear_11_bias_0 = const()[name = tensor("linear_11_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1849728)))]; 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_12_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 c_layer_0_tmp_begin_0 = const()[name = tensor("c_layer_0_tmp_begin_0"), val = tensor([1, 0, 0])]; tensor c_layer_0_tmp_end_0 = const()[name = tensor("c_layer_0_tmp_end_0"), val = tensor([0, 0, 0])]; tensor c_layer_0_tmp_begin_mask_0 = const()[name = tensor("c_layer_0_tmp_begin_mask_0"), val = tensor([false, true, true])]; tensor c_layer_0_tmp_end_mask_0 = const()[name = tensor("c_layer_0_tmp_end_mask_0"), val = tensor([true, true, true])]; tensor c_layer_0_tmp = slice_by_index(begin = c_layer_0_tmp_begin_0, begin_mask = c_layer_0_tmp_begin_mask_0, end = c_layer_0_tmp_end_0, end_mask = c_layer_0_tmp_end_mask_0, x = while_loop_0_1)[name = tensor("c_layer_0_tmp")]; tensor slice_by_index_13 = const()[name = tensor("slice_by_index_13"), val = tensor([100])]; tensor while_loop_1_loop_vars0_0 = const()[name = tensor("while_loop_1_loop_vars0_0"), val = tensor([0])]; tensor while_loop_1_0, tensor while_loop_1_1 = while_loop(loop_vars = (while_loop_1_loop_vars0_0, concat_12))[name = tensor("while_loop_1")] (tensor while_loop_1_loop_vars0_0_x0_1_1_1_0, tensor concat_14_x0_1_1_1_0) { tensor less_3 = less(x = while_loop_1_loop_vars0_0_x0_1_1_1_0, y = slice_by_index_13)[name = tensor("less_3")]; } -> (less_3) (tensor while_loop_1_loop_vars0_0_x0_1_1_1_1, tensor concat_14_x0_1_1_1_1) { tensor gather_6_batch_dims_0 = const()[name = tensor("gather_6_batch_dims_0"), val = tensor(0)]; tensor gather_6_validate_indices_0 = const()[name = tensor("gather_6_validate_indices_0"), val = tensor(false)]; tensor greater_equal_2_y_0 = const()[name = tensor("greater_equal_2_y_0"), val = tensor(0)]; tensor greater_equal_2 = greater_equal(x = while_loop_1_loop_vars0_0_x0_1_1_1_1, y = greater_equal_2_y_0)[name = tensor("greater_equal_2")]; tensor slice_by_index_37 = const()[name = tensor("slice_by_index_37"), val = tensor(100)]; tensor add_22 = add(x = while_loop_1_loop_vars0_0_x0_1_1_1_1, y = slice_by_index_37)[name = tensor("add_22")]; tensor select_2 = select(a = while_loop_1_loop_vars0_0_x0_1_1_1_1, b = add_22, cond = greater_equal_2)[name = tensor("select_2")]; tensor gather_6_axis_1 = const()[name = tensor("gather_6_axis_1"), val = tensor(0)]; tensor gather_6 = gather(axis = gather_6_axis_1, batch_dims = gather_6_batch_dims_0, indices = select_2, validate_indices = gather_6_validate_indices_0, x = c_layer_0_tmp)[name = tensor("gather_6")]; tensor gather_7_batch_dims_0 = const()[name = tensor("gather_7_batch_dims_0"), val = tensor(0)]; tensor gather_7_validate_indices_0 = const()[name = tensor("gather_7_validate_indices_0"), val = tensor(false)]; tensor slice_by_index_38 = const()[name = tensor("slice_by_index_38"), val = tensor(101)]; tensor add_23 = add(x = while_loop_1_loop_vars0_0_x0_1_1_1_1, y = slice_by_index_38)[name = tensor("add_23")]; tensor select_3 = select(a = while_loop_1_loop_vars0_0_x0_1_1_1_1, b = add_23, cond = greater_equal_2)[name = tensor("select_3")]; tensor gather_7_axis_1 = const()[name = tensor("gather_7_axis_1"), val = tensor(0)]; tensor gather_7 = gather(axis = gather_7_axis_1, batch_dims = gather_7_batch_dims_0, indices = select_3, validate_indices = gather_7_validate_indices_0, x = concat_14_x0_1_1_1_1)[name = tensor("gather_7")]; tensor squeeze_6_axes_0 = const()[name = tensor("squeeze_6_axes_0"), val = tensor([0])]; tensor squeeze_6 = squeeze(axes = squeeze_6_axes_0, x = gather_6)[name = tensor("squeeze_6")]; tensor squeeze_7_axes_0 = const()[name = tensor("squeeze_7_axes_0"), val = tensor([0])]; tensor squeeze_7 = squeeze(axes = squeeze_7_axes_0, x = gather_7)[name = tensor("squeeze_7")]; tensor linear_18_weight_0 = const()[name = tensor("linear_18_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1850816)))]; tensor linear_18_bias_0 = const()[name = tensor("linear_18_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2113024)))]; tensor linear_18 = linear(bias = linear_18_bias_0, weight = linear_18_weight_0, x = squeeze_6)[name = tensor("linear_18")]; tensor linear_19_weight_0 = const()[name = tensor("linear_19_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2114112)))]; tensor linear_19_bias_0 = const()[name = tensor("linear_19_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2376320)))]; tensor linear_19 = linear(bias = linear_19_bias_0, weight = linear_19_weight_0, x = squeeze_7)[name = tensor("linear_19")]; tensor add_15 = add(x = linear_18, y = linear_19)[name = tensor("add_15")]; tensor sigmoid_6 = sigmoid(x = add_15)[name = tensor("sigmoid_6")]; tensor linear_20_weight_0 = const()[name = tensor("linear_20_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2377408)))]; tensor linear_20_bias_0 = const()[name = tensor("linear_20_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2639616)))]; tensor linear_20 = linear(bias = linear_20_bias_0, weight = linear_20_weight_0, x = squeeze_6)[name = tensor("linear_20")]; tensor linear_21_weight_0 = const()[name = tensor("linear_21_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2640704)))]; tensor linear_21_bias_0 = const()[name = tensor("linear_21_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2902912)))]; tensor linear_21 = linear(bias = linear_21_bias_0, weight = linear_21_weight_0, x = squeeze_7)[name = tensor("linear_21")]; tensor add_16 = add(x = linear_20, y = linear_21)[name = tensor("add_16")]; tensor sigmoid_7 = sigmoid(x = add_16)[name = tensor("sigmoid_7")]; tensor linear_22_weight_0 = const()[name = tensor("linear_22_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2904000)))]; tensor linear_22_bias_0 = const()[name = tensor("linear_22_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3166208)))]; tensor linear_22 = linear(bias = linear_22_bias_0, weight = linear_22_weight_0, x = squeeze_6)[name = tensor("linear_22")]; tensor linear_23_weight_0 = const()[name = tensor("linear_23_weight_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3167296)))]; tensor linear_23_bias_0 = const()[name = tensor("linear_23_bias_0"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3429504)))]; tensor linear_23 = linear(bias = linear_23_bias_0, weight = linear_23_weight_0, x = squeeze_7)[name = tensor("linear_23")]; tensor mul_9 = mul(x = sigmoid_6, y = linear_23)[name = tensor("mul_9")]; tensor add_17 = add(x = linear_22, y = mul_9)[name = tensor("add_17")]; tensor tanh_3 = tanh(x = add_17)[name = tensor("tanh_3")]; tensor sub_3_x_0 = const()[name = tensor("sub_3_x_0"), val = tensor(0x1p+0)]; tensor sub_3 = sub(x = sub_3_x_0, y = sigmoid_7)[name = tensor("sub_3")]; tensor mul_10 = mul(x = sub_3, y = tanh_3)[name = tensor("mul_10")]; tensor mul_11 = mul(x = sigmoid_7, y = squeeze_7)[name = tensor("mul_11")]; tensor add_18 = add(x = mul_10, y = mul_11)[name = tensor("add_18")]; tensor add_19_y_0 = const()[name = tensor("add_19_y_0"), val = tensor(1)]; tensor add_19 = add(x = while_loop_1_loop_vars0_0_x0_1_1_1_1, y = add_19_y_0)[name = tensor("add_19")]; tensor expand_dims_3_axes_0 = const()[name = tensor("expand_dims_3_axes_0"), val = tensor([0])]; tensor expand_dims_3 = expand_dims(axes = expand_dims_3_axes_0, x = add_18)[name = tensor("expand_dims_3")]; tensor scatter_3_axis_0 = const()[name = tensor("scatter_3_axis_0"), val = tensor(0)]; tensor scatter_3_mode_0 = const()[name = tensor("scatter_3_mode_0"), val = tensor("add")]; tensor scatter_3_validate_indices_0 = const()[name = tensor("scatter_3_validate_indices_0"), val = tensor(false)]; tensor scatter_3 = scatter(axis = scatter_3_axis_0, data = concat_14_x0_1_1_1_1, indices = add_19, mode = scatter_3_mode_0, updates = expand_dims_3, validate_indices = scatter_3_validate_indices_0)[name = tensor("scatter_3")]; } -> (add_19, scatter_3); tensor c_tmp_begin_0 = const()[name = tensor("c_tmp_begin_0"), val = tensor([1, 0, 0])]; tensor c_tmp_end_0 = const()[name = tensor("c_tmp_end_0"), val = tensor([0, 0, 0])]; tensor c_tmp_begin_mask_0 = const()[name = tensor("c_tmp_begin_mask_0"), val = tensor([false, true, true])]; tensor c_tmp_end_mask_0 = const()[name = tensor("c_tmp_end_mask_0"), val = tensor([true, true, true])]; tensor c_tmp = slice_by_index(begin = c_tmp_begin_0, begin_mask = c_tmp_begin_mask_0, end = c_tmp_end_0, end_mask = c_tmp_end_mask_0, x = while_loop_1_1)[name = tensor("c_tmp")]; tensor c_perm_0 = const()[name = tensor("c_perm_0"), val = tensor([1, 0, 2])]; tensor var_70 = const()[name = tensor("op_70"), val = tensor([1, 100, 16, 32])]; tensor var_71 = reshape(shape = var_70, x = emb)[name = tensor("op_71")]; tensor transpose_3_perm_0 = const()[name = tensor("transpose_3_perm_0"), val = tensor([2, 0, 1, 3])]; tensor concat_19 = const()[name = tensor("concat_19"), val = tensor([16, 100, 32])]; tensor transpose_3 = transpose(perm = transpose_3_perm_0, x = var_71)[name = tensor("transpose_11")]; tensor reshape_3 = reshape(shape = concat_19, x = transpose_3)[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 = decoder_df_skip_weight)[name = tensor("matmul_1")]; tensor concat_24 = const()[name = tensor("concat_24"), val = tensor([16, 1, 100, 16])]; tensor reshape_5 = reshape(shape = concat_24, x = matmul_1)[name = tensor("reshape_5")]; tensor x_3_perm_0 = const()[name = tensor("x_3_perm_0"), val = tensor([1, 2, 0, 3])]; tensor concat_25 = const()[name = tensor("concat_25"), val = tensor([1, 100, 256])]; tensor x_3 = transpose(perm = x_3_perm_0, x = reshape_5)[name = tensor("transpose_10")]; tensor var_74 = reshape(shape = concat_25, x = x_3)[name = tensor("op_74")]; tensor c = transpose(perm = c_perm_0, x = c_tmp)[name = tensor("transpose_12")]; tensor input_13 = add(x = c, y = var_74)[name = tensor("input_13")]; tensor const_5 = const()[name = tensor("const_5"), val = tensor(0x0p+0)]; tensor input_5_pad_0 = const()[name = tensor("input_5_pad_0"), val = tensor([0, 0, 0, 0, 4, 0, 0, 0])]; tensor input_5_mode_0 = const()[name = tensor("input_5_mode_0"), val = tensor("constant")]; tensor input_5 = pad(constant_val = const_5, mode = input_5_mode_0, pad = input_5_pad_0, x = c0)[name = tensor("input_5")]; tensor input_7_pad_type_0 = const()[name = tensor("input_7_pad_type_0"), val = tensor("valid")]; tensor input_7_groups_0 = const()[name = tensor("input_7_groups_0"), val = tensor(2)]; tensor input_7_strides_0 = const()[name = tensor("input_7_strides_0"), val = tensor([1, 1])]; tensor input_7_pad_0 = const()[name = tensor("input_7_pad_0"), val = tensor([0, 0, 0, 0])]; tensor input_7_dilations_0 = const()[name = tensor("input_7_dilations_0"), val = tensor([1, 1])]; tensor input_7 = conv(dilations = input_7_dilations_0, groups = input_7_groups_0, pad = input_7_pad_0, pad_type = input_7_pad_type_0, strides = input_7_strides_0, weight = decoder_df_convp_1_weight, x = input_5)[name = tensor("input_7")]; tensor input_9_pad_type_0 = const()[name = tensor("input_9_pad_type_0"), val = tensor("valid")]; tensor input_9_strides_0 = const()[name = tensor("input_9_strides_0"), val = tensor([1, 1])]; tensor input_9_pad_0 = const()[name = tensor("input_9_pad_0"), val = tensor([0, 0, 0, 0])]; 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_8 = const()[name = tensor("const_8"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3430592)))]; tensor const_9 = const()[name = tensor("const_9"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3431104)))]; tensor input_11 = conv(bias = const_9, 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_8, x = input_7)[name = tensor("input_11")]; tensor var_110 = relu(x = input_11)[name = tensor("op_110")]; tensor var_115 = const()[name = tensor("op_115"), val = tensor([0, 2, 3, 1])]; tensor alpha = linear(bias = decoder_df_fc_a_0_bias, weight = decoder_df_fc_a_0_weight, x = input_13)[name = tensor("linear_24")]; tensor var_129 = const()[name = tensor("op_129"), val = tensor([1, 100, 16, 16])]; tensor var_130 = reshape(shape = var_129, x = input_13)[name = tensor("op_130")]; 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_130)[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 = decoder_df_out_0_weight)[name = tensor("matmul_2")]; tensor concat_35 = const()[name = tensor("concat_35"), val = tensor([16, 1, 100, 60])]; 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, 960])]; tensor x = transpose(perm = x_perm_0, x = reshape_8)[name = tensor("transpose_7")]; tensor input = reshape(shape = concat_36, x = x)[name = tensor("input")]; tensor var_134 = tanh(x = input)[name = tensor("op_134")]; tensor concat_37 = const()[name = tensor("concat_37"), val = tensor([1, 100, 96, 10])]; tensor var_139 = reshape(shape = concat_37, x = var_134)[name = tensor("op_139")]; tensor c0_1 = transpose(perm = var_115, x = var_110)[name = tensor("transpose_9")]; tensor df_coefficients = add(x = var_139, y = c0_1)[name = tensor("op_141")]; } -> (df_coefficients, alpha); }