| program(1.0) | |
| [buildInfo = dict<tensor<string, []>, tensor<string, []>>({{"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<ios17>(tensor<fp32, [1, 64, 100, 32]> e0, tensor<fp32, [1, 64, 100, 16]> e1, tensor<fp32, [1, 64, 100, 8]> e2, tensor<fp32, [1, 64, 100, 8]> e3, tensor<fp32, [1, 100, 512]> emb) { | |
| tensor<fp32, [16, 32, 16]> decoder_emb_gru_linear_in_0_weight = const()[name = tensor<string, []>("decoder_emb_gru_linear_in_0_weight"), val = tensor<fp32, [16, 32, 16]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(64)))]; | |
| tensor<fp32, [16, 16, 32]> decoder_emb_gru_linear_out_0_weight = const()[name = tensor<string, []>("decoder_emb_gru_linear_out_0_weight"), val = tensor<fp32, [16, 16, 32]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(32896)))]; | |
| tensor<fp32, [64, 1, 1, 3]> decoder_convt3_0_weight = const()[name = tensor<string, []>("decoder_convt3_0_weight"), val = tensor<fp32, [64, 1, 1, 3]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(65728)))]; | |
| tensor<fp32, [64, 1, 1, 3]> decoder_convt2_0_weight = const()[name = tensor<string, []>("decoder_convt2_0_weight"), val = tensor<fp32, [64, 1, 1, 3]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(66560)))]; | |
| tensor<fp32, [64, 1, 1, 3]> decoder_convt1_0_weight = const()[name = tensor<string, []>("decoder_convt1_0_weight"), val = tensor<fp32, [64, 1, 1, 3]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(67392)))]; | |
| tensor<int32, [4]> var_63 = const()[name = tensor<string, []>("op_63"), val = tensor<int32, [4]>([1, 100, 16, 32])]; | |
| tensor<fp32, [1, 100, 16, 32]> var_64 = reshape(shape = var_63, x = emb)[name = tensor<string, []>("op_64")]; | |
| tensor<int32, [4]> transpose_0_perm_0 = const()[name = tensor<string, []>("transpose_0_perm_0"), val = tensor<int32, [4]>([2, 0, 1, 3])]; | |
| tensor<int32, [3]> concat_4 = const()[name = tensor<string, []>("concat_4"), val = tensor<int32, [3]>([16, 100, 32])]; | |
| tensor<fp32, [16, 1, 100, 32]> transpose_0 = transpose(perm = transpose_0_perm_0, x = var_64)[name = tensor<string, []>("transpose_11")]; | |
| tensor<fp32, [16, 100, 32]> reshape_0 = reshape(shape = concat_4, x = transpose_0)[name = tensor<string, []>("reshape_0")]; | |
| tensor<bool, []> matmul_0_transpose_x_0 = const()[name = tensor<string, []>("matmul_0_transpose_x_0"), val = tensor<bool, []>(false)]; | |
| tensor<bool, []> matmul_0_transpose_y_0 = const()[name = tensor<string, []>("matmul_0_transpose_y_0"), val = tensor<bool, []>(false)]; | |
| tensor<fp32, [16, 100, 16]> matmul_0 = matmul(transpose_x = matmul_0_transpose_x_0, transpose_y = matmul_0_transpose_y_0, x = reshape_0, y = decoder_emb_gru_linear_in_0_weight)[name = tensor<string, []>("matmul_0")]; | |
| tensor<int32, [4]> concat_9 = const()[name = tensor<string, []>("concat_9"), val = tensor<int32, [4]>([16, 1, 100, 16])]; | |
| tensor<fp32, [16, 1, 100, 16]> reshape_2 = reshape(shape = concat_9, x = matmul_0)[name = tensor<string, []>("reshape_2")]; | |
| tensor<int32, [4]> x_1_perm_0 = const()[name = tensor<string, []>("x_1_perm_0"), val = tensor<int32, [4]>([1, 2, 0, 3])]; | |
| tensor<int32, [3]> concat_10 = const()[name = tensor<string, []>("concat_10"), val = tensor<int32, [3]>([1, 100, 256])]; | |
| tensor<fp32, [1, 100, 16, 16]> x_1 = transpose(perm = x_1_perm_0, x = reshape_2)[name = tensor<string, []>("transpose_10")]; | |
| tensor<fp32, [1, 100, 256]> input_1 = reshape(shape = concat_10, x = x_1)[name = tensor<string, []>("input_1")]; | |
| tensor<fp32, [1, 100, 256]> input_3 = relu(x = input_1)[name = tensor<string, []>("input_3")]; | |
| tensor<int32, [3]> transpose_2_perm_0 = const()[name = tensor<string, []>("transpose_2_perm_0"), val = tensor<int32, [3]>([1, 0, 2])]; | |
| tensor<int32, [1]> slice_by_index_10 = const()[name = tensor<string, []>("slice_by_index_10"), val = tensor<int32, [1]>([100])]; | |
| tensor<fp32, [101, 1, 256]> concat_12 = const()[name = tensor<string, []>("concat_12"), val = tensor<fp32, [101, 1, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(68224)))]; | |
| tensor<int32, [1]> while_loop_0_loop_vars0_0 = const()[name = tensor<string, []>("while_loop_0_loop_vars0_0"), val = tensor<int32, [1]>([0])]; | |
| tensor<fp32, [100, 1, 256]> transpose_2 = transpose(perm = transpose_2_perm_0, x = input_3)[name = tensor<string, []>("transpose_9")]; | |
| tensor<int32, [1]> while_loop_0_0, tensor<fp32, [101, 1, 256]> while_loop_0_1 = while_loop(loop_vars = (while_loop_0_loop_vars0_0, concat_12))[name = tensor<string, []>("while_loop_0")] | |
| (tensor<int32, [1]> while_loop_0_loop_vars0_0_x0_1_1_1_0, tensor<fp32, [101, 1, 256]> concat_12_x0_1_1_1_0) { | |
| tensor<bool, [1]> less_1 = less(x = while_loop_0_loop_vars0_0_x0_1_1_1_0, y = slice_by_index_10)[name = tensor<string, []>("less_1")]; | |
| } -> (less_1) | |
| (tensor<int32, [1]> while_loop_0_loop_vars0_0_x0_1_1_1_1, tensor<fp32, [101, 1, 256]> concat_12_x0_1_1_1_1) { | |
| tensor<int32, []> gather_2_batch_dims_0 = const()[name = tensor<string, []>("gather_2_batch_dims_0"), val = tensor<int32, []>(0)]; | |
| tensor<bool, []> gather_2_validate_indices_0 = const()[name = tensor<string, []>("gather_2_validate_indices_0"), val = tensor<bool, []>(false)]; | |
| tensor<int32, []> greater_equal_0_y_0 = const()[name = tensor<string, []>("greater_equal_0_y_0"), val = tensor<int32, []>(0)]; | |
| tensor<bool, [1]> 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<string, []>("greater_equal_0")]; | |
| tensor<int32, []> slice_by_index_26 = const()[name = tensor<string, []>("slice_by_index_26"), val = tensor<int32, []>(100)]; | |
| tensor<int32, [1]> add_20 = add(x = while_loop_0_loop_vars0_0_x0_1_1_1_1, y = slice_by_index_26)[name = tensor<string, []>("add_20")]; | |
| tensor<int32, [1]> select_0 = select(a = while_loop_0_loop_vars0_0_x0_1_1_1_1, b = add_20, cond = greater_equal_0)[name = tensor<string, []>("select_0")]; | |
| tensor<int32, []> gather_2_axis_1 = const()[name = tensor<string, []>("gather_2_axis_1"), val = tensor<int32, []>(0)]; | |
| tensor<fp32, [1, 1, 256]> 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<string, []>("gather_2")]; | |
| tensor<int32, []> gather_3_batch_dims_0 = const()[name = tensor<string, []>("gather_3_batch_dims_0"), val = tensor<int32, []>(0)]; | |
| tensor<bool, []> gather_3_validate_indices_0 = const()[name = tensor<string, []>("gather_3_validate_indices_0"), val = tensor<bool, []>(false)]; | |
| tensor<int32, []> slice_by_index_27 = const()[name = tensor<string, []>("slice_by_index_27"), val = tensor<int32, []>(101)]; | |
| tensor<int32, [1]> add_21 = add(x = while_loop_0_loop_vars0_0_x0_1_1_1_1, y = slice_by_index_27)[name = tensor<string, []>("add_21")]; | |
| tensor<int32, [1]> select_1 = select(a = while_loop_0_loop_vars0_0_x0_1_1_1_1, b = add_21, cond = greater_equal_0)[name = tensor<string, []>("select_1")]; | |
| tensor<int32, []> gather_3_axis_1 = const()[name = tensor<string, []>("gather_3_axis_1"), val = tensor<int32, []>(0)]; | |
| tensor<fp32, [1, 1, 256]> 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<string, []>("gather_3")]; | |
| tensor<int32, [1]> squeeze_2_axes_0 = const()[name = tensor<string, []>("squeeze_2_axes_0"), val = tensor<int32, [1]>([0])]; | |
| tensor<fp32, [1, 256]> squeeze_2 = squeeze(axes = squeeze_2_axes_0, x = gather_2)[name = tensor<string, []>("squeeze_2")]; | |
| tensor<int32, [1]> squeeze_3_axes_0 = const()[name = tensor<string, []>("squeeze_3_axes_0"), val = tensor<int32, [1]>([0])]; | |
| tensor<fp32, [1, 256]> squeeze_3 = squeeze(axes = squeeze_3_axes_0, x = gather_3)[name = tensor<string, []>("squeeze_3")]; | |
| tensor<fp32, [256, 256]> linear_6_weight_0 = const()[name = tensor<string, []>("linear_6_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(171712)))]; | |
| tensor<fp32, [256]> linear_6_bias_0 = const()[name = tensor<string, []>("linear_6_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(433920)))]; | |
| tensor<fp32, [1, 256]> linear_6 = linear(bias = linear_6_bias_0, weight = linear_6_weight_0, x = squeeze_2)[name = tensor<string, []>("linear_6")]; | |
| tensor<fp32, [256, 256]> linear_7_weight_0 = const()[name = tensor<string, []>("linear_7_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(435008)))]; | |
| tensor<fp32, [256]> linear_7_bias_0 = const()[name = tensor<string, []>("linear_7_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(697216)))]; | |
| tensor<fp32, [1, 256]> linear_7 = linear(bias = linear_7_bias_0, weight = linear_7_weight_0, x = squeeze_3)[name = tensor<string, []>("linear_7")]; | |
| tensor<fp32, [1, 256]> add_5 = add(x = linear_6, y = linear_7)[name = tensor<string, []>("add_5")]; | |
| tensor<fp32, [1, 256]> sigmoid_2 = sigmoid(x = add_5)[name = tensor<string, []>("sigmoid_2")]; | |
| tensor<fp32, [256, 256]> linear_8_weight_0 = const()[name = tensor<string, []>("linear_8_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(698304)))]; | |
| tensor<fp32, [256]> linear_8_bias_0 = const()[name = tensor<string, []>("linear_8_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(960512)))]; | |
| tensor<fp32, [1, 256]> linear_8 = linear(bias = linear_8_bias_0, weight = linear_8_weight_0, x = squeeze_2)[name = tensor<string, []>("linear_8")]; | |
| tensor<fp32, [256, 256]> linear_9_weight_0 = const()[name = tensor<string, []>("linear_9_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(961600)))]; | |
| tensor<fp32, [256]> linear_9_bias_0 = const()[name = tensor<string, []>("linear_9_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1223808)))]; | |
| tensor<fp32, [1, 256]> linear_9 = linear(bias = linear_9_bias_0, weight = linear_9_weight_0, x = squeeze_3)[name = tensor<string, []>("linear_9")]; | |
| tensor<fp32, [1, 256]> add_6 = add(x = linear_8, y = linear_9)[name = tensor<string, []>("add_6")]; | |
| tensor<fp32, [1, 256]> sigmoid_3 = sigmoid(x = add_6)[name = tensor<string, []>("sigmoid_3")]; | |
| tensor<fp32, [256, 256]> linear_10_weight_0 = const()[name = tensor<string, []>("linear_10_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1224896)))]; | |
| tensor<fp32, [256]> linear_10_bias_0 = const()[name = tensor<string, []>("linear_10_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1487104)))]; | |
| tensor<fp32, [1, 256]> linear_10 = linear(bias = linear_10_bias_0, weight = linear_10_weight_0, x = squeeze_2)[name = tensor<string, []>("linear_10")]; | |
| tensor<fp32, [256, 256]> linear_11_weight_0 = const()[name = tensor<string, []>("linear_11_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1488192)))]; | |
| tensor<fp32, [256]> linear_11_bias_0 = const()[name = tensor<string, []>("linear_11_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1750400)))]; | |
| tensor<fp32, [1, 256]> linear_11 = linear(bias = linear_11_bias_0, weight = linear_11_weight_0, x = squeeze_3)[name = tensor<string, []>("linear_11")]; | |
| tensor<fp32, [1, 256]> mul_3 = mul(x = sigmoid_2, y = linear_11)[name = tensor<string, []>("mul_3")]; | |
| tensor<fp32, [1, 256]> add_7 = add(x = linear_10, y = mul_3)[name = tensor<string, []>("add_7")]; | |
| tensor<fp32, [1, 256]> tanh_1 = tanh(x = add_7)[name = tensor<string, []>("tanh_1")]; | |
| tensor<fp32, []> sub_1_x_0 = const()[name = tensor<string, []>("sub_1_x_0"), val = tensor<fp32, []>(0x1p+0)]; | |
| tensor<fp32, [1, 256]> sub_1 = sub(x = sub_1_x_0, y = sigmoid_3)[name = tensor<string, []>("sub_1")]; | |
| tensor<fp32, [1, 256]> mul_4 = mul(x = sub_1, y = tanh_1)[name = tensor<string, []>("mul_4")]; | |
| tensor<fp32, [1, 256]> mul_5 = mul(x = sigmoid_3, y = squeeze_3)[name = tensor<string, []>("mul_5")]; | |
| tensor<fp32, [1, 256]> add_8 = add(x = mul_4, y = mul_5)[name = tensor<string, []>("add_8")]; | |
| tensor<int32, []> add_9_y_0 = const()[name = tensor<string, []>("add_9_y_0"), val = tensor<int32, []>(1)]; | |
| tensor<int32, [1]> add_9 = add(x = while_loop_0_loop_vars0_0_x0_1_1_1_1, y = add_9_y_0)[name = tensor<string, []>("add_9")]; | |
| tensor<int32, [1]> expand_dims_1_axes_0 = const()[name = tensor<string, []>("expand_dims_1_axes_0"), val = tensor<int32, [1]>([0])]; | |
| tensor<fp32, [1, 1, 256]> expand_dims_1 = expand_dims(axes = expand_dims_1_axes_0, x = add_8)[name = tensor<string, []>("expand_dims_1")]; | |
| tensor<int32, []> scatter_1_axis_0 = const()[name = tensor<string, []>("scatter_1_axis_0"), val = tensor<int32, []>(0)]; | |
| tensor<string, []> scatter_1_mode_0 = const()[name = tensor<string, []>("scatter_1_mode_0"), val = tensor<string, []>("add")]; | |
| tensor<bool, []> scatter_1_validate_indices_0 = const()[name = tensor<string, []>("scatter_1_validate_indices_0"), val = tensor<bool, []>(false)]; | |
| tensor<fp32, [101, 1, 256]> 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<string, []>("scatter_1")]; | |
| } -> (add_9, scatter_1); | |
| tensor<int32, [3]> x_3_layer_0_tmp_begin_0 = const()[name = tensor<string, []>("x_3_layer_0_tmp_begin_0"), val = tensor<int32, [3]>([1, 0, 0])]; | |
| tensor<int32, [3]> x_3_layer_0_tmp_end_0 = const()[name = tensor<string, []>("x_3_layer_0_tmp_end_0"), val = tensor<int32, [3]>([0, 0, 0])]; | |
| tensor<bool, [3]> x_3_layer_0_tmp_begin_mask_0 = const()[name = tensor<string, []>("x_3_layer_0_tmp_begin_mask_0"), val = tensor<bool, [3]>([false, true, true])]; | |
| tensor<bool, [3]> x_3_layer_0_tmp_end_mask_0 = const()[name = tensor<string, []>("x_3_layer_0_tmp_end_mask_0"), val = tensor<bool, [3]>([true, true, true])]; | |
| tensor<fp32, [100, 1, 256]> x_3_layer_0_tmp = slice_by_index(begin = x_3_layer_0_tmp_begin_0, begin_mask = x_3_layer_0_tmp_begin_mask_0, end = x_3_layer_0_tmp_end_0, end_mask = x_3_layer_0_tmp_end_mask_0, x = while_loop_0_1)[name = tensor<string, []>("x_3_layer_0_tmp")]; | |
| tensor<int32, [1]> slice_by_index_13 = const()[name = tensor<string, []>("slice_by_index_13"), val = tensor<int32, [1]>([100])]; | |
| tensor<int32, [1]> while_loop_1_loop_vars0_0 = const()[name = tensor<string, []>("while_loop_1_loop_vars0_0"), val = tensor<int32, [1]>([0])]; | |
| tensor<int32, [1]> while_loop_1_0, tensor<fp32, [101, 1, 256]> while_loop_1_1 = while_loop(loop_vars = (while_loop_1_loop_vars0_0, concat_12))[name = tensor<string, []>("while_loop_1")] | |
| (tensor<int32, [1]> while_loop_1_loop_vars0_0_x0_1_1_1_0, tensor<fp32, [101, 1, 256]> concat_14_x0_1_1_1_0) { | |
| tensor<bool, [1]> less_3 = less(x = while_loop_1_loop_vars0_0_x0_1_1_1_0, y = slice_by_index_13)[name = tensor<string, []>("less_3")]; | |
| } -> (less_3) | |
| (tensor<int32, [1]> while_loop_1_loop_vars0_0_x0_1_1_1_1, tensor<fp32, [101, 1, 256]> concat_14_x0_1_1_1_1) { | |
| tensor<int32, []> gather_6_batch_dims_0 = const()[name = tensor<string, []>("gather_6_batch_dims_0"), val = tensor<int32, []>(0)]; | |
| tensor<bool, []> gather_6_validate_indices_0 = const()[name = tensor<string, []>("gather_6_validate_indices_0"), val = tensor<bool, []>(false)]; | |
| tensor<int32, []> greater_equal_2_y_0 = const()[name = tensor<string, []>("greater_equal_2_y_0"), val = tensor<int32, []>(0)]; | |
| tensor<bool, [1]> 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<string, []>("greater_equal_2")]; | |
| tensor<int32, []> slice_by_index_28 = const()[name = tensor<string, []>("slice_by_index_28"), val = tensor<int32, []>(100)]; | |
| tensor<int32, [1]> add_22 = add(x = while_loop_1_loop_vars0_0_x0_1_1_1_1, y = slice_by_index_28)[name = tensor<string, []>("add_22")]; | |
| tensor<int32, [1]> select_2 = select(a = while_loop_1_loop_vars0_0_x0_1_1_1_1, b = add_22, cond = greater_equal_2)[name = tensor<string, []>("select_2")]; | |
| tensor<int32, []> gather_6_axis_1 = const()[name = tensor<string, []>("gather_6_axis_1"), val = tensor<int32, []>(0)]; | |
| tensor<fp32, [1, 1, 256]> 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 = x_3_layer_0_tmp)[name = tensor<string, []>("gather_6")]; | |
| tensor<int32, []> gather_7_batch_dims_0 = const()[name = tensor<string, []>("gather_7_batch_dims_0"), val = tensor<int32, []>(0)]; | |
| tensor<bool, []> gather_7_validate_indices_0 = const()[name = tensor<string, []>("gather_7_validate_indices_0"), val = tensor<bool, []>(false)]; | |
| tensor<int32, []> slice_by_index_29 = const()[name = tensor<string, []>("slice_by_index_29"), val = tensor<int32, []>(101)]; | |
| tensor<int32, [1]> add_23 = add(x = while_loop_1_loop_vars0_0_x0_1_1_1_1, y = slice_by_index_29)[name = tensor<string, []>("add_23")]; | |
| tensor<int32, [1]> select_3 = select(a = while_loop_1_loop_vars0_0_x0_1_1_1_1, b = add_23, cond = greater_equal_2)[name = tensor<string, []>("select_3")]; | |
| tensor<int32, []> gather_7_axis_1 = const()[name = tensor<string, []>("gather_7_axis_1"), val = tensor<int32, []>(0)]; | |
| tensor<fp32, [1, 1, 256]> 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<string, []>("gather_7")]; | |
| tensor<int32, [1]> squeeze_6_axes_0 = const()[name = tensor<string, []>("squeeze_6_axes_0"), val = tensor<int32, [1]>([0])]; | |
| tensor<fp32, [1, 256]> squeeze_6 = squeeze(axes = squeeze_6_axes_0, x = gather_6)[name = tensor<string, []>("squeeze_6")]; | |
| tensor<int32, [1]> squeeze_7_axes_0 = const()[name = tensor<string, []>("squeeze_7_axes_0"), val = tensor<int32, [1]>([0])]; | |
| tensor<fp32, [1, 256]> squeeze_7 = squeeze(axes = squeeze_7_axes_0, x = gather_7)[name = tensor<string, []>("squeeze_7")]; | |
| tensor<fp32, [256, 256]> linear_18_weight_0 = const()[name = tensor<string, []>("linear_18_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(1751488)))]; | |
| tensor<fp32, [256]> linear_18_bias_0 = const()[name = tensor<string, []>("linear_18_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2013696)))]; | |
| tensor<fp32, [1, 256]> linear_18 = linear(bias = linear_18_bias_0, weight = linear_18_weight_0, x = squeeze_6)[name = tensor<string, []>("linear_18")]; | |
| tensor<fp32, [256, 256]> linear_19_weight_0 = const()[name = tensor<string, []>("linear_19_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2014784)))]; | |
| tensor<fp32, [256]> linear_19_bias_0 = const()[name = tensor<string, []>("linear_19_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2276992)))]; | |
| tensor<fp32, [1, 256]> linear_19 = linear(bias = linear_19_bias_0, weight = linear_19_weight_0, x = squeeze_7)[name = tensor<string, []>("linear_19")]; | |
| tensor<fp32, [1, 256]> add_15 = add(x = linear_18, y = linear_19)[name = tensor<string, []>("add_15")]; | |
| tensor<fp32, [1, 256]> sigmoid_6 = sigmoid(x = add_15)[name = tensor<string, []>("sigmoid_6")]; | |
| tensor<fp32, [256, 256]> linear_20_weight_0 = const()[name = tensor<string, []>("linear_20_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2278080)))]; | |
| tensor<fp32, [256]> linear_20_bias_0 = const()[name = tensor<string, []>("linear_20_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2540288)))]; | |
| tensor<fp32, [1, 256]> linear_20 = linear(bias = linear_20_bias_0, weight = linear_20_weight_0, x = squeeze_6)[name = tensor<string, []>("linear_20")]; | |
| tensor<fp32, [256, 256]> linear_21_weight_0 = const()[name = tensor<string, []>("linear_21_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2541376)))]; | |
| tensor<fp32, [256]> linear_21_bias_0 = const()[name = tensor<string, []>("linear_21_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2803584)))]; | |
| tensor<fp32, [1, 256]> linear_21 = linear(bias = linear_21_bias_0, weight = linear_21_weight_0, x = squeeze_7)[name = tensor<string, []>("linear_21")]; | |
| tensor<fp32, [1, 256]> add_16 = add(x = linear_20, y = linear_21)[name = tensor<string, []>("add_16")]; | |
| tensor<fp32, [1, 256]> sigmoid_7 = sigmoid(x = add_16)[name = tensor<string, []>("sigmoid_7")]; | |
| tensor<fp32, [256, 256]> linear_22_weight_0 = const()[name = tensor<string, []>("linear_22_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(2804672)))]; | |
| tensor<fp32, [256]> linear_22_bias_0 = const()[name = tensor<string, []>("linear_22_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3066880)))]; | |
| tensor<fp32, [1, 256]> linear_22 = linear(bias = linear_22_bias_0, weight = linear_22_weight_0, x = squeeze_6)[name = tensor<string, []>("linear_22")]; | |
| tensor<fp32, [256, 256]> linear_23_weight_0 = const()[name = tensor<string, []>("linear_23_weight_0"), val = tensor<fp32, [256, 256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3067968)))]; | |
| tensor<fp32, [256]> linear_23_bias_0 = const()[name = tensor<string, []>("linear_23_bias_0"), val = tensor<fp32, [256]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3330176)))]; | |
| tensor<fp32, [1, 256]> linear_23 = linear(bias = linear_23_bias_0, weight = linear_23_weight_0, x = squeeze_7)[name = tensor<string, []>("linear_23")]; | |
| tensor<fp32, [1, 256]> mul_9 = mul(x = sigmoid_6, y = linear_23)[name = tensor<string, []>("mul_9")]; | |
| tensor<fp32, [1, 256]> add_17 = add(x = linear_22, y = mul_9)[name = tensor<string, []>("add_17")]; | |
| tensor<fp32, [1, 256]> tanh_3 = tanh(x = add_17)[name = tensor<string, []>("tanh_3")]; | |
| tensor<fp32, []> sub_3_x_0 = const()[name = tensor<string, []>("sub_3_x_0"), val = tensor<fp32, []>(0x1p+0)]; | |
| tensor<fp32, [1, 256]> sub_3 = sub(x = sub_3_x_0, y = sigmoid_7)[name = tensor<string, []>("sub_3")]; | |
| tensor<fp32, [1, 256]> mul_10 = mul(x = sub_3, y = tanh_3)[name = tensor<string, []>("mul_10")]; | |
| tensor<fp32, [1, 256]> mul_11 = mul(x = sigmoid_7, y = squeeze_7)[name = tensor<string, []>("mul_11")]; | |
| tensor<fp32, [1, 256]> add_18 = add(x = mul_10, y = mul_11)[name = tensor<string, []>("add_18")]; | |
| tensor<int32, []> add_19_y_0 = const()[name = tensor<string, []>("add_19_y_0"), val = tensor<int32, []>(1)]; | |
| tensor<int32, [1]> add_19 = add(x = while_loop_1_loop_vars0_0_x0_1_1_1_1, y = add_19_y_0)[name = tensor<string, []>("add_19")]; | |
| tensor<int32, [1]> expand_dims_3_axes_0 = const()[name = tensor<string, []>("expand_dims_3_axes_0"), val = tensor<int32, [1]>([0])]; | |
| tensor<fp32, [1, 1, 256]> expand_dims_3 = expand_dims(axes = expand_dims_3_axes_0, x = add_18)[name = tensor<string, []>("expand_dims_3")]; | |
| tensor<int32, []> scatter_3_axis_0 = const()[name = tensor<string, []>("scatter_3_axis_0"), val = tensor<int32, []>(0)]; | |
| tensor<string, []> scatter_3_mode_0 = const()[name = tensor<string, []>("scatter_3_mode_0"), val = tensor<string, []>("add")]; | |
| tensor<bool, []> scatter_3_validate_indices_0 = const()[name = tensor<string, []>("scatter_3_validate_indices_0"), val = tensor<bool, []>(false)]; | |
| tensor<fp32, [101, 1, 256]> 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<string, []>("scatter_3")]; | |
| } -> (add_19, scatter_3); | |
| tensor<int32, [3]> x_3_tmp_begin_0 = const()[name = tensor<string, []>("x_3_tmp_begin_0"), val = tensor<int32, [3]>([1, 0, 0])]; | |
| tensor<int32, [3]> x_3_tmp_end_0 = const()[name = tensor<string, []>("x_3_tmp_end_0"), val = tensor<int32, [3]>([0, 0, 0])]; | |
| tensor<bool, [3]> x_3_tmp_begin_mask_0 = const()[name = tensor<string, []>("x_3_tmp_begin_mask_0"), val = tensor<bool, [3]>([false, true, true])]; | |
| tensor<bool, [3]> x_3_tmp_end_mask_0 = const()[name = tensor<string, []>("x_3_tmp_end_mask_0"), val = tensor<bool, [3]>([true, true, true])]; | |
| tensor<fp32, [100, 1, 256]> x_3_tmp = slice_by_index(begin = x_3_tmp_begin_0, begin_mask = x_3_tmp_begin_mask_0, end = x_3_tmp_end_0, end_mask = x_3_tmp_end_mask_0, x = while_loop_1_1)[name = tensor<string, []>("x_3_tmp")]; | |
| tensor<int32, [3]> x_3_perm_0 = const()[name = tensor<string, []>("x_3_perm_0"), val = tensor<int32, [3]>([1, 0, 2])]; | |
| tensor<int32, [4]> var_87 = const()[name = tensor<string, []>("op_87"), val = tensor<int32, [4]>([1, 100, 16, 16])]; | |
| tensor<fp32, [1, 100, 256]> x_3 = transpose(perm = x_3_perm_0, x = x_3_tmp)[name = tensor<string, []>("transpose_8")]; | |
| tensor<fp32, [1, 100, 16, 16]> var_88 = reshape(shape = var_87, x = x_3)[name = tensor<string, []>("op_88")]; | |
| tensor<int32, [4]> transpose_3_perm_0 = const()[name = tensor<string, []>("transpose_3_perm_0"), val = tensor<int32, [4]>([2, 0, 1, 3])]; | |
| tensor<int32, [3]> concat_19 = const()[name = tensor<string, []>("concat_19"), val = tensor<int32, [3]>([16, 100, 16])]; | |
| tensor<fp32, [16, 1, 100, 16]> transpose_3 = transpose(perm = transpose_3_perm_0, x = var_88)[name = tensor<string, []>("transpose_7")]; | |
| tensor<fp32, [16, 100, 16]> reshape_3 = reshape(shape = concat_19, x = transpose_3)[name = tensor<string, []>("reshape_3")]; | |
| tensor<bool, []> matmul_1_transpose_x_0 = const()[name = tensor<string, []>("matmul_1_transpose_x_0"), val = tensor<bool, []>(false)]; | |
| tensor<bool, []> matmul_1_transpose_y_0 = const()[name = tensor<string, []>("matmul_1_transpose_y_0"), val = tensor<bool, []>(false)]; | |
| tensor<fp32, [16, 100, 32]> matmul_1 = matmul(transpose_x = matmul_1_transpose_x_0, transpose_y = matmul_1_transpose_y_0, x = reshape_3, y = decoder_emb_gru_linear_out_0_weight)[name = tensor<string, []>("matmul_1")]; | |
| tensor<int32, [4]> concat_24 = const()[name = tensor<string, []>("concat_24"), val = tensor<int32, [4]>([16, 1, 100, 32])]; | |
| tensor<fp32, [16, 1, 100, 32]> reshape_5 = reshape(shape = concat_24, x = matmul_1)[name = tensor<string, []>("reshape_5")]; | |
| tensor<int32, [4]> x_perm_0 = const()[name = tensor<string, []>("x_perm_0"), val = tensor<int32, [4]>([1, 2, 0, 3])]; | |
| tensor<int32, [3]> concat_25 = const()[name = tensor<string, []>("concat_25"), val = tensor<int32, [3]>([1, 100, 512])]; | |
| tensor<fp32, [1, 100, 16, 32]> x = transpose(perm = x_perm_0, x = reshape_5)[name = tensor<string, []>("transpose_6")]; | |
| tensor<fp32, [1, 100, 512]> input_5 = reshape(shape = concat_25, x = x)[name = tensor<string, []>("input_5")]; | |
| tensor<fp32, [1, 100, 512]> var_92 = relu(x = input_5)[name = tensor<string, []>("op_92")]; | |
| tensor<int32, [4]> concat_26 = const()[name = tensor<string, []>("concat_26"), val = tensor<int32, [4]>([1, 100, 8, 64])]; | |
| tensor<fp32, [1, 100, 8, 64]> var_97 = reshape(shape = concat_26, x = var_92)[name = tensor<string, []>("op_97")]; | |
| tensor<int32, [4]> var_102 = const()[name = tensor<string, []>("op_102"), val = tensor<int32, [4]>([0, 3, 1, 2])]; | |
| tensor<string, []> input_7_pad_type_0 = const()[name = tensor<string, []>("input_7_pad_type_0"), val = tensor<string, []>("valid")]; | |
| tensor<int32, []> input_7_groups_0 = const()[name = tensor<string, []>("input_7_groups_0"), val = tensor<int32, []>(64)]; | |
| tensor<int32, [2]> input_7_strides_0 = const()[name = tensor<string, []>("input_7_strides_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [4]> input_7_pad_0 = const()[name = tensor<string, []>("input_7_pad_0"), val = tensor<int32, [4]>([0, 0, 0, 0])]; | |
| tensor<int32, [2]> input_7_dilations_0 = const()[name = tensor<string, []>("input_7_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<fp32, [64, 1, 1, 1]> const_5 = const()[name = tensor<string, []>("const_5"), val = tensor<fp32, [64, 1, 1, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3331264)))]; | |
| tensor<fp32, [64]> const_6 = const()[name = tensor<string, []>("const_6"), val = tensor<fp32, [64]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3331584)))]; | |
| tensor<fp32, [1, 64, 100, 8]> input_9 = conv(bias = const_6, 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 = const_5, x = e3)[name = tensor<string, []>("input_9")]; | |
| tensor<fp32, [1, 64, 100, 8]> var_125 = relu(x = input_9)[name = tensor<string, []>("op_125")]; | |
| tensor<fp32, [1, 64, 100, 8]> emb_out = transpose(perm = var_102, x = var_97)[name = tensor<string, []>("transpose_5")]; | |
| tensor<fp32, [1, 64, 100, 8]> input_11 = add(x = var_125, y = emb_out)[name = tensor<string, []>("input_11")]; | |
| tensor<string, []> input_13_pad_type_0 = const()[name = tensor<string, []>("input_13_pad_type_0"), val = tensor<string, []>("custom")]; | |
| tensor<int32, [4]> input_13_pad_0 = const()[name = tensor<string, []>("input_13_pad_0"), val = tensor<int32, [4]>([0, 0, 1, 1])]; | |
| tensor<int32, []> input_13_groups_0 = const()[name = tensor<string, []>("input_13_groups_0"), val = tensor<int32, []>(64)]; | |
| tensor<int32, [2]> input_13_strides_0 = const()[name = tensor<string, []>("input_13_strides_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [2]> input_13_dilations_0 = const()[name = tensor<string, []>("input_13_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<fp32, [1, 64, 100, 8]> input_13 = conv(dilations = input_13_dilations_0, groups = input_13_groups_0, pad = input_13_pad_0, pad_type = input_13_pad_type_0, strides = input_13_strides_0, weight = decoder_convt3_0_weight, x = input_11)[name = tensor<string, []>("input_13")]; | |
| tensor<string, []> input_15_pad_type_0 = const()[name = tensor<string, []>("input_15_pad_type_0"), val = tensor<string, []>("valid")]; | |
| tensor<int32, [2]> input_15_strides_0 = const()[name = tensor<string, []>("input_15_strides_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [4]> input_15_pad_0 = const()[name = tensor<string, []>("input_15_pad_0"), val = tensor<int32, [4]>([0, 0, 0, 0])]; | |
| tensor<int32, [2]> input_15_dilations_0 = const()[name = tensor<string, []>("input_15_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, []> input_15_groups_0 = const()[name = tensor<string, []>("input_15_groups_0"), val = tensor<int32, []>(1)]; | |
| tensor<fp32, [64, 64, 1, 1]> const_7 = const()[name = tensor<string, []>("const_7"), val = tensor<fp32, [64, 64, 1, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3331904)))]; | |
| tensor<fp32, [64]> const_8 = const()[name = tensor<string, []>("const_8"), val = tensor<fp32, [64]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3348352)))]; | |
| tensor<fp32, [1, 64, 100, 8]> input_17 = conv(bias = const_8, dilations = input_15_dilations_0, groups = input_15_groups_0, pad = input_15_pad_0, pad_type = input_15_pad_type_0, strides = input_15_strides_0, weight = const_7, x = input_13)[name = tensor<string, []>("input_17")]; | |
| tensor<fp32, [1, 64, 100, 8]> e3_1 = relu(x = input_17)[name = tensor<string, []>("e3")]; | |
| tensor<string, []> input_19_pad_type_0 = const()[name = tensor<string, []>("input_19_pad_type_0"), val = tensor<string, []>("valid")]; | |
| tensor<int32, []> input_19_groups_0 = const()[name = tensor<string, []>("input_19_groups_0"), val = tensor<int32, []>(64)]; | |
| tensor<int32, [2]> input_19_strides_0 = const()[name = tensor<string, []>("input_19_strides_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [4]> input_19_pad_0 = const()[name = tensor<string, []>("input_19_pad_0"), val = tensor<int32, [4]>([0, 0, 0, 0])]; | |
| tensor<int32, [2]> input_19_dilations_0 = const()[name = tensor<string, []>("input_19_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<fp32, [64, 1, 1, 1]> const_9 = const()[name = tensor<string, []>("const_9"), val = tensor<fp32, [64, 1, 1, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3348672)))]; | |
| tensor<fp32, [64]> const_10 = const()[name = tensor<string, []>("const_10"), val = tensor<fp32, [64]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3348992)))]; | |
| tensor<fp32, [1, 64, 100, 8]> input_21 = conv(bias = const_10, 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_9, x = e2)[name = tensor<string, []>("input_21")]; | |
| tensor<fp32, [1, 64, 100, 8]> var_178 = relu(x = input_21)[name = tensor<string, []>("op_178")]; | |
| tensor<fp32, [1, 64, 100, 8]> input_23 = add(x = var_178, y = e3_1)[name = tensor<string, []>("input_23")]; | |
| tensor<string, []> conv_transpose_0_pad_type_0 = const()[name = tensor<string, []>("conv_transpose_0_pad_type_0"), val = tensor<string, []>("custom")]; | |
| tensor<int32, [4]> conv_transpose_0_pad_0 = const()[name = tensor<string, []>("conv_transpose_0_pad_0"), val = tensor<int32, [4]>([0, 0, 0, 0])]; | |
| tensor<int32, [2]> conv_transpose_0_strides_0 = const()[name = tensor<string, []>("conv_transpose_0_strides_0"), val = tensor<int32, [2]>([1, 2])]; | |
| tensor<int32, []> conv_transpose_0_groups_0 = const()[name = tensor<string, []>("conv_transpose_0_groups_0"), val = tensor<int32, []>(64)]; | |
| tensor<int32, [2]> conv_transpose_0_dilations_0 = const()[name = tensor<string, []>("conv_transpose_0_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [4]> conv_transpose_0_has_output_shape_output_shape_0 = const()[name = tensor<string, []>("conv_transpose_0_has_output_shape_output_shape_0"), val = tensor<int32, [4]>([1, 64, 100, 17])]; | |
| tensor<fp32, [1, 64, 100, 17]> conv_transpose_0_has_output_shape = conv_transpose(dilations = conv_transpose_0_dilations_0, groups = conv_transpose_0_groups_0, output_shape = conv_transpose_0_has_output_shape_output_shape_0, pad = conv_transpose_0_pad_0, pad_type = conv_transpose_0_pad_type_0, strides = conv_transpose_0_strides_0, weight = decoder_convt2_0_weight, x = input_23)[name = tensor<string, []>("conv_transpose_0_has_output_shape")]; | |
| tensor<int32, [2]> input_25_crop_height_0 = const()[name = tensor<string, []>("input_25_crop_height_0"), val = tensor<int32, [2]>([0, 0])]; | |
| tensor<int32, [2]> input_25_crop_width_0 = const()[name = tensor<string, []>("input_25_crop_width_0"), val = tensor<int32, [2]>([1, 0])]; | |
| tensor<fp32, [1, 64, 100, 16]> input_25 = crop(crop_height = input_25_crop_height_0, crop_width = input_25_crop_width_0, x = conv_transpose_0_has_output_shape)[name = tensor<string, []>("input_25")]; | |
| tensor<string, []> input_27_pad_type_0 = const()[name = tensor<string, []>("input_27_pad_type_0"), val = tensor<string, []>("valid")]; | |
| tensor<int32, [2]> input_27_strides_0 = const()[name = tensor<string, []>("input_27_strides_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [4]> input_27_pad_0 = const()[name = tensor<string, []>("input_27_pad_0"), val = tensor<int32, [4]>([0, 0, 0, 0])]; | |
| tensor<int32, [2]> input_27_dilations_0 = const()[name = tensor<string, []>("input_27_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, []> input_27_groups_0 = const()[name = tensor<string, []>("input_27_groups_0"), val = tensor<int32, []>(1)]; | |
| tensor<fp32, [64, 64, 1, 1]> const_11 = const()[name = tensor<string, []>("const_11"), val = tensor<fp32, [64, 64, 1, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3349312)))]; | |
| tensor<fp32, [64]> const_12 = const()[name = tensor<string, []>("const_12"), val = tensor<fp32, [64]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3365760)))]; | |
| tensor<fp32, [1, 64, 100, 16]> input_29 = conv(bias = const_12, 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_11, x = input_25)[name = tensor<string, []>("input_29")]; | |
| tensor<fp32, [1, 64, 100, 16]> e2_1 = relu(x = input_29)[name = tensor<string, []>("e2")]; | |
| tensor<string, []> input_31_pad_type_0 = const()[name = tensor<string, []>("input_31_pad_type_0"), val = tensor<string, []>("valid")]; | |
| tensor<int32, []> input_31_groups_0 = const()[name = tensor<string, []>("input_31_groups_0"), val = tensor<int32, []>(64)]; | |
| tensor<int32, [2]> input_31_strides_0 = const()[name = tensor<string, []>("input_31_strides_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [4]> input_31_pad_0 = const()[name = tensor<string, []>("input_31_pad_0"), val = tensor<int32, [4]>([0, 0, 0, 0])]; | |
| tensor<int32, [2]> input_31_dilations_0 = const()[name = tensor<string, []>("input_31_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<fp32, [64, 1, 1, 1]> const_13 = const()[name = tensor<string, []>("const_13"), val = tensor<fp32, [64, 1, 1, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3366080)))]; | |
| tensor<fp32, [64]> const_14 = const()[name = tensor<string, []>("const_14"), val = tensor<fp32, [64]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3366400)))]; | |
| tensor<fp32, [1, 64, 100, 16]> input_33 = conv(bias = const_14, dilations = input_31_dilations_0, groups = input_31_groups_0, pad = input_31_pad_0, pad_type = input_31_pad_type_0, strides = input_31_strides_0, weight = const_13, x = e1)[name = tensor<string, []>("input_33")]; | |
| tensor<fp32, [1, 64, 100, 16]> var_232 = relu(x = input_33)[name = tensor<string, []>("op_232")]; | |
| tensor<fp32, [1, 64, 100, 16]> input_35 = add(x = var_232, y = e2_1)[name = tensor<string, []>("input_35")]; | |
| tensor<string, []> conv_transpose_1_pad_type_0 = const()[name = tensor<string, []>("conv_transpose_1_pad_type_0"), val = tensor<string, []>("custom")]; | |
| tensor<int32, [4]> conv_transpose_1_pad_0 = const()[name = tensor<string, []>("conv_transpose_1_pad_0"), val = tensor<int32, [4]>([0, 0, 0, 0])]; | |
| tensor<int32, [2]> conv_transpose_1_strides_0 = const()[name = tensor<string, []>("conv_transpose_1_strides_0"), val = tensor<int32, [2]>([1, 2])]; | |
| tensor<int32, []> conv_transpose_1_groups_0 = const()[name = tensor<string, []>("conv_transpose_1_groups_0"), val = tensor<int32, []>(64)]; | |
| tensor<int32, [2]> conv_transpose_1_dilations_0 = const()[name = tensor<string, []>("conv_transpose_1_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [4]> conv_transpose_1_has_output_shape_output_shape_0 = const()[name = tensor<string, []>("conv_transpose_1_has_output_shape_output_shape_0"), val = tensor<int32, [4]>([1, 64, 100, 33])]; | |
| tensor<fp32, [1, 64, 100, 33]> conv_transpose_1_has_output_shape = conv_transpose(dilations = conv_transpose_1_dilations_0, groups = conv_transpose_1_groups_0, output_shape = conv_transpose_1_has_output_shape_output_shape_0, pad = conv_transpose_1_pad_0, pad_type = conv_transpose_1_pad_type_0, strides = conv_transpose_1_strides_0, weight = decoder_convt1_0_weight, x = input_35)[name = tensor<string, []>("conv_transpose_1_has_output_shape")]; | |
| tensor<int32, [2]> input_37_crop_height_0 = const()[name = tensor<string, []>("input_37_crop_height_0"), val = tensor<int32, [2]>([0, 0])]; | |
| tensor<int32, [2]> input_37_crop_width_0 = const()[name = tensor<string, []>("input_37_crop_width_0"), val = tensor<int32, [2]>([1, 0])]; | |
| tensor<fp32, [1, 64, 100, 32]> input_37 = crop(crop_height = input_37_crop_height_0, crop_width = input_37_crop_width_0, x = conv_transpose_1_has_output_shape)[name = tensor<string, []>("input_37")]; | |
| tensor<string, []> input_39_pad_type_0 = const()[name = tensor<string, []>("input_39_pad_type_0"), val = tensor<string, []>("valid")]; | |
| tensor<int32, [2]> input_39_strides_0 = const()[name = tensor<string, []>("input_39_strides_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [4]> input_39_pad_0 = const()[name = tensor<string, []>("input_39_pad_0"), val = tensor<int32, [4]>([0, 0, 0, 0])]; | |
| tensor<int32, [2]> input_39_dilations_0 = const()[name = tensor<string, []>("input_39_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, []> input_39_groups_0 = const()[name = tensor<string, []>("input_39_groups_0"), val = tensor<int32, []>(1)]; | |
| tensor<fp32, [64, 64, 1, 1]> const_15 = const()[name = tensor<string, []>("const_15"), val = tensor<fp32, [64, 64, 1, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3366720)))]; | |
| tensor<fp32, [64]> const_16 = const()[name = tensor<string, []>("const_16"), val = tensor<fp32, [64]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3383168)))]; | |
| tensor<fp32, [1, 64, 100, 32]> input_41 = conv(bias = const_16, dilations = input_39_dilations_0, groups = input_39_groups_0, pad = input_39_pad_0, pad_type = input_39_pad_type_0, strides = input_39_strides_0, weight = const_15, x = input_37)[name = tensor<string, []>("input_41")]; | |
| tensor<fp32, [1, 64, 100, 32]> e1_1 = relu(x = input_41)[name = tensor<string, []>("e1")]; | |
| tensor<string, []> input_43_pad_type_0 = const()[name = tensor<string, []>("input_43_pad_type_0"), val = tensor<string, []>("valid")]; | |
| tensor<int32, []> input_43_groups_0 = const()[name = tensor<string, []>("input_43_groups_0"), val = tensor<int32, []>(64)]; | |
| tensor<int32, [2]> input_43_strides_0 = const()[name = tensor<string, []>("input_43_strides_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [4]> input_43_pad_0 = const()[name = tensor<string, []>("input_43_pad_0"), val = tensor<int32, [4]>([0, 0, 0, 0])]; | |
| tensor<int32, [2]> input_43_dilations_0 = const()[name = tensor<string, []>("input_43_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<fp32, [64, 1, 1, 1]> const_17 = const()[name = tensor<string, []>("const_17"), val = tensor<fp32, [64, 1, 1, 1]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3383488)))]; | |
| tensor<fp32, [64]> const_18 = const()[name = tensor<string, []>("const_18"), val = tensor<fp32, [64]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3383808)))]; | |
| tensor<fp32, [1, 64, 100, 32]> input_45 = conv(bias = const_18, 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_17, x = e0)[name = tensor<string, []>("input_45")]; | |
| tensor<fp32, [1, 64, 100, 32]> var_286 = relu(x = input_45)[name = tensor<string, []>("op_286")]; | |
| tensor<fp32, [1, 64, 100, 32]> input_47 = add(x = var_286, y = e1_1)[name = tensor<string, []>("input_47")]; | |
| tensor<string, []> input_pad_type_0 = const()[name = tensor<string, []>("input_pad_type_0"), val = tensor<string, []>("custom")]; | |
| tensor<int32, [4]> input_pad_0 = const()[name = tensor<string, []>("input_pad_0"), val = tensor<int32, [4]>([0, 0, 1, 1])]; | |
| tensor<int32, [2]> input_strides_0 = const()[name = tensor<string, []>("input_strides_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, [2]> input_dilations_0 = const()[name = tensor<string, []>("input_dilations_0"), val = tensor<int32, [2]>([1, 1])]; | |
| tensor<int32, []> input_groups_0 = const()[name = tensor<string, []>("input_groups_0"), val = tensor<int32, []>(1)]; | |
| tensor<fp32, [1, 64, 1, 3]> const_19 = const()[name = tensor<string, []>("const_19"), val = tensor<fp32, [1, 64, 1, 3]>(BLOBFILE(path = tensor<string, []>("@model_path/weights/weight.bin"), offset = tensor<uint64, []>(3384128)))]; | |
| tensor<fp32, [1]> const_20 = const()[name = tensor<string, []>("const_20"), val = tensor<fp32, [1]>([-0x1.0ebe88p+0])]; | |
| tensor<fp32, [1, 1, 100, 32]> erb_gains = conv(bias = const_20, dilations = input_dilations_0, groups = input_groups_0, pad = input_pad_0, pad_type = input_pad_type_0, strides = input_strides_0, weight = const_19, x = input_47)[name = tensor<string, []>("erb_gains")]; | |
| } -> (erb_gains); | |
| } |