diff --git "a/optimized/ami/400ms/ls_eend_ami_400ms.mlmodelc/model.mil" "b/optimized/ami/400ms/ls_eend_ami_400ms.mlmodelc/model.mil" new file mode 100644--- /dev/null +++ "b/optimized/ami/400ms/ls_eend_ami_400ms.mlmodelc/model.mil" @@ -0,0 +1,1326 @@ +program(1.0) +[buildInfo = dict, tensor>({{"coremlc-component-MIL", "3510.2.1"}, {"coremlc-version", "3500.32.1"}, {"coremltools-component-torch", "2.6.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})] +{ + func main(tensor cnn_window, tensor dec_kv, tensor dec_scale, tensor enc_conv_cache, tensor enc_kv, tensor enc_scale, tensor features, tensor valid_mask) { + tensor encoder_cnn_bias = const()[name = tensor("encoder_cnn_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(64)))]; + tensor encoder_cnn_weight = const()[name = tensor("encoder_cnn_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1152)))]; + tensor encoder__causal_mask = const()[name = tensor("encoder__causal_mask"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4981952)))]; + tensor encoder__t_index = const()[name = tensor("encoder__t_index"), val = tensor([0x1p+0, 0x1p+1, 0x1.8p+1, 0x1p+2])]; + tensor encoder_input_projection_linear_bias = const()[name = tensor("encoder_input_projection_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4982080)))]; + tensor encoder_input_projection_linear_weight = const()[name = tensor("encoder_input_projection_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4983168)))]; + tensor encoder_pre_layer_norm_bias = const()[name = tensor("encoder_pre_layer_norm_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(5336512)))]; + tensor encoder_pre_layer_norm_weight = const()[name = tensor("encoder_pre_layer_norm_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(5337600)))]; + tensor encoder_ffn1_0_module_sequential_0_bias = const()[name = tensor("encoder_ffn1_0_module_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(5338688)))]; + tensor encoder_ffn1_0_module_sequential_0_weight = const()[name = tensor("encoder_ffn1_0_module_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(5339776)))]; + tensor encoder_ffn1_0_module_sequential_1_linear_bias = const()[name = tensor("encoder_ffn1_0_module_sequential_1_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(5340864)))]; + tensor encoder_ffn1_0_module_sequential_1_linear_weight = const()[name = tensor("encoder_ffn1_0_module_sequential_1_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(5345024)))]; + tensor encoder_ffn1_0_module_sequential_4_linear_bias = const()[name = tensor("encoder_ffn1_0_module_sequential_4_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6393664)))]; + tensor encoder_ffn1_0_module_sequential_4_linear_weight = const()[name = tensor("encoder_ffn1_0_module_sequential_4_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6394752)))]; + tensor encoder_ret_lns_0_bias = const()[name = tensor("encoder_ret_lns_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7443392)))]; + tensor encoder_ret_lns_0_weight = const()[name = tensor("encoder_ret_lns_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7444480)))]; + tensor encoder_q_proj_0_bias = const()[name = tensor("encoder_q_proj_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7445568)))]; + tensor encoder_q_proj_0_weight = const()[name = tensor("encoder_q_proj_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7446656)))]; + tensor encoder_k_proj_0_bias = const()[name = tensor("encoder_k_proj_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7708864)))]; + tensor encoder_k_proj_0_weight = const()[name = tensor("encoder_k_proj_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7709952)))]; + tensor encoder_v_proj_0_bias = const()[name = tensor("encoder_v_proj_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7972160)))]; + tensor encoder_v_proj_0_weight = const()[name = tensor("encoder_v_proj_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7973248)))]; + tensor encoder_g_proj_0_bias = const()[name = tensor("encoder_g_proj_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8235456)))]; + tensor encoder_g_proj_0_weight = const()[name = tensor("encoder_g_proj_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8236544)))]; + tensor encoder_out_proj_0_bias = const()[name = tensor("encoder_out_proj_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8498752)))]; + tensor encoder_out_proj_0_weight = const()[name = tensor("encoder_out_proj_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8499840)))]; + tensor encoder_conv_module_0_sequential_0_bias = const()[name = tensor("encoder_conv_module_0_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8762048)))]; + tensor encoder_conv_module_0_sequential_0_weight = const()[name = tensor("encoder_conv_module_0_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8763136)))]; + tensor encoder_conv_module_0_sequential_2_conv_bias = const()[name = tensor("encoder_conv_module_0_sequential_2_conv_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8764224)))]; + tensor encoder_conv_module_0_sequential_2_conv_weight = const()[name = tensor("encoder_conv_module_0_sequential_2_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8766336)))]; + tensor encoder_conv_module_0_sequential_4_conv_weight = const()[name = tensor("encoder_conv_module_0_sequential_4_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9290688)))]; + tensor encoder_conv_module_0_sequential_5_running_var = const()[name = tensor("encoder_conv_module_0_sequential_5_running_var"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9307136)))]; + tensor encoder_conv_module_0_sequential_5_running_mean = const()[name = tensor("encoder_conv_module_0_sequential_5_running_mean"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9308224)))]; + tensor encoder_conv_module_0_sequential_5_bias = const()[name = tensor("encoder_conv_module_0_sequential_5_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9309312)))]; + tensor encoder_conv_module_0_sequential_5_weight = const()[name = tensor("encoder_conv_module_0_sequential_5_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9310400)))]; + tensor encoder_conv_module_0_sequential_7_conv_bias = const()[name = tensor("encoder_conv_module_0_sequential_7_conv_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9311488)))]; + tensor encoder_conv_module_0_sequential_7_conv_weight = const()[name = tensor("encoder_conv_module_0_sequential_7_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9312576)))]; + tensor encoder_ffn2_0_module_sequential_0_bias = const()[name = tensor("encoder_ffn2_0_module_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9574784)))]; + tensor encoder_ffn2_0_module_sequential_0_weight = const()[name = tensor("encoder_ffn2_0_module_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9575872)))]; + tensor encoder_ffn2_0_module_sequential_1_linear_bias = const()[name = tensor("encoder_ffn2_0_module_sequential_1_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9576960)))]; + tensor encoder_ffn2_0_module_sequential_1_linear_weight = const()[name = tensor("encoder_ffn2_0_module_sequential_1_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9581120)))]; + tensor encoder_ffn2_0_module_sequential_4_linear_bias = const()[name = tensor("encoder_ffn2_0_module_sequential_4_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10629760)))]; + tensor encoder_ffn2_0_module_sequential_4_linear_weight = const()[name = tensor("encoder_ffn2_0_module_sequential_4_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10630848)))]; + tensor encoder_layer_norm_0_bias = const()[name = tensor("encoder_layer_norm_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11679488)))]; + tensor encoder_layer_norm_0_weight = const()[name = tensor("encoder_layer_norm_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11680576)))]; + tensor encoder_ffn1_1_module_sequential_0_bias = const()[name = tensor("encoder_ffn1_1_module_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11681664)))]; + tensor encoder_ffn1_1_module_sequential_0_weight = const()[name = tensor("encoder_ffn1_1_module_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11682752)))]; + tensor encoder_ffn1_1_module_sequential_1_linear_bias = const()[name = tensor("encoder_ffn1_1_module_sequential_1_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11683840)))]; + tensor encoder_ffn1_1_module_sequential_1_linear_weight = const()[name = tensor("encoder_ffn1_1_module_sequential_1_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11688000)))]; + tensor encoder_ffn1_1_module_sequential_4_linear_bias = const()[name = tensor("encoder_ffn1_1_module_sequential_4_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(12736640)))]; + tensor encoder_ffn1_1_module_sequential_4_linear_weight = const()[name = tensor("encoder_ffn1_1_module_sequential_4_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(12737728)))]; + tensor encoder_ret_lns_1_bias = const()[name = tensor("encoder_ret_lns_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(13786368)))]; + tensor encoder_ret_lns_1_weight = const()[name = tensor("encoder_ret_lns_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(13787456)))]; + tensor encoder_q_proj_1_bias = const()[name = tensor("encoder_q_proj_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(13788544)))]; + tensor encoder_q_proj_1_weight = const()[name = tensor("encoder_q_proj_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(13789632)))]; + tensor encoder_k_proj_1_bias = const()[name = tensor("encoder_k_proj_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14051840)))]; + tensor encoder_k_proj_1_weight = const()[name = tensor("encoder_k_proj_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14052928)))]; + tensor encoder_v_proj_1_bias = const()[name = tensor("encoder_v_proj_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14315136)))]; + tensor encoder_v_proj_1_weight = const()[name = tensor("encoder_v_proj_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14316224)))]; + tensor encoder_g_proj_1_bias = const()[name = tensor("encoder_g_proj_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14578432)))]; + tensor encoder_g_proj_1_weight = const()[name = tensor("encoder_g_proj_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14579520)))]; + tensor encoder_out_proj_1_bias = const()[name = tensor("encoder_out_proj_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14841728)))]; + tensor encoder_out_proj_1_weight = const()[name = tensor("encoder_out_proj_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14842816)))]; + tensor encoder_conv_module_1_sequential_0_bias = const()[name = tensor("encoder_conv_module_1_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15105024)))]; + tensor encoder_conv_module_1_sequential_0_weight = const()[name = tensor("encoder_conv_module_1_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15106112)))]; + tensor encoder_conv_module_1_sequential_2_conv_bias = const()[name = tensor("encoder_conv_module_1_sequential_2_conv_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15107200)))]; + tensor encoder_conv_module_1_sequential_2_conv_weight = const()[name = tensor("encoder_conv_module_1_sequential_2_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15109312)))]; + tensor encoder_conv_module_1_sequential_4_conv_weight = const()[name = tensor("encoder_conv_module_1_sequential_4_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15633664)))]; + tensor encoder_conv_module_1_sequential_5_running_var = const()[name = tensor("encoder_conv_module_1_sequential_5_running_var"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15650112)))]; + tensor encoder_conv_module_1_sequential_5_running_mean = const()[name = tensor("encoder_conv_module_1_sequential_5_running_mean"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15651200)))]; + tensor encoder_conv_module_1_sequential_5_bias = const()[name = tensor("encoder_conv_module_1_sequential_5_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15652288)))]; + tensor encoder_conv_module_1_sequential_5_weight = const()[name = tensor("encoder_conv_module_1_sequential_5_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15653376)))]; + tensor encoder_conv_module_1_sequential_7_conv_bias = const()[name = tensor("encoder_conv_module_1_sequential_7_conv_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15654464)))]; + tensor encoder_conv_module_1_sequential_7_conv_weight = const()[name = tensor("encoder_conv_module_1_sequential_7_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15655552)))]; + tensor encoder_ffn2_1_module_sequential_0_bias = const()[name = tensor("encoder_ffn2_1_module_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15917760)))]; + tensor encoder_ffn2_1_module_sequential_0_weight = const()[name = tensor("encoder_ffn2_1_module_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15918848)))]; + tensor encoder_ffn2_1_module_sequential_1_linear_bias = const()[name = tensor("encoder_ffn2_1_module_sequential_1_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15919936)))]; + tensor encoder_ffn2_1_module_sequential_1_linear_weight = const()[name = tensor("encoder_ffn2_1_module_sequential_1_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15924096)))]; + tensor encoder_ffn2_1_module_sequential_4_linear_bias = const()[name = tensor("encoder_ffn2_1_module_sequential_4_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(16972736)))]; + tensor encoder_ffn2_1_module_sequential_4_linear_weight = const()[name = tensor("encoder_ffn2_1_module_sequential_4_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(16973824)))]; + tensor encoder_layer_norm_1_bias = const()[name = tensor("encoder_layer_norm_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18022464)))]; + tensor encoder_layer_norm_1_weight = const()[name = tensor("encoder_layer_norm_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18023552)))]; + tensor encoder_ffn1_2_module_sequential_0_bias = const()[name = tensor("encoder_ffn1_2_module_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18024640)))]; + tensor encoder_ffn1_2_module_sequential_0_weight = const()[name = tensor("encoder_ffn1_2_module_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18025728)))]; + tensor encoder_ffn1_2_module_sequential_1_linear_bias = const()[name = tensor("encoder_ffn1_2_module_sequential_1_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18026816)))]; + tensor encoder_ffn1_2_module_sequential_1_linear_weight = const()[name = tensor("encoder_ffn1_2_module_sequential_1_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18030976)))]; + tensor encoder_ffn1_2_module_sequential_4_linear_bias = const()[name = tensor("encoder_ffn1_2_module_sequential_4_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(19079616)))]; + tensor encoder_ffn1_2_module_sequential_4_linear_weight = const()[name = tensor("encoder_ffn1_2_module_sequential_4_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(19080704)))]; + tensor encoder_ret_lns_2_bias = const()[name = tensor("encoder_ret_lns_2_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20129344)))]; + tensor encoder_ret_lns_2_weight = const()[name = tensor("encoder_ret_lns_2_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20130432)))]; + tensor encoder_q_proj_2_bias = const()[name = tensor("encoder_q_proj_2_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20131520)))]; + tensor encoder_q_proj_2_weight = const()[name = tensor("encoder_q_proj_2_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20132608)))]; + tensor encoder_k_proj_2_bias = const()[name = tensor("encoder_k_proj_2_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20394816)))]; + tensor encoder_k_proj_2_weight = const()[name = tensor("encoder_k_proj_2_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20395904)))]; + tensor encoder_v_proj_2_bias = const()[name = tensor("encoder_v_proj_2_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20658112)))]; + tensor encoder_v_proj_2_weight = const()[name = tensor("encoder_v_proj_2_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20659200)))]; + tensor encoder_g_proj_2_bias = const()[name = tensor("encoder_g_proj_2_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20921408)))]; + tensor encoder_g_proj_2_weight = const()[name = tensor("encoder_g_proj_2_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20922496)))]; + tensor encoder_out_proj_2_bias = const()[name = tensor("encoder_out_proj_2_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21184704)))]; + tensor encoder_out_proj_2_weight = const()[name = tensor("encoder_out_proj_2_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21185792)))]; + tensor encoder_conv_module_2_sequential_0_bias = const()[name = tensor("encoder_conv_module_2_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21448000)))]; + tensor encoder_conv_module_2_sequential_0_weight = const()[name = tensor("encoder_conv_module_2_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21449088)))]; + tensor encoder_conv_module_2_sequential_2_conv_bias = const()[name = tensor("encoder_conv_module_2_sequential_2_conv_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21450176)))]; + tensor encoder_conv_module_2_sequential_2_conv_weight = const()[name = tensor("encoder_conv_module_2_sequential_2_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21452288)))]; + tensor encoder_conv_module_2_sequential_4_conv_weight = const()[name = tensor("encoder_conv_module_2_sequential_4_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21976640)))]; + tensor encoder_conv_module_2_sequential_5_running_var = const()[name = tensor("encoder_conv_module_2_sequential_5_running_var"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21993088)))]; + tensor encoder_conv_module_2_sequential_5_running_mean = const()[name = tensor("encoder_conv_module_2_sequential_5_running_mean"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21994176)))]; + tensor encoder_conv_module_2_sequential_5_bias = const()[name = tensor("encoder_conv_module_2_sequential_5_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21995264)))]; + tensor encoder_conv_module_2_sequential_5_weight = const()[name = tensor("encoder_conv_module_2_sequential_5_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21996352)))]; + tensor encoder_conv_module_2_sequential_7_conv_bias = const()[name = tensor("encoder_conv_module_2_sequential_7_conv_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21997440)))]; + tensor encoder_conv_module_2_sequential_7_conv_weight = const()[name = tensor("encoder_conv_module_2_sequential_7_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21998528)))]; + tensor encoder_ffn2_2_module_sequential_0_bias = const()[name = tensor("encoder_ffn2_2_module_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(22260736)))]; + tensor encoder_ffn2_2_module_sequential_0_weight = const()[name = tensor("encoder_ffn2_2_module_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(22261824)))]; + tensor encoder_ffn2_2_module_sequential_1_linear_bias = const()[name = tensor("encoder_ffn2_2_module_sequential_1_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(22262912)))]; + tensor encoder_ffn2_2_module_sequential_1_linear_weight = const()[name = tensor("encoder_ffn2_2_module_sequential_1_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(22267072)))]; + tensor encoder_ffn2_2_module_sequential_4_linear_bias = const()[name = tensor("encoder_ffn2_2_module_sequential_4_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(23315712)))]; + tensor encoder_ffn2_2_module_sequential_4_linear_weight = const()[name = tensor("encoder_ffn2_2_module_sequential_4_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(23316800)))]; + tensor encoder_layer_norm_2_bias = const()[name = tensor("encoder_layer_norm_2_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(24365440)))]; + tensor encoder_layer_norm_2_weight = const()[name = tensor("encoder_layer_norm_2_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(24366528)))]; + tensor encoder_ffn1_3_module_sequential_0_bias = const()[name = tensor("encoder_ffn1_3_module_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(24367616)))]; + tensor encoder_ffn1_3_module_sequential_0_weight = const()[name = tensor("encoder_ffn1_3_module_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(24368704)))]; + tensor encoder_ffn1_3_module_sequential_1_linear_bias = const()[name = tensor("encoder_ffn1_3_module_sequential_1_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(24369792)))]; + tensor encoder_ffn1_3_module_sequential_1_linear_weight = const()[name = tensor("encoder_ffn1_3_module_sequential_1_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(24373952)))]; + tensor encoder_ffn1_3_module_sequential_4_linear_bias = const()[name = tensor("encoder_ffn1_3_module_sequential_4_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(25422592)))]; + tensor encoder_ffn1_3_module_sequential_4_linear_weight = const()[name = tensor("encoder_ffn1_3_module_sequential_4_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(25423680)))]; + tensor encoder_ret_lns_3_bias = const()[name = tensor("encoder_ret_lns_3_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(26472320)))]; + tensor encoder_ret_lns_3_weight = const()[name = tensor("encoder_ret_lns_3_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(26473408)))]; + tensor encoder_q_proj_3_bias = const()[name = tensor("encoder_q_proj_3_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(26474496)))]; + tensor encoder_q_proj_3_weight = const()[name = tensor("encoder_q_proj_3_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(26475584)))]; + tensor encoder_k_proj_3_bias = const()[name = tensor("encoder_k_proj_3_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(26737792)))]; + tensor encoder_k_proj_3_weight = const()[name = tensor("encoder_k_proj_3_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(26738880)))]; + tensor encoder_v_proj_3_bias = const()[name = tensor("encoder_v_proj_3_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(27001088)))]; + tensor encoder_v_proj_3_weight = const()[name = tensor("encoder_v_proj_3_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(27002176)))]; + tensor encoder_g_proj_3_bias = const()[name = tensor("encoder_g_proj_3_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(27264384)))]; + tensor encoder_g_proj_3_weight = const()[name = tensor("encoder_g_proj_3_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(27265472)))]; + tensor encoder_out_proj_3_bias = const()[name = tensor("encoder_out_proj_3_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(27527680)))]; + tensor encoder_out_proj_3_weight = const()[name = tensor("encoder_out_proj_3_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(27528768)))]; + tensor encoder_conv_module_3_sequential_0_bias = const()[name = tensor("encoder_conv_module_3_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(27790976)))]; + tensor encoder_conv_module_3_sequential_0_weight = const()[name = tensor("encoder_conv_module_3_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(27792064)))]; + tensor encoder_conv_module_3_sequential_2_conv_bias = const()[name = tensor("encoder_conv_module_3_sequential_2_conv_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(27793152)))]; + tensor encoder_conv_module_3_sequential_2_conv_weight = const()[name = tensor("encoder_conv_module_3_sequential_2_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(27795264)))]; + tensor encoder_conv_module_3_sequential_4_conv_weight = const()[name = tensor("encoder_conv_module_3_sequential_4_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28319616)))]; + tensor encoder_conv_module_3_sequential_5_running_var = const()[name = tensor("encoder_conv_module_3_sequential_5_running_var"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28336064)))]; + tensor encoder_conv_module_3_sequential_5_running_mean = const()[name = tensor("encoder_conv_module_3_sequential_5_running_mean"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28337152)))]; + tensor encoder_conv_module_3_sequential_5_bias = const()[name = tensor("encoder_conv_module_3_sequential_5_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28338240)))]; + tensor encoder_conv_module_3_sequential_5_weight = const()[name = tensor("encoder_conv_module_3_sequential_5_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28339328)))]; + tensor encoder_conv_module_3_sequential_7_conv_bias = const()[name = tensor("encoder_conv_module_3_sequential_7_conv_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28340416)))]; + tensor encoder_conv_module_3_sequential_7_conv_weight = const()[name = tensor("encoder_conv_module_3_sequential_7_conv_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28341504)))]; + tensor encoder_ffn2_3_module_sequential_0_bias = const()[name = tensor("encoder_ffn2_3_module_sequential_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28603712)))]; + tensor encoder_ffn2_3_module_sequential_0_weight = const()[name = tensor("encoder_ffn2_3_module_sequential_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28604800)))]; + tensor encoder_ffn2_3_module_sequential_1_linear_bias = const()[name = tensor("encoder_ffn2_3_module_sequential_1_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28605888)))]; + tensor encoder_ffn2_3_module_sequential_1_linear_weight = const()[name = tensor("encoder_ffn2_3_module_sequential_1_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(28610048)))]; + tensor encoder_ffn2_3_module_sequential_4_linear_bias = const()[name = tensor("encoder_ffn2_3_module_sequential_4_linear_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(29658688)))]; + tensor encoder_ffn2_3_module_sequential_4_linear_weight = const()[name = tensor("encoder_ffn2_3_module_sequential_4_linear_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(29659776)))]; + tensor encoder_layer_norm_3_bias = const()[name = tensor("encoder_layer_norm_3_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(30708416)))]; + tensor encoder_layer_norm_3_weight = const()[name = tensor("encoder_layer_norm_3_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(30709504)))]; + tensor decoder__causal_mask = const()[name = tensor("decoder__causal_mask"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(30710592)))]; + tensor decoder_convert_bias = const()[name = tensor("decoder_convert_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(30710720)))]; + tensor decoder_convert_weight = const()[name = tensor("decoder_convert_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(30711808)))]; + tensor decoder_q_proj_0_bias = const()[name = tensor("decoder_q_proj_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(31236160)))]; + tensor decoder_q_proj_0_weight = const()[name = tensor("decoder_q_proj_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(31237248)))]; + tensor decoder_k_proj_0_bias = const()[name = tensor("decoder_k_proj_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(31499456)))]; + tensor decoder_k_proj_0_weight = const()[name = tensor("decoder_k_proj_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(31500544)))]; + tensor decoder_v_proj_0_bias = const()[name = tensor("decoder_v_proj_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(31762752)))]; + tensor decoder_v_proj_0_weight = const()[name = tensor("decoder_v_proj_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(31763840)))]; + tensor decoder_g_proj_0_bias = const()[name = tensor("decoder_g_proj_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(32026048)))]; + tensor decoder_g_proj_0_weight = const()[name = tensor("decoder_g_proj_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(32027136)))]; + tensor decoder_out_proj_0_bias = const()[name = tensor("decoder_out_proj_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(32289344)))]; + tensor decoder_out_proj_0_weight = const()[name = tensor("decoder_out_proj_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(32290432)))]; + tensor decoder_norm11_0_bias = const()[name = tensor("decoder_norm11_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(32552640)))]; + tensor decoder_norm11_0_weight = const()[name = tensor("decoder_norm11_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(32553728)))]; + tensor decoder_self_attn2_0_out_proj_bias = const()[name = tensor("decoder_self_attn2_0_out_proj_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(32554816)))]; + tensor decoder_self_attn2_0_out_proj_weight = const()[name = tensor("decoder_self_attn2_0_out_proj_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(32555904)))]; + tensor decoder_self_attn2_0_in_proj_bias = const()[name = tensor("decoder_self_attn2_0_in_proj_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(32818112)))]; + tensor decoder_self_attn2_0_in_proj_weight = const()[name = tensor("decoder_self_attn2_0_in_proj_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(32821248)))]; + tensor decoder_norm21_0_bias = const()[name = tensor("decoder_norm21_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(33607744)))]; + tensor decoder_norm21_0_weight = const()[name = tensor("decoder_norm21_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(33608832)))]; + tensor decoder_linear1_0_bias = const()[name = tensor("decoder_linear1_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(33609920)))]; + tensor decoder_linear1_0_weight = const()[name = tensor("decoder_linear1_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(33618176)))]; + tensor decoder_linear2_0_bias = const()[name = tensor("decoder_linear2_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(35715392)))]; + tensor decoder_linear2_0_weight = const()[name = tensor("decoder_linear2_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(35716480)))]; + tensor decoder_norm22_0_bias = const()[name = tensor("decoder_norm22_0_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(37813696)))]; + tensor decoder_norm22_0_weight = const()[name = tensor("decoder_norm22_0_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(37814784)))]; + tensor decoder_q_proj_1_bias = const()[name = tensor("decoder_q_proj_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(37815872)))]; + tensor decoder_q_proj_1_weight = const()[name = tensor("decoder_q_proj_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(37816960)))]; + tensor decoder_k_proj_1_bias = const()[name = tensor("decoder_k_proj_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(38079168)))]; + tensor decoder_k_proj_1_weight = const()[name = tensor("decoder_k_proj_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(38080256)))]; + tensor decoder_v_proj_1_bias = const()[name = tensor("decoder_v_proj_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(38342464)))]; + tensor decoder_v_proj_1_weight = const()[name = tensor("decoder_v_proj_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(38343552)))]; + tensor decoder_g_proj_1_bias = const()[name = tensor("decoder_g_proj_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(38605760)))]; + tensor decoder_g_proj_1_weight = const()[name = tensor("decoder_g_proj_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(38606848)))]; + tensor decoder_out_proj_1_bias = const()[name = tensor("decoder_out_proj_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(38869056)))]; + tensor decoder_out_proj_1_weight = const()[name = tensor("decoder_out_proj_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(38870144)))]; + tensor decoder_norm11_1_bias = const()[name = tensor("decoder_norm11_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(39132352)))]; + tensor decoder_norm11_1_weight = const()[name = tensor("decoder_norm11_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(39133440)))]; + tensor decoder_self_attn2_1_out_proj_bias = const()[name = tensor("decoder_self_attn2_1_out_proj_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(39134528)))]; + tensor decoder_self_attn2_1_out_proj_weight = const()[name = tensor("decoder_self_attn2_1_out_proj_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(39135616)))]; + tensor decoder_self_attn2_1_in_proj_bias = const()[name = tensor("decoder_self_attn2_1_in_proj_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(39397824)))]; + tensor decoder_self_attn2_1_in_proj_weight = const()[name = tensor("decoder_self_attn2_1_in_proj_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(39400960)))]; + tensor decoder_norm21_1_bias = const()[name = tensor("decoder_norm21_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(40187456)))]; + tensor decoder_norm21_1_weight = const()[name = tensor("decoder_norm21_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(40188544)))]; + tensor decoder_linear1_1_bias = const()[name = tensor("decoder_linear1_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(40189632)))]; + tensor decoder_linear1_1_weight = const()[name = tensor("decoder_linear1_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(40197888)))]; + tensor decoder_linear2_1_bias = const()[name = tensor("decoder_linear2_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(42295104)))]; + tensor decoder_linear2_1_weight = const()[name = tensor("decoder_linear2_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(42296192)))]; + tensor decoder_norm22_1_bias = const()[name = tensor("decoder_norm22_1_bias"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(44393408)))]; + tensor decoder_norm22_1_weight = const()[name = tensor("decoder_norm22_1_weight"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(44394496)))]; + tensor var_11 = const()[name = tensor("op_11"), val = tensor(0x1.5798eep-27)]; + tensor var_17 = const()[name = tensor("op_17"), val = tensor(0x1.0c6f7ap-20)]; + tensor var_20 = const()[name = tensor("op_20"), val = tensor(2)]; + tensor var_23 = const()[name = tensor("op_23"), val = tensor(0)]; + tensor var_26 = const()[name = tensor("op_26"), val = tensor(1)]; + tensor var_28 = const()[name = tensor("op_28"), val = tensor(0x1.4f8b58p-17)]; + tensor var_29 = const()[name = tensor("op_29"), val = tensor(true)]; + tensor input_1 = linear(bias = encoder_input_projection_linear_bias, weight = encoder_input_projection_linear_weight, x = features)[name = tensor("linear_0")]; + tensor input_3_axes_0 = const()[name = tensor("input_3_axes_0"), val = tensor([-1])]; + tensor input_3 = layer_norm(axes = input_3_axes_0, beta = encoder_pre_layer_norm_bias, epsilon = var_28, gamma = encoder_pre_layer_norm_weight, x = input_1)[name = tensor("input_3")]; + tensor input_5_axes_0 = const()[name = tensor("input_5_axes_0"), val = tensor([-1])]; + tensor input_5 = layer_norm(axes = input_5_axes_0, beta = encoder_ffn1_0_module_sequential_0_bias, epsilon = var_28, gamma = encoder_ffn1_0_module_sequential_0_weight, x = input_3)[name = tensor("input_5")]; + tensor inputs_1 = linear(bias = encoder_ffn1_0_module_sequential_1_linear_bias, weight = encoder_ffn1_0_module_sequential_1_linear_weight, x = input_5)[name = tensor("linear_1")]; + tensor input_7 = silu(x = inputs_1)[name = tensor("input_7")]; + tensor input_11 = linear(bias = encoder_ffn1_0_module_sequential_4_linear_bias, weight = encoder_ffn1_0_module_sequential_4_linear_weight, x = input_7)[name = tensor("linear_2")]; + tensor var_147 = const()[name = tensor("op_147"), val = tensor(0x1p-1)]; + tensor var_148 = mul(x = input_11, y = var_147)[name = tensor("op_148")]; + tensor input_13 = add(x = var_148, y = input_3)[name = tensor("input_13")]; + tensor x_1_axes_0 = const()[name = tensor("x_1_axes_0"), val = tensor([-1])]; + tensor x_1 = layer_norm(axes = x_1_axes_0, beta = encoder_ret_lns_0_bias, epsilon = var_28, gamma = encoder_ret_lns_0_weight, x = input_13)[name = tensor("x_1")]; + tensor prev_kv_1_begin_0 = const()[name = tensor("prev_kv_1_begin_0"), val = tensor([0, 0, 0, 0, 0])]; + tensor prev_kv_1_end_0 = const()[name = tensor("prev_kv_1_end_0"), val = tensor([1, 1, 4, 64, 64])]; + tensor prev_kv_1_end_mask_0 = const()[name = tensor("prev_kv_1_end_mask_0"), val = tensor([false, true, true, true, true])]; + tensor prev_kv_1_squeeze_mask_0 = const()[name = tensor("prev_kv_1_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; + tensor prev_kv_1 = slice_by_index(begin = prev_kv_1_begin_0, end = prev_kv_1_end_0, end_mask = prev_kv_1_end_mask_0, squeeze_mask = prev_kv_1_squeeze_mask_0, x = enc_kv)[name = tensor("prev_kv_1")]; + tensor prev_scale_1_begin_0 = const()[name = tensor("prev_scale_1_begin_0"), val = tensor([0, 0])]; + tensor prev_scale_1_end_0 = const()[name = tensor("prev_scale_1_end_0"), val = tensor([1, 1])]; + tensor prev_scale_1_end_mask_0 = const()[name = tensor("prev_scale_1_end_mask_0"), val = tensor([false, true])]; + tensor prev_scale_1_squeeze_mask_0 = const()[name = tensor("prev_scale_1_squeeze_mask_0"), val = tensor([true, false])]; + tensor prev_scale_1 = slice_by_index(begin = prev_scale_1_begin_0, end = prev_scale_1_end_0, end_mask = prev_scale_1_end_mask_0, squeeze_mask = prev_scale_1_squeeze_mask_0, x = enc_scale)[name = tensor("prev_scale_1")]; + tensor var_162 = linear(bias = encoder_q_proj_0_bias, weight = encoder_q_proj_0_weight, x = x_1)[name = tensor("linear_3")]; + tensor var_163 = const()[name = tensor("op_163"), val = tensor([1, 4, 4, 64])]; + tensor var_164 = reshape(shape = var_163, x = var_162)[name = tensor("op_164")]; + tensor q_1_perm_0 = const()[name = tensor("q_1_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_168 = linear(bias = encoder_k_proj_0_bias, weight = encoder_k_proj_0_weight, x = x_1)[name = tensor("linear_4")]; + tensor var_169 = const()[name = tensor("op_169"), val = tensor(0x1p-3)]; + tensor var_170 = mul(x = var_168, y = var_169)[name = tensor("op_170")]; + tensor var_171 = const()[name = tensor("op_171"), val = tensor([1, 4, 4, 64])]; + tensor var_172 = reshape(shape = var_171, x = var_170)[name = tensor("op_172")]; + tensor k_1_perm_0 = const()[name = tensor("k_1_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_176 = linear(bias = encoder_v_proj_0_bias, weight = encoder_v_proj_0_weight, x = x_1)[name = tensor("linear_5")]; + tensor var_177 = const()[name = tensor("op_177"), val = tensor([1, 4, 4, 64])]; + tensor var_178 = reshape(shape = var_177, x = var_176)[name = tensor("op_178")]; + tensor v_1_perm_0 = const()[name = tensor("v_1_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor input_17 = linear(bias = encoder_g_proj_0_bias, weight = encoder_g_proj_0_weight, x = x_1)[name = tensor("linear_6")]; + tensor sqrt_s0_1 = sqrt(x = prev_scale_1)[name = tensor("sqrt_s0_1")]; + tensor s_t_1 = add(x = prev_scale_1, y = encoder__t_index)[name = tensor("s_t_1")]; + tensor sqrt_s_t_1 = sqrt(x = s_t_1)[name = tensor("sqrt_s_t_1")]; + tensor qk_1_transpose_x_1 = const()[name = tensor("qk_1_transpose_x_1"), val = tensor(false)]; + tensor qk_1_transpose_y_1 = const()[name = tensor("qk_1_transpose_y_1"), val = tensor(true)]; + tensor k_1 = transpose(perm = k_1_perm_0, x = var_172)[name = tensor("transpose_57")]; + tensor q_1 = transpose(perm = q_1_perm_0, x = var_164)[name = tensor("transpose_58")]; + tensor qk_1 = matmul(transpose_x = qk_1_transpose_x_1, transpose_y = qk_1_transpose_y_1, x = q_1, y = k_1)[name = tensor("qk_1")]; + tensor var_188 = const()[name = tensor("op_188"), val = tensor([4, 1])]; + tensor var_189 = reshape(shape = var_188, x = sqrt_s_t_1)[name = tensor("op_189")]; + tensor M_1 = real_div(x = encoder__causal_mask, y = var_189)[name = tensor("M_1")]; + tensor var_191 = mul(x = qk_1, y = M_1)[name = tensor("op_191")]; + tensor inner_1_transpose_x_0 = const()[name = tensor("inner_1_transpose_x_0"), val = tensor(false)]; + tensor inner_1_transpose_y_0 = const()[name = tensor("inner_1_transpose_y_0"), val = tensor(false)]; + tensor v_1 = transpose(perm = v_1_perm_0, x = var_178)[name = tensor("transpose_56")]; + tensor inner_1 = matmul(transpose_x = inner_1_transpose_x_0, transpose_y = inner_1_transpose_y_0, x = var_191, y = v_1)[name = tensor("inner_1")]; + tensor var_193_transpose_x_0 = const()[name = tensor("op_193_transpose_x_0"), val = tensor(false)]; + tensor var_193_transpose_y_0 = const()[name = tensor("op_193_transpose_y_0"), val = tensor(false)]; + tensor var_193 = matmul(transpose_x = var_193_transpose_x_0, transpose_y = var_193_transpose_y_0, x = q_1, y = prev_kv_1)[name = tensor("op_193")]; + tensor var_194 = real_div(x = sqrt_s0_1, y = sqrt_s_t_1)[name = tensor("op_194")]; + tensor var_195 = const()[name = tensor("op_195"), val = tensor([1, 1, 4, 1])]; + tensor var_196 = reshape(shape = var_195, x = var_194)[name = tensor("op_196")]; + tensor cross_1 = mul(x = var_193, y = var_196)[name = tensor("cross_1")]; + tensor out_1 = add(x = inner_1, y = cross_1)[name = tensor("out_1")]; + tensor var_199 = mul(x = prev_kv_1, y = sqrt_s0_1)[name = tensor("op_199")]; + tensor var_201_transpose_x_1 = const()[name = tensor("op_201_transpose_x_1"), val = tensor(true)]; + tensor var_201_transpose_y_1 = const()[name = tensor("op_201_transpose_y_1"), val = tensor(false)]; + tensor var_201 = matmul(transpose_x = var_201_transpose_x_1, transpose_y = var_201_transpose_y_1, x = k_1, y = v_1)[name = tensor("op_201")]; + tensor new_kv_unnorm_1 = add(x = var_199, y = var_201)[name = tensor("new_kv_unnorm_1")]; + tensor var_203 = const()[name = tensor("op_203"), val = tensor(0x1p+2)]; + tensor new_scale_1 = add(x = prev_scale_1, y = var_203)[name = tensor("new_scale_1")]; + tensor var_205 = sqrt(x = new_scale_1)[name = tensor("op_205")]; + tensor var_206 = real_div(x = new_kv_unnorm_1, y = var_205)[name = tensor("op_206")]; + tensor var_207_perm_0 = const()[name = tensor("op_207_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor out_3_axes_0 = const()[name = tensor("out_3_axes_0"), val = tensor([-1])]; + tensor var_207 = transpose(perm = var_207_perm_0, x = out_1)[name = tensor("transpose_55")]; + tensor out_3 = layer_norm(axes = out_3_axes_0, epsilon = var_17, x = var_207)[name = tensor("out_3")]; + tensor var_211 = const()[name = tensor("op_211"), val = tensor([1, 4, 256])]; + tensor out_5 = reshape(shape = var_211, x = out_3)[name = tensor("out_5")]; + tensor var_213 = silu(x = input_17)[name = tensor("op_213")]; + tensor input_19 = mul(x = var_213, y = out_5)[name = tensor("input_19")]; + tensor ret_out_1 = linear(bias = encoder_out_proj_0_bias, weight = encoder_out_proj_0_weight, x = input_19)[name = tensor("linear_7")]; + tensor x_3 = add(x = input_13, y = ret_out_1)[name = tensor("x_3")]; + tensor window_1_begin_0 = const()[name = tensor("window_1_begin_0"), val = tensor([0, 0, 0, 0])]; + tensor window_1_end_0 = const()[name = tensor("window_1_end_0"), val = tensor([1, 1, 16, 256])]; + tensor window_1_end_mask_0 = const()[name = tensor("window_1_end_mask_0"), val = tensor([false, true, true, true])]; + tensor window_1_squeeze_mask_0 = const()[name = tensor("window_1_squeeze_mask_0"), val = tensor([true, false, false, false])]; + tensor window_1 = slice_by_index(begin = window_1_begin_0, end = window_1_end_0, end_mask = window_1_end_mask_0, squeeze_mask = window_1_squeeze_mask_0, x = enc_conv_cache)[name = tensor("window_1")]; + tensor var_221_begin_0 = const()[name = tensor("op_221_begin_0"), val = tensor([0, 0, 0])]; + tensor var_221_end_0 = const()[name = tensor("op_221_end_0"), val = tensor([1, 1, 256])]; + tensor var_221_end_mask_0 = const()[name = tensor("op_221_end_mask_0"), val = tensor([true, false, true])]; + tensor var_221 = slice_by_index(begin = var_221_begin_0, end = var_221_end_0, end_mask = var_221_end_mask_0, x = x_3)[name = tensor("op_221")]; + tensor var_224_begin_0 = const()[name = tensor("op_224_begin_0"), val = tensor([0, 1, 0])]; + tensor var_224_end_0 = const()[name = tensor("op_224_end_0"), val = tensor([1, 16, 256])]; + tensor var_224_end_mask_0 = const()[name = tensor("op_224_end_mask_0"), val = tensor([true, true, true])]; + tensor var_224 = slice_by_index(begin = var_224_begin_0, end = var_224_end_0, end_mask = var_224_end_mask_0, x = window_1)[name = tensor("op_224")]; + tensor window_3_interleave_0 = const()[name = tensor("window_3_interleave_0"), val = tensor(false)]; + tensor window_3 = concat(axis = var_26, interleave = window_3_interleave_0, values = (var_224, var_221))[name = tensor("window_3")]; + tensor var_229_begin_0 = const()[name = tensor("op_229_begin_0"), val = tensor([0, 1, 0])]; + tensor var_229_end_0 = const()[name = tensor("op_229_end_0"), val = tensor([1, 2, 256])]; + tensor var_229_end_mask_0 = const()[name = tensor("op_229_end_mask_0"), val = tensor([true, false, true])]; + tensor var_229 = slice_by_index(begin = var_229_begin_0, end = var_229_end_0, end_mask = var_229_end_mask_0, x = x_3)[name = tensor("op_229")]; + tensor var_232_begin_0 = const()[name = tensor("op_232_begin_0"), val = tensor([0, 1, 0])]; + tensor var_232_end_0 = const()[name = tensor("op_232_end_0"), val = tensor([1, 16, 256])]; + tensor var_232_end_mask_0 = const()[name = tensor("op_232_end_mask_0"), val = tensor([true, true, true])]; + tensor var_232 = slice_by_index(begin = var_232_begin_0, end = var_232_end_0, end_mask = var_232_end_mask_0, x = window_3)[name = tensor("op_232")]; + tensor window_5_interleave_0 = const()[name = tensor("window_5_interleave_0"), val = tensor(false)]; + tensor window_5 = concat(axis = var_26, interleave = window_5_interleave_0, values = (var_232, var_229))[name = tensor("window_5")]; + tensor var_237_begin_0 = const()[name = tensor("op_237_begin_0"), val = tensor([0, 2, 0])]; + tensor var_237_end_0 = const()[name = tensor("op_237_end_0"), val = tensor([1, 3, 256])]; + tensor var_237_end_mask_0 = const()[name = tensor("op_237_end_mask_0"), val = tensor([true, false, true])]; + tensor var_237 = slice_by_index(begin = var_237_begin_0, end = var_237_end_0, end_mask = var_237_end_mask_0, x = x_3)[name = tensor("op_237")]; + tensor var_240_begin_0 = const()[name = tensor("op_240_begin_0"), val = tensor([0, 1, 0])]; + tensor var_240_end_0 = const()[name = tensor("op_240_end_0"), val = tensor([1, 16, 256])]; + tensor var_240_end_mask_0 = const()[name = tensor("op_240_end_mask_0"), val = tensor([true, true, true])]; + tensor var_240 = slice_by_index(begin = var_240_begin_0, end = var_240_end_0, end_mask = var_240_end_mask_0, x = window_5)[name = tensor("op_240")]; + tensor window_7_interleave_0 = const()[name = tensor("window_7_interleave_0"), val = tensor(false)]; + tensor window_7 = concat(axis = var_26, interleave = window_7_interleave_0, values = (var_240, var_237))[name = tensor("window_7")]; + tensor var_245_begin_0 = const()[name = tensor("op_245_begin_0"), val = tensor([0, 3, 0])]; + tensor var_245_end_0 = const()[name = tensor("op_245_end_0"), val = tensor([1, 1, 256])]; + tensor var_245_end_mask_0 = const()[name = tensor("op_245_end_mask_0"), val = tensor([true, true, true])]; + tensor var_245 = slice_by_index(begin = var_245_begin_0, end = var_245_end_0, end_mask = var_245_end_mask_0, x = x_3)[name = tensor("op_245")]; + tensor var_248_begin_0 = const()[name = tensor("op_248_begin_0"), val = tensor([0, 1, 0])]; + tensor var_248_end_0 = const()[name = tensor("op_248_end_0"), val = tensor([1, 16, 256])]; + tensor var_248_end_mask_0 = const()[name = tensor("op_248_end_mask_0"), val = tensor([true, true, true])]; + tensor var_248 = slice_by_index(begin = var_248_begin_0, end = var_248_end_0, end_mask = var_248_end_mask_0, x = window_7)[name = tensor("op_248")]; + tensor window_9_interleave_0 = const()[name = tensor("window_9_interleave_0"), val = tensor(false)]; + tensor window_9 = concat(axis = var_26, interleave = window_9_interleave_0, values = (var_248, var_245))[name = tensor("window_9")]; + tensor input_21_interleave_0 = const()[name = tensor("input_21_interleave_0"), val = tensor(false)]; + tensor input_21 = concat(axis = var_23, interleave = input_21_interleave_0, values = (window_3, window_5, window_7, window_9))[name = tensor("input_21")]; + tensor x_5_axes_0 = const()[name = tensor("x_5_axes_0"), val = tensor([-1])]; + tensor x_5 = layer_norm(axes = x_5_axes_0, beta = encoder_conv_module_0_sequential_0_bias, epsilon = var_28, gamma = encoder_conv_module_0_sequential_0_weight, x = input_21)[name = tensor("x_5")]; + tensor input_23_perm_0 = const()[name = tensor("input_23_perm_0"), val = tensor([0, 2, 1])]; + tensor inputs_3_pad_type_0 = const()[name = tensor("inputs_3_pad_type_0"), val = tensor("valid")]; + tensor inputs_3_strides_0 = const()[name = tensor("inputs_3_strides_0"), val = tensor([1])]; + tensor inputs_3_pad_0 = const()[name = tensor("inputs_3_pad_0"), val = tensor([0, 0])]; + tensor inputs_3_dilations_0 = const()[name = tensor("inputs_3_dilations_0"), val = tensor([1])]; + tensor inputs_3_groups_0 = const()[name = tensor("inputs_3_groups_0"), val = tensor(1)]; + tensor input_23 = transpose(perm = input_23_perm_0, x = x_5)[name = tensor("transpose_54")]; + tensor inputs_3 = conv(bias = encoder_conv_module_0_sequential_2_conv_bias, dilations = inputs_3_dilations_0, groups = inputs_3_groups_0, pad = inputs_3_pad_0, pad_type = inputs_3_pad_type_0, strides = inputs_3_strides_0, weight = encoder_conv_module_0_sequential_2_conv_weight, x = input_23)[name = tensor("inputs_3")]; + tensor var_273_split_sizes_0 = const()[name = tensor("op_273_split_sizes_0"), val = tensor([256, 256])]; + tensor var_273_axis_0 = const()[name = tensor("op_273_axis_0"), val = tensor(1)]; + tensor var_273_0, tensor var_273_1 = split(axis = var_273_axis_0, split_sizes = var_273_split_sizes_0, x = inputs_3)[name = tensor("op_273")]; + tensor var_275 = sigmoid(x = var_273_1)[name = tensor("op_275")]; + tensor inputs_5 = mul(x = var_273_0, y = var_275)[name = tensor("inputs_5")]; + tensor outputs_aug_1_pad_type_0 = const()[name = tensor("outputs_aug_1_pad_type_0"), val = tensor("custom")]; + tensor outputs_aug_1_pad_0 = const()[name = tensor("outputs_aug_1_pad_0"), val = tensor([15, 15])]; + tensor outputs_aug_1_groups_0 = const()[name = tensor("outputs_aug_1_groups_0"), val = tensor(256)]; + tensor outputs_aug_1_strides_0 = const()[name = tensor("outputs_aug_1_strides_0"), val = tensor([1])]; + tensor outputs_aug_1_dilations_0 = const()[name = tensor("outputs_aug_1_dilations_0"), val = tensor([1])]; + tensor outputs_aug_1 = conv(dilations = outputs_aug_1_dilations_0, groups = outputs_aug_1_groups_0, pad = outputs_aug_1_pad_0, pad_type = outputs_aug_1_pad_type_0, strides = outputs_aug_1_strides_0, weight = encoder_conv_module_0_sequential_4_conv_weight, x = inputs_5)[name = tensor("outputs_aug_1")]; + tensor input_25_begin_0 = const()[name = tensor("input_25_begin_0"), val = tensor([0, 0, 0])]; + tensor input_25_end_0 = const()[name = tensor("input_25_end_0"), val = tensor([4, 256, 16])]; + tensor input_25_end_mask_0 = const()[name = tensor("input_25_end_mask_0"), val = tensor([true, true, false])]; + tensor input_25 = slice_by_index(begin = input_25_begin_0, end = input_25_end_0, end_mask = input_25_end_mask_0, x = outputs_aug_1)[name = tensor("input_25")]; + tensor inputs_7 = batch_norm(beta = encoder_conv_module_0_sequential_5_bias, epsilon = var_28, gamma = encoder_conv_module_0_sequential_5_weight, mean = encoder_conv_module_0_sequential_5_running_mean, variance = encoder_conv_module_0_sequential_5_running_var, x = input_25)[name = tensor("inputs_7")]; + tensor input_27 = silu(x = inputs_7)[name = tensor("input_27")]; + tensor input_29_pad_type_0 = const()[name = tensor("input_29_pad_type_0"), val = tensor("valid")]; + tensor input_29_strides_0 = const()[name = tensor("input_29_strides_0"), val = tensor([1])]; + tensor input_29_pad_0 = const()[name = tensor("input_29_pad_0"), val = tensor([0, 0])]; + tensor input_29_dilations_0 = const()[name = tensor("input_29_dilations_0"), val = tensor([1])]; + tensor input_29_groups_0 = const()[name = tensor("input_29_groups_0"), val = tensor(1)]; + tensor input_29 = conv(bias = encoder_conv_module_0_sequential_7_conv_bias, dilations = input_29_dilations_0, groups = input_29_groups_0, pad = input_29_pad_0, pad_type = input_29_pad_type_0, strides = input_29_strides_0, weight = encoder_conv_module_0_sequential_7_conv_weight, x = input_27)[name = tensor("input_29")]; + tensor conv_out_1_perm_0 = const()[name = tensor("conv_out_1_perm_0"), val = tensor([0, 2, 1])]; + tensor var_306_begin_0 = const()[name = tensor("op_306_begin_0"), val = tensor([0, -1, 0])]; + tensor var_306_end_0 = const()[name = tensor("op_306_end_0"), val = tensor([4, 16, 256])]; + tensor var_306_end_mask_0 = const()[name = tensor("op_306_end_mask_0"), val = tensor([true, true, true])]; + tensor conv_out_1 = transpose(perm = conv_out_1_perm_0, x = input_29)[name = tensor("transpose_53")]; + tensor var_306 = slice_by_index(begin = var_306_begin_0, end = var_306_end_0, end_mask = var_306_end_mask_0, x = conv_out_1)[name = tensor("op_306")]; + tensor var_308_perm_0 = const()[name = tensor("op_308_perm_0"), val = tensor([1, 0, 2])]; + tensor var_308 = transpose(perm = var_308_perm_0, x = var_306)[name = tensor("transpose_52")]; + tensor input_31 = add(x = x_3, y = var_308)[name = tensor("input_31")]; + tensor input_33_axes_0 = const()[name = tensor("input_33_axes_0"), val = tensor([-1])]; + tensor input_33 = layer_norm(axes = input_33_axes_0, beta = encoder_ffn2_0_module_sequential_0_bias, epsilon = var_28, gamma = encoder_ffn2_0_module_sequential_0_weight, x = input_31)[name = tensor("input_33")]; + tensor inputs_9 = linear(bias = encoder_ffn2_0_module_sequential_1_linear_bias, weight = encoder_ffn2_0_module_sequential_1_linear_weight, x = input_33)[name = tensor("linear_8")]; + tensor input_35 = silu(x = inputs_9)[name = tensor("input_35")]; + tensor input_39 = linear(bias = encoder_ffn2_0_module_sequential_4_linear_bias, weight = encoder_ffn2_0_module_sequential_4_linear_weight, x = input_35)[name = tensor("linear_9")]; + tensor var_331 = const()[name = tensor("op_331"), val = tensor(0x1p-1)]; + tensor var_332 = mul(x = input_39, y = var_331)[name = tensor("op_332")]; + tensor input_41 = add(x = var_332, y = input_31)[name = tensor("input_41")]; + tensor input_43_axes_0 = const()[name = tensor("input_43_axes_0"), val = tensor([-1])]; + tensor input_43 = layer_norm(axes = input_43_axes_0, beta = encoder_layer_norm_0_bias, epsilon = var_28, gamma = encoder_layer_norm_0_weight, x = input_41)[name = tensor("input_43")]; + tensor input_45_axes_0 = const()[name = tensor("input_45_axes_0"), val = tensor([-1])]; + tensor input_45 = layer_norm(axes = input_45_axes_0, beta = encoder_ffn1_1_module_sequential_0_bias, epsilon = var_28, gamma = encoder_ffn1_1_module_sequential_0_weight, x = input_43)[name = tensor("input_45")]; + tensor inputs_11 = linear(bias = encoder_ffn1_1_module_sequential_1_linear_bias, weight = encoder_ffn1_1_module_sequential_1_linear_weight, x = input_45)[name = tensor("linear_10")]; + tensor input_47 = silu(x = inputs_11)[name = tensor("input_47")]; + tensor input_51 = linear(bias = encoder_ffn1_1_module_sequential_4_linear_bias, weight = encoder_ffn1_1_module_sequential_4_linear_weight, x = input_47)[name = tensor("linear_11")]; + tensor var_361 = const()[name = tensor("op_361"), val = tensor(0x1p-1)]; + tensor var_362 = mul(x = input_51, y = var_361)[name = tensor("op_362")]; + tensor input_53 = add(x = var_362, y = input_43)[name = tensor("input_53")]; + tensor x_7_axes_0 = const()[name = tensor("x_7_axes_0"), val = tensor([-1])]; + tensor x_7 = layer_norm(axes = x_7_axes_0, beta = encoder_ret_lns_1_bias, epsilon = var_28, gamma = encoder_ret_lns_1_weight, x = input_53)[name = tensor("x_7")]; + tensor prev_kv_3_begin_0 = const()[name = tensor("prev_kv_3_begin_0"), val = tensor([1, 0, 0, 0, 0])]; + tensor prev_kv_3_end_0 = const()[name = tensor("prev_kv_3_end_0"), val = tensor([2, 1, 4, 64, 64])]; + tensor prev_kv_3_end_mask_0 = const()[name = tensor("prev_kv_3_end_mask_0"), val = tensor([false, true, true, true, true])]; + tensor prev_kv_3_squeeze_mask_0 = const()[name = tensor("prev_kv_3_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; + tensor prev_kv_3 = slice_by_index(begin = prev_kv_3_begin_0, end = prev_kv_3_end_0, end_mask = prev_kv_3_end_mask_0, squeeze_mask = prev_kv_3_squeeze_mask_0, x = enc_kv)[name = tensor("prev_kv_3")]; + tensor prev_scale_3_begin_0 = const()[name = tensor("prev_scale_3_begin_0"), val = tensor([1, 0])]; + tensor prev_scale_3_end_0 = const()[name = tensor("prev_scale_3_end_0"), val = tensor([2, 1])]; + tensor prev_scale_3_end_mask_0 = const()[name = tensor("prev_scale_3_end_mask_0"), val = tensor([false, true])]; + tensor prev_scale_3_squeeze_mask_0 = const()[name = tensor("prev_scale_3_squeeze_mask_0"), val = tensor([true, false])]; + tensor prev_scale_3 = slice_by_index(begin = prev_scale_3_begin_0, end = prev_scale_3_end_0, end_mask = prev_scale_3_end_mask_0, squeeze_mask = prev_scale_3_squeeze_mask_0, x = enc_scale)[name = tensor("prev_scale_3")]; + tensor var_376 = linear(bias = encoder_q_proj_1_bias, weight = encoder_q_proj_1_weight, x = x_7)[name = tensor("linear_12")]; + tensor var_377 = const()[name = tensor("op_377"), val = tensor([1, 4, 4, 64])]; + tensor var_378 = reshape(shape = var_377, x = var_376)[name = tensor("op_378")]; + tensor q_3_perm_0 = const()[name = tensor("q_3_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_382 = linear(bias = encoder_k_proj_1_bias, weight = encoder_k_proj_1_weight, x = x_7)[name = tensor("linear_13")]; + tensor var_383 = const()[name = tensor("op_383"), val = tensor(0x1p-3)]; + tensor var_384 = mul(x = var_382, y = var_383)[name = tensor("op_384")]; + tensor var_385 = const()[name = tensor("op_385"), val = tensor([1, 4, 4, 64])]; + tensor var_386 = reshape(shape = var_385, x = var_384)[name = tensor("op_386")]; + tensor k_3_perm_0 = const()[name = tensor("k_3_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_390 = linear(bias = encoder_v_proj_1_bias, weight = encoder_v_proj_1_weight, x = x_7)[name = tensor("linear_14")]; + tensor var_391 = const()[name = tensor("op_391"), val = tensor([1, 4, 4, 64])]; + tensor var_392 = reshape(shape = var_391, x = var_390)[name = tensor("op_392")]; + tensor v_3_perm_0 = const()[name = tensor("v_3_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor input_57 = linear(bias = encoder_g_proj_1_bias, weight = encoder_g_proj_1_weight, x = x_7)[name = tensor("linear_15")]; + tensor sqrt_s0_3 = sqrt(x = prev_scale_3)[name = tensor("sqrt_s0_3")]; + tensor s_t_3 = add(x = prev_scale_3, y = encoder__t_index)[name = tensor("s_t_3")]; + tensor sqrt_s_t_3 = sqrt(x = s_t_3)[name = tensor("sqrt_s_t_3")]; + tensor qk_3_transpose_x_1 = const()[name = tensor("qk_3_transpose_x_1"), val = tensor(false)]; + tensor qk_3_transpose_y_1 = const()[name = tensor("qk_3_transpose_y_1"), val = tensor(true)]; + tensor k_3 = transpose(perm = k_3_perm_0, x = var_386)[name = tensor("transpose_50")]; + tensor q_3 = transpose(perm = q_3_perm_0, x = var_378)[name = tensor("transpose_51")]; + tensor qk_3 = matmul(transpose_x = qk_3_transpose_x_1, transpose_y = qk_3_transpose_y_1, x = q_3, y = k_3)[name = tensor("qk_3")]; + tensor var_402 = const()[name = tensor("op_402"), val = tensor([4, 1])]; + tensor var_403 = reshape(shape = var_402, x = sqrt_s_t_3)[name = tensor("op_403")]; + tensor M_3 = real_div(x = encoder__causal_mask, y = var_403)[name = tensor("M_3")]; + tensor var_405 = mul(x = qk_3, y = M_3)[name = tensor("op_405")]; + tensor inner_3_transpose_x_0 = const()[name = tensor("inner_3_transpose_x_0"), val = tensor(false)]; + tensor inner_3_transpose_y_0 = const()[name = tensor("inner_3_transpose_y_0"), val = tensor(false)]; + tensor v_3 = transpose(perm = v_3_perm_0, x = var_392)[name = tensor("transpose_49")]; + tensor inner_3 = matmul(transpose_x = inner_3_transpose_x_0, transpose_y = inner_3_transpose_y_0, x = var_405, y = v_3)[name = tensor("inner_3")]; + tensor var_407_transpose_x_0 = const()[name = tensor("op_407_transpose_x_0"), val = tensor(false)]; + tensor var_407_transpose_y_0 = const()[name = tensor("op_407_transpose_y_0"), val = tensor(false)]; + tensor var_407 = matmul(transpose_x = var_407_transpose_x_0, transpose_y = var_407_transpose_y_0, x = q_3, y = prev_kv_3)[name = tensor("op_407")]; + tensor var_408 = real_div(x = sqrt_s0_3, y = sqrt_s_t_3)[name = tensor("op_408")]; + tensor var_409 = const()[name = tensor("op_409"), val = tensor([1, 1, 4, 1])]; + tensor var_410 = reshape(shape = var_409, x = var_408)[name = tensor("op_410")]; + tensor cross_3 = mul(x = var_407, y = var_410)[name = tensor("cross_3")]; + tensor out_7 = add(x = inner_3, y = cross_3)[name = tensor("out_7")]; + tensor var_413 = mul(x = prev_kv_3, y = sqrt_s0_3)[name = tensor("op_413")]; + tensor var_415_transpose_x_1 = const()[name = tensor("op_415_transpose_x_1"), val = tensor(true)]; + tensor var_415_transpose_y_1 = const()[name = tensor("op_415_transpose_y_1"), val = tensor(false)]; + tensor var_415 = matmul(transpose_x = var_415_transpose_x_1, transpose_y = var_415_transpose_y_1, x = k_3, y = v_3)[name = tensor("op_415")]; + tensor new_kv_unnorm_3 = add(x = var_413, y = var_415)[name = tensor("new_kv_unnorm_3")]; + tensor var_417 = const()[name = tensor("op_417"), val = tensor(0x1p+2)]; + tensor new_scale_3 = add(x = prev_scale_3, y = var_417)[name = tensor("new_scale_3")]; + tensor var_419 = sqrt(x = new_scale_3)[name = tensor("op_419")]; + tensor var_420 = real_div(x = new_kv_unnorm_3, y = var_419)[name = tensor("op_420")]; + tensor var_421_perm_0 = const()[name = tensor("op_421_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor out_9_axes_0 = const()[name = tensor("out_9_axes_0"), val = tensor([-1])]; + tensor var_421 = transpose(perm = var_421_perm_0, x = out_7)[name = tensor("transpose_48")]; + tensor out_9 = layer_norm(axes = out_9_axes_0, epsilon = var_17, x = var_421)[name = tensor("out_9")]; + tensor var_425 = const()[name = tensor("op_425"), val = tensor([1, 4, 256])]; + tensor out_11 = reshape(shape = var_425, x = out_9)[name = tensor("out_11")]; + tensor var_427 = silu(x = input_57)[name = tensor("op_427")]; + tensor input_59 = mul(x = var_427, y = out_11)[name = tensor("input_59")]; + tensor ret_out_3 = linear(bias = encoder_out_proj_1_bias, weight = encoder_out_proj_1_weight, x = input_59)[name = tensor("linear_16")]; + tensor x_9 = add(x = input_53, y = ret_out_3)[name = tensor("x_9")]; + tensor window_11_begin_0 = const()[name = tensor("window_11_begin_0"), val = tensor([1, 0, 0, 0])]; + tensor window_11_end_0 = const()[name = tensor("window_11_end_0"), val = tensor([2, 1, 16, 256])]; + tensor window_11_end_mask_0 = const()[name = tensor("window_11_end_mask_0"), val = tensor([false, true, true, true])]; + tensor window_11_squeeze_mask_0 = const()[name = tensor("window_11_squeeze_mask_0"), val = tensor([true, false, false, false])]; + tensor window_11 = slice_by_index(begin = window_11_begin_0, end = window_11_end_0, end_mask = window_11_end_mask_0, squeeze_mask = window_11_squeeze_mask_0, x = enc_conv_cache)[name = tensor("window_11")]; + tensor var_435_begin_0 = const()[name = tensor("op_435_begin_0"), val = tensor([0, 0, 0])]; + tensor var_435_end_0 = const()[name = tensor("op_435_end_0"), val = tensor([1, 1, 256])]; + tensor var_435_end_mask_0 = const()[name = tensor("op_435_end_mask_0"), val = tensor([true, false, true])]; + tensor var_435 = slice_by_index(begin = var_435_begin_0, end = var_435_end_0, end_mask = var_435_end_mask_0, x = x_9)[name = tensor("op_435")]; + tensor var_438_begin_0 = const()[name = tensor("op_438_begin_0"), val = tensor([0, 1, 0])]; + tensor var_438_end_0 = const()[name = tensor("op_438_end_0"), val = tensor([1, 16, 256])]; + tensor var_438_end_mask_0 = const()[name = tensor("op_438_end_mask_0"), val = tensor([true, true, true])]; + tensor var_438 = slice_by_index(begin = var_438_begin_0, end = var_438_end_0, end_mask = var_438_end_mask_0, x = window_11)[name = tensor("op_438")]; + tensor window_13_interleave_0 = const()[name = tensor("window_13_interleave_0"), val = tensor(false)]; + tensor window_13 = concat(axis = var_26, interleave = window_13_interleave_0, values = (var_438, var_435))[name = tensor("window_13")]; + tensor var_443_begin_0 = const()[name = tensor("op_443_begin_0"), val = tensor([0, 1, 0])]; + tensor var_443_end_0 = const()[name = tensor("op_443_end_0"), val = tensor([1, 2, 256])]; + tensor var_443_end_mask_0 = const()[name = tensor("op_443_end_mask_0"), val = tensor([true, false, true])]; + tensor var_443 = slice_by_index(begin = var_443_begin_0, end = var_443_end_0, end_mask = var_443_end_mask_0, x = x_9)[name = tensor("op_443")]; + tensor var_446_begin_0 = const()[name = tensor("op_446_begin_0"), val = tensor([0, 1, 0])]; + tensor var_446_end_0 = const()[name = tensor("op_446_end_0"), val = tensor([1, 16, 256])]; + tensor var_446_end_mask_0 = const()[name = tensor("op_446_end_mask_0"), val = tensor([true, true, true])]; + tensor var_446 = slice_by_index(begin = var_446_begin_0, end = var_446_end_0, end_mask = var_446_end_mask_0, x = window_13)[name = tensor("op_446")]; + tensor window_15_interleave_0 = const()[name = tensor("window_15_interleave_0"), val = tensor(false)]; + tensor window_15 = concat(axis = var_26, interleave = window_15_interleave_0, values = (var_446, var_443))[name = tensor("window_15")]; + tensor var_451_begin_0 = const()[name = tensor("op_451_begin_0"), val = tensor([0, 2, 0])]; + tensor var_451_end_0 = const()[name = tensor("op_451_end_0"), val = tensor([1, 3, 256])]; + tensor var_451_end_mask_0 = const()[name = tensor("op_451_end_mask_0"), val = tensor([true, false, true])]; + tensor var_451 = slice_by_index(begin = var_451_begin_0, end = var_451_end_0, end_mask = var_451_end_mask_0, x = x_9)[name = tensor("op_451")]; + tensor var_454_begin_0 = const()[name = tensor("op_454_begin_0"), val = tensor([0, 1, 0])]; + tensor var_454_end_0 = const()[name = tensor("op_454_end_0"), val = tensor([1, 16, 256])]; + tensor var_454_end_mask_0 = const()[name = tensor("op_454_end_mask_0"), val = tensor([true, true, true])]; + tensor var_454 = slice_by_index(begin = var_454_begin_0, end = var_454_end_0, end_mask = var_454_end_mask_0, x = window_15)[name = tensor("op_454")]; + tensor window_17_interleave_0 = const()[name = tensor("window_17_interleave_0"), val = tensor(false)]; + tensor window_17 = concat(axis = var_26, interleave = window_17_interleave_0, values = (var_454, var_451))[name = tensor("window_17")]; + tensor var_459_begin_0 = const()[name = tensor("op_459_begin_0"), val = tensor([0, 3, 0])]; + tensor var_459_end_0 = const()[name = tensor("op_459_end_0"), val = tensor([1, 1, 256])]; + tensor var_459_end_mask_0 = const()[name = tensor("op_459_end_mask_0"), val = tensor([true, true, true])]; + tensor var_459 = slice_by_index(begin = var_459_begin_0, end = var_459_end_0, end_mask = var_459_end_mask_0, x = x_9)[name = tensor("op_459")]; + tensor var_462_begin_0 = const()[name = tensor("op_462_begin_0"), val = tensor([0, 1, 0])]; + tensor var_462_end_0 = const()[name = tensor("op_462_end_0"), val = tensor([1, 16, 256])]; + tensor var_462_end_mask_0 = const()[name = tensor("op_462_end_mask_0"), val = tensor([true, true, true])]; + tensor var_462 = slice_by_index(begin = var_462_begin_0, end = var_462_end_0, end_mask = var_462_end_mask_0, x = window_17)[name = tensor("op_462")]; + tensor window_19_interleave_0 = const()[name = tensor("window_19_interleave_0"), val = tensor(false)]; + tensor window_19 = concat(axis = var_26, interleave = window_19_interleave_0, values = (var_462, var_459))[name = tensor("window_19")]; + tensor input_61_interleave_0 = const()[name = tensor("input_61_interleave_0"), val = tensor(false)]; + tensor input_61 = concat(axis = var_23, interleave = input_61_interleave_0, values = (window_13, window_15, window_17, window_19))[name = tensor("input_61")]; + tensor x_11_axes_0 = const()[name = tensor("x_11_axes_0"), val = tensor([-1])]; + tensor x_11 = layer_norm(axes = x_11_axes_0, beta = encoder_conv_module_1_sequential_0_bias, epsilon = var_28, gamma = encoder_conv_module_1_sequential_0_weight, x = input_61)[name = tensor("x_11")]; + tensor input_63_perm_0 = const()[name = tensor("input_63_perm_0"), val = tensor([0, 2, 1])]; + tensor inputs_13_pad_type_0 = const()[name = tensor("inputs_13_pad_type_0"), val = tensor("valid")]; + tensor inputs_13_strides_0 = const()[name = tensor("inputs_13_strides_0"), val = tensor([1])]; + tensor inputs_13_pad_0 = const()[name = tensor("inputs_13_pad_0"), val = tensor([0, 0])]; + tensor inputs_13_dilations_0 = const()[name = tensor("inputs_13_dilations_0"), val = tensor([1])]; + tensor inputs_13_groups_0 = const()[name = tensor("inputs_13_groups_0"), val = tensor(1)]; + tensor input_63 = transpose(perm = input_63_perm_0, x = x_11)[name = tensor("transpose_47")]; + tensor inputs_13 = conv(bias = encoder_conv_module_1_sequential_2_conv_bias, dilations = inputs_13_dilations_0, groups = inputs_13_groups_0, pad = inputs_13_pad_0, pad_type = inputs_13_pad_type_0, strides = inputs_13_strides_0, weight = encoder_conv_module_1_sequential_2_conv_weight, x = input_63)[name = tensor("inputs_13")]; + tensor var_487_split_sizes_0 = const()[name = tensor("op_487_split_sizes_0"), val = tensor([256, 256])]; + tensor var_487_axis_0 = const()[name = tensor("op_487_axis_0"), val = tensor(1)]; + tensor var_487_0, tensor var_487_1 = split(axis = var_487_axis_0, split_sizes = var_487_split_sizes_0, x = inputs_13)[name = tensor("op_487")]; + tensor var_489 = sigmoid(x = var_487_1)[name = tensor("op_489")]; + tensor inputs_15 = mul(x = var_487_0, y = var_489)[name = tensor("inputs_15")]; + tensor outputs_aug_3_pad_type_0 = const()[name = tensor("outputs_aug_3_pad_type_0"), val = tensor("custom")]; + tensor outputs_aug_3_pad_0 = const()[name = tensor("outputs_aug_3_pad_0"), val = tensor([15, 15])]; + tensor outputs_aug_3_groups_0 = const()[name = tensor("outputs_aug_3_groups_0"), val = tensor(256)]; + tensor outputs_aug_3_strides_0 = const()[name = tensor("outputs_aug_3_strides_0"), val = tensor([1])]; + tensor outputs_aug_3_dilations_0 = const()[name = tensor("outputs_aug_3_dilations_0"), val = tensor([1])]; + tensor outputs_aug_3 = conv(dilations = outputs_aug_3_dilations_0, groups = outputs_aug_3_groups_0, pad = outputs_aug_3_pad_0, pad_type = outputs_aug_3_pad_type_0, strides = outputs_aug_3_strides_0, weight = encoder_conv_module_1_sequential_4_conv_weight, x = inputs_15)[name = tensor("outputs_aug_3")]; + tensor input_65_begin_0 = const()[name = tensor("input_65_begin_0"), val = tensor([0, 0, 0])]; + tensor input_65_end_0 = const()[name = tensor("input_65_end_0"), val = tensor([4, 256, 16])]; + tensor input_65_end_mask_0 = const()[name = tensor("input_65_end_mask_0"), val = tensor([true, true, false])]; + tensor input_65 = slice_by_index(begin = input_65_begin_0, end = input_65_end_0, end_mask = input_65_end_mask_0, x = outputs_aug_3)[name = tensor("input_65")]; + tensor inputs_17 = batch_norm(beta = encoder_conv_module_1_sequential_5_bias, epsilon = var_28, gamma = encoder_conv_module_1_sequential_5_weight, mean = encoder_conv_module_1_sequential_5_running_mean, variance = encoder_conv_module_1_sequential_5_running_var, x = input_65)[name = tensor("inputs_17")]; + tensor input_67 = silu(x = inputs_17)[name = tensor("input_67")]; + tensor input_69_pad_type_0 = const()[name = tensor("input_69_pad_type_0"), val = tensor("valid")]; + tensor input_69_strides_0 = const()[name = tensor("input_69_strides_0"), val = tensor([1])]; + tensor input_69_pad_0 = const()[name = tensor("input_69_pad_0"), val = tensor([0, 0])]; + tensor input_69_dilations_0 = const()[name = tensor("input_69_dilations_0"), val = tensor([1])]; + tensor input_69_groups_0 = const()[name = tensor("input_69_groups_0"), val = tensor(1)]; + tensor input_69 = conv(bias = encoder_conv_module_1_sequential_7_conv_bias, dilations = input_69_dilations_0, groups = input_69_groups_0, pad = input_69_pad_0, pad_type = input_69_pad_type_0, strides = input_69_strides_0, weight = encoder_conv_module_1_sequential_7_conv_weight, x = input_67)[name = tensor("input_69")]; + tensor conv_out_3_perm_0 = const()[name = tensor("conv_out_3_perm_0"), val = tensor([0, 2, 1])]; + tensor var_520_begin_0 = const()[name = tensor("op_520_begin_0"), val = tensor([0, -1, 0])]; + tensor var_520_end_0 = const()[name = tensor("op_520_end_0"), val = tensor([4, 16, 256])]; + tensor var_520_end_mask_0 = const()[name = tensor("op_520_end_mask_0"), val = tensor([true, true, true])]; + tensor conv_out_3 = transpose(perm = conv_out_3_perm_0, x = input_69)[name = tensor("transpose_46")]; + tensor var_520 = slice_by_index(begin = var_520_begin_0, end = var_520_end_0, end_mask = var_520_end_mask_0, x = conv_out_3)[name = tensor("op_520")]; + tensor var_522_perm_0 = const()[name = tensor("op_522_perm_0"), val = tensor([1, 0, 2])]; + tensor var_522 = transpose(perm = var_522_perm_0, x = var_520)[name = tensor("transpose_45")]; + tensor input_71 = add(x = x_9, y = var_522)[name = tensor("input_71")]; + tensor input_73_axes_0 = const()[name = tensor("input_73_axes_0"), val = tensor([-1])]; + tensor input_73 = layer_norm(axes = input_73_axes_0, beta = encoder_ffn2_1_module_sequential_0_bias, epsilon = var_28, gamma = encoder_ffn2_1_module_sequential_0_weight, x = input_71)[name = tensor("input_73")]; + tensor inputs_19 = linear(bias = encoder_ffn2_1_module_sequential_1_linear_bias, weight = encoder_ffn2_1_module_sequential_1_linear_weight, x = input_73)[name = tensor("linear_17")]; + tensor input_75 = silu(x = inputs_19)[name = tensor("input_75")]; + tensor input_79 = linear(bias = encoder_ffn2_1_module_sequential_4_linear_bias, weight = encoder_ffn2_1_module_sequential_4_linear_weight, x = input_75)[name = tensor("linear_18")]; + tensor var_545 = const()[name = tensor("op_545"), val = tensor(0x1p-1)]; + tensor var_546 = mul(x = input_79, y = var_545)[name = tensor("op_546")]; + tensor input_81 = add(x = var_546, y = input_71)[name = tensor("input_81")]; + tensor input_83_axes_0 = const()[name = tensor("input_83_axes_0"), val = tensor([-1])]; + tensor input_83 = layer_norm(axes = input_83_axes_0, beta = encoder_layer_norm_1_bias, epsilon = var_28, gamma = encoder_layer_norm_1_weight, x = input_81)[name = tensor("input_83")]; + tensor input_85_axes_0 = const()[name = tensor("input_85_axes_0"), val = tensor([-1])]; + tensor input_85 = layer_norm(axes = input_85_axes_0, beta = encoder_ffn1_2_module_sequential_0_bias, epsilon = var_28, gamma = encoder_ffn1_2_module_sequential_0_weight, x = input_83)[name = tensor("input_85")]; + tensor inputs_21 = linear(bias = encoder_ffn1_2_module_sequential_1_linear_bias, weight = encoder_ffn1_2_module_sequential_1_linear_weight, x = input_85)[name = tensor("linear_19")]; + tensor input_87 = silu(x = inputs_21)[name = tensor("input_87")]; + tensor input_91 = linear(bias = encoder_ffn1_2_module_sequential_4_linear_bias, weight = encoder_ffn1_2_module_sequential_4_linear_weight, x = input_87)[name = tensor("linear_20")]; + tensor var_575 = const()[name = tensor("op_575"), val = tensor(0x1p-1)]; + tensor var_576 = mul(x = input_91, y = var_575)[name = tensor("op_576")]; + tensor input_93 = add(x = var_576, y = input_83)[name = tensor("input_93")]; + tensor x_13_axes_0 = const()[name = tensor("x_13_axes_0"), val = tensor([-1])]; + tensor x_13 = layer_norm(axes = x_13_axes_0, beta = encoder_ret_lns_2_bias, epsilon = var_28, gamma = encoder_ret_lns_2_weight, x = input_93)[name = tensor("x_13")]; + tensor prev_kv_5_begin_0 = const()[name = tensor("prev_kv_5_begin_0"), val = tensor([2, 0, 0, 0, 0])]; + tensor prev_kv_5_end_0 = const()[name = tensor("prev_kv_5_end_0"), val = tensor([3, 1, 4, 64, 64])]; + tensor prev_kv_5_end_mask_0 = const()[name = tensor("prev_kv_5_end_mask_0"), val = tensor([false, true, true, true, true])]; + tensor prev_kv_5_squeeze_mask_0 = const()[name = tensor("prev_kv_5_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; + tensor prev_kv_5 = slice_by_index(begin = prev_kv_5_begin_0, end = prev_kv_5_end_0, end_mask = prev_kv_5_end_mask_0, squeeze_mask = prev_kv_5_squeeze_mask_0, x = enc_kv)[name = tensor("prev_kv_5")]; + tensor prev_scale_5_begin_0 = const()[name = tensor("prev_scale_5_begin_0"), val = tensor([2, 0])]; + tensor prev_scale_5_end_0 = const()[name = tensor("prev_scale_5_end_0"), val = tensor([3, 1])]; + tensor prev_scale_5_end_mask_0 = const()[name = tensor("prev_scale_5_end_mask_0"), val = tensor([false, true])]; + tensor prev_scale_5_squeeze_mask_0 = const()[name = tensor("prev_scale_5_squeeze_mask_0"), val = tensor([true, false])]; + tensor prev_scale_5 = slice_by_index(begin = prev_scale_5_begin_0, end = prev_scale_5_end_0, end_mask = prev_scale_5_end_mask_0, squeeze_mask = prev_scale_5_squeeze_mask_0, x = enc_scale)[name = tensor("prev_scale_5")]; + tensor var_590 = linear(bias = encoder_q_proj_2_bias, weight = encoder_q_proj_2_weight, x = x_13)[name = tensor("linear_21")]; + tensor var_591 = const()[name = tensor("op_591"), val = tensor([1, 4, 4, 64])]; + tensor var_592 = reshape(shape = var_591, x = var_590)[name = tensor("op_592")]; + tensor q_5_perm_0 = const()[name = tensor("q_5_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_596 = linear(bias = encoder_k_proj_2_bias, weight = encoder_k_proj_2_weight, x = x_13)[name = tensor("linear_22")]; + tensor var_597 = const()[name = tensor("op_597"), val = tensor(0x1p-3)]; + tensor var_598 = mul(x = var_596, y = var_597)[name = tensor("op_598")]; + tensor var_599 = const()[name = tensor("op_599"), val = tensor([1, 4, 4, 64])]; + tensor var_600 = reshape(shape = var_599, x = var_598)[name = tensor("op_600")]; + tensor k_5_perm_0 = const()[name = tensor("k_5_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_604 = linear(bias = encoder_v_proj_2_bias, weight = encoder_v_proj_2_weight, x = x_13)[name = tensor("linear_23")]; + tensor var_605 = const()[name = tensor("op_605"), val = tensor([1, 4, 4, 64])]; + tensor var_606 = reshape(shape = var_605, x = var_604)[name = tensor("op_606")]; + tensor v_5_perm_0 = const()[name = tensor("v_5_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor input_97 = linear(bias = encoder_g_proj_2_bias, weight = encoder_g_proj_2_weight, x = x_13)[name = tensor("linear_24")]; + tensor sqrt_s0_5 = sqrt(x = prev_scale_5)[name = tensor("sqrt_s0_5")]; + tensor s_t_5 = add(x = prev_scale_5, y = encoder__t_index)[name = tensor("s_t_5")]; + tensor sqrt_s_t_5 = sqrt(x = s_t_5)[name = tensor("sqrt_s_t_5")]; + tensor qk_5_transpose_x_1 = const()[name = tensor("qk_5_transpose_x_1"), val = tensor(false)]; + tensor qk_5_transpose_y_1 = const()[name = tensor("qk_5_transpose_y_1"), val = tensor(true)]; + tensor k_5 = transpose(perm = k_5_perm_0, x = var_600)[name = tensor("transpose_43")]; + tensor q_5 = transpose(perm = q_5_perm_0, x = var_592)[name = tensor("transpose_44")]; + tensor qk_5 = matmul(transpose_x = qk_5_transpose_x_1, transpose_y = qk_5_transpose_y_1, x = q_5, y = k_5)[name = tensor("qk_5")]; + tensor var_616 = const()[name = tensor("op_616"), val = tensor([4, 1])]; + tensor var_617 = reshape(shape = var_616, x = sqrt_s_t_5)[name = tensor("op_617")]; + tensor M_5 = real_div(x = encoder__causal_mask, y = var_617)[name = tensor("M_5")]; + tensor var_619 = mul(x = qk_5, y = M_5)[name = tensor("op_619")]; + tensor inner_5_transpose_x_0 = const()[name = tensor("inner_5_transpose_x_0"), val = tensor(false)]; + tensor inner_5_transpose_y_0 = const()[name = tensor("inner_5_transpose_y_0"), val = tensor(false)]; + tensor v_5 = transpose(perm = v_5_perm_0, x = var_606)[name = tensor("transpose_42")]; + tensor inner_5 = matmul(transpose_x = inner_5_transpose_x_0, transpose_y = inner_5_transpose_y_0, x = var_619, y = v_5)[name = tensor("inner_5")]; + tensor var_621_transpose_x_0 = const()[name = tensor("op_621_transpose_x_0"), val = tensor(false)]; + tensor var_621_transpose_y_0 = const()[name = tensor("op_621_transpose_y_0"), val = tensor(false)]; + tensor var_621 = matmul(transpose_x = var_621_transpose_x_0, transpose_y = var_621_transpose_y_0, x = q_5, y = prev_kv_5)[name = tensor("op_621")]; + tensor var_622 = real_div(x = sqrt_s0_5, y = sqrt_s_t_5)[name = tensor("op_622")]; + tensor var_623 = const()[name = tensor("op_623"), val = tensor([1, 1, 4, 1])]; + tensor var_624 = reshape(shape = var_623, x = var_622)[name = tensor("op_624")]; + tensor cross_5 = mul(x = var_621, y = var_624)[name = tensor("cross_5")]; + tensor out_13 = add(x = inner_5, y = cross_5)[name = tensor("out_13")]; + tensor var_627 = mul(x = prev_kv_5, y = sqrt_s0_5)[name = tensor("op_627")]; + tensor var_629_transpose_x_1 = const()[name = tensor("op_629_transpose_x_1"), val = tensor(true)]; + tensor var_629_transpose_y_1 = const()[name = tensor("op_629_transpose_y_1"), val = tensor(false)]; + tensor var_629 = matmul(transpose_x = var_629_transpose_x_1, transpose_y = var_629_transpose_y_1, x = k_5, y = v_5)[name = tensor("op_629")]; + tensor new_kv_unnorm_5 = add(x = var_627, y = var_629)[name = tensor("new_kv_unnorm_5")]; + tensor var_631 = const()[name = tensor("op_631"), val = tensor(0x1p+2)]; + tensor new_scale_5 = add(x = prev_scale_5, y = var_631)[name = tensor("new_scale_5")]; + tensor var_633 = sqrt(x = new_scale_5)[name = tensor("op_633")]; + tensor var_634 = real_div(x = new_kv_unnorm_5, y = var_633)[name = tensor("op_634")]; + tensor var_635_perm_0 = const()[name = tensor("op_635_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor out_15_axes_0 = const()[name = tensor("out_15_axes_0"), val = tensor([-1])]; + tensor var_635 = transpose(perm = var_635_perm_0, x = out_13)[name = tensor("transpose_41")]; + tensor out_15 = layer_norm(axes = out_15_axes_0, epsilon = var_17, x = var_635)[name = tensor("out_15")]; + tensor var_639 = const()[name = tensor("op_639"), val = tensor([1, 4, 256])]; + tensor out_17 = reshape(shape = var_639, x = out_15)[name = tensor("out_17")]; + tensor var_641 = silu(x = input_97)[name = tensor("op_641")]; + tensor input_99 = mul(x = var_641, y = out_17)[name = tensor("input_99")]; + tensor ret_out_5 = linear(bias = encoder_out_proj_2_bias, weight = encoder_out_proj_2_weight, x = input_99)[name = tensor("linear_25")]; + tensor x_15 = add(x = input_93, y = ret_out_5)[name = tensor("x_15")]; + tensor window_21_begin_0 = const()[name = tensor("window_21_begin_0"), val = tensor([2, 0, 0, 0])]; + tensor window_21_end_0 = const()[name = tensor("window_21_end_0"), val = tensor([3, 1, 16, 256])]; + tensor window_21_end_mask_0 = const()[name = tensor("window_21_end_mask_0"), val = tensor([false, true, true, true])]; + tensor window_21_squeeze_mask_0 = const()[name = tensor("window_21_squeeze_mask_0"), val = tensor([true, false, false, false])]; + tensor window_21 = slice_by_index(begin = window_21_begin_0, end = window_21_end_0, end_mask = window_21_end_mask_0, squeeze_mask = window_21_squeeze_mask_0, x = enc_conv_cache)[name = tensor("window_21")]; + tensor var_649_begin_0 = const()[name = tensor("op_649_begin_0"), val = tensor([0, 0, 0])]; + tensor var_649_end_0 = const()[name = tensor("op_649_end_0"), val = tensor([1, 1, 256])]; + tensor var_649_end_mask_0 = const()[name = tensor("op_649_end_mask_0"), val = tensor([true, false, true])]; + tensor var_649 = slice_by_index(begin = var_649_begin_0, end = var_649_end_0, end_mask = var_649_end_mask_0, x = x_15)[name = tensor("op_649")]; + tensor var_652_begin_0 = const()[name = tensor("op_652_begin_0"), val = tensor([0, 1, 0])]; + tensor var_652_end_0 = const()[name = tensor("op_652_end_0"), val = tensor([1, 16, 256])]; + tensor var_652_end_mask_0 = const()[name = tensor("op_652_end_mask_0"), val = tensor([true, true, true])]; + tensor var_652 = slice_by_index(begin = var_652_begin_0, end = var_652_end_0, end_mask = var_652_end_mask_0, x = window_21)[name = tensor("op_652")]; + tensor window_23_interleave_0 = const()[name = tensor("window_23_interleave_0"), val = tensor(false)]; + tensor window_23 = concat(axis = var_26, interleave = window_23_interleave_0, values = (var_652, var_649))[name = tensor("window_23")]; + tensor var_657_begin_0 = const()[name = tensor("op_657_begin_0"), val = tensor([0, 1, 0])]; + tensor var_657_end_0 = const()[name = tensor("op_657_end_0"), val = tensor([1, 2, 256])]; + tensor var_657_end_mask_0 = const()[name = tensor("op_657_end_mask_0"), val = tensor([true, false, true])]; + tensor var_657 = slice_by_index(begin = var_657_begin_0, end = var_657_end_0, end_mask = var_657_end_mask_0, x = x_15)[name = tensor("op_657")]; + tensor var_660_begin_0 = const()[name = tensor("op_660_begin_0"), val = tensor([0, 1, 0])]; + tensor var_660_end_0 = const()[name = tensor("op_660_end_0"), val = tensor([1, 16, 256])]; + tensor var_660_end_mask_0 = const()[name = tensor("op_660_end_mask_0"), val = tensor([true, true, true])]; + tensor var_660 = slice_by_index(begin = var_660_begin_0, end = var_660_end_0, end_mask = var_660_end_mask_0, x = window_23)[name = tensor("op_660")]; + tensor window_25_interleave_0 = const()[name = tensor("window_25_interleave_0"), val = tensor(false)]; + tensor window_25 = concat(axis = var_26, interleave = window_25_interleave_0, values = (var_660, var_657))[name = tensor("window_25")]; + tensor var_665_begin_0 = const()[name = tensor("op_665_begin_0"), val = tensor([0, 2, 0])]; + tensor var_665_end_0 = const()[name = tensor("op_665_end_0"), val = tensor([1, 3, 256])]; + tensor var_665_end_mask_0 = const()[name = tensor("op_665_end_mask_0"), val = tensor([true, false, true])]; + tensor var_665 = slice_by_index(begin = var_665_begin_0, end = var_665_end_0, end_mask = var_665_end_mask_0, x = x_15)[name = tensor("op_665")]; + tensor var_668_begin_0 = const()[name = tensor("op_668_begin_0"), val = tensor([0, 1, 0])]; + tensor var_668_end_0 = const()[name = tensor("op_668_end_0"), val = tensor([1, 16, 256])]; + tensor var_668_end_mask_0 = const()[name = tensor("op_668_end_mask_0"), val = tensor([true, true, true])]; + tensor var_668 = slice_by_index(begin = var_668_begin_0, end = var_668_end_0, end_mask = var_668_end_mask_0, x = window_25)[name = tensor("op_668")]; + tensor window_27_interleave_0 = const()[name = tensor("window_27_interleave_0"), val = tensor(false)]; + tensor window_27 = concat(axis = var_26, interleave = window_27_interleave_0, values = (var_668, var_665))[name = tensor("window_27")]; + tensor var_673_begin_0 = const()[name = tensor("op_673_begin_0"), val = tensor([0, 3, 0])]; + tensor var_673_end_0 = const()[name = tensor("op_673_end_0"), val = tensor([1, 1, 256])]; + tensor var_673_end_mask_0 = const()[name = tensor("op_673_end_mask_0"), val = tensor([true, true, true])]; + tensor var_673 = slice_by_index(begin = var_673_begin_0, end = var_673_end_0, end_mask = var_673_end_mask_0, x = x_15)[name = tensor("op_673")]; + tensor var_676_begin_0 = const()[name = tensor("op_676_begin_0"), val = tensor([0, 1, 0])]; + tensor var_676_end_0 = const()[name = tensor("op_676_end_0"), val = tensor([1, 16, 256])]; + tensor var_676_end_mask_0 = const()[name = tensor("op_676_end_mask_0"), val = tensor([true, true, true])]; + tensor var_676 = slice_by_index(begin = var_676_begin_0, end = var_676_end_0, end_mask = var_676_end_mask_0, x = window_27)[name = tensor("op_676")]; + tensor window_29_interleave_0 = const()[name = tensor("window_29_interleave_0"), val = tensor(false)]; + tensor window_29 = concat(axis = var_26, interleave = window_29_interleave_0, values = (var_676, var_673))[name = tensor("window_29")]; + tensor input_101_interleave_0 = const()[name = tensor("input_101_interleave_0"), val = tensor(false)]; + tensor input_101 = concat(axis = var_23, interleave = input_101_interleave_0, values = (window_23, window_25, window_27, window_29))[name = tensor("input_101")]; + tensor x_17_axes_0 = const()[name = tensor("x_17_axes_0"), val = tensor([-1])]; + tensor x_17 = layer_norm(axes = x_17_axes_0, beta = encoder_conv_module_2_sequential_0_bias, epsilon = var_28, gamma = encoder_conv_module_2_sequential_0_weight, x = input_101)[name = tensor("x_17")]; + tensor input_103_perm_0 = const()[name = tensor("input_103_perm_0"), val = tensor([0, 2, 1])]; + tensor inputs_23_pad_type_0 = const()[name = tensor("inputs_23_pad_type_0"), val = tensor("valid")]; + tensor inputs_23_strides_0 = const()[name = tensor("inputs_23_strides_0"), val = tensor([1])]; + tensor inputs_23_pad_0 = const()[name = tensor("inputs_23_pad_0"), val = tensor([0, 0])]; + tensor inputs_23_dilations_0 = const()[name = tensor("inputs_23_dilations_0"), val = tensor([1])]; + tensor inputs_23_groups_0 = const()[name = tensor("inputs_23_groups_0"), val = tensor(1)]; + tensor input_103 = transpose(perm = input_103_perm_0, x = x_17)[name = tensor("transpose_40")]; + tensor inputs_23 = conv(bias = encoder_conv_module_2_sequential_2_conv_bias, dilations = inputs_23_dilations_0, groups = inputs_23_groups_0, pad = inputs_23_pad_0, pad_type = inputs_23_pad_type_0, strides = inputs_23_strides_0, weight = encoder_conv_module_2_sequential_2_conv_weight, x = input_103)[name = tensor("inputs_23")]; + tensor var_701_split_sizes_0 = const()[name = tensor("op_701_split_sizes_0"), val = tensor([256, 256])]; + tensor var_701_axis_0 = const()[name = tensor("op_701_axis_0"), val = tensor(1)]; + tensor var_701_0, tensor var_701_1 = split(axis = var_701_axis_0, split_sizes = var_701_split_sizes_0, x = inputs_23)[name = tensor("op_701")]; + tensor var_703 = sigmoid(x = var_701_1)[name = tensor("op_703")]; + tensor inputs_25 = mul(x = var_701_0, y = var_703)[name = tensor("inputs_25")]; + tensor outputs_aug_5_pad_type_0 = const()[name = tensor("outputs_aug_5_pad_type_0"), val = tensor("custom")]; + tensor outputs_aug_5_pad_0 = const()[name = tensor("outputs_aug_5_pad_0"), val = tensor([15, 15])]; + tensor outputs_aug_5_groups_0 = const()[name = tensor("outputs_aug_5_groups_0"), val = tensor(256)]; + tensor outputs_aug_5_strides_0 = const()[name = tensor("outputs_aug_5_strides_0"), val = tensor([1])]; + tensor outputs_aug_5_dilations_0 = const()[name = tensor("outputs_aug_5_dilations_0"), val = tensor([1])]; + tensor outputs_aug_5 = conv(dilations = outputs_aug_5_dilations_0, groups = outputs_aug_5_groups_0, pad = outputs_aug_5_pad_0, pad_type = outputs_aug_5_pad_type_0, strides = outputs_aug_5_strides_0, weight = encoder_conv_module_2_sequential_4_conv_weight, x = inputs_25)[name = tensor("outputs_aug_5")]; + tensor input_105_begin_0 = const()[name = tensor("input_105_begin_0"), val = tensor([0, 0, 0])]; + tensor input_105_end_0 = const()[name = tensor("input_105_end_0"), val = tensor([4, 256, 16])]; + tensor input_105_end_mask_0 = const()[name = tensor("input_105_end_mask_0"), val = tensor([true, true, false])]; + tensor input_105 = slice_by_index(begin = input_105_begin_0, end = input_105_end_0, end_mask = input_105_end_mask_0, x = outputs_aug_5)[name = tensor("input_105")]; + tensor inputs_27 = batch_norm(beta = encoder_conv_module_2_sequential_5_bias, epsilon = var_28, gamma = encoder_conv_module_2_sequential_5_weight, mean = encoder_conv_module_2_sequential_5_running_mean, variance = encoder_conv_module_2_sequential_5_running_var, x = input_105)[name = tensor("inputs_27")]; + tensor input_107 = silu(x = inputs_27)[name = tensor("input_107")]; + tensor input_109_pad_type_0 = const()[name = tensor("input_109_pad_type_0"), val = tensor("valid")]; + tensor input_109_strides_0 = const()[name = tensor("input_109_strides_0"), val = tensor([1])]; + tensor input_109_pad_0 = const()[name = tensor("input_109_pad_0"), val = tensor([0, 0])]; + tensor input_109_dilations_0 = const()[name = tensor("input_109_dilations_0"), val = tensor([1])]; + tensor input_109_groups_0 = const()[name = tensor("input_109_groups_0"), val = tensor(1)]; + tensor input_109 = conv(bias = encoder_conv_module_2_sequential_7_conv_bias, dilations = input_109_dilations_0, groups = input_109_groups_0, pad = input_109_pad_0, pad_type = input_109_pad_type_0, strides = input_109_strides_0, weight = encoder_conv_module_2_sequential_7_conv_weight, x = input_107)[name = tensor("input_109")]; + tensor conv_out_5_perm_0 = const()[name = tensor("conv_out_5_perm_0"), val = tensor([0, 2, 1])]; + tensor var_734_begin_0 = const()[name = tensor("op_734_begin_0"), val = tensor([0, -1, 0])]; + tensor var_734_end_0 = const()[name = tensor("op_734_end_0"), val = tensor([4, 16, 256])]; + tensor var_734_end_mask_0 = const()[name = tensor("op_734_end_mask_0"), val = tensor([true, true, true])]; + tensor conv_out_5 = transpose(perm = conv_out_5_perm_0, x = input_109)[name = tensor("transpose_39")]; + tensor var_734 = slice_by_index(begin = var_734_begin_0, end = var_734_end_0, end_mask = var_734_end_mask_0, x = conv_out_5)[name = tensor("op_734")]; + tensor var_736_perm_0 = const()[name = tensor("op_736_perm_0"), val = tensor([1, 0, 2])]; + tensor var_736 = transpose(perm = var_736_perm_0, x = var_734)[name = tensor("transpose_38")]; + tensor input_111 = add(x = x_15, y = var_736)[name = tensor("input_111")]; + tensor input_113_axes_0 = const()[name = tensor("input_113_axes_0"), val = tensor([-1])]; + tensor input_113 = layer_norm(axes = input_113_axes_0, beta = encoder_ffn2_2_module_sequential_0_bias, epsilon = var_28, gamma = encoder_ffn2_2_module_sequential_0_weight, x = input_111)[name = tensor("input_113")]; + tensor inputs_29 = linear(bias = encoder_ffn2_2_module_sequential_1_linear_bias, weight = encoder_ffn2_2_module_sequential_1_linear_weight, x = input_113)[name = tensor("linear_26")]; + tensor input_115 = silu(x = inputs_29)[name = tensor("input_115")]; + tensor input_119 = linear(bias = encoder_ffn2_2_module_sequential_4_linear_bias, weight = encoder_ffn2_2_module_sequential_4_linear_weight, x = input_115)[name = tensor("linear_27")]; + tensor var_759 = const()[name = tensor("op_759"), val = tensor(0x1p-1)]; + tensor var_760 = mul(x = input_119, y = var_759)[name = tensor("op_760")]; + tensor input_121 = add(x = var_760, y = input_111)[name = tensor("input_121")]; + tensor input_123_axes_0 = const()[name = tensor("input_123_axes_0"), val = tensor([-1])]; + tensor input_123 = layer_norm(axes = input_123_axes_0, beta = encoder_layer_norm_2_bias, epsilon = var_28, gamma = encoder_layer_norm_2_weight, x = input_121)[name = tensor("input_123")]; + tensor input_125_axes_0 = const()[name = tensor("input_125_axes_0"), val = tensor([-1])]; + tensor input_125 = layer_norm(axes = input_125_axes_0, beta = encoder_ffn1_3_module_sequential_0_bias, epsilon = var_28, gamma = encoder_ffn1_3_module_sequential_0_weight, x = input_123)[name = tensor("input_125")]; + tensor inputs_31 = linear(bias = encoder_ffn1_3_module_sequential_1_linear_bias, weight = encoder_ffn1_3_module_sequential_1_linear_weight, x = input_125)[name = tensor("linear_28")]; + tensor input_127 = silu(x = inputs_31)[name = tensor("input_127")]; + tensor input_131 = linear(bias = encoder_ffn1_3_module_sequential_4_linear_bias, weight = encoder_ffn1_3_module_sequential_4_linear_weight, x = input_127)[name = tensor("linear_29")]; + tensor var_789 = const()[name = tensor("op_789"), val = tensor(0x1p-1)]; + tensor var_790 = mul(x = input_131, y = var_789)[name = tensor("op_790")]; + tensor input_133 = add(x = var_790, y = input_123)[name = tensor("input_133")]; + tensor x_19_axes_0 = const()[name = tensor("x_19_axes_0"), val = tensor([-1])]; + tensor x_19 = layer_norm(axes = x_19_axes_0, beta = encoder_ret_lns_3_bias, epsilon = var_28, gamma = encoder_ret_lns_3_weight, x = input_133)[name = tensor("x_19")]; + tensor prev_kv_7_begin_0 = const()[name = tensor("prev_kv_7_begin_0"), val = tensor([3, 0, 0, 0, 0])]; + tensor prev_kv_7_end_0 = const()[name = tensor("prev_kv_7_end_0"), val = tensor([4, 1, 4, 64, 64])]; + tensor prev_kv_7_end_mask_0 = const()[name = tensor("prev_kv_7_end_mask_0"), val = tensor([false, true, true, true, true])]; + tensor prev_kv_7_squeeze_mask_0 = const()[name = tensor("prev_kv_7_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; + tensor prev_kv_7 = slice_by_index(begin = prev_kv_7_begin_0, end = prev_kv_7_end_0, end_mask = prev_kv_7_end_mask_0, squeeze_mask = prev_kv_7_squeeze_mask_0, x = enc_kv)[name = tensor("prev_kv_7")]; + tensor prev_scale_7_begin_0 = const()[name = tensor("prev_scale_7_begin_0"), val = tensor([3, 0])]; + tensor prev_scale_7_end_0 = const()[name = tensor("prev_scale_7_end_0"), val = tensor([4, 1])]; + tensor prev_scale_7_end_mask_0 = const()[name = tensor("prev_scale_7_end_mask_0"), val = tensor([false, true])]; + tensor prev_scale_7_squeeze_mask_0 = const()[name = tensor("prev_scale_7_squeeze_mask_0"), val = tensor([true, false])]; + tensor prev_scale_7 = slice_by_index(begin = prev_scale_7_begin_0, end = prev_scale_7_end_0, end_mask = prev_scale_7_end_mask_0, squeeze_mask = prev_scale_7_squeeze_mask_0, x = enc_scale)[name = tensor("prev_scale_7")]; + tensor var_804 = linear(bias = encoder_q_proj_3_bias, weight = encoder_q_proj_3_weight, x = x_19)[name = tensor("linear_30")]; + tensor var_805 = const()[name = tensor("op_805"), val = tensor([1, 4, 4, 64])]; + tensor var_806 = reshape(shape = var_805, x = var_804)[name = tensor("op_806")]; + tensor q_7_perm_0 = const()[name = tensor("q_7_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_810 = linear(bias = encoder_k_proj_3_bias, weight = encoder_k_proj_3_weight, x = x_19)[name = tensor("linear_31")]; + tensor var_811 = const()[name = tensor("op_811"), val = tensor(0x1p-3)]; + tensor var_812 = mul(x = var_810, y = var_811)[name = tensor("op_812")]; + tensor var_813 = const()[name = tensor("op_813"), val = tensor([1, 4, 4, 64])]; + tensor var_814 = reshape(shape = var_813, x = var_812)[name = tensor("op_814")]; + tensor k_7_perm_0 = const()[name = tensor("k_7_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_818 = linear(bias = encoder_v_proj_3_bias, weight = encoder_v_proj_3_weight, x = x_19)[name = tensor("linear_32")]; + tensor var_819 = const()[name = tensor("op_819"), val = tensor([1, 4, 4, 64])]; + tensor var_820 = reshape(shape = var_819, x = var_818)[name = tensor("op_820")]; + tensor v_7_perm_0 = const()[name = tensor("v_7_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor input_137 = linear(bias = encoder_g_proj_3_bias, weight = encoder_g_proj_3_weight, x = x_19)[name = tensor("linear_33")]; + tensor sqrt_s0_7 = sqrt(x = prev_scale_7)[name = tensor("sqrt_s0_7")]; + tensor s_t_7 = add(x = prev_scale_7, y = encoder__t_index)[name = tensor("s_t_7")]; + tensor sqrt_s_t_7 = sqrt(x = s_t_7)[name = tensor("sqrt_s_t_7")]; + tensor qk_7_transpose_x_1 = const()[name = tensor("qk_7_transpose_x_1"), val = tensor(false)]; + tensor qk_7_transpose_y_1 = const()[name = tensor("qk_7_transpose_y_1"), val = tensor(true)]; + tensor k_7 = transpose(perm = k_7_perm_0, x = var_814)[name = tensor("transpose_36")]; + tensor q_7 = transpose(perm = q_7_perm_0, x = var_806)[name = tensor("transpose_37")]; + tensor qk_7 = matmul(transpose_x = qk_7_transpose_x_1, transpose_y = qk_7_transpose_y_1, x = q_7, y = k_7)[name = tensor("qk_7")]; + tensor var_830 = const()[name = tensor("op_830"), val = tensor([4, 1])]; + tensor var_831 = reshape(shape = var_830, x = sqrt_s_t_7)[name = tensor("op_831")]; + tensor M_7 = real_div(x = encoder__causal_mask, y = var_831)[name = tensor("M_7")]; + tensor var_833 = mul(x = qk_7, y = M_7)[name = tensor("op_833")]; + tensor inner_7_transpose_x_0 = const()[name = tensor("inner_7_transpose_x_0"), val = tensor(false)]; + tensor inner_7_transpose_y_0 = const()[name = tensor("inner_7_transpose_y_0"), val = tensor(false)]; + tensor v_7 = transpose(perm = v_7_perm_0, x = var_820)[name = tensor("transpose_35")]; + tensor inner_7 = matmul(transpose_x = inner_7_transpose_x_0, transpose_y = inner_7_transpose_y_0, x = var_833, y = v_7)[name = tensor("inner_7")]; + tensor var_835_transpose_x_0 = const()[name = tensor("op_835_transpose_x_0"), val = tensor(false)]; + tensor var_835_transpose_y_0 = const()[name = tensor("op_835_transpose_y_0"), val = tensor(false)]; + tensor var_835 = matmul(transpose_x = var_835_transpose_x_0, transpose_y = var_835_transpose_y_0, x = q_7, y = prev_kv_7)[name = tensor("op_835")]; + tensor var_836 = real_div(x = sqrt_s0_7, y = sqrt_s_t_7)[name = tensor("op_836")]; + tensor var_837 = const()[name = tensor("op_837"), val = tensor([1, 1, 4, 1])]; + tensor var_838 = reshape(shape = var_837, x = var_836)[name = tensor("op_838")]; + tensor cross_7 = mul(x = var_835, y = var_838)[name = tensor("cross_7")]; + tensor out_19 = add(x = inner_7, y = cross_7)[name = tensor("out_19")]; + tensor var_841 = mul(x = prev_kv_7, y = sqrt_s0_7)[name = tensor("op_841")]; + tensor var_843_transpose_x_1 = const()[name = tensor("op_843_transpose_x_1"), val = tensor(true)]; + tensor var_843_transpose_y_1 = const()[name = tensor("op_843_transpose_y_1"), val = tensor(false)]; + tensor var_843 = matmul(transpose_x = var_843_transpose_x_1, transpose_y = var_843_transpose_y_1, x = k_7, y = v_7)[name = tensor("op_843")]; + tensor new_kv_unnorm_7 = add(x = var_841, y = var_843)[name = tensor("new_kv_unnorm_7")]; + tensor var_845 = const()[name = tensor("op_845"), val = tensor(0x1p+2)]; + tensor new_scale_7 = add(x = prev_scale_7, y = var_845)[name = tensor("new_scale_7")]; + tensor var_847 = sqrt(x = new_scale_7)[name = tensor("op_847")]; + tensor nkv_1 = real_div(x = new_kv_unnorm_7, y = var_847)[name = tensor("nkv_1")]; + tensor var_849_perm_0 = const()[name = tensor("op_849_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor out_21_axes_0 = const()[name = tensor("out_21_axes_0"), val = tensor([-1])]; + tensor var_849 = transpose(perm = var_849_perm_0, x = out_19)[name = tensor("transpose_34")]; + tensor out_21 = layer_norm(axes = out_21_axes_0, epsilon = var_17, x = var_849)[name = tensor("out_21")]; + tensor var_853 = const()[name = tensor("op_853"), val = tensor([1, 4, 256])]; + tensor out_23 = reshape(shape = var_853, x = out_21)[name = tensor("out_23")]; + tensor var_855 = silu(x = input_137)[name = tensor("op_855")]; + tensor input_139 = mul(x = var_855, y = out_23)[name = tensor("input_139")]; + tensor ret_out_7 = linear(bias = encoder_out_proj_3_bias, weight = encoder_out_proj_3_weight, x = input_139)[name = tensor("linear_34")]; + tensor x_21 = add(x = input_133, y = ret_out_7)[name = tensor("x_21")]; + tensor window_31_begin_0 = const()[name = tensor("window_31_begin_0"), val = tensor([3, 0, 0, 0])]; + tensor window_31_end_0 = const()[name = tensor("window_31_end_0"), val = tensor([4, 1, 16, 256])]; + tensor window_31_end_mask_0 = const()[name = tensor("window_31_end_mask_0"), val = tensor([false, true, true, true])]; + tensor window_31_squeeze_mask_0 = const()[name = tensor("window_31_squeeze_mask_0"), val = tensor([true, false, false, false])]; + tensor window_31 = slice_by_index(begin = window_31_begin_0, end = window_31_end_0, end_mask = window_31_end_mask_0, squeeze_mask = window_31_squeeze_mask_0, x = enc_conv_cache)[name = tensor("window_31")]; + tensor var_863_begin_0 = const()[name = tensor("op_863_begin_0"), val = tensor([0, 0, 0])]; + tensor var_863_end_0 = const()[name = tensor("op_863_end_0"), val = tensor([1, 1, 256])]; + tensor var_863_end_mask_0 = const()[name = tensor("op_863_end_mask_0"), val = tensor([true, false, true])]; + tensor var_863 = slice_by_index(begin = var_863_begin_0, end = var_863_end_0, end_mask = var_863_end_mask_0, x = x_21)[name = tensor("op_863")]; + tensor var_866_begin_0 = const()[name = tensor("op_866_begin_0"), val = tensor([0, 1, 0])]; + tensor var_866_end_0 = const()[name = tensor("op_866_end_0"), val = tensor([1, 16, 256])]; + tensor var_866_end_mask_0 = const()[name = tensor("op_866_end_mask_0"), val = tensor([true, true, true])]; + tensor var_866 = slice_by_index(begin = var_866_begin_0, end = var_866_end_0, end_mask = var_866_end_mask_0, x = window_31)[name = tensor("op_866")]; + tensor window_33_interleave_0 = const()[name = tensor("window_33_interleave_0"), val = tensor(false)]; + tensor window_33 = concat(axis = var_26, interleave = window_33_interleave_0, values = (var_866, var_863))[name = tensor("window_33")]; + tensor var_871_begin_0 = const()[name = tensor("op_871_begin_0"), val = tensor([0, 1, 0])]; + tensor var_871_end_0 = const()[name = tensor("op_871_end_0"), val = tensor([1, 2, 256])]; + tensor var_871_end_mask_0 = const()[name = tensor("op_871_end_mask_0"), val = tensor([true, false, true])]; + tensor var_871 = slice_by_index(begin = var_871_begin_0, end = var_871_end_0, end_mask = var_871_end_mask_0, x = x_21)[name = tensor("op_871")]; + tensor var_874_begin_0 = const()[name = tensor("op_874_begin_0"), val = tensor([0, 1, 0])]; + tensor var_874_end_0 = const()[name = tensor("op_874_end_0"), val = tensor([1, 16, 256])]; + tensor var_874_end_mask_0 = const()[name = tensor("op_874_end_mask_0"), val = tensor([true, true, true])]; + tensor var_874 = slice_by_index(begin = var_874_begin_0, end = var_874_end_0, end_mask = var_874_end_mask_0, x = window_33)[name = tensor("op_874")]; + tensor window_35_interleave_0 = const()[name = tensor("window_35_interleave_0"), val = tensor(false)]; + tensor window_35 = concat(axis = var_26, interleave = window_35_interleave_0, values = (var_874, var_871))[name = tensor("window_35")]; + tensor var_879_begin_0 = const()[name = tensor("op_879_begin_0"), val = tensor([0, 2, 0])]; + tensor var_879_end_0 = const()[name = tensor("op_879_end_0"), val = tensor([1, 3, 256])]; + tensor var_879_end_mask_0 = const()[name = tensor("op_879_end_mask_0"), val = tensor([true, false, true])]; + tensor var_879 = slice_by_index(begin = var_879_begin_0, end = var_879_end_0, end_mask = var_879_end_mask_0, x = x_21)[name = tensor("op_879")]; + tensor var_882_begin_0 = const()[name = tensor("op_882_begin_0"), val = tensor([0, 1, 0])]; + tensor var_882_end_0 = const()[name = tensor("op_882_end_0"), val = tensor([1, 16, 256])]; + tensor var_882_end_mask_0 = const()[name = tensor("op_882_end_mask_0"), val = tensor([true, true, true])]; + tensor var_882 = slice_by_index(begin = var_882_begin_0, end = var_882_end_0, end_mask = var_882_end_mask_0, x = window_35)[name = tensor("op_882")]; + tensor window_37_interleave_0 = const()[name = tensor("window_37_interleave_0"), val = tensor(false)]; + tensor window_37 = concat(axis = var_26, interleave = window_37_interleave_0, values = (var_882, var_879))[name = tensor("window_37")]; + tensor var_887_begin_0 = const()[name = tensor("op_887_begin_0"), val = tensor([0, 3, 0])]; + tensor var_887_end_0 = const()[name = tensor("op_887_end_0"), val = tensor([1, 1, 256])]; + tensor var_887_end_mask_0 = const()[name = tensor("op_887_end_mask_0"), val = tensor([true, true, true])]; + tensor var_887 = slice_by_index(begin = var_887_begin_0, end = var_887_end_0, end_mask = var_887_end_mask_0, x = x_21)[name = tensor("op_887")]; + tensor var_890_begin_0 = const()[name = tensor("op_890_begin_0"), val = tensor([0, 1, 0])]; + tensor var_890_end_0 = const()[name = tensor("op_890_end_0"), val = tensor([1, 16, 256])]; + tensor var_890_end_mask_0 = const()[name = tensor("op_890_end_mask_0"), val = tensor([true, true, true])]; + tensor var_890 = slice_by_index(begin = var_890_begin_0, end = var_890_end_0, end_mask = var_890_end_mask_0, x = window_37)[name = tensor("op_890")]; + tensor window_interleave_0 = const()[name = tensor("window_interleave_0"), val = tensor(false)]; + tensor window = concat(axis = var_26, interleave = window_interleave_0, values = (var_890, var_887))[name = tensor("window")]; + tensor input_141_interleave_0 = const()[name = tensor("input_141_interleave_0"), val = tensor(false)]; + tensor input_141 = concat(axis = var_23, interleave = input_141_interleave_0, values = (window_33, window_35, window_37, window))[name = tensor("input_141")]; + tensor x_23_axes_0 = const()[name = tensor("x_23_axes_0"), val = tensor([-1])]; + tensor x_23 = layer_norm(axes = x_23_axes_0, beta = encoder_conv_module_3_sequential_0_bias, epsilon = var_28, gamma = encoder_conv_module_3_sequential_0_weight, x = input_141)[name = tensor("x_23")]; + tensor input_143_perm_0 = const()[name = tensor("input_143_perm_0"), val = tensor([0, 2, 1])]; + tensor inputs_33_pad_type_0 = const()[name = tensor("inputs_33_pad_type_0"), val = tensor("valid")]; + tensor inputs_33_strides_0 = const()[name = tensor("inputs_33_strides_0"), val = tensor([1])]; + tensor inputs_33_pad_0 = const()[name = tensor("inputs_33_pad_0"), val = tensor([0, 0])]; + tensor inputs_33_dilations_0 = const()[name = tensor("inputs_33_dilations_0"), val = tensor([1])]; + tensor inputs_33_groups_0 = const()[name = tensor("inputs_33_groups_0"), val = tensor(1)]; + tensor input_143 = transpose(perm = input_143_perm_0, x = x_23)[name = tensor("transpose_33")]; + tensor inputs_33 = conv(bias = encoder_conv_module_3_sequential_2_conv_bias, dilations = inputs_33_dilations_0, groups = inputs_33_groups_0, pad = inputs_33_pad_0, pad_type = inputs_33_pad_type_0, strides = inputs_33_strides_0, weight = encoder_conv_module_3_sequential_2_conv_weight, x = input_143)[name = tensor("inputs_33")]; + tensor var_915_split_sizes_0 = const()[name = tensor("op_915_split_sizes_0"), val = tensor([256, 256])]; + tensor var_915_axis_0 = const()[name = tensor("op_915_axis_0"), val = tensor(1)]; + tensor var_915_0, tensor var_915_1 = split(axis = var_915_axis_0, split_sizes = var_915_split_sizes_0, x = inputs_33)[name = tensor("op_915")]; + tensor var_917 = sigmoid(x = var_915_1)[name = tensor("op_917")]; + tensor inputs_35 = mul(x = var_915_0, y = var_917)[name = tensor("inputs_35")]; + tensor outputs_aug_pad_type_0 = const()[name = tensor("outputs_aug_pad_type_0"), val = tensor("custom")]; + tensor outputs_aug_pad_0 = const()[name = tensor("outputs_aug_pad_0"), val = tensor([15, 15])]; + tensor outputs_aug_groups_0 = const()[name = tensor("outputs_aug_groups_0"), val = tensor(256)]; + tensor outputs_aug_strides_0 = const()[name = tensor("outputs_aug_strides_0"), val = tensor([1])]; + tensor outputs_aug_dilations_0 = const()[name = tensor("outputs_aug_dilations_0"), val = tensor([1])]; + tensor outputs_aug = conv(dilations = outputs_aug_dilations_0, groups = outputs_aug_groups_0, pad = outputs_aug_pad_0, pad_type = outputs_aug_pad_type_0, strides = outputs_aug_strides_0, weight = encoder_conv_module_3_sequential_4_conv_weight, x = inputs_35)[name = tensor("outputs_aug")]; + tensor input_145_begin_0 = const()[name = tensor("input_145_begin_0"), val = tensor([0, 0, 0])]; + tensor input_145_end_0 = const()[name = tensor("input_145_end_0"), val = tensor([4, 256, 16])]; + tensor input_145_end_mask_0 = const()[name = tensor("input_145_end_mask_0"), val = tensor([true, true, false])]; + tensor input_145 = slice_by_index(begin = input_145_begin_0, end = input_145_end_0, end_mask = input_145_end_mask_0, x = outputs_aug)[name = tensor("input_145")]; + tensor inputs_37 = batch_norm(beta = encoder_conv_module_3_sequential_5_bias, epsilon = var_28, gamma = encoder_conv_module_3_sequential_5_weight, mean = encoder_conv_module_3_sequential_5_running_mean, variance = encoder_conv_module_3_sequential_5_running_var, x = input_145)[name = tensor("inputs_37")]; + tensor input_147 = silu(x = inputs_37)[name = tensor("input_147")]; + tensor input_149_pad_type_0 = const()[name = tensor("input_149_pad_type_0"), val = tensor("valid")]; + tensor input_149_strides_0 = const()[name = tensor("input_149_strides_0"), val = tensor([1])]; + tensor input_149_pad_0 = const()[name = tensor("input_149_pad_0"), val = tensor([0, 0])]; + tensor input_149_dilations_0 = const()[name = tensor("input_149_dilations_0"), val = tensor([1])]; + tensor input_149_groups_0 = const()[name = tensor("input_149_groups_0"), val = tensor(1)]; + tensor input_149 = conv(bias = encoder_conv_module_3_sequential_7_conv_bias, dilations = input_149_dilations_0, groups = input_149_groups_0, pad = input_149_pad_0, pad_type = input_149_pad_type_0, strides = input_149_strides_0, weight = encoder_conv_module_3_sequential_7_conv_weight, x = input_147)[name = tensor("input_149")]; + tensor conv_out_7_perm_0 = const()[name = tensor("conv_out_7_perm_0"), val = tensor([0, 2, 1])]; + tensor var_948_begin_0 = const()[name = tensor("op_948_begin_0"), val = tensor([0, -1, 0])]; + tensor var_948_end_0 = const()[name = tensor("op_948_end_0"), val = tensor([4, 16, 256])]; + tensor var_948_end_mask_0 = const()[name = tensor("op_948_end_mask_0"), val = tensor([true, true, true])]; + tensor conv_out_7 = transpose(perm = conv_out_7_perm_0, x = input_149)[name = tensor("transpose_32")]; + tensor var_948 = slice_by_index(begin = var_948_begin_0, end = var_948_end_0, end_mask = var_948_end_mask_0, x = conv_out_7)[name = tensor("op_948")]; + tensor var_950_perm_0 = const()[name = tensor("op_950_perm_0"), val = tensor([1, 0, 2])]; + tensor var_950 = transpose(perm = var_950_perm_0, x = var_948)[name = tensor("transpose_31")]; + tensor input_151 = add(x = x_21, y = var_950)[name = tensor("input_151")]; + tensor input_153_axes_0 = const()[name = tensor("input_153_axes_0"), val = tensor([-1])]; + tensor input_153 = layer_norm(axes = input_153_axes_0, beta = encoder_ffn2_3_module_sequential_0_bias, epsilon = var_28, gamma = encoder_ffn2_3_module_sequential_0_weight, x = input_151)[name = tensor("input_153")]; + tensor inputs = linear(bias = encoder_ffn2_3_module_sequential_1_linear_bias, weight = encoder_ffn2_3_module_sequential_1_linear_weight, x = input_153)[name = tensor("linear_35")]; + tensor input_155 = silu(x = inputs)[name = tensor("input_155")]; + tensor input_159 = linear(bias = encoder_ffn2_3_module_sequential_4_linear_bias, weight = encoder_ffn2_3_module_sequential_4_linear_weight, x = input_155)[name = tensor("linear_36")]; + tensor var_973 = const()[name = tensor("op_973"), val = tensor(0x1p-1)]; + tensor var_974 = mul(x = input_159, y = var_973)[name = tensor("op_974")]; + tensor input_161 = add(x = var_974, y = input_151)[name = tensor("input_161")]; + tensor x_25_axes_0 = const()[name = tensor("x_25_axes_0"), val = tensor([-1])]; + tensor x_25 = layer_norm(axes = x_25_axes_0, beta = encoder_layer_norm_3_bias, epsilon = var_28, gamma = encoder_layer_norm_3_weight, x = input_161)[name = tensor("x_25")]; + tensor x_ct_perm_0 = const()[name = tensor("x_ct_perm_0"), val = tensor([0, 2, 1])]; + tensor cat_interleave_0 = const()[name = tensor("cat_interleave_0"), val = tensor(false)]; + tensor x_ct = transpose(perm = x_ct_perm_0, x = x_25)[name = tensor("transpose_30")]; + tensor cat = concat(axis = var_20, interleave = cat_interleave_0, values = (cnn_window, x_ct))[name = tensor("cat")]; + tensor conv_out_pad_type_0 = const()[name = tensor("conv_out_pad_type_0"), val = tensor("valid")]; + tensor conv_out_strides_0 = const()[name = tensor("conv_out_strides_0"), val = tensor([1])]; + tensor conv_out_pad_0 = const()[name = tensor("conv_out_pad_0"), val = tensor([0, 0])]; + tensor conv_out_dilations_0 = const()[name = tensor("conv_out_dilations_0"), val = tensor([1])]; + tensor conv_out_groups_0 = const()[name = tensor("conv_out_groups_0"), val = tensor(1)]; + tensor conv_out = conv(bias = encoder_cnn_bias, dilations = conv_out_dilations_0, groups = conv_out_groups_0, pad = conv_out_pad_0, pad_type = conv_out_pad_type_0, strides = conv_out_strides_0, weight = encoder_cnn_weight, x = cat)[name = tensor("conv_out")]; + tensor var_992_begin_0 = const()[name = tensor("op_992_begin_0"), val = tensor([0, 0, 4])]; + tensor var_992_end_0 = const()[name = tensor("op_992_end_0"), val = tensor([1, 256, 22])]; + tensor var_992_end_mask_0 = const()[name = tensor("op_992_end_mask_0"), val = tensor([true, true, true])]; + tensor cnn_window_new = slice_by_index(begin = var_992_begin_0, end = var_992_end_0, end_mask = var_992_end_mask_0, x = cat)[name = tensor("op_992")]; + tensor input_163_perm_0 = const()[name = tensor("input_163_perm_0"), val = tensor([0, 2, 1])]; + tensor var_994 = const()[name = tensor("op_994"), val = tensor([-1])]; + tensor input_163 = transpose(perm = input_163_perm_0, x = conv_out)[name = tensor("transpose_29")]; + tensor var_995 = reduce_l2_norm(axes = var_994, keep_dims = var_29, x = input_163)[name = tensor("op_995")]; + tensor const_12 = const()[name = tensor("const_12"), val = tensor(0x1.fffffep+127)]; + tensor clip_0 = clip(alpha = var_11, beta = const_12, x = var_995)[name = tensor("clip_0")]; + tensor emb = real_div(x = input_163, y = clip_0)[name = tensor("emb")]; + tensor var_999_axis_0 = const()[name = tensor("op_999_axis_0"), val = tensor(0)]; + tensor enc_kv_new = stack(axis = var_999_axis_0, values = (var_206, var_420, var_634, nkv_1))[name = tensor("op_999")]; + tensor var_1001_axis_0 = const()[name = tensor("op_1001_axis_0"), val = tensor(0)]; + tensor enc_scale_new = stack(axis = var_1001_axis_0, values = (new_scale_1, new_scale_3, new_scale_5, new_scale_7))[name = tensor("op_1001")]; + tensor var_1003_axis_0 = const()[name = tensor("op_1003_axis_0"), val = tensor(0)]; + tensor enc_conv_cache_new = stack(axis = var_1003_axis_0, values = (window_9, window_19, window_29, window))[name = tensor("op_1003")]; + tensor var_1012 = const()[name = tensor("op_1012"), val = tensor(0x1.5798eep-27)]; + tensor var_1017 = const()[name = tensor("op_1017"), val = tensor(0x1.4f8b58p-17)]; + tensor var_1019 = const()[name = tensor("op_1019"), val = tensor(0x1.0c6f7ap-20)]; + tensor var_1020 = const()[name = tensor("op_1020"), val = tensor(true)]; + tensor var_1022 = const()[name = tensor("op_1022"), val = tensor(0x1p+0)]; + tensor var_1026 = const()[name = tensor("op_1026"), val = tensor(-1)]; + tensor var_1032 = const()[name = tensor("op_1032"), val = tensor(0)]; + tensor pos = const()[name = tensor("pos"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(44395584)))]; + tensor var_1094_axes_0 = const()[name = tensor("op_1094_axes_0"), val = tensor([2])]; + tensor var_1094 = expand_dims(axes = var_1094_axes_0, x = emb)[name = tensor("op_1094")]; + tensor emb_exp_reps_0 = const()[name = tensor("emb_exp_reps_0"), val = tensor([1, 1, 6, 1])]; + tensor emb_exp = tile(reps = emb_exp_reps_0, x = var_1094)[name = tensor("emb_exp")]; + tensor input_165_interleave_0 = const()[name = tensor("input_165_interleave_0"), val = tensor(false)]; + tensor input_165 = concat(axis = var_1026, interleave = input_165_interleave_0, values = (emb_exp, pos))[name = tensor("input_165")]; + tensor x_27 = linear(bias = decoder_convert_bias, weight = decoder_convert_weight, x = input_165)[name = tensor("linear_37")]; + tensor var_1102_perm_0 = const()[name = tensor("op_1102_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_1106 = const()[name = tensor("op_1106"), val = tensor([6, 4, 256])]; + tensor var_1102 = transpose(perm = var_1102_perm_0, x = x_27)[name = tensor("transpose_28")]; + tensor x_29 = reshape(shape = var_1106, x = var_1102)[name = tensor("x_29")]; + tensor prev_kv_9_begin_0 = const()[name = tensor("prev_kv_9_begin_0"), val = tensor([0, 0, 0, 0, 0])]; + tensor prev_kv_9_end_0 = const()[name = tensor("prev_kv_9_end_0"), val = tensor([1, 6, 4, 64, 64])]; + tensor prev_kv_9_end_mask_0 = const()[name = tensor("prev_kv_9_end_mask_0"), val = tensor([false, true, true, true, true])]; + tensor prev_kv_9_squeeze_mask_0 = const()[name = tensor("prev_kv_9_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; + tensor prev_kv_9 = slice_by_index(begin = prev_kv_9_begin_0, end = prev_kv_9_end_0, end_mask = prev_kv_9_end_mask_0, squeeze_mask = prev_kv_9_squeeze_mask_0, x = dec_kv)[name = tensor("prev_kv_9")]; + tensor prev_scale_9_begin_0 = const()[name = tensor("prev_scale_9_begin_0"), val = tensor([0, 0])]; + tensor prev_scale_9_end_0 = const()[name = tensor("prev_scale_9_end_0"), val = tensor([1, 1])]; + tensor prev_scale_9_end_mask_0 = const()[name = tensor("prev_scale_9_end_mask_0"), val = tensor([false, true])]; + tensor prev_scale_9_squeeze_mask_0 = const()[name = tensor("prev_scale_9_squeeze_mask_0"), val = tensor([true, false])]; + tensor prev_scale_9 = slice_by_index(begin = prev_scale_9_begin_0, end = prev_scale_9_end_0, end_mask = prev_scale_9_end_mask_0, squeeze_mask = prev_scale_9_squeeze_mask_0, x = dec_scale)[name = tensor("prev_scale_9")]; + tensor var_1114 = linear(bias = decoder_q_proj_0_bias, weight = decoder_q_proj_0_weight, x = x_29)[name = tensor("linear_38")]; + tensor var_1115 = const()[name = tensor("op_1115"), val = tensor([6, 4, 4, 64])]; + tensor var_1116 = reshape(shape = var_1115, x = var_1114)[name = tensor("op_1116")]; + tensor q_9_perm_0 = const()[name = tensor("q_9_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_1120 = linear(bias = decoder_k_proj_0_bias, weight = decoder_k_proj_0_weight, x = x_29)[name = tensor("linear_39")]; + tensor var_1121 = const()[name = tensor("op_1121"), val = tensor(0x1p-3)]; + tensor var_1122 = mul(x = var_1120, y = var_1121)[name = tensor("op_1122")]; + tensor var_1123 = const()[name = tensor("op_1123"), val = tensor([6, 4, 4, 64])]; + tensor var_1124 = reshape(shape = var_1123, x = var_1122)[name = tensor("op_1124")]; + tensor k_9_perm_0 = const()[name = tensor("k_9_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_1128 = linear(bias = decoder_v_proj_0_bias, weight = decoder_v_proj_0_weight, x = x_29)[name = tensor("linear_40")]; + tensor var_1129 = const()[name = tensor("op_1129"), val = tensor([6, 4, 4, 64])]; + tensor var_1130 = reshape(shape = var_1129, x = var_1128)[name = tensor("op_1130")]; + tensor v_9_perm_0 = const()[name = tensor("v_9_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor input_169 = linear(bias = decoder_g_proj_0_bias, weight = decoder_g_proj_0_weight, x = x_29)[name = tensor("linear_41")]; + tensor cumsum_mask_1_exclusive_0 = const()[name = tensor("cumsum_mask_1_exclusive_0"), val = tensor(false)]; + tensor cumsum_mask_1_reverse_0 = const()[name = tensor("cumsum_mask_1_reverse_0"), val = tensor(false)]; + tensor cumsum_mask_1 = cumsum(axis = var_1032, exclusive = cumsum_mask_1_exclusive_0, reverse = cumsum_mask_1_reverse_0, x = valid_mask)[name = tensor("cumsum_mask_1")]; + tensor sqrt_s0_9 = sqrt(x = prev_scale_9)[name = tensor("sqrt_s0_9")]; + tensor s_t_9 = add(x = prev_scale_9, y = cumsum_mask_1)[name = tensor("s_t_9")]; + tensor const_20 = const()[name = tensor("const_20"), val = tensor(0x1.fffffep+127)]; + tensor clip_1 = clip(alpha = var_1022, beta = const_20, x = s_t_9)[name = tensor("clip_1")]; + tensor sqrt_s_t_9 = sqrt(x = clip_1)[name = tensor("sqrt_s_t_9")]; + tensor qk_9_transpose_x_1 = const()[name = tensor("qk_9_transpose_x_1"), val = tensor(false)]; + tensor qk_9_transpose_y_1 = const()[name = tensor("qk_9_transpose_y_1"), val = tensor(true)]; + tensor k_9 = transpose(perm = k_9_perm_0, x = var_1124)[name = tensor("transpose_26")]; + tensor q_9 = transpose(perm = q_9_perm_0, x = var_1116)[name = tensor("transpose_27")]; + tensor qk_9 = matmul(transpose_x = qk_9_transpose_x_1, transpose_y = qk_9_transpose_y_1, x = q_9, y = k_9)[name = tensor("qk_9")]; + tensor var_1142 = const()[name = tensor("op_1142"), val = tensor([1, 4])]; + tensor var_1143 = reshape(shape = var_1142, x = valid_mask)[name = tensor("op_1143")]; + tensor causal_with_valid_1 = mul(x = decoder__causal_mask, y = var_1143)[name = tensor("causal_with_valid_1")]; + tensor var_1145 = const()[name = tensor("op_1145"), val = tensor([4, 1])]; + tensor var_1146 = reshape(shape = var_1145, x = sqrt_s_t_9)[name = tensor("op_1146")]; + tensor M_9 = real_div(x = causal_with_valid_1, y = var_1146)[name = tensor("M_9")]; + tensor var_1148 = mul(x = qk_9, y = M_9)[name = tensor("op_1148")]; + tensor inner_9_transpose_x_0 = const()[name = tensor("inner_9_transpose_x_0"), val = tensor(false)]; + tensor inner_9_transpose_y_0 = const()[name = tensor("inner_9_transpose_y_0"), val = tensor(false)]; + tensor v_9 = transpose(perm = v_9_perm_0, x = var_1130)[name = tensor("transpose_25")]; + tensor inner_9 = matmul(transpose_x = inner_9_transpose_x_0, transpose_y = inner_9_transpose_y_0, x = var_1148, y = v_9)[name = tensor("inner_9")]; + tensor var_1150_transpose_x_0 = const()[name = tensor("op_1150_transpose_x_0"), val = tensor(false)]; + tensor var_1150_transpose_y_0 = const()[name = tensor("op_1150_transpose_y_0"), val = tensor(false)]; + tensor var_1150 = matmul(transpose_x = var_1150_transpose_x_0, transpose_y = var_1150_transpose_y_0, x = q_9, y = prev_kv_9)[name = tensor("op_1150")]; + tensor var_1151 = real_div(x = sqrt_s0_9, y = sqrt_s_t_9)[name = tensor("op_1151")]; + tensor var_1152 = const()[name = tensor("op_1152"), val = tensor([1, 1, 4, 1])]; + tensor var_1153 = reshape(shape = var_1152, x = var_1151)[name = tensor("op_1153")]; + tensor cross_9 = mul(x = var_1150, y = var_1153)[name = tensor("cross_9")]; + tensor out_25 = add(x = inner_9, y = cross_9)[name = tensor("out_25")]; + tensor var_1156 = const()[name = tensor("op_1156"), val = tensor([1, 1, 4, 1])]; + tensor var_1157 = reshape(shape = var_1156, x = valid_mask)[name = tensor("op_1157")]; + tensor v_masked_1 = mul(x = v_9, y = var_1157)[name = tensor("v_masked_1")]; + tensor var_1159 = mul(x = prev_kv_9, y = sqrt_s0_9)[name = tensor("op_1159")]; + tensor var_1161_transpose_x_1 = const()[name = tensor("op_1161_transpose_x_1"), val = tensor(true)]; + tensor var_1161_transpose_y_1 = const()[name = tensor("op_1161_transpose_y_1"), val = tensor(false)]; + tensor var_1161 = matmul(transpose_x = var_1161_transpose_x_1, transpose_y = var_1161_transpose_y_1, x = k_9, y = v_masked_1)[name = tensor("op_1161")]; + tensor new_kv_unnorm_9 = add(x = var_1159, y = var_1161)[name = tensor("new_kv_unnorm_9")]; + tensor var_1163_keep_dims_0 = const()[name = tensor("op_1163_keep_dims_0"), val = tensor(false)]; + tensor var_1163 = reduce_sum(keep_dims = var_1163_keep_dims_0, x = valid_mask)[name = tensor("op_1163")]; + tensor var_1164 = const()[name = tensor("op_1164"), val = tensor([1])]; + tensor var_1165 = reshape(shape = var_1164, x = var_1163)[name = tensor("op_1165")]; + tensor new_scale_9 = add(x = prev_scale_9, y = var_1165)[name = tensor("new_scale_9")]; + tensor const_21 = const()[name = tensor("const_21"), val = tensor(0x1.fffffep+127)]; + tensor clip_2 = clip(alpha = var_1022, beta = const_21, x = new_scale_9)[name = tensor("clip_2")]; + tensor sqrt_new_scale_1 = sqrt(x = clip_2)[name = tensor("sqrt_new_scale_1")]; + tensor var_1169 = real_div(x = new_kv_unnorm_9, y = sqrt_new_scale_1)[name = tensor("op_1169")]; + tensor var_1170_perm_0 = const()[name = tensor("op_1170_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor out_27_axes_0 = const()[name = tensor("out_27_axes_0"), val = tensor([-1])]; + tensor var_1170 = transpose(perm = var_1170_perm_0, x = out_25)[name = tensor("transpose_24")]; + tensor out_27 = layer_norm(axes = out_27_axes_0, epsilon = var_1019, x = var_1170)[name = tensor("out_27")]; + tensor var_1174 = const()[name = tensor("op_1174"), val = tensor([6, 4, 256])]; + tensor out_29 = reshape(shape = var_1174, x = out_27)[name = tensor("out_29")]; + tensor var_1176 = silu(x = input_169)[name = tensor("op_1176")]; + tensor input_171 = mul(x = var_1176, y = out_29)[name = tensor("input_171")]; + tensor ret_out_9 = linear(bias = decoder_out_proj_0_bias, weight = decoder_out_proj_0_weight, x = input_171)[name = tensor("linear_42")]; + tensor input_173 = add(x = x_29, y = ret_out_9)[name = tensor("input_173")]; + tensor xt_1_axes_0 = const()[name = tensor("xt_1_axes_0"), val = tensor([-1])]; + tensor xt_1 = layer_norm(axes = xt_1_axes_0, beta = decoder_norm11_0_bias, epsilon = var_1017, gamma = decoder_norm11_0_weight, x = input_173)[name = tensor("xt_1")]; + tensor var_1186 = const()[name = tensor("op_1186"), val = tensor([1, 6, 4, 256])]; + tensor var_1187 = reshape(shape = var_1186, x = xt_1)[name = tensor("op_1187")]; + tensor var_1188_perm_0 = const()[name = tensor("op_1188_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_1191 = const()[name = tensor("op_1191"), val = tensor([4, 6, 256])]; + tensor var_1188 = transpose(perm = var_1188_perm_0, x = var_1187)[name = tensor("transpose_23")]; + tensor query_1 = reshape(shape = var_1191, x = var_1188)[name = tensor("query_1")]; + tensor query_3_perm_0 = const()[name = tensor("query_3_perm_0"), val = tensor([1, 0, 2])]; + tensor query_3 = transpose(perm = query_3_perm_0, x = query_1)[name = tensor("transpose_22")]; + tensor var_1214 = linear(bias = decoder_self_attn2_0_in_proj_bias, weight = decoder_self_attn2_0_in_proj_weight, x = query_3)[name = tensor("linear_43")]; + tensor concat_1 = const()[name = tensor("concat_1"), val = tensor([6, 4, 3, 256])]; + tensor var_1216 = reshape(shape = concat_1, x = var_1214)[name = tensor("op_1216")]; + tensor var_1217_axes_0 = const()[name = tensor("op_1217_axes_0"), val = tensor([0])]; + tensor var_1217 = expand_dims(axes = var_1217_axes_0, x = var_1216)[name = tensor("op_1217")]; + tensor var_1218_perm_0 = const()[name = tensor("op_1218_perm_0"), val = tensor([-2, 1, 2, 0, 4])]; + tensor var_1219_axes_0 = const()[name = tensor("op_1219_axes_0"), val = tensor([-2])]; + tensor var_1218 = transpose(perm = var_1218_perm_0, x = var_1217)[name = tensor("transpose_21")]; + tensor var_1219 = squeeze(axes = var_1219_axes_0, x = var_1218)[name = tensor("op_1219")]; + tensor q_11_begin_0 = const()[name = tensor("q_11_begin_0"), val = tensor([0, 0, 0, 0])]; + tensor q_11_end_0 = const()[name = tensor("q_11_end_0"), val = tensor([1, 6, 4, 256])]; + tensor q_11_end_mask_0 = const()[name = tensor("q_11_end_mask_0"), val = tensor([false, true, true, true])]; + tensor q_11_squeeze_mask_0 = const()[name = tensor("q_11_squeeze_mask_0"), val = tensor([true, false, false, false])]; + tensor q_11 = slice_by_index(begin = q_11_begin_0, end = q_11_end_0, end_mask = q_11_end_mask_0, squeeze_mask = q_11_squeeze_mask_0, x = var_1219)[name = tensor("q_11")]; + tensor k_11_begin_0 = const()[name = tensor("k_11_begin_0"), val = tensor([1, 0, 0, 0])]; + tensor k_11_end_0 = const()[name = tensor("k_11_end_0"), val = tensor([2, 6, 4, 256])]; + tensor k_11_end_mask_0 = const()[name = tensor("k_11_end_mask_0"), val = tensor([false, true, true, true])]; + tensor k_11_squeeze_mask_0 = const()[name = tensor("k_11_squeeze_mask_0"), val = tensor([true, false, false, false])]; + tensor k_11 = slice_by_index(begin = k_11_begin_0, end = k_11_end_0, end_mask = k_11_end_mask_0, squeeze_mask = k_11_squeeze_mask_0, x = var_1219)[name = tensor("k_11")]; + tensor v_11_begin_0 = const()[name = tensor("v_11_begin_0"), val = tensor([2, 0, 0, 0])]; + tensor v_11_end_0 = const()[name = tensor("v_11_end_0"), val = tensor([3, 6, 4, 256])]; + tensor v_11_end_mask_0 = const()[name = tensor("v_11_end_mask_0"), val = tensor([false, true, true, true])]; + tensor v_11_squeeze_mask_0 = const()[name = tensor("v_11_squeeze_mask_0"), val = tensor([true, false, false, false])]; + tensor v_11 = slice_by_index(begin = v_11_begin_0, end = v_11_end_0, end_mask = v_11_end_mask_0, squeeze_mask = v_11_squeeze_mask_0, x = var_1219)[name = tensor("v_11")]; + tensor var_1227 = const()[name = tensor("op_1227"), val = tensor([6, 16, 64])]; + tensor var_1228 = reshape(shape = var_1227, x = q_11)[name = tensor("op_1228")]; + tensor q_13_perm_0 = const()[name = tensor("q_13_perm_0"), val = tensor([1, 0, 2])]; + tensor var_1234 = const()[name = tensor("op_1234"), val = tensor([6, 16, 64])]; + tensor var_1235 = reshape(shape = var_1234, x = k_11)[name = tensor("op_1235")]; + tensor k_13_perm_0 = const()[name = tensor("k_13_perm_0"), val = tensor([1, 0, 2])]; + tensor var_1241 = const()[name = tensor("op_1241"), val = tensor([6, 16, 64])]; + tensor var_1242 = reshape(shape = var_1241, x = v_11)[name = tensor("op_1242")]; + tensor v_13_perm_0 = const()[name = tensor("v_13_perm_0"), val = tensor([1, 0, 2])]; + tensor var_1245 = const()[name = tensor("op_1245"), val = tensor([4, 4, 6, 64])]; + tensor q_13 = transpose(perm = q_13_perm_0, x = var_1228)[name = tensor("transpose_20")]; + tensor q_15 = reshape(shape = var_1245, x = q_13)[name = tensor("q_15")]; + tensor var_1247 = const()[name = tensor("op_1247"), val = tensor([4, 4, 6, 64])]; + tensor k_13 = transpose(perm = k_13_perm_0, x = var_1235)[name = tensor("transpose_19")]; + tensor k_15 = reshape(shape = var_1247, x = k_13)[name = tensor("k_15")]; + tensor var_1249 = const()[name = tensor("op_1249"), val = tensor([4, 4, 6, 64])]; + tensor v_13 = transpose(perm = v_13_perm_0, x = var_1242)[name = tensor("transpose_18")]; + tensor v_15 = reshape(shape = var_1249, x = v_13)[name = tensor("v_15")]; + tensor mul_1_y_0 = const()[name = tensor("mul_1_y_0"), val = tensor(0x1p-3)]; + tensor mul_1 = mul(x = q_15, y = mul_1_y_0)[name = tensor("mul_1")]; + tensor matmul_0_transpose_y_0 = const()[name = tensor("matmul_0_transpose_y_0"), val = tensor(true)]; + tensor matmul_0_transpose_x_0 = const()[name = tensor("matmul_0_transpose_x_0"), val = tensor(false)]; + tensor matmul_0 = matmul(transpose_x = matmul_0_transpose_x_0, transpose_y = matmul_0_transpose_y_0, x = mul_1, y = k_15)[name = tensor("matmul_0")]; + tensor softmax_0_axis_0 = const()[name = tensor("softmax_0_axis_0"), val = tensor(-1)]; + tensor softmax_0 = softmax(axis = softmax_0_axis_0, x = matmul_0)[name = tensor("softmax_0")]; + tensor attn_output_1_transpose_x_0 = const()[name = tensor("attn_output_1_transpose_x_0"), val = tensor(false)]; + tensor attn_output_1_transpose_y_0 = const()[name = tensor("attn_output_1_transpose_y_0"), val = tensor(false)]; + tensor attn_output_1 = matmul(transpose_x = attn_output_1_transpose_x_0, transpose_y = attn_output_1_transpose_y_0, x = softmax_0, y = v_15)[name = tensor("attn_output_1")]; + tensor var_1252 = const()[name = tensor("op_1252"), val = tensor([2, 0, 1, 3])]; + tensor var_1257 = const()[name = tensor("op_1257"), val = tensor([24, 256])]; + tensor var_1253 = transpose(perm = var_1252, x = attn_output_1)[name = tensor("transpose_17")]; + tensor attn_output_3 = reshape(shape = var_1257, x = var_1253)[name = tensor("attn_output_3")]; + tensor attn_output_5 = linear(bias = decoder_self_attn2_0_out_proj_bias, weight = decoder_self_attn2_0_out_proj_weight, x = attn_output_3)[name = tensor("linear_44")]; + tensor var_1261 = const()[name = tensor("op_1261"), val = tensor([6, 4, 256])]; + tensor attn_output_7 = reshape(shape = var_1261, x = attn_output_5)[name = tensor("attn_output_7")]; + tensor sa2_out_1_perm_0 = const()[name = tensor("sa2_out_1_perm_0"), val = tensor([1, 0, 2])]; + tensor sa2_out_1 = transpose(perm = sa2_out_1_perm_0, x = attn_output_7)[name = tensor("transpose_16")]; + tensor input_175 = add(x = query_1, y = sa2_out_1)[name = tensor("input_175")]; + tensor input_177_axes_0 = const()[name = tensor("input_177_axes_0"), val = tensor([-1])]; + tensor input_177 = layer_norm(axes = input_177_axes_0, beta = decoder_norm21_0_bias, epsilon = var_1017, gamma = decoder_norm21_0_weight, x = input_175)[name = tensor("input_177")]; + tensor input_179 = linear(bias = decoder_linear1_0_bias, weight = decoder_linear1_0_weight, x = input_177)[name = tensor("linear_45")]; + tensor input_181 = relu(x = input_179)[name = tensor("input_181")]; + tensor ffn_out_1 = linear(bias = decoder_linear2_0_bias, weight = decoder_linear2_0_weight, x = input_181)[name = tensor("linear_46")]; + tensor input_183 = add(x = input_177, y = ffn_out_1)[name = tensor("input_183")]; + tensor xt_3_axes_0 = const()[name = tensor("xt_3_axes_0"), val = tensor([-1])]; + tensor xt_3 = layer_norm(axes = xt_3_axes_0, beta = decoder_norm22_0_bias, epsilon = var_1017, gamma = decoder_norm22_0_weight, x = input_183)[name = tensor("xt_3")]; + tensor var_1281 = const()[name = tensor("op_1281"), val = tensor([1, 4, 6, 256])]; + tensor x_31 = reshape(shape = var_1281, x = xt_3)[name = tensor("x_31")]; + tensor var_1283_perm_0 = const()[name = tensor("op_1283_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_1287 = const()[name = tensor("op_1287"), val = tensor([6, 4, 256])]; + tensor var_1283 = transpose(perm = var_1283_perm_0, x = x_31)[name = tensor("transpose_15")]; + tensor x = reshape(shape = var_1287, x = var_1283)[name = tensor("x")]; + tensor prev_kv_begin_0 = const()[name = tensor("prev_kv_begin_0"), val = tensor([1, 0, 0, 0, 0])]; + tensor prev_kv_end_0 = const()[name = tensor("prev_kv_end_0"), val = tensor([2, 6, 4, 64, 64])]; + tensor prev_kv_end_mask_0 = const()[name = tensor("prev_kv_end_mask_0"), val = tensor([false, true, true, true, true])]; + tensor prev_kv_squeeze_mask_0 = const()[name = tensor("prev_kv_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; + tensor prev_kv = slice_by_index(begin = prev_kv_begin_0, end = prev_kv_end_0, end_mask = prev_kv_end_mask_0, squeeze_mask = prev_kv_squeeze_mask_0, x = dec_kv)[name = tensor("prev_kv")]; + tensor prev_scale_begin_0 = const()[name = tensor("prev_scale_begin_0"), val = tensor([1, 0])]; + tensor prev_scale_end_0 = const()[name = tensor("prev_scale_end_0"), val = tensor([2, 1])]; + tensor prev_scale_end_mask_0 = const()[name = tensor("prev_scale_end_mask_0"), val = tensor([false, true])]; + tensor prev_scale_squeeze_mask_0 = const()[name = tensor("prev_scale_squeeze_mask_0"), val = tensor([true, false])]; + tensor prev_scale = slice_by_index(begin = prev_scale_begin_0, end = prev_scale_end_0, end_mask = prev_scale_end_mask_0, squeeze_mask = prev_scale_squeeze_mask_0, x = dec_scale)[name = tensor("prev_scale")]; + tensor var_1295 = linear(bias = decoder_q_proj_1_bias, weight = decoder_q_proj_1_weight, x = x)[name = tensor("linear_47")]; + tensor var_1296 = const()[name = tensor("op_1296"), val = tensor([6, 4, 4, 64])]; + tensor var_1297 = reshape(shape = var_1296, x = var_1295)[name = tensor("op_1297")]; + tensor q_17_perm_0 = const()[name = tensor("q_17_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_1301 = linear(bias = decoder_k_proj_1_bias, weight = decoder_k_proj_1_weight, x = x)[name = tensor("linear_48")]; + tensor var_1302 = const()[name = tensor("op_1302"), val = tensor(0x1p-3)]; + tensor var_1303 = mul(x = var_1301, y = var_1302)[name = tensor("op_1303")]; + tensor var_1304 = const()[name = tensor("op_1304"), val = tensor([6, 4, 4, 64])]; + tensor var_1305 = reshape(shape = var_1304, x = var_1303)[name = tensor("op_1305")]; + tensor k_17_perm_0 = const()[name = tensor("k_17_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_1309 = linear(bias = decoder_v_proj_1_bias, weight = decoder_v_proj_1_weight, x = x)[name = tensor("linear_49")]; + tensor var_1310 = const()[name = tensor("op_1310"), val = tensor([6, 4, 4, 64])]; + tensor var_1311 = reshape(shape = var_1310, x = var_1309)[name = tensor("op_1311")]; + tensor v_17_perm_0 = const()[name = tensor("v_17_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor input_187 = linear(bias = decoder_g_proj_1_bias, weight = decoder_g_proj_1_weight, x = x)[name = tensor("linear_50")]; + tensor sqrt_s0 = sqrt(x = prev_scale)[name = tensor("sqrt_s0")]; + tensor s_t = add(x = prev_scale, y = cumsum_mask_1)[name = tensor("s_t")]; + tensor const_32 = const()[name = tensor("const_32"), val = tensor(0x1.fffffep+127)]; + tensor clip_3 = clip(alpha = var_1022, beta = const_32, x = s_t)[name = tensor("clip_3")]; + tensor sqrt_s_t = sqrt(x = clip_3)[name = tensor("sqrt_s_t")]; + tensor qk_transpose_x_1 = const()[name = tensor("qk_transpose_x_1"), val = tensor(false)]; + tensor qk_transpose_y_1 = const()[name = tensor("qk_transpose_y_1"), val = tensor(true)]; + tensor k_17 = transpose(perm = k_17_perm_0, x = var_1305)[name = tensor("transpose_13")]; + tensor q_17 = transpose(perm = q_17_perm_0, x = var_1297)[name = tensor("transpose_14")]; + tensor qk = matmul(transpose_x = qk_transpose_x_1, transpose_y = qk_transpose_y_1, x = q_17, y = k_17)[name = tensor("qk")]; + tensor var_1326 = const()[name = tensor("op_1326"), val = tensor([4, 1])]; + tensor var_1327 = reshape(shape = var_1326, x = sqrt_s_t)[name = tensor("op_1327")]; + tensor M = real_div(x = causal_with_valid_1, y = var_1327)[name = tensor("M")]; + tensor var_1329 = mul(x = qk, y = M)[name = tensor("op_1329")]; + tensor inner_transpose_x_0 = const()[name = tensor("inner_transpose_x_0"), val = tensor(false)]; + tensor inner_transpose_y_0 = const()[name = tensor("inner_transpose_y_0"), val = tensor(false)]; + tensor v_17 = transpose(perm = v_17_perm_0, x = var_1311)[name = tensor("transpose_12")]; + tensor inner = matmul(transpose_x = inner_transpose_x_0, transpose_y = inner_transpose_y_0, x = var_1329, y = v_17)[name = tensor("inner")]; + tensor var_1331_transpose_x_0 = const()[name = tensor("op_1331_transpose_x_0"), val = tensor(false)]; + tensor var_1331_transpose_y_0 = const()[name = tensor("op_1331_transpose_y_0"), val = tensor(false)]; + tensor var_1331 = matmul(transpose_x = var_1331_transpose_x_0, transpose_y = var_1331_transpose_y_0, x = q_17, y = prev_kv)[name = tensor("op_1331")]; + tensor var_1332 = real_div(x = sqrt_s0, y = sqrt_s_t)[name = tensor("op_1332")]; + tensor var_1333 = const()[name = tensor("op_1333"), val = tensor([1, 1, 4, 1])]; + tensor var_1334 = reshape(shape = var_1333, x = var_1332)[name = tensor("op_1334")]; + tensor cross = mul(x = var_1331, y = var_1334)[name = tensor("cross")]; + tensor out_31 = add(x = inner, y = cross)[name = tensor("out_31")]; + tensor v_masked = mul(x = v_17, y = var_1157)[name = tensor("v_masked")]; + tensor var_1340 = mul(x = prev_kv, y = sqrt_s0)[name = tensor("op_1340")]; + tensor var_1342_transpose_x_1 = const()[name = tensor("op_1342_transpose_x_1"), val = tensor(true)]; + tensor var_1342_transpose_y_1 = const()[name = tensor("op_1342_transpose_y_1"), val = tensor(false)]; + tensor var_1342 = matmul(transpose_x = var_1342_transpose_x_1, transpose_y = var_1342_transpose_y_1, x = k_17, y = v_masked)[name = tensor("op_1342")]; + tensor new_kv_unnorm = add(x = var_1340, y = var_1342)[name = tensor("new_kv_unnorm")]; + tensor new_scale = add(x = prev_scale, y = var_1165)[name = tensor("new_scale")]; + tensor const_33 = const()[name = tensor("const_33"), val = tensor(0x1.fffffep+127)]; + tensor clip_4 = clip(alpha = var_1022, beta = const_33, x = new_scale)[name = tensor("clip_4")]; + tensor sqrt_new_scale = sqrt(x = clip_4)[name = tensor("sqrt_new_scale")]; + tensor nkv = real_div(x = new_kv_unnorm, y = sqrt_new_scale)[name = tensor("nkv")]; + tensor var_1351_perm_0 = const()[name = tensor("op_1351_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor out_33_axes_0 = const()[name = tensor("out_33_axes_0"), val = tensor([-1])]; + tensor var_1351 = transpose(perm = var_1351_perm_0, x = out_31)[name = tensor("transpose_11")]; + tensor out_33 = layer_norm(axes = out_33_axes_0, epsilon = var_1019, x = var_1351)[name = tensor("out_33")]; + tensor var_1355 = const()[name = tensor("op_1355"), val = tensor([6, 4, 256])]; + tensor out = reshape(shape = var_1355, x = out_33)[name = tensor("out")]; + tensor var_1357 = silu(x = input_187)[name = tensor("op_1357")]; + tensor input_189 = mul(x = var_1357, y = out)[name = tensor("input_189")]; + tensor ret_out = linear(bias = decoder_out_proj_1_bias, weight = decoder_out_proj_1_weight, x = input_189)[name = tensor("linear_51")]; + tensor input_191 = add(x = x, y = ret_out)[name = tensor("input_191")]; + tensor xt_5_axes_0 = const()[name = tensor("xt_5_axes_0"), val = tensor([-1])]; + tensor xt_5 = layer_norm(axes = xt_5_axes_0, beta = decoder_norm11_1_bias, epsilon = var_1017, gamma = decoder_norm11_1_weight, x = input_191)[name = tensor("xt_5")]; + tensor var_1367 = const()[name = tensor("op_1367"), val = tensor([1, 6, 4, 256])]; + tensor var_1368 = reshape(shape = var_1367, x = xt_5)[name = tensor("op_1368")]; + tensor var_1369_perm_0 = const()[name = tensor("op_1369_perm_0"), val = tensor([0, 2, 1, 3])]; + tensor var_1372 = const()[name = tensor("op_1372"), val = tensor([4, 6, 256])]; + tensor var_1369 = transpose(perm = var_1369_perm_0, x = var_1368)[name = tensor("transpose_10")]; + tensor query_5 = reshape(shape = var_1372, x = var_1369)[name = tensor("query_5")]; + tensor query_perm_0 = const()[name = tensor("query_perm_0"), val = tensor([1, 0, 2])]; + tensor query = transpose(perm = query_perm_0, x = query_5)[name = tensor("transpose_9")]; + tensor var_1395 = linear(bias = decoder_self_attn2_1_in_proj_bias, weight = decoder_self_attn2_1_in_proj_weight, x = query)[name = tensor("linear_52")]; + tensor concat_2 = const()[name = tensor("concat_2"), val = tensor([6, 4, 3, 256])]; + tensor var_1397 = reshape(shape = concat_2, x = var_1395)[name = tensor("op_1397")]; + tensor var_1398_axes_0 = const()[name = tensor("op_1398_axes_0"), val = tensor([0])]; + tensor var_1398 = expand_dims(axes = var_1398_axes_0, x = var_1397)[name = tensor("op_1398")]; + tensor var_1399_perm_0 = const()[name = tensor("op_1399_perm_0"), val = tensor([-2, 1, 2, 0, 4])]; + tensor var_1400_axes_0 = const()[name = tensor("op_1400_axes_0"), val = tensor([-2])]; + tensor var_1399 = transpose(perm = var_1399_perm_0, x = var_1398)[name = tensor("transpose_8")]; + tensor var_1400 = squeeze(axes = var_1400_axes_0, x = var_1399)[name = tensor("op_1400")]; + tensor q_19_begin_0 = const()[name = tensor("q_19_begin_0"), val = tensor([0, 0, 0, 0])]; + tensor q_19_end_0 = const()[name = tensor("q_19_end_0"), val = tensor([1, 6, 4, 256])]; + tensor q_19_end_mask_0 = const()[name = tensor("q_19_end_mask_0"), val = tensor([false, true, true, true])]; + tensor q_19_squeeze_mask_0 = const()[name = tensor("q_19_squeeze_mask_0"), val = tensor([true, false, false, false])]; + tensor q_19 = slice_by_index(begin = q_19_begin_0, end = q_19_end_0, end_mask = q_19_end_mask_0, squeeze_mask = q_19_squeeze_mask_0, x = var_1400)[name = tensor("q_19")]; + tensor k_19_begin_0 = const()[name = tensor("k_19_begin_0"), val = tensor([1, 0, 0, 0])]; + tensor k_19_end_0 = const()[name = tensor("k_19_end_0"), val = tensor([2, 6, 4, 256])]; + tensor k_19_end_mask_0 = const()[name = tensor("k_19_end_mask_0"), val = tensor([false, true, true, true])]; + tensor k_19_squeeze_mask_0 = const()[name = tensor("k_19_squeeze_mask_0"), val = tensor([true, false, false, false])]; + tensor k_19 = slice_by_index(begin = k_19_begin_0, end = k_19_end_0, end_mask = k_19_end_mask_0, squeeze_mask = k_19_squeeze_mask_0, x = var_1400)[name = tensor("k_19")]; + tensor v_19_begin_0 = const()[name = tensor("v_19_begin_0"), val = tensor([2, 0, 0, 0])]; + tensor v_19_end_0 = const()[name = tensor("v_19_end_0"), val = tensor([3, 6, 4, 256])]; + tensor v_19_end_mask_0 = const()[name = tensor("v_19_end_mask_0"), val = tensor([false, true, true, true])]; + tensor v_19_squeeze_mask_0 = const()[name = tensor("v_19_squeeze_mask_0"), val = tensor([true, false, false, false])]; + tensor v_19 = slice_by_index(begin = v_19_begin_0, end = v_19_end_0, end_mask = v_19_end_mask_0, squeeze_mask = v_19_squeeze_mask_0, x = var_1400)[name = tensor("v_19")]; + tensor var_1408 = const()[name = tensor("op_1408"), val = tensor([6, 16, 64])]; + tensor var_1409 = reshape(shape = var_1408, x = q_19)[name = tensor("op_1409")]; + tensor q_21_perm_0 = const()[name = tensor("q_21_perm_0"), val = tensor([1, 0, 2])]; + tensor var_1415 = const()[name = tensor("op_1415"), val = tensor([6, 16, 64])]; + tensor var_1416 = reshape(shape = var_1415, x = k_19)[name = tensor("op_1416")]; + tensor k_21_perm_0 = const()[name = tensor("k_21_perm_0"), val = tensor([1, 0, 2])]; + tensor var_1422 = const()[name = tensor("op_1422"), val = tensor([6, 16, 64])]; + tensor var_1423 = reshape(shape = var_1422, x = v_19)[name = tensor("op_1423")]; + tensor v_21_perm_0 = const()[name = tensor("v_21_perm_0"), val = tensor([1, 0, 2])]; + tensor var_1426 = const()[name = tensor("op_1426"), val = tensor([4, 4, 6, 64])]; + tensor q_21 = transpose(perm = q_21_perm_0, x = var_1409)[name = tensor("transpose_7")]; + tensor q = reshape(shape = var_1426, x = q_21)[name = tensor("q")]; + tensor var_1428 = const()[name = tensor("op_1428"), val = tensor([4, 4, 6, 64])]; + tensor k_21 = transpose(perm = k_21_perm_0, x = var_1416)[name = tensor("transpose_6")]; + tensor k = reshape(shape = var_1428, x = k_21)[name = tensor("k")]; + tensor var_1430 = const()[name = tensor("op_1430"), val = tensor([4, 4, 6, 64])]; + tensor v_21 = transpose(perm = v_21_perm_0, x = var_1423)[name = tensor("transpose_5")]; + tensor v = reshape(shape = var_1430, x = v_21)[name = tensor("v")]; + tensor mul_3_y_0 = const()[name = tensor("mul_3_y_0"), val = tensor(0x1p-3)]; + tensor mul_3 = mul(x = q, y = mul_3_y_0)[name = tensor("mul_3")]; + tensor matmul_1_transpose_y_0 = const()[name = tensor("matmul_1_transpose_y_0"), val = tensor(true)]; + tensor matmul_1_transpose_x_0 = const()[name = tensor("matmul_1_transpose_x_0"), val = tensor(false)]; + tensor matmul_1 = matmul(transpose_x = matmul_1_transpose_x_0, transpose_y = matmul_1_transpose_y_0, x = mul_3, y = k)[name = tensor("matmul_1")]; + tensor softmax_1_axis_0 = const()[name = tensor("softmax_1_axis_0"), val = tensor(-1)]; + tensor softmax_1 = softmax(axis = softmax_1_axis_0, x = matmul_1)[name = tensor("softmax_1")]; + tensor attn_output_9_transpose_x_0 = const()[name = tensor("attn_output_9_transpose_x_0"), val = tensor(false)]; + tensor attn_output_9_transpose_y_0 = const()[name = tensor("attn_output_9_transpose_y_0"), val = tensor(false)]; + tensor attn_output_9 = matmul(transpose_x = attn_output_9_transpose_x_0, transpose_y = attn_output_9_transpose_y_0, x = softmax_1, y = v)[name = tensor("attn_output_9")]; + tensor var_1433 = const()[name = tensor("op_1433"), val = tensor([2, 0, 1, 3])]; + tensor var_1438 = const()[name = tensor("op_1438"), val = tensor([24, 256])]; + tensor var_1434 = transpose(perm = var_1433, x = attn_output_9)[name = tensor("transpose_4")]; + tensor attn_output_11 = reshape(shape = var_1438, x = var_1434)[name = tensor("attn_output_11")]; + tensor attn_output_13 = linear(bias = decoder_self_attn2_1_out_proj_bias, weight = decoder_self_attn2_1_out_proj_weight, x = attn_output_11)[name = tensor("linear_53")]; + tensor var_1442 = const()[name = tensor("op_1442"), val = tensor([6, 4, 256])]; + tensor attn_output = reshape(shape = var_1442, x = attn_output_13)[name = tensor("attn_output")]; + tensor sa2_out_perm_0 = const()[name = tensor("sa2_out_perm_0"), val = tensor([1, 0, 2])]; + tensor sa2_out = transpose(perm = sa2_out_perm_0, x = attn_output)[name = tensor("transpose_3")]; + tensor input_193 = add(x = query_5, y = sa2_out)[name = tensor("input_193")]; + tensor input_195_axes_0 = const()[name = tensor("input_195_axes_0"), val = tensor([-1])]; + tensor input_195 = layer_norm(axes = input_195_axes_0, beta = decoder_norm21_1_bias, epsilon = var_1017, gamma = decoder_norm21_1_weight, x = input_193)[name = tensor("input_195")]; + tensor input_197 = linear(bias = decoder_linear1_1_bias, weight = decoder_linear1_1_weight, x = input_195)[name = tensor("linear_54")]; + tensor input_199 = relu(x = input_197)[name = tensor("input_199")]; + tensor ffn_out = linear(bias = decoder_linear2_1_bias, weight = decoder_linear2_1_weight, x = input_199)[name = tensor("linear_55")]; + tensor input_201 = add(x = input_195, y = ffn_out)[name = tensor("input_201")]; + tensor xt_axes_0 = const()[name = tensor("xt_axes_0"), val = tensor([-1])]; + tensor xt = layer_norm(axes = xt_axes_0, beta = decoder_norm22_1_bias, epsilon = var_1017, gamma = decoder_norm22_1_weight, x = input_201)[name = tensor("xt")]; + tensor var_1462 = const()[name = tensor("op_1462"), val = tensor([1, 4, 6, 256])]; + tensor input = reshape(shape = var_1462, x = xt)[name = tensor("input")]; + tensor var_1464 = const()[name = tensor("op_1464"), val = tensor([-1])]; + tensor var_1465 = reduce_l2_norm(axes = var_1464, keep_dims = var_1020, x = input)[name = tensor("op_1465")]; + tensor const_42 = const()[name = tensor("const_42"), val = tensor(0x1.fffffep+127)]; + tensor clip_5 = clip(alpha = var_1012, beta = const_42, x = var_1465)[name = tensor("clip_5")]; + tensor var_1467 = real_div(x = input, y = clip_5)[name = tensor("op_1467")]; + tensor concat_6 = const()[name = tensor("concat_6"), val = tensor([4, 1, 256])]; + tensor reshape_0 = reshape(shape = concat_6, x = emb)[name = tensor("reshape_0")]; + tensor transpose_1_perm_0 = const()[name = tensor("transpose_1_perm_0"), val = tensor([0, 1, 3, 2])]; + tensor concat_7 = const()[name = tensor("concat_7"), val = tensor([4, 256, 6])]; + tensor transpose_1 = transpose(perm = transpose_1_perm_0, x = var_1467)[name = tensor("transpose_2")]; + tensor reshape_1 = reshape(shape = concat_7, x = transpose_1)[name = tensor("reshape_1")]; + 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_0, y = reshape_1)[name = tensor("matmul_2")]; + tensor concat_10 = const()[name = tensor("concat_10"), val = tensor([1, 4, 6])]; + tensor reshape_2 = reshape(shape = concat_10, x = matmul_2)[name = tensor("reshape_2")]; + tensor output_begin_0 = const()[name = tensor("output_begin_0"), val = tensor([0, 0, 1])]; + tensor output_end_0 = const()[name = tensor("output_end_0"), val = tensor([1, 4, 5])]; + tensor output_end_mask_0 = const()[name = tensor("output_end_mask_0"), val = tensor([true, true, false])]; + tensor output = slice_by_index(begin = output_begin_0, end = output_end_0, end_mask = output_end_mask_0, x = reshape_2)[name = tensor("output")]; + tensor probs = sigmoid(x = output)[name = tensor("op_1471")]; + tensor var_1473_axis_0 = const()[name = tensor("op_1473_axis_0"), val = tensor(0)]; + tensor dec_kv_new = stack(axis = var_1473_axis_0, values = (var_1169, nkv))[name = tensor("op_1473")]; + tensor var_1475_axis_0 = const()[name = tensor("op_1475_axis_0"), val = tensor(0)]; + tensor dec_scale_new = stack(axis = var_1475_axis_0, values = (new_scale_9, new_scale))[name = tensor("op_1475")]; + } -> (probs, enc_kv_new, enc_scale_new, enc_conv_cache_new, cnn_window_new, dec_kv_new, dec_scale_new); +} \ No newline at end of file