program(1.0) [buildInfo = dict, tensor>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1.2"}, {"coremltools-component-torch", "2.7.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})] { func main(tensor codes) [FlexibleShapeInformation = tuple, dict, tensor>>, tuple, dict, list, ?>>>>((("DefaultShapes", {{"codes", [16, 1, 57]}}), ("RangeDims", {{"codes", [[16, 16], [1, 1], [1, 125]]}})))] { tensor input_1_begin_0 = const()[name = tensor("input_1_begin_0"), val = tensor([0, 0, 0])]; tensor input_1_end_0 = const()[name = tensor("input_1_end_0"), val = tensor([1, 1, 0])]; tensor input_1_end_mask_0 = const()[name = tensor("input_1_end_mask_0"), val = tensor([false, true, true])]; tensor input_1_squeeze_mask_0 = const()[name = tensor("input_1_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_1 = slice_by_index(begin = input_1_begin_0, end = input_1_end_0, end_mask = input_1_end_mask_0, squeeze_mask = input_1_squeeze_mask_0, x = codes)[name = tensor("input_1")]; tensor var_118_batch_dims_0 = const()[name = tensor("op_118_batch_dims_0"), val = tensor(0)]; tensor var_118_validate_indices_0 = const()[name = tensor("op_118_validate_indices_0"), val = tensor(false)]; tensor quantizers_0_codebook_weight_to_fp16 = const()[name = tensor("quantizers_0_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(64)))]; tensor input_1_to_int16_dtype_0 = const()[name = tensor("input_1_to_int16_dtype_0"), val = tensor("int16")]; tensor cast_291_dtype_0 = const()[name = tensor("cast_291_dtype_0"), val = tensor("int32")]; tensor greater_equal_0_y_0 = const()[name = tensor("greater_equal_0_y_0"), val = tensor(0)]; tensor input_1_to_int16 = cast(dtype = input_1_to_int16_dtype_0, x = input_1)[name = tensor("cast_419")]; tensor cast_291 = cast(dtype = cast_291_dtype_0, x = input_1_to_int16)[name = tensor("cast_418")]; tensor greater_equal_0 = greater_equal(x = cast_291, y = greater_equal_0_y_0)[name = tensor("greater_equal_0")]; tensor slice_by_index_0 = const()[name = tensor("slice_by_index_0"), val = tensor(1024)]; tensor add_12 = add(x = cast_291, y = slice_by_index_0)[name = tensor("add_12")]; tensor select_0 = select(a = cast_291, b = add_12, cond = greater_equal_0)[name = tensor("select_0")]; tensor var_118_cast_fp16_cast_uint16_axis_0 = const()[name = tensor("op_118_cast_fp16_cast_uint16_axis_0"), val = tensor(0)]; tensor select_0_to_int16_dtype_0 = const()[name = tensor("select_0_to_int16_dtype_0"), val = tensor("int16")]; tensor select_0_to_int16 = cast(dtype = select_0_to_int16_dtype_0, x = select_0)[name = tensor("cast_417")]; tensor var_118_cast_fp16_cast_uint16_cast_uint16 = gather(axis = var_118_cast_fp16_cast_uint16_axis_0, batch_dims = var_118_batch_dims_0, indices = select_0_to_int16, validate_indices = var_118_validate_indices_0, x = quantizers_0_codebook_weight_to_fp16)[name = tensor("op_118_cast_fp16_cast_uint16_cast_uint16")]; tensor input_3_perm_0 = const()[name = tensor("input_3_perm_0"), val = tensor([0, 2, 1])]; tensor emb_1_pad_type_0 = const()[name = tensor("emb_1_pad_type_0"), val = tensor("valid")]; tensor emb_1_strides_0 = const()[name = tensor("emb_1_strides_0"), val = tensor([1])]; tensor emb_1_pad_0 = const()[name = tensor("emb_1_pad_0"), val = tensor([0, 0])]; tensor emb_1_dilations_0 = const()[name = tensor("emb_1_dilations_0"), val = tensor([1])]; tensor emb_1_groups_0 = const()[name = tensor("emb_1_groups_0"), val = tensor(1)]; tensor quantizers_0_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_0_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(16512)))]; tensor quantizers_0_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_0_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(24768)))]; tensor input_3_cast_fp16 = transpose(perm = input_3_perm_0, x = var_118_cast_fp16_cast_uint16_cast_uint16)[name = tensor("transpose_53")]; tensor emb_1_cast_fp16 = conv(bias = quantizers_0_out_proj_bias_to_fp16, dilations = emb_1_dilations_0, groups = emb_1_groups_0, pad = emb_1_pad_0, pad_type = emb_1_pad_type_0, strides = emb_1_strides_0, weight = quantizers_0_out_proj_weight_to_fp16, x = input_3_cast_fp16)[name = tensor("emb_1_cast_fp16")]; tensor input_5_begin_0 = const()[name = tensor("input_5_begin_0"), val = tensor([1, 0, 0])]; tensor input_5_end_0 = const()[name = tensor("input_5_end_0"), val = tensor([2, 1, 0])]; tensor input_5_end_mask_0 = const()[name = tensor("input_5_end_mask_0"), val = tensor([false, true, true])]; tensor input_5_squeeze_mask_0 = const()[name = tensor("input_5_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_5 = slice_by_index(begin = input_5_begin_0, end = input_5_end_0, end_mask = input_5_end_mask_0, squeeze_mask = input_5_squeeze_mask_0, x = codes)[name = tensor("input_5")]; tensor var_139_axis_0 = const()[name = tensor("op_139_axis_0"), val = tensor(0)]; tensor var_139_batch_dims_0 = const()[name = tensor("op_139_batch_dims_0"), val = tensor(0)]; tensor var_139_validate_indices_0 = const()[name = tensor("op_139_validate_indices_0"), val = tensor(false)]; tensor quantizers_1_codebook_weight_to_fp16 = const()[name = tensor("quantizers_1_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(25856)))]; tensor input_5_to_uint16_dtype_0 = const()[name = tensor("input_5_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_5_to_uint16 = cast(dtype = input_5_to_uint16_dtype_0, x = input_5)[name = tensor("cast_416")]; tensor var_139_cast_fp16_cast_uint16 = gather(axis = var_139_axis_0, batch_dims = var_139_batch_dims_0, indices = input_5_to_uint16, validate_indices = var_139_validate_indices_0, x = quantizers_1_codebook_weight_to_fp16)[name = tensor("op_139_cast_fp16_cast_uint16")]; tensor input_7_perm_0 = const()[name = tensor("input_7_perm_0"), val = tensor([0, 2, 1])]; tensor z_1_pad_type_0 = const()[name = tensor("z_1_pad_type_0"), val = tensor("valid")]; tensor z_1_strides_0 = const()[name = tensor("z_1_strides_0"), val = tensor([1])]; tensor z_1_pad_0 = const()[name = tensor("z_1_pad_0"), val = tensor([0, 0])]; tensor z_1_dilations_0 = const()[name = tensor("z_1_dilations_0"), val = tensor([1])]; tensor z_1_groups_0 = const()[name = tensor("z_1_groups_0"), val = tensor(1)]; tensor quantizers_1_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_1_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(42304)))]; tensor quantizers_1_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_1_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(50560)))]; tensor input_7_cast_fp16 = transpose(perm = input_7_perm_0, x = var_139_cast_fp16_cast_uint16)[name = tensor("transpose_52")]; tensor z_1_cast_fp16 = conv(bias = quantizers_1_out_proj_bias_to_fp16, dilations = z_1_dilations_0, groups = z_1_groups_0, pad = z_1_pad_0, pad_type = z_1_pad_type_0, strides = z_1_strides_0, weight = quantizers_1_out_proj_weight_to_fp16, x = input_7_cast_fp16)[name = tensor("z_1_cast_fp16")]; tensor emb_3_cast_fp16 = add(x = emb_1_cast_fp16, y = z_1_cast_fp16)[name = tensor("emb_3_cast_fp16")]; tensor input_9_begin_0 = const()[name = tensor("input_9_begin_0"), val = tensor([2, 0, 0])]; tensor input_9_end_0 = const()[name = tensor("input_9_end_0"), val = tensor([3, 1, 0])]; tensor input_9_end_mask_0 = const()[name = tensor("input_9_end_mask_0"), val = tensor([false, true, true])]; tensor input_9_squeeze_mask_0 = const()[name = tensor("input_9_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_9 = slice_by_index(begin = input_9_begin_0, end = input_9_end_0, end_mask = input_9_end_mask_0, squeeze_mask = input_9_squeeze_mask_0, x = codes)[name = tensor("input_9")]; tensor var_162_axis_0 = const()[name = tensor("op_162_axis_0"), val = tensor(0)]; tensor var_162_batch_dims_0 = const()[name = tensor("op_162_batch_dims_0"), val = tensor(0)]; tensor var_162_validate_indices_0 = const()[name = tensor("op_162_validate_indices_0"), val = tensor(false)]; tensor quantizers_2_codebook_weight_to_fp16 = const()[name = tensor("quantizers_2_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(51648)))]; tensor input_9_to_uint16_dtype_0 = const()[name = tensor("input_9_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_9_to_uint16 = cast(dtype = input_9_to_uint16_dtype_0, x = input_9)[name = tensor("cast_415")]; tensor var_162_cast_fp16_cast_uint16 = gather(axis = var_162_axis_0, batch_dims = var_162_batch_dims_0, indices = input_9_to_uint16, validate_indices = var_162_validate_indices_0, x = quantizers_2_codebook_weight_to_fp16)[name = tensor("op_162_cast_fp16_cast_uint16")]; tensor input_11_perm_0 = const()[name = tensor("input_11_perm_0"), val = tensor([0, 2, 1])]; tensor z_3_pad_type_0 = const()[name = tensor("z_3_pad_type_0"), val = tensor("valid")]; tensor z_3_strides_0 = const()[name = tensor("z_3_strides_0"), val = tensor([1])]; tensor z_3_pad_0 = const()[name = tensor("z_3_pad_0"), val = tensor([0, 0])]; tensor z_3_dilations_0 = const()[name = tensor("z_3_dilations_0"), val = tensor([1])]; tensor z_3_groups_0 = const()[name = tensor("z_3_groups_0"), val = tensor(1)]; tensor quantizers_2_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_2_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(68096)))]; tensor quantizers_2_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_2_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(76352)))]; tensor input_11_cast_fp16 = transpose(perm = input_11_perm_0, x = var_162_cast_fp16_cast_uint16)[name = tensor("transpose_51")]; tensor z_3_cast_fp16 = conv(bias = quantizers_2_out_proj_bias_to_fp16, dilations = z_3_dilations_0, groups = z_3_groups_0, pad = z_3_pad_0, pad_type = z_3_pad_type_0, strides = z_3_strides_0, weight = quantizers_2_out_proj_weight_to_fp16, x = input_11_cast_fp16)[name = tensor("z_3_cast_fp16")]; tensor emb_5_cast_fp16 = add(x = emb_3_cast_fp16, y = z_3_cast_fp16)[name = tensor("emb_5_cast_fp16")]; tensor input_13_begin_0 = const()[name = tensor("input_13_begin_0"), val = tensor([3, 0, 0])]; tensor input_13_end_0 = const()[name = tensor("input_13_end_0"), val = tensor([4, 1, 0])]; tensor input_13_end_mask_0 = const()[name = tensor("input_13_end_mask_0"), val = tensor([false, true, true])]; tensor input_13_squeeze_mask_0 = const()[name = tensor("input_13_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_13 = slice_by_index(begin = input_13_begin_0, end = input_13_end_0, end_mask = input_13_end_mask_0, squeeze_mask = input_13_squeeze_mask_0, x = codes)[name = tensor("input_13")]; tensor var_185_axis_0 = const()[name = tensor("op_185_axis_0"), val = tensor(0)]; tensor var_185_batch_dims_0 = const()[name = tensor("op_185_batch_dims_0"), val = tensor(0)]; tensor var_185_validate_indices_0 = const()[name = tensor("op_185_validate_indices_0"), val = tensor(false)]; tensor quantizers_3_codebook_weight_to_fp16 = const()[name = tensor("quantizers_3_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(77440)))]; tensor input_13_to_uint16_dtype_0 = const()[name = tensor("input_13_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_13_to_uint16 = cast(dtype = input_13_to_uint16_dtype_0, x = input_13)[name = tensor("cast_414")]; tensor var_185_cast_fp16_cast_uint16 = gather(axis = var_185_axis_0, batch_dims = var_185_batch_dims_0, indices = input_13_to_uint16, validate_indices = var_185_validate_indices_0, x = quantizers_3_codebook_weight_to_fp16)[name = tensor("op_185_cast_fp16_cast_uint16")]; tensor input_15_perm_0 = const()[name = tensor("input_15_perm_0"), val = tensor([0, 2, 1])]; tensor z_5_pad_type_0 = const()[name = tensor("z_5_pad_type_0"), val = tensor("valid")]; tensor z_5_strides_0 = const()[name = tensor("z_5_strides_0"), val = tensor([1])]; tensor z_5_pad_0 = const()[name = tensor("z_5_pad_0"), val = tensor([0, 0])]; tensor z_5_dilations_0 = const()[name = tensor("z_5_dilations_0"), val = tensor([1])]; tensor z_5_groups_0 = const()[name = tensor("z_5_groups_0"), val = tensor(1)]; tensor quantizers_3_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_3_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(93888)))]; tensor quantizers_3_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_3_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(102144)))]; tensor input_15_cast_fp16 = transpose(perm = input_15_perm_0, x = var_185_cast_fp16_cast_uint16)[name = tensor("transpose_50")]; tensor z_5_cast_fp16 = conv(bias = quantizers_3_out_proj_bias_to_fp16, dilations = z_5_dilations_0, groups = z_5_groups_0, pad = z_5_pad_0, pad_type = z_5_pad_type_0, strides = z_5_strides_0, weight = quantizers_3_out_proj_weight_to_fp16, x = input_15_cast_fp16)[name = tensor("z_5_cast_fp16")]; tensor emb_7_cast_fp16 = add(x = emb_5_cast_fp16, y = z_5_cast_fp16)[name = tensor("emb_7_cast_fp16")]; tensor input_17_begin_0 = const()[name = tensor("input_17_begin_0"), val = tensor([4, 0, 0])]; tensor input_17_end_0 = const()[name = tensor("input_17_end_0"), val = tensor([5, 1, 0])]; tensor input_17_end_mask_0 = const()[name = tensor("input_17_end_mask_0"), val = tensor([false, true, true])]; tensor input_17_squeeze_mask_0 = const()[name = tensor("input_17_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_17 = slice_by_index(begin = input_17_begin_0, end = input_17_end_0, end_mask = input_17_end_mask_0, squeeze_mask = input_17_squeeze_mask_0, x = codes)[name = tensor("input_17")]; tensor var_208_axis_0 = const()[name = tensor("op_208_axis_0"), val = tensor(0)]; tensor var_208_batch_dims_0 = const()[name = tensor("op_208_batch_dims_0"), val = tensor(0)]; tensor var_208_validate_indices_0 = const()[name = tensor("op_208_validate_indices_0"), val = tensor(false)]; tensor quantizers_4_codebook_weight_to_fp16 = const()[name = tensor("quantizers_4_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(103232)))]; tensor input_17_to_uint16_dtype_0 = const()[name = tensor("input_17_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_17_to_uint16 = cast(dtype = input_17_to_uint16_dtype_0, x = input_17)[name = tensor("cast_413")]; tensor var_208_cast_fp16_cast_uint16 = gather(axis = var_208_axis_0, batch_dims = var_208_batch_dims_0, indices = input_17_to_uint16, validate_indices = var_208_validate_indices_0, x = quantizers_4_codebook_weight_to_fp16)[name = tensor("op_208_cast_fp16_cast_uint16")]; tensor input_19_perm_0 = const()[name = tensor("input_19_perm_0"), val = tensor([0, 2, 1])]; tensor z_7_pad_type_0 = const()[name = tensor("z_7_pad_type_0"), val = tensor("valid")]; tensor z_7_strides_0 = const()[name = tensor("z_7_strides_0"), val = tensor([1])]; tensor z_7_pad_0 = const()[name = tensor("z_7_pad_0"), val = tensor([0, 0])]; tensor z_7_dilations_0 = const()[name = tensor("z_7_dilations_0"), val = tensor([1])]; tensor z_7_groups_0 = const()[name = tensor("z_7_groups_0"), val = tensor(1)]; tensor quantizers_4_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_4_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(119680)))]; tensor quantizers_4_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_4_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(127936)))]; tensor input_19_cast_fp16 = transpose(perm = input_19_perm_0, x = var_208_cast_fp16_cast_uint16)[name = tensor("transpose_49")]; tensor z_7_cast_fp16 = conv(bias = quantizers_4_out_proj_bias_to_fp16, dilations = z_7_dilations_0, groups = z_7_groups_0, pad = z_7_pad_0, pad_type = z_7_pad_type_0, strides = z_7_strides_0, weight = quantizers_4_out_proj_weight_to_fp16, x = input_19_cast_fp16)[name = tensor("z_7_cast_fp16")]; tensor emb_9_cast_fp16 = add(x = emb_7_cast_fp16, y = z_7_cast_fp16)[name = tensor("emb_9_cast_fp16")]; tensor input_21_begin_0 = const()[name = tensor("input_21_begin_0"), val = tensor([5, 0, 0])]; tensor input_21_end_0 = const()[name = tensor("input_21_end_0"), val = tensor([6, 1, 0])]; tensor input_21_end_mask_0 = const()[name = tensor("input_21_end_mask_0"), val = tensor([false, true, true])]; tensor input_21_squeeze_mask_0 = const()[name = tensor("input_21_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_21 = slice_by_index(begin = input_21_begin_0, end = input_21_end_0, end_mask = input_21_end_mask_0, squeeze_mask = input_21_squeeze_mask_0, x = codes)[name = tensor("input_21")]; tensor var_231_axis_0 = const()[name = tensor("op_231_axis_0"), val = tensor(0)]; tensor var_231_batch_dims_0 = const()[name = tensor("op_231_batch_dims_0"), val = tensor(0)]; tensor var_231_validate_indices_0 = const()[name = tensor("op_231_validate_indices_0"), val = tensor(false)]; tensor quantizers_5_codebook_weight_to_fp16 = const()[name = tensor("quantizers_5_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(129024)))]; tensor input_21_to_uint16_dtype_0 = const()[name = tensor("input_21_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_21_to_uint16 = cast(dtype = input_21_to_uint16_dtype_0, x = input_21)[name = tensor("cast_412")]; tensor var_231_cast_fp16_cast_uint16 = gather(axis = var_231_axis_0, batch_dims = var_231_batch_dims_0, indices = input_21_to_uint16, validate_indices = var_231_validate_indices_0, x = quantizers_5_codebook_weight_to_fp16)[name = tensor("op_231_cast_fp16_cast_uint16")]; tensor input_23_perm_0 = const()[name = tensor("input_23_perm_0"), val = tensor([0, 2, 1])]; tensor z_9_pad_type_0 = const()[name = tensor("z_9_pad_type_0"), val = tensor("valid")]; tensor z_9_strides_0 = const()[name = tensor("z_9_strides_0"), val = tensor([1])]; tensor z_9_pad_0 = const()[name = tensor("z_9_pad_0"), val = tensor([0, 0])]; tensor z_9_dilations_0 = const()[name = tensor("z_9_dilations_0"), val = tensor([1])]; tensor z_9_groups_0 = const()[name = tensor("z_9_groups_0"), val = tensor(1)]; tensor quantizers_5_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_5_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(145472)))]; tensor quantizers_5_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_5_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(153728)))]; tensor input_23_cast_fp16 = transpose(perm = input_23_perm_0, x = var_231_cast_fp16_cast_uint16)[name = tensor("transpose_48")]; tensor z_9_cast_fp16 = conv(bias = quantizers_5_out_proj_bias_to_fp16, dilations = z_9_dilations_0, groups = z_9_groups_0, pad = z_9_pad_0, pad_type = z_9_pad_type_0, strides = z_9_strides_0, weight = quantizers_5_out_proj_weight_to_fp16, x = input_23_cast_fp16)[name = tensor("z_9_cast_fp16")]; tensor emb_11_cast_fp16 = add(x = emb_9_cast_fp16, y = z_9_cast_fp16)[name = tensor("emb_11_cast_fp16")]; tensor input_25_begin_0 = const()[name = tensor("input_25_begin_0"), val = tensor([6, 0, 0])]; tensor input_25_end_0 = const()[name = tensor("input_25_end_0"), val = tensor([7, 1, 0])]; tensor input_25_end_mask_0 = const()[name = tensor("input_25_end_mask_0"), val = tensor([false, true, true])]; tensor input_25_squeeze_mask_0 = const()[name = tensor("input_25_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_25 = slice_by_index(begin = input_25_begin_0, end = input_25_end_0, end_mask = input_25_end_mask_0, squeeze_mask = input_25_squeeze_mask_0, x = codes)[name = tensor("input_25")]; tensor var_254_axis_0 = const()[name = tensor("op_254_axis_0"), val = tensor(0)]; tensor var_254_batch_dims_0 = const()[name = tensor("op_254_batch_dims_0"), val = tensor(0)]; tensor var_254_validate_indices_0 = const()[name = tensor("op_254_validate_indices_0"), val = tensor(false)]; tensor quantizers_6_codebook_weight_to_fp16 = const()[name = tensor("quantizers_6_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(154816)))]; tensor input_25_to_uint16_dtype_0 = const()[name = tensor("input_25_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_25_to_uint16 = cast(dtype = input_25_to_uint16_dtype_0, x = input_25)[name = tensor("cast_411")]; tensor var_254_cast_fp16_cast_uint16 = gather(axis = var_254_axis_0, batch_dims = var_254_batch_dims_0, indices = input_25_to_uint16, validate_indices = var_254_validate_indices_0, x = quantizers_6_codebook_weight_to_fp16)[name = tensor("op_254_cast_fp16_cast_uint16")]; tensor input_27_perm_0 = const()[name = tensor("input_27_perm_0"), val = tensor([0, 2, 1])]; tensor z_11_pad_type_0 = const()[name = tensor("z_11_pad_type_0"), val = tensor("valid")]; tensor z_11_strides_0 = const()[name = tensor("z_11_strides_0"), val = tensor([1])]; tensor z_11_pad_0 = const()[name = tensor("z_11_pad_0"), val = tensor([0, 0])]; tensor z_11_dilations_0 = const()[name = tensor("z_11_dilations_0"), val = tensor([1])]; tensor z_11_groups_0 = const()[name = tensor("z_11_groups_0"), val = tensor(1)]; tensor quantizers_6_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_6_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(171264)))]; tensor quantizers_6_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_6_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(179520)))]; tensor input_27_cast_fp16 = transpose(perm = input_27_perm_0, x = var_254_cast_fp16_cast_uint16)[name = tensor("transpose_47")]; tensor z_11_cast_fp16 = conv(bias = quantizers_6_out_proj_bias_to_fp16, dilations = z_11_dilations_0, groups = z_11_groups_0, pad = z_11_pad_0, pad_type = z_11_pad_type_0, strides = z_11_strides_0, weight = quantizers_6_out_proj_weight_to_fp16, x = input_27_cast_fp16)[name = tensor("z_11_cast_fp16")]; tensor emb_13_cast_fp16 = add(x = emb_11_cast_fp16, y = z_11_cast_fp16)[name = tensor("emb_13_cast_fp16")]; tensor input_29_begin_0 = const()[name = tensor("input_29_begin_0"), val = tensor([7, 0, 0])]; tensor input_29_end_0 = const()[name = tensor("input_29_end_0"), val = tensor([8, 1, 0])]; tensor input_29_end_mask_0 = const()[name = tensor("input_29_end_mask_0"), val = tensor([false, true, true])]; tensor input_29_squeeze_mask_0 = const()[name = tensor("input_29_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_29 = slice_by_index(begin = input_29_begin_0, end = input_29_end_0, end_mask = input_29_end_mask_0, squeeze_mask = input_29_squeeze_mask_0, x = codes)[name = tensor("input_29")]; tensor var_277_axis_0 = const()[name = tensor("op_277_axis_0"), val = tensor(0)]; tensor var_277_batch_dims_0 = const()[name = tensor("op_277_batch_dims_0"), val = tensor(0)]; tensor var_277_validate_indices_0 = const()[name = tensor("op_277_validate_indices_0"), val = tensor(false)]; tensor quantizers_7_codebook_weight_to_fp16 = const()[name = tensor("quantizers_7_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(180608)))]; tensor input_29_to_uint16_dtype_0 = const()[name = tensor("input_29_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_29_to_uint16 = cast(dtype = input_29_to_uint16_dtype_0, x = input_29)[name = tensor("cast_410")]; tensor var_277_cast_fp16_cast_uint16 = gather(axis = var_277_axis_0, batch_dims = var_277_batch_dims_0, indices = input_29_to_uint16, validate_indices = var_277_validate_indices_0, x = quantizers_7_codebook_weight_to_fp16)[name = tensor("op_277_cast_fp16_cast_uint16")]; tensor input_31_perm_0 = const()[name = tensor("input_31_perm_0"), val = tensor([0, 2, 1])]; tensor z_13_pad_type_0 = const()[name = tensor("z_13_pad_type_0"), val = tensor("valid")]; tensor z_13_strides_0 = const()[name = tensor("z_13_strides_0"), val = tensor([1])]; tensor z_13_pad_0 = const()[name = tensor("z_13_pad_0"), val = tensor([0, 0])]; tensor z_13_dilations_0 = const()[name = tensor("z_13_dilations_0"), val = tensor([1])]; tensor z_13_groups_0 = const()[name = tensor("z_13_groups_0"), val = tensor(1)]; tensor quantizers_7_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_7_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(197056)))]; tensor quantizers_7_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_7_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(205312)))]; tensor input_31_cast_fp16 = transpose(perm = input_31_perm_0, x = var_277_cast_fp16_cast_uint16)[name = tensor("transpose_46")]; tensor z_13_cast_fp16 = conv(bias = quantizers_7_out_proj_bias_to_fp16, dilations = z_13_dilations_0, groups = z_13_groups_0, pad = z_13_pad_0, pad_type = z_13_pad_type_0, strides = z_13_strides_0, weight = quantizers_7_out_proj_weight_to_fp16, x = input_31_cast_fp16)[name = tensor("z_13_cast_fp16")]; tensor emb_15_cast_fp16 = add(x = emb_13_cast_fp16, y = z_13_cast_fp16)[name = tensor("emb_15_cast_fp16")]; tensor input_33_begin_0 = const()[name = tensor("input_33_begin_0"), val = tensor([8, 0, 0])]; tensor input_33_end_0 = const()[name = tensor("input_33_end_0"), val = tensor([9, 1, 0])]; tensor input_33_end_mask_0 = const()[name = tensor("input_33_end_mask_0"), val = tensor([false, true, true])]; tensor input_33_squeeze_mask_0 = const()[name = tensor("input_33_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_33 = slice_by_index(begin = input_33_begin_0, end = input_33_end_0, end_mask = input_33_end_mask_0, squeeze_mask = input_33_squeeze_mask_0, x = codes)[name = tensor("input_33")]; tensor var_300_axis_0 = const()[name = tensor("op_300_axis_0"), val = tensor(0)]; tensor var_300_batch_dims_0 = const()[name = tensor("op_300_batch_dims_0"), val = tensor(0)]; tensor var_300_validate_indices_0 = const()[name = tensor("op_300_validate_indices_0"), val = tensor(false)]; tensor quantizers_8_codebook_weight_to_fp16 = const()[name = tensor("quantizers_8_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(206400)))]; tensor input_33_to_uint16_dtype_0 = const()[name = tensor("input_33_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_33_to_uint16 = cast(dtype = input_33_to_uint16_dtype_0, x = input_33)[name = tensor("cast_409")]; tensor var_300_cast_fp16_cast_uint16 = gather(axis = var_300_axis_0, batch_dims = var_300_batch_dims_0, indices = input_33_to_uint16, validate_indices = var_300_validate_indices_0, x = quantizers_8_codebook_weight_to_fp16)[name = tensor("op_300_cast_fp16_cast_uint16")]; tensor input_35_perm_0 = const()[name = tensor("input_35_perm_0"), val = tensor([0, 2, 1])]; tensor z_15_pad_type_0 = const()[name = tensor("z_15_pad_type_0"), val = tensor("valid")]; tensor z_15_strides_0 = const()[name = tensor("z_15_strides_0"), val = tensor([1])]; tensor z_15_pad_0 = const()[name = tensor("z_15_pad_0"), val = tensor([0, 0])]; tensor z_15_dilations_0 = const()[name = tensor("z_15_dilations_0"), val = tensor([1])]; tensor z_15_groups_0 = const()[name = tensor("z_15_groups_0"), val = tensor(1)]; tensor quantizers_8_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_8_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(222848)))]; tensor quantizers_8_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_8_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(231104)))]; tensor input_35_cast_fp16 = transpose(perm = input_35_perm_0, x = var_300_cast_fp16_cast_uint16)[name = tensor("transpose_45")]; tensor z_15_cast_fp16 = conv(bias = quantizers_8_out_proj_bias_to_fp16, dilations = z_15_dilations_0, groups = z_15_groups_0, pad = z_15_pad_0, pad_type = z_15_pad_type_0, strides = z_15_strides_0, weight = quantizers_8_out_proj_weight_to_fp16, x = input_35_cast_fp16)[name = tensor("z_15_cast_fp16")]; tensor emb_17_cast_fp16 = add(x = emb_15_cast_fp16, y = z_15_cast_fp16)[name = tensor("emb_17_cast_fp16")]; tensor input_37_begin_0 = const()[name = tensor("input_37_begin_0"), val = tensor([9, 0, 0])]; tensor input_37_end_0 = const()[name = tensor("input_37_end_0"), val = tensor([10, 1, 0])]; tensor input_37_end_mask_0 = const()[name = tensor("input_37_end_mask_0"), val = tensor([false, true, true])]; tensor input_37_squeeze_mask_0 = const()[name = tensor("input_37_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_37 = slice_by_index(begin = input_37_begin_0, end = input_37_end_0, end_mask = input_37_end_mask_0, squeeze_mask = input_37_squeeze_mask_0, x = codes)[name = tensor("input_37")]; tensor var_323_axis_0 = const()[name = tensor("op_323_axis_0"), val = tensor(0)]; tensor var_323_batch_dims_0 = const()[name = tensor("op_323_batch_dims_0"), val = tensor(0)]; tensor var_323_validate_indices_0 = const()[name = tensor("op_323_validate_indices_0"), val = tensor(false)]; tensor quantizers_9_codebook_weight_to_fp16 = const()[name = tensor("quantizers_9_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(232192)))]; tensor input_37_to_uint16_dtype_0 = const()[name = tensor("input_37_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_37_to_uint16 = cast(dtype = input_37_to_uint16_dtype_0, x = input_37)[name = tensor("cast_408")]; tensor var_323_cast_fp16_cast_uint16 = gather(axis = var_323_axis_0, batch_dims = var_323_batch_dims_0, indices = input_37_to_uint16, validate_indices = var_323_validate_indices_0, x = quantizers_9_codebook_weight_to_fp16)[name = tensor("op_323_cast_fp16_cast_uint16")]; tensor input_39_perm_0 = const()[name = tensor("input_39_perm_0"), val = tensor([0, 2, 1])]; tensor z_17_pad_type_0 = const()[name = tensor("z_17_pad_type_0"), val = tensor("valid")]; tensor z_17_strides_0 = const()[name = tensor("z_17_strides_0"), val = tensor([1])]; tensor z_17_pad_0 = const()[name = tensor("z_17_pad_0"), val = tensor([0, 0])]; tensor z_17_dilations_0 = const()[name = tensor("z_17_dilations_0"), val = tensor([1])]; tensor z_17_groups_0 = const()[name = tensor("z_17_groups_0"), val = tensor(1)]; tensor quantizers_9_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_9_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(248640)))]; tensor quantizers_9_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_9_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(256896)))]; tensor input_39_cast_fp16 = transpose(perm = input_39_perm_0, x = var_323_cast_fp16_cast_uint16)[name = tensor("transpose_44")]; tensor z_17_cast_fp16 = conv(bias = quantizers_9_out_proj_bias_to_fp16, dilations = z_17_dilations_0, groups = z_17_groups_0, pad = z_17_pad_0, pad_type = z_17_pad_type_0, strides = z_17_strides_0, weight = quantizers_9_out_proj_weight_to_fp16, x = input_39_cast_fp16)[name = tensor("z_17_cast_fp16")]; tensor emb_19_cast_fp16 = add(x = emb_17_cast_fp16, y = z_17_cast_fp16)[name = tensor("emb_19_cast_fp16")]; tensor input_41_begin_0 = const()[name = tensor("input_41_begin_0"), val = tensor([10, 0, 0])]; tensor input_41_end_0 = const()[name = tensor("input_41_end_0"), val = tensor([11, 1, 0])]; tensor input_41_end_mask_0 = const()[name = tensor("input_41_end_mask_0"), val = tensor([false, true, true])]; tensor input_41_squeeze_mask_0 = const()[name = tensor("input_41_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_41 = slice_by_index(begin = input_41_begin_0, end = input_41_end_0, end_mask = input_41_end_mask_0, squeeze_mask = input_41_squeeze_mask_0, x = codes)[name = tensor("input_41")]; tensor var_346_axis_0 = const()[name = tensor("op_346_axis_0"), val = tensor(0)]; tensor var_346_batch_dims_0 = const()[name = tensor("op_346_batch_dims_0"), val = tensor(0)]; tensor var_346_validate_indices_0 = const()[name = tensor("op_346_validate_indices_0"), val = tensor(false)]; tensor quantizers_10_codebook_weight_to_fp16 = const()[name = tensor("quantizers_10_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(257984)))]; tensor input_41_to_uint16_dtype_0 = const()[name = tensor("input_41_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_41_to_uint16 = cast(dtype = input_41_to_uint16_dtype_0, x = input_41)[name = tensor("cast_407")]; tensor var_346_cast_fp16_cast_uint16 = gather(axis = var_346_axis_0, batch_dims = var_346_batch_dims_0, indices = input_41_to_uint16, validate_indices = var_346_validate_indices_0, x = quantizers_10_codebook_weight_to_fp16)[name = tensor("op_346_cast_fp16_cast_uint16")]; tensor input_43_perm_0 = const()[name = tensor("input_43_perm_0"), val = tensor([0, 2, 1])]; tensor z_19_pad_type_0 = const()[name = tensor("z_19_pad_type_0"), val = tensor("valid")]; tensor z_19_strides_0 = const()[name = tensor("z_19_strides_0"), val = tensor([1])]; tensor z_19_pad_0 = const()[name = tensor("z_19_pad_0"), val = tensor([0, 0])]; tensor z_19_dilations_0 = const()[name = tensor("z_19_dilations_0"), val = tensor([1])]; tensor z_19_groups_0 = const()[name = tensor("z_19_groups_0"), val = tensor(1)]; tensor quantizers_10_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_10_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(274432)))]; tensor quantizers_10_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_10_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(282688)))]; tensor input_43_cast_fp16 = transpose(perm = input_43_perm_0, x = var_346_cast_fp16_cast_uint16)[name = tensor("transpose_43")]; tensor z_19_cast_fp16 = conv(bias = quantizers_10_out_proj_bias_to_fp16, dilations = z_19_dilations_0, groups = z_19_groups_0, pad = z_19_pad_0, pad_type = z_19_pad_type_0, strides = z_19_strides_0, weight = quantizers_10_out_proj_weight_to_fp16, x = input_43_cast_fp16)[name = tensor("z_19_cast_fp16")]; tensor emb_21_cast_fp16 = add(x = emb_19_cast_fp16, y = z_19_cast_fp16)[name = tensor("emb_21_cast_fp16")]; tensor input_45_begin_0 = const()[name = tensor("input_45_begin_0"), val = tensor([11, 0, 0])]; tensor input_45_end_0 = const()[name = tensor("input_45_end_0"), val = tensor([12, 1, 0])]; tensor input_45_end_mask_0 = const()[name = tensor("input_45_end_mask_0"), val = tensor([false, true, true])]; tensor input_45_squeeze_mask_0 = const()[name = tensor("input_45_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_45 = slice_by_index(begin = input_45_begin_0, end = input_45_end_0, end_mask = input_45_end_mask_0, squeeze_mask = input_45_squeeze_mask_0, x = codes)[name = tensor("input_45")]; tensor var_369_axis_0 = const()[name = tensor("op_369_axis_0"), val = tensor(0)]; tensor var_369_batch_dims_0 = const()[name = tensor("op_369_batch_dims_0"), val = tensor(0)]; tensor var_369_validate_indices_0 = const()[name = tensor("op_369_validate_indices_0"), val = tensor(false)]; tensor quantizers_11_codebook_weight_to_fp16 = const()[name = tensor("quantizers_11_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(283776)))]; tensor input_45_to_uint16_dtype_0 = const()[name = tensor("input_45_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_45_to_uint16 = cast(dtype = input_45_to_uint16_dtype_0, x = input_45)[name = tensor("cast_406")]; tensor var_369_cast_fp16_cast_uint16 = gather(axis = var_369_axis_0, batch_dims = var_369_batch_dims_0, indices = input_45_to_uint16, validate_indices = var_369_validate_indices_0, x = quantizers_11_codebook_weight_to_fp16)[name = tensor("op_369_cast_fp16_cast_uint16")]; tensor input_47_perm_0 = const()[name = tensor("input_47_perm_0"), val = tensor([0, 2, 1])]; tensor z_21_pad_type_0 = const()[name = tensor("z_21_pad_type_0"), val = tensor("valid")]; tensor z_21_strides_0 = const()[name = tensor("z_21_strides_0"), val = tensor([1])]; tensor z_21_pad_0 = const()[name = tensor("z_21_pad_0"), val = tensor([0, 0])]; tensor z_21_dilations_0 = const()[name = tensor("z_21_dilations_0"), val = tensor([1])]; tensor z_21_groups_0 = const()[name = tensor("z_21_groups_0"), val = tensor(1)]; tensor quantizers_11_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_11_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(300224)))]; tensor quantizers_11_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_11_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(308480)))]; tensor input_47_cast_fp16 = transpose(perm = input_47_perm_0, x = var_369_cast_fp16_cast_uint16)[name = tensor("transpose_42")]; tensor z_21_cast_fp16 = conv(bias = quantizers_11_out_proj_bias_to_fp16, dilations = z_21_dilations_0, groups = z_21_groups_0, pad = z_21_pad_0, pad_type = z_21_pad_type_0, strides = z_21_strides_0, weight = quantizers_11_out_proj_weight_to_fp16, x = input_47_cast_fp16)[name = tensor("z_21_cast_fp16")]; tensor emb_23_cast_fp16 = add(x = emb_21_cast_fp16, y = z_21_cast_fp16)[name = tensor("emb_23_cast_fp16")]; tensor input_49_begin_0 = const()[name = tensor("input_49_begin_0"), val = tensor([12, 0, 0])]; tensor input_49_end_0 = const()[name = tensor("input_49_end_0"), val = tensor([13, 1, 0])]; tensor input_49_end_mask_0 = const()[name = tensor("input_49_end_mask_0"), val = tensor([false, true, true])]; tensor input_49_squeeze_mask_0 = const()[name = tensor("input_49_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_49 = slice_by_index(begin = input_49_begin_0, end = input_49_end_0, end_mask = input_49_end_mask_0, squeeze_mask = input_49_squeeze_mask_0, x = codes)[name = tensor("input_49")]; tensor var_392_axis_0 = const()[name = tensor("op_392_axis_0"), val = tensor(0)]; tensor var_392_batch_dims_0 = const()[name = tensor("op_392_batch_dims_0"), val = tensor(0)]; tensor var_392_validate_indices_0 = const()[name = tensor("op_392_validate_indices_0"), val = tensor(false)]; tensor quantizers_12_codebook_weight_to_fp16 = const()[name = tensor("quantizers_12_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(309568)))]; tensor input_49_to_uint16_dtype_0 = const()[name = tensor("input_49_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_49_to_uint16 = cast(dtype = input_49_to_uint16_dtype_0, x = input_49)[name = tensor("cast_405")]; tensor var_392_cast_fp16_cast_uint16 = gather(axis = var_392_axis_0, batch_dims = var_392_batch_dims_0, indices = input_49_to_uint16, validate_indices = var_392_validate_indices_0, x = quantizers_12_codebook_weight_to_fp16)[name = tensor("op_392_cast_fp16_cast_uint16")]; tensor input_51_perm_0 = const()[name = tensor("input_51_perm_0"), val = tensor([0, 2, 1])]; tensor z_23_pad_type_0 = const()[name = tensor("z_23_pad_type_0"), val = tensor("valid")]; tensor z_23_strides_0 = const()[name = tensor("z_23_strides_0"), val = tensor([1])]; tensor z_23_pad_0 = const()[name = tensor("z_23_pad_0"), val = tensor([0, 0])]; tensor z_23_dilations_0 = const()[name = tensor("z_23_dilations_0"), val = tensor([1])]; tensor z_23_groups_0 = const()[name = tensor("z_23_groups_0"), val = tensor(1)]; tensor quantizers_12_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_12_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(326016)))]; tensor quantizers_12_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_12_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(334272)))]; tensor input_51_cast_fp16 = transpose(perm = input_51_perm_0, x = var_392_cast_fp16_cast_uint16)[name = tensor("transpose_41")]; tensor z_23_cast_fp16 = conv(bias = quantizers_12_out_proj_bias_to_fp16, dilations = z_23_dilations_0, groups = z_23_groups_0, pad = z_23_pad_0, pad_type = z_23_pad_type_0, strides = z_23_strides_0, weight = quantizers_12_out_proj_weight_to_fp16, x = input_51_cast_fp16)[name = tensor("z_23_cast_fp16")]; tensor emb_25_cast_fp16 = add(x = emb_23_cast_fp16, y = z_23_cast_fp16)[name = tensor("emb_25_cast_fp16")]; tensor input_53_begin_0 = const()[name = tensor("input_53_begin_0"), val = tensor([13, 0, 0])]; tensor input_53_end_0 = const()[name = tensor("input_53_end_0"), val = tensor([14, 1, 0])]; tensor input_53_end_mask_0 = const()[name = tensor("input_53_end_mask_0"), val = tensor([false, true, true])]; tensor input_53_squeeze_mask_0 = const()[name = tensor("input_53_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_53 = slice_by_index(begin = input_53_begin_0, end = input_53_end_0, end_mask = input_53_end_mask_0, squeeze_mask = input_53_squeeze_mask_0, x = codes)[name = tensor("input_53")]; tensor var_415_axis_0 = const()[name = tensor("op_415_axis_0"), val = tensor(0)]; tensor var_415_batch_dims_0 = const()[name = tensor("op_415_batch_dims_0"), val = tensor(0)]; tensor var_415_validate_indices_0 = const()[name = tensor("op_415_validate_indices_0"), val = tensor(false)]; tensor quantizers_13_codebook_weight_to_fp16 = const()[name = tensor("quantizers_13_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(335360)))]; tensor input_53_to_uint16_dtype_0 = const()[name = tensor("input_53_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_53_to_uint16 = cast(dtype = input_53_to_uint16_dtype_0, x = input_53)[name = tensor("cast_404")]; tensor var_415_cast_fp16_cast_uint16 = gather(axis = var_415_axis_0, batch_dims = var_415_batch_dims_0, indices = input_53_to_uint16, validate_indices = var_415_validate_indices_0, x = quantizers_13_codebook_weight_to_fp16)[name = tensor("op_415_cast_fp16_cast_uint16")]; tensor input_55_perm_0 = const()[name = tensor("input_55_perm_0"), val = tensor([0, 2, 1])]; tensor z_25_pad_type_0 = const()[name = tensor("z_25_pad_type_0"), val = tensor("valid")]; tensor z_25_strides_0 = const()[name = tensor("z_25_strides_0"), val = tensor([1])]; tensor z_25_pad_0 = const()[name = tensor("z_25_pad_0"), val = tensor([0, 0])]; tensor z_25_dilations_0 = const()[name = tensor("z_25_dilations_0"), val = tensor([1])]; tensor z_25_groups_0 = const()[name = tensor("z_25_groups_0"), val = tensor(1)]; tensor quantizers_13_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_13_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(351808)))]; tensor quantizers_13_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_13_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(360064)))]; tensor input_55_cast_fp16 = transpose(perm = input_55_perm_0, x = var_415_cast_fp16_cast_uint16)[name = tensor("transpose_40")]; tensor z_25_cast_fp16 = conv(bias = quantizers_13_out_proj_bias_to_fp16, dilations = z_25_dilations_0, groups = z_25_groups_0, pad = z_25_pad_0, pad_type = z_25_pad_type_0, strides = z_25_strides_0, weight = quantizers_13_out_proj_weight_to_fp16, x = input_55_cast_fp16)[name = tensor("z_25_cast_fp16")]; tensor emb_27_cast_fp16 = add(x = emb_25_cast_fp16, y = z_25_cast_fp16)[name = tensor("emb_27_cast_fp16")]; tensor input_57_begin_0 = const()[name = tensor("input_57_begin_0"), val = tensor([14, 0, 0])]; tensor input_57_end_0 = const()[name = tensor("input_57_end_0"), val = tensor([15, 1, 0])]; tensor input_57_end_mask_0 = const()[name = tensor("input_57_end_mask_0"), val = tensor([false, true, true])]; tensor input_57_squeeze_mask_0 = const()[name = tensor("input_57_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_57 = slice_by_index(begin = input_57_begin_0, end = input_57_end_0, end_mask = input_57_end_mask_0, squeeze_mask = input_57_squeeze_mask_0, x = codes)[name = tensor("input_57")]; tensor var_438_axis_0 = const()[name = tensor("op_438_axis_0"), val = tensor(0)]; tensor var_438_batch_dims_0 = const()[name = tensor("op_438_batch_dims_0"), val = tensor(0)]; tensor var_438_validate_indices_0 = const()[name = tensor("op_438_validate_indices_0"), val = tensor(false)]; tensor quantizers_14_codebook_weight_to_fp16 = const()[name = tensor("quantizers_14_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(361152)))]; tensor input_57_to_uint16_dtype_0 = const()[name = tensor("input_57_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_57_to_uint16 = cast(dtype = input_57_to_uint16_dtype_0, x = input_57)[name = tensor("cast_403")]; tensor var_438_cast_fp16_cast_uint16 = gather(axis = var_438_axis_0, batch_dims = var_438_batch_dims_0, indices = input_57_to_uint16, validate_indices = var_438_validate_indices_0, x = quantizers_14_codebook_weight_to_fp16)[name = tensor("op_438_cast_fp16_cast_uint16")]; tensor input_59_perm_0 = const()[name = tensor("input_59_perm_0"), val = tensor([0, 2, 1])]; tensor z_27_pad_type_0 = const()[name = tensor("z_27_pad_type_0"), val = tensor("valid")]; tensor z_27_strides_0 = const()[name = tensor("z_27_strides_0"), val = tensor([1])]; tensor z_27_pad_0 = const()[name = tensor("z_27_pad_0"), val = tensor([0, 0])]; tensor z_27_dilations_0 = const()[name = tensor("z_27_dilations_0"), val = tensor([1])]; tensor z_27_groups_0 = const()[name = tensor("z_27_groups_0"), val = tensor(1)]; tensor quantizers_14_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_14_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(377600)))]; tensor quantizers_14_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_14_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(385856)))]; tensor input_59_cast_fp16 = transpose(perm = input_59_perm_0, x = var_438_cast_fp16_cast_uint16)[name = tensor("transpose_39")]; tensor z_27_cast_fp16 = conv(bias = quantizers_14_out_proj_bias_to_fp16, dilations = z_27_dilations_0, groups = z_27_groups_0, pad = z_27_pad_0, pad_type = z_27_pad_type_0, strides = z_27_strides_0, weight = quantizers_14_out_proj_weight_to_fp16, x = input_59_cast_fp16)[name = tensor("z_27_cast_fp16")]; tensor emb_cast_fp16 = add(x = emb_27_cast_fp16, y = z_27_cast_fp16)[name = tensor("emb_cast_fp16")]; tensor input_61_begin_0 = const()[name = tensor("input_61_begin_0"), val = tensor([15, 0, 0])]; tensor input_61_end_0 = const()[name = tensor("input_61_end_0"), val = tensor([16, 1, 0])]; tensor input_61_end_mask_0 = const()[name = tensor("input_61_end_mask_0"), val = tensor([false, true, true])]; tensor input_61_squeeze_mask_0 = const()[name = tensor("input_61_squeeze_mask_0"), val = tensor([true, false, false])]; tensor input_61 = slice_by_index(begin = input_61_begin_0, end = input_61_end_0, end_mask = input_61_end_mask_0, squeeze_mask = input_61_squeeze_mask_0, x = codes)[name = tensor("input_61")]; tensor var_461_axis_0 = const()[name = tensor("op_461_axis_0"), val = tensor(0)]; tensor var_461_batch_dims_0 = const()[name = tensor("op_461_batch_dims_0"), val = tensor(0)]; tensor var_461_validate_indices_0 = const()[name = tensor("op_461_validate_indices_0"), val = tensor(false)]; tensor quantizers_15_codebook_weight_to_fp16 = const()[name = tensor("quantizers_15_codebook_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(386944)))]; tensor input_61_to_uint16_dtype_0 = const()[name = tensor("input_61_to_uint16_dtype_0"), val = tensor("uint16")]; tensor input_61_to_uint16 = cast(dtype = input_61_to_uint16_dtype_0, x = input_61)[name = tensor("cast_402")]; tensor var_461_cast_fp16_cast_uint16 = gather(axis = var_461_axis_0, batch_dims = var_461_batch_dims_0, indices = input_61_to_uint16, validate_indices = var_461_validate_indices_0, x = quantizers_15_codebook_weight_to_fp16)[name = tensor("op_461_cast_fp16_cast_uint16")]; tensor input_63_perm_0 = const()[name = tensor("input_63_perm_0"), val = tensor([0, 2, 1])]; tensor z_pad_type_0 = const()[name = tensor("z_pad_type_0"), val = tensor("valid")]; tensor z_strides_0 = const()[name = tensor("z_strides_0"), val = tensor([1])]; tensor z_pad_0 = const()[name = tensor("z_pad_0"), val = tensor([0, 0])]; tensor z_dilations_0 = const()[name = tensor("z_dilations_0"), val = tensor([1])]; tensor z_groups_0 = const()[name = tensor("z_groups_0"), val = tensor(1)]; tensor quantizers_15_out_proj_weight_to_fp16 = const()[name = tensor("quantizers_15_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(403392)))]; tensor quantizers_15_out_proj_bias_to_fp16 = const()[name = tensor("quantizers_15_out_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(411648)))]; tensor input_63_cast_fp16 = transpose(perm = input_63_perm_0, x = var_461_cast_fp16_cast_uint16)[name = tensor("transpose_38")]; tensor z_cast_fp16 = conv(bias = quantizers_15_out_proj_bias_to_fp16, dilations = z_dilations_0, groups = z_groups_0, pad = z_pad_0, pad_type = z_pad_type_0, strides = z_strides_0, weight = quantizers_15_out_proj_weight_to_fp16, x = input_63_cast_fp16)[name = tensor("z_cast_fp16")]; tensor input_65_cast_fp16 = add(x = emb_cast_fp16, y = z_cast_fp16)[name = tensor("input_65_cast_fp16")]; tensor x_1_pad_type_0 = const()[name = tensor("x_1_pad_type_0"), val = tensor("valid")]; tensor x_1_strides_0 = const()[name = tensor("x_1_strides_0"), val = tensor([1])]; tensor x_1_pad_0 = const()[name = tensor("x_1_pad_0"), val = tensor([0, 0])]; tensor x_1_dilations_0 = const()[name = tensor("x_1_dilations_0"), val = tensor([1])]; tensor x_1_groups_0 = const()[name = tensor("x_1_groups_0"), val = tensor(1)]; tensor output_proj_weight_to_fp16 = const()[name = tensor("output_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(412736)))]; tensor output_proj_bias_to_fp16 = const()[name = tensor("output_proj_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1199232)))]; tensor x_1_cast_fp16 = conv(bias = output_proj_bias_to_fp16, dilations = x_1_dilations_0, groups = x_1_groups_0, pad = x_1_pad_0, pad_type = x_1_pad_type_0, strides = x_1_strides_0, weight = output_proj_weight_to_fp16, x = input_65_cast_fp16)[name = tensor("x_1_cast_fp16")]; tensor var_496_begin_0 = const()[name = tensor("op_496_begin_0"), val = tensor([0, 0, 0])]; tensor var_496_end_0 = const()[name = tensor("op_496_end_0"), val = tensor([1, 1, 0])]; tensor var_496_end_mask_0 = const()[name = tensor("op_496_end_mask_0"), val = tensor([true, false, true])]; tensor var_496_squeeze_mask_0 = const()[name = tensor("op_496_squeeze_mask_0"), val = tensor([false, true, false])]; tensor var_496_cast_fp16 = slice_by_index(begin = var_496_begin_0, end = var_496_end_0, end_mask = var_496_end_mask_0, squeeze_mask = var_496_squeeze_mask_0, x = x_1_cast_fp16)[name = tensor("op_496_cast_fp16")]; tensor fill_like_0_value_0_to_fp16 = const()[name = tensor("fill_like_0_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_like_0_cast_fp16 = fill_like(ref_tensor = var_496_cast_fp16, value = fill_like_0_value_0_to_fp16)[name = tensor("fill_like_0_cast_fp16")]; tensor var_512_axes_0 = const()[name = tensor("op_512_axes_0"), val = tensor([-1])]; tensor var_512_keep_dims_0 = const()[name = tensor("op_512_keep_dims_0"), val = tensor(false)]; tensor var_512_cast_fp16 = reduce_sum(axes = var_512_axes_0, keep_dims = var_512_keep_dims_0, x = fill_like_0_cast_fp16)[name = tensor("op_512_cast_fp16")]; tensor input_lengths_1_dtype_0 = const()[name = tensor("input_lengths_1_dtype_0"), val = tensor("int32")]; tensor concat_0x = const()[name = tensor("concat_0x"), val = tensor([1, 192, 4, -1])]; tensor var_533_cast_fp16 = reshape(shape = concat_0x, x = x_1_cast_fp16)[name = tensor("op_533_cast_fp16")]; tensor var_534 = const()[name = tensor("op_534"), val = tensor([0, 1, 3, 2])]; tensor concat_1x = const()[name = tensor("concat_1x"), val = tensor([1, 192, -1])]; tensor var_535_cast_fp16 = transpose(perm = var_534, x = var_533_cast_fp16)[name = tensor("transpose_37")]; tensor x_3_cast_fp16 = reshape(shape = concat_1x, x = var_535_cast_fp16)[name = tensor("x_3_cast_fp16")]; tensor var_541 = const()[name = tensor("op_541"), val = tensor(4)]; tensor var_512_cast_fp16_to_int32 = cast(dtype = input_lengths_1_dtype_0, x = var_512_cast_fp16)[name = tensor("cast_401")]; tensor input_lengths_3 = mul(x = var_512_cast_fp16_to_int32, y = var_541)[name = tensor("input_lengths_3")]; tensor var_547 = const()[name = tensor("op_547"), val = tensor(0)]; tensor var_554 = const()[name = tensor("op_554"), val = tensor(500)]; tensor var_565 = const()[name = tensor("op_565"), val = tensor(1)]; tensor input_67_perm_0 = const()[name = tensor("input_67_perm_0"), val = tensor([0, 2, 1])]; tensor decoder_1_input_proj_weight_to_fp16 = const()[name = tensor("decoder_1_input_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1200832)))]; tensor linear_0_bias_0_to_fp16 = const()[name = tensor("linear_0_bias_0_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1299200)))]; tensor input_67_cast_fp16 = transpose(perm = input_67_perm_0, x = x_3_cast_fp16)[name = tensor("transpose_36")]; tensor linear_0_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_1_input_proj_weight_to_fp16, x = input_67_cast_fp16)[name = tensor("linear_0_cast_fp16")]; tensor x_7_axes_0 = const()[name = tensor("x_7_axes_0"), val = tensor([-1])]; tensor decoder_1_transformer_layers_0_norm1_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_0_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1299776)))]; tensor decoder_1_transformer_layers_0_norm1_bias_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_0_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1300352)))]; tensor var_561_to_fp16 = const()[name = tensor("op_561_to_fp16"), val = tensor(0x1.5p-17)]; tensor x_7_cast_fp16 = layer_norm(axes = x_7_axes_0, beta = decoder_1_transformer_layers_0_norm1_bias_to_fp16, epsilon = var_561_to_fp16, gamma = decoder_1_transformer_layers_0_norm1_weight_to_fp16, x = linear_0_cast_fp16)[name = tensor("x_7_cast_fp16")]; tensor var_593_shape_cast_fp16 = shape(x = x_7_cast_fp16)[name = tensor("op_593_shape_cast_fp16")]; tensor gather_4_axis_0 = const()[name = tensor("gather_4_axis_0"), val = tensor(0)]; tensor gather_4_batch_dims_0 = const()[name = tensor("gather_4_batch_dims_0"), val = tensor(0)]; tensor gather_4_validate_indices_0 = const()[name = tensor("gather_4_validate_indices_0"), val = tensor(false)]; tensor var_593_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_593_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_4_indices_0_to_uint16 = const()[name = tensor("gather_4_indices_0_to_uint16"), val = tensor(1)]; tensor var_593_shape_cast_fp16_to_uint16 = cast(dtype = var_593_shape_cast_fp16_to_uint16_dtype_0, x = var_593_shape_cast_fp16)[name = tensor("cast_400")]; tensor gather_4_cast_uint16 = gather(axis = gather_4_axis_0, batch_dims = gather_4_batch_dims_0, indices = gather_4_indices_0_to_uint16, validate_indices = gather_4_validate_indices_0, x = var_593_shape_cast_fp16_to_uint16)[name = tensor("gather_4_cast_uint16")]; tensor gather_4_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_4_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_1_transformer_layers_0_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_0_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1300928)))]; tensor linear_1_bias_0_to_fp16 = const()[name = tensor("linear_1_bias_0_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1694208)))]; tensor linear_1_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_1_transformer_layers_0_self_attn_in_proj_weight_to_fp16, x = x_7_cast_fp16)[name = tensor("linear_1_cast_fp16")]; tensor concat_2x = const()[name = tensor("concat_2x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_600_cast_fp16 = reshape(shape = concat_2x, x = linear_1_cast_fp16)[name = tensor("op_600_cast_fp16")]; tensor var_601 = const()[name = tensor("op_601"), val = tensor([2, 0, 3, 1, 4])]; tensor q_1_begin_0 = const()[name = tensor("q_1_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_1_end_0 = const()[name = tensor("q_1_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_1_end_mask_0 = const()[name = tensor("q_1_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_1_squeeze_mask_0 = const()[name = tensor("q_1_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_3_cast_fp16 = transpose(perm = var_601, x = var_600_cast_fp16)[name = tensor("transpose_35")]; tensor q_1_cast_fp16 = slice_by_index(begin = q_1_begin_0, end = q_1_end_0, end_mask = q_1_end_mask_0, squeeze_mask = q_1_squeeze_mask_0, x = projected_3_cast_fp16)[name = tensor("q_1_cast_fp16")]; tensor k_1_begin_0 = const()[name = tensor("k_1_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_1_end_0 = const()[name = tensor("k_1_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_1_end_mask_0 = const()[name = tensor("k_1_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_1_squeeze_mask_0 = const()[name = tensor("k_1_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_1_cast_fp16 = slice_by_index(begin = k_1_begin_0, end = k_1_end_0, end_mask = k_1_end_mask_0, squeeze_mask = k_1_squeeze_mask_0, x = projected_3_cast_fp16)[name = tensor("k_1_cast_fp16")]; tensor v_1_begin_0 = const()[name = tensor("v_1_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_1_end_0 = const()[name = tensor("v_1_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_1_end_mask_0 = const()[name = tensor("v_1_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_1_squeeze_mask_0 = const()[name = tensor("v_1_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_1_cast_fp16 = slice_by_index(begin = v_1_begin_0, end = v_1_end_0, end_mask = v_1_end_mask_0, squeeze_mask = v_1_squeeze_mask_0, x = projected_3_cast_fp16)[name = tensor("v_1_cast_fp16")]; tensor var_613_shape_cast_fp16 = shape(x = q_1_cast_fp16)[name = tensor("op_613_shape_cast_fp16")]; tensor gather_8_axis_0 = const()[name = tensor("gather_8_axis_0"), val = tensor(0)]; tensor gather_8_batch_dims_0 = const()[name = tensor("gather_8_batch_dims_0"), val = tensor(0)]; tensor gather_8_validate_indices_0 = const()[name = tensor("gather_8_validate_indices_0"), val = tensor(false)]; tensor var_613_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_613_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_8_indices_0_to_uint16 = const()[name = tensor("gather_8_indices_0_to_uint16"), val = tensor(2)]; tensor var_613_shape_cast_fp16_to_uint16 = cast(dtype = var_613_shape_cast_fp16_to_uint16_dtype_0, x = var_613_shape_cast_fp16)[name = tensor("cast_398")]; tensor gather_8_cast_uint16 = gather(axis = gather_8_axis_0, batch_dims = gather_8_batch_dims_0, indices = gather_8_indices_0_to_uint16, validate_indices = gather_8_validate_indices_0, x = var_613_shape_cast_fp16_to_uint16)[name = tensor("gather_8_cast_uint16")]; tensor gather_8_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_8_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_3 = const()[name = tensor("const_3"), val = tensor(0)]; tensor const_4 = const()[name = tensor("const_4"), val = tensor(1)]; tensor gather_8_cast_uint16_to_int32 = cast(dtype = gather_8_cast_uint16_to_int32_dtype_0, x = gather_8_cast_uint16)[name = tensor("cast_397")]; tensor var_617 = range_1d(end = gather_8_cast_uint16_to_int32, start = const_3, step = const_4)[name = tensor("op_617")]; tensor var_616_to_fp16 = const()[name = tensor("op_616_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_617_promoted_to_fp16_dtype_0 = const()[name = tensor("op_617_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_617_to_fp16 = cast(dtype = var_617_promoted_to_fp16_dtype_0, x = var_617)[name = tensor("cast_396")]; tensor ts_1_cast_fp16 = add(x = var_616_to_fp16, y = var_617_to_fp16)[name = tensor("ts_1_cast_fp16")]; tensor var_620 = const()[name = tensor("op_620"), val = tensor([1, 1, -1, 1])]; tensor ts_3_cast_fp16 = reshape(shape = var_620, x = ts_1_cast_fp16)[name = tensor("ts_3_cast_fp16")]; tensor concat_3x = const()[name = tensor("concat_3x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_1_cast_fp16 = reshape(shape = concat_3x, x = q_1_cast_fp16)[name = tensor("q2_1_cast_fp16")]; tensor concat_4x = const()[name = tensor("concat_4x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_1_cast_fp16 = reshape(shape = concat_4x, x = k_1_cast_fp16)[name = tensor("k2_1_cast_fp16")]; tensor qr_1_begin_0 = const()[name = tensor("qr_1_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_1_end_0 = const()[name = tensor("qr_1_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_1_end_mask_0 = const()[name = tensor("qr_1_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_1_squeeze_mask_0 = const()[name = tensor("qr_1_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_1_cast_fp16 = slice_by_index(begin = qr_1_begin_0, end = qr_1_end_0, end_mask = qr_1_end_mask_0, squeeze_mask = qr_1_squeeze_mask_0, x = q2_1_cast_fp16)[name = tensor("qr_1_cast_fp16")]; tensor qi_1_begin_0 = const()[name = tensor("qi_1_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_1_end_0 = const()[name = tensor("qi_1_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_1_end_mask_0 = const()[name = tensor("qi_1_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_1_squeeze_mask_0 = const()[name = tensor("qi_1_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_1_cast_fp16 = slice_by_index(begin = qi_1_begin_0, end = qi_1_end_0, end_mask = qi_1_end_mask_0, squeeze_mask = qi_1_squeeze_mask_0, x = q2_1_cast_fp16)[name = tensor("qi_1_cast_fp16")]; tensor kr_1_begin_0 = const()[name = tensor("kr_1_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_1_end_0 = const()[name = tensor("kr_1_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_1_end_mask_0 = const()[name = tensor("kr_1_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_1_squeeze_mask_0 = const()[name = tensor("kr_1_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_1_cast_fp16 = slice_by_index(begin = kr_1_begin_0, end = kr_1_end_0, end_mask = kr_1_end_mask_0, squeeze_mask = kr_1_squeeze_mask_0, x = k2_1_cast_fp16)[name = tensor("kr_1_cast_fp16")]; tensor ki_1_begin_0 = const()[name = tensor("ki_1_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_1_end_0 = const()[name = tensor("ki_1_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_1_end_mask_0 = const()[name = tensor("ki_1_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_1_squeeze_mask_0 = const()[name = tensor("ki_1_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_1_cast_fp16 = slice_by_index(begin = ki_1_begin_0, end = ki_1_end_0, end_mask = ki_1_end_mask_0, squeeze_mask = ki_1_squeeze_mask_0, x = k2_1_cast_fp16)[name = tensor("ki_1_cast_fp16")]; tensor freqs_1_to_fp16 = const()[name = tensor("freqs_1_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1695808)))]; tensor var_633_cast_fp16 = mul(x = freqs_1_to_fp16, y = ts_3_cast_fp16)[name = tensor("op_633_cast_fp16")]; tensor rotr_1_cast_fp16 = cos(x = var_633_cast_fp16)[name = tensor("rotr_1_cast_fp16")]; tensor roti_1_cast_fp16 = sin(x = var_633_cast_fp16)[name = tensor("roti_1_cast_fp16")]; tensor var_637_cast_fp16 = mul(x = qr_1_cast_fp16, y = rotr_1_cast_fp16)[name = tensor("op_637_cast_fp16")]; tensor var_638_cast_fp16 = mul(x = qi_1_cast_fp16, y = roti_1_cast_fp16)[name = tensor("op_638_cast_fp16")]; tensor var_639_cast_fp16 = sub(x = var_637_cast_fp16, y = var_638_cast_fp16)[name = tensor("op_639_cast_fp16")]; tensor var_640_cast_fp16 = mul(x = qr_1_cast_fp16, y = roti_1_cast_fp16)[name = tensor("op_640_cast_fp16")]; tensor var_641_cast_fp16 = mul(x = qi_1_cast_fp16, y = rotr_1_cast_fp16)[name = tensor("op_641_cast_fp16")]; tensor var_642_cast_fp16 = add(x = var_640_cast_fp16, y = var_641_cast_fp16)[name = tensor("op_642_cast_fp16")]; tensor qo_1_axis_0 = const()[name = tensor("qo_1_axis_0"), val = tensor(-1)]; tensor qo_1_cast_fp16 = stack(axis = qo_1_axis_0, values = (var_639_cast_fp16, var_642_cast_fp16))[name = tensor("qo_1_cast_fp16")]; tensor var_645_cast_fp16 = mul(x = kr_1_cast_fp16, y = rotr_1_cast_fp16)[name = tensor("op_645_cast_fp16")]; tensor var_646_cast_fp16 = mul(x = ki_1_cast_fp16, y = roti_1_cast_fp16)[name = tensor("op_646_cast_fp16")]; tensor var_647_cast_fp16 = sub(x = var_645_cast_fp16, y = var_646_cast_fp16)[name = tensor("op_647_cast_fp16")]; tensor var_648_cast_fp16 = mul(x = kr_1_cast_fp16, y = roti_1_cast_fp16)[name = tensor("op_648_cast_fp16")]; tensor var_649_cast_fp16 = mul(x = ki_1_cast_fp16, y = rotr_1_cast_fp16)[name = tensor("op_649_cast_fp16")]; tensor var_650_cast_fp16 = add(x = var_648_cast_fp16, y = var_649_cast_fp16)[name = tensor("op_650_cast_fp16")]; tensor ko_1_axis_0 = const()[name = tensor("ko_1_axis_0"), val = tensor(-1)]; tensor ko_1_cast_fp16 = stack(axis = ko_1_axis_0, values = (var_647_cast_fp16, var_650_cast_fp16))[name = tensor("ko_1_cast_fp16")]; tensor concat_5x = const()[name = tensor("concat_5x"), val = tensor([1, 4, -1, 64])]; tensor q_3_cast_fp16 = reshape(shape = concat_5x, x = qo_1_cast_fp16)[name = tensor("q_3_cast_fp16")]; tensor concat_6x = const()[name = tensor("concat_6x"), val = tensor([1, 4, -1, 64])]; tensor k_3_cast_fp16 = reshape(shape = concat_6x, x = ko_1_cast_fp16)[name = tensor("k_3_cast_fp16")]; tensor const_5 = const()[name = tensor("const_5"), val = tensor(0)]; tensor const_6 = const()[name = tensor("const_6"), val = tensor(1)]; tensor gather_4_cast_uint16_to_int32 = cast(dtype = gather_4_cast_uint16_to_int32_dtype_0, x = gather_4_cast_uint16)[name = tensor("cast_399")]; tensor positions_1 = range_1d(end = gather_4_cast_uint16_to_int32, start = const_5, step = const_6)[name = tensor("positions_1")]; tensor concat_7x = const()[name = tensor("concat_7x"), val = tensor([1, 1, -1])]; tensor var_662 = reshape(shape = concat_7x, x = positions_1)[name = tensor("op_662")]; tensor var_663 = const()[name = tensor("op_663"), val = tensor([-1, 1, 1])]; tensor var_664 = reshape(shape = var_663, x = input_lengths_3)[name = tensor("op_664")]; tensor valid_k_1 = less(x = var_662, y = var_664)[name = tensor("valid_k_1")]; tensor concat_8x = const()[name = tensor("concat_8x"), val = tensor([1, -1, 1])]; tensor var_667 = reshape(shape = concat_8x, x = positions_1)[name = tensor("op_667")]; tensor delta_1 = sub(x = var_667, y = var_662)[name = tensor("delta_1")]; tensor concat_10_axis_0 = const()[name = tensor("concat_10_axis_0"), val = tensor(0)]; tensor concat_10_interleave_0 = const()[name = tensor("concat_10_interleave_0"), val = tensor(false)]; tensor concat_10 = concat(axis = concat_10_axis_0, interleave = concat_10_interleave_0, values = (var_565, gather_4_cast_uint16_to_int32, gather_4_cast_uint16_to_int32))[name = tensor("concat_10")]; tensor fill_0_value_0_to_fp16 = const()[name = tensor("fill_0_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_0_cast_fp16 = fill(shape = concat_10, value = fill_0_value_0_to_fp16)[name = tensor("fill_0_cast_fp16")]; tensor attn_bias_1_dtype_0 = const()[name = tensor("attn_bias_1_dtype_0"), val = tensor("bool")]; tensor var_673 = greater_equal(x = delta_1, y = var_547)[name = tensor("op_673")]; tensor fill_0_cast_fp16_to_bool = cast(dtype = attn_bias_1_dtype_0, x = fill_0_cast_fp16)[name = tensor("cast_395")]; tensor attn_bias_3 = logical_and(x = fill_0_cast_fp16_to_bool, y = var_673)[name = tensor("attn_bias_3")]; tensor var_675 = less(x = delta_1, y = var_554)[name = tensor("op_675")]; tensor attn_bias_5 = logical_and(x = attn_bias_3, y = var_675)[name = tensor("attn_bias_5")]; tensor var_677 = logical_and(x = attn_bias_5, y = valid_k_1)[name = tensor("op_677")]; tensor var_679_axes_0 = const()[name = tensor("op_679_axes_0"), val = tensor([1])]; tensor var_679 = expand_dims(axes = var_679_axes_0, x = var_677)[name = tensor("op_679")]; tensor sub_0_x_0_to_fp16 = const()[name = tensor("sub_0_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_35_to_fp16_dtype_0 = const()[name = tensor("cast_35_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_679_to_fp16 = cast(dtype = cast_35_to_fp16_dtype_0, x = var_679)[name = tensor("cast_394")]; tensor sub_0_cast_fp16 = sub(x = sub_0_x_0_to_fp16, y = var_679_to_fp16)[name = tensor("sub_0_cast_fp16")]; tensor mul_0_x_0_to_fp16 = const()[name = tensor("mul_0_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_0_cast_fp16 = mul(x = mul_0_x_0_to_fp16, y = sub_0_cast_fp16)[name = tensor("mul_0_cast_fp16")]; tensor mul_1_y_0_to_fp16 = const()[name = tensor("mul_1_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_1_cast_fp16 = mul(x = q_3_cast_fp16, y = mul_1_y_0_to_fp16)[name = tensor("mul_1_cast_fp16")]; 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_cast_fp16 = matmul(transpose_x = matmul_0_transpose_x_0, transpose_y = matmul_0_transpose_y_0, x = mul_1_cast_fp16, y = k_3_cast_fp16)[name = tensor("matmul_0_cast_fp16")]; tensor add_0_cast_fp16 = add(x = matmul_0_cast_fp16, y = mul_0_cast_fp16)[name = tensor("add_0_cast_fp16")]; tensor add_0_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_0_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_0_axis_0 = const()[name = tensor("softmax_0_axis_0"), val = tensor(-1)]; tensor add_0_cast_fp16_to_fp32 = cast(dtype = add_0_cast_fp16_to_fp32_dtype_0, x = add_0_cast_fp16)[name = tensor("cast_393")]; tensor softmax_0 = softmax(axis = softmax_0_axis_0, x = add_0_cast_fp16_to_fp32)[name = tensor("softmax_0")]; tensor out_1_transpose_x_0 = const()[name = tensor("out_1_transpose_x_0"), val = tensor(false)]; tensor out_1_transpose_y_0 = const()[name = tensor("out_1_transpose_y_0"), val = tensor(false)]; tensor softmax_0_to_fp16_dtype_0 = const()[name = tensor("softmax_0_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_0_to_fp16 = cast(dtype = softmax_0_to_fp16_dtype_0, x = softmax_0)[name = tensor("cast_392")]; tensor out_1_cast_fp16 = matmul(transpose_x = out_1_transpose_x_0, transpose_y = out_1_transpose_y_0, x = softmax_0_to_fp16, y = v_1_cast_fp16)[name = tensor("out_1_cast_fp16")]; tensor concat_11x = const()[name = tensor("concat_11x"), val = tensor([1, -1])]; tensor var_685 = reshape(shape = concat_11x, x = positions_1)[name = tensor("op_685")]; tensor var_686 = const()[name = tensor("op_686"), val = tensor([-1, 1])]; tensor var_687 = reshape(shape = var_686, x = input_lengths_3)[name = tensor("op_687")]; tensor var_688 = less(x = var_685, y = var_687)[name = tensor("op_688")]; tensor concat_12x = const()[name = tensor("concat_12x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_1 = reshape(shape = concat_12x, x = var_688)[name = tensor("valid_q_1")]; tensor var_692_to_fp16 = const()[name = tensor("op_692_to_fp16"), val = tensor(0x0p+0)]; tensor out_3_cast_fp16 = select(a = out_1_cast_fp16, b = var_692_to_fp16, cond = valid_q_1)[name = tensor("out_3_cast_fp16")]; tensor var_694_perm_0 = const()[name = tensor("op_694_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_13x = const()[name = tensor("concat_13x"), val = tensor([1, -1, 256])]; tensor var_694_cast_fp16 = transpose(perm = var_694_perm_0, x = out_3_cast_fp16)[name = tensor("transpose_34")]; tensor input_69_cast_fp16 = reshape(shape = concat_13x, x = var_694_cast_fp16)[name = tensor("input_69_cast_fp16")]; tensor decoder_1_transformer_layers_0_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_0_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1695936)))]; tensor linear_2_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_1_transformer_layers_0_self_attn_out_proj_weight_to_fp16, x = input_69_cast_fp16)[name = tensor("linear_2_cast_fp16")]; tensor decoder_1_transformer_layers_0_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_0_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1827072)))]; tensor var_700_cast_fp16 = mul(x = decoder_1_transformer_layers_0_layer_scale_1_scale_to_fp16, y = linear_2_cast_fp16)[name = tensor("op_700_cast_fp16")]; tensor input_71_cast_fp16 = add(x = linear_0_cast_fp16, y = var_700_cast_fp16)[name = tensor("input_71_cast_fp16")]; tensor input_73_axes_0 = const()[name = tensor("input_73_axes_0"), val = tensor([-1])]; tensor decoder_1_transformer_layers_0_norm2_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_0_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1827648)))]; tensor decoder_1_transformer_layers_0_norm2_bias_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_0_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1828224)))]; tensor input_73_cast_fp16 = layer_norm(axes = input_73_axes_0, beta = decoder_1_transformer_layers_0_norm2_bias_to_fp16, epsilon = var_561_to_fp16, gamma = decoder_1_transformer_layers_0_norm2_weight_to_fp16, x = input_71_cast_fp16)[name = tensor("input_73_cast_fp16")]; tensor decoder_1_transformer_layers_0_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_0_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1828800)))]; tensor linear_3_bias_0_to_fp16 = const()[name = tensor("linear_3_bias_0_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2353152)))]; tensor linear_3_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_1_transformer_layers_0_ffn_0_weight_to_fp16, x = input_73_cast_fp16)[name = tensor("linear_3_cast_fp16")]; tensor input_77_mode_0 = const()[name = tensor("input_77_mode_0"), val = tensor("EXACT")]; tensor input_77_cast_fp16 = gelu(mode = input_77_mode_0, x = linear_3_cast_fp16)[name = tensor("input_77_cast_fp16")]; tensor decoder_1_transformer_layers_0_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_0_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2355264)))]; tensor linear_4_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_1_transformer_layers_0_ffn_2_weight_to_fp16, x = input_77_cast_fp16)[name = tensor("linear_4_cast_fp16")]; tensor decoder_1_transformer_layers_0_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_0_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2879616)))]; tensor var_715_cast_fp16 = mul(x = decoder_1_transformer_layers_0_layer_scale_2_scale_to_fp16, y = linear_4_cast_fp16)[name = tensor("op_715_cast_fp16")]; tensor input_79_cast_fp16 = add(x = input_71_cast_fp16, y = var_715_cast_fp16)[name = tensor("input_79_cast_fp16")]; tensor x_13_axes_0 = const()[name = tensor("x_13_axes_0"), val = tensor([-1])]; tensor decoder_1_transformer_layers_1_norm1_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_1_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2880192)))]; tensor decoder_1_transformer_layers_1_norm1_bias_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_1_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2880768)))]; tensor x_13_cast_fp16 = layer_norm(axes = x_13_axes_0, beta = decoder_1_transformer_layers_1_norm1_bias_to_fp16, epsilon = var_561_to_fp16, gamma = decoder_1_transformer_layers_1_norm1_weight_to_fp16, x = input_79_cast_fp16)[name = tensor("x_13_cast_fp16")]; tensor var_730_shape_cast_fp16 = shape(x = x_13_cast_fp16)[name = tensor("op_730_shape_cast_fp16")]; tensor gather_14_axis_0 = const()[name = tensor("gather_14_axis_0"), val = tensor(0)]; tensor gather_14_batch_dims_0 = const()[name = tensor("gather_14_batch_dims_0"), val = tensor(0)]; tensor gather_14_validate_indices_0 = const()[name = tensor("gather_14_validate_indices_0"), val = tensor(false)]; tensor var_730_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_730_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_14_indices_0_to_uint16 = const()[name = tensor("gather_14_indices_0_to_uint16"), val = tensor(1)]; tensor var_730_shape_cast_fp16_to_uint16 = cast(dtype = var_730_shape_cast_fp16_to_uint16_dtype_0, x = var_730_shape_cast_fp16)[name = tensor("cast_391")]; tensor gather_14_cast_uint16 = gather(axis = gather_14_axis_0, batch_dims = gather_14_batch_dims_0, indices = gather_14_indices_0_to_uint16, validate_indices = gather_14_validate_indices_0, x = var_730_shape_cast_fp16_to_uint16)[name = tensor("gather_14_cast_uint16")]; tensor gather_14_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_14_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_1_transformer_layers_1_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_1_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(2881344)))]; tensor linear_5_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_1_transformer_layers_1_self_attn_in_proj_weight_to_fp16, x = x_13_cast_fp16)[name = tensor("linear_5_cast_fp16")]; tensor concat_14x = const()[name = tensor("concat_14x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_737_cast_fp16 = reshape(shape = concat_14x, x = linear_5_cast_fp16)[name = tensor("op_737_cast_fp16")]; tensor var_738 = const()[name = tensor("op_738"), val = tensor([2, 0, 3, 1, 4])]; tensor q_5_begin_0 = const()[name = tensor("q_5_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_5_end_0 = const()[name = tensor("q_5_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_5_end_mask_0 = const()[name = tensor("q_5_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_5_squeeze_mask_0 = const()[name = tensor("q_5_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_7_cast_fp16 = transpose(perm = var_738, x = var_737_cast_fp16)[name = tensor("transpose_33")]; tensor q_5_cast_fp16 = slice_by_index(begin = q_5_begin_0, end = q_5_end_0, end_mask = q_5_end_mask_0, squeeze_mask = q_5_squeeze_mask_0, x = projected_7_cast_fp16)[name = tensor("q_5_cast_fp16")]; tensor k_5_begin_0 = const()[name = tensor("k_5_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_5_end_0 = const()[name = tensor("k_5_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_5_end_mask_0 = const()[name = tensor("k_5_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_5_squeeze_mask_0 = const()[name = tensor("k_5_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_5_cast_fp16 = slice_by_index(begin = k_5_begin_0, end = k_5_end_0, end_mask = k_5_end_mask_0, squeeze_mask = k_5_squeeze_mask_0, x = projected_7_cast_fp16)[name = tensor("k_5_cast_fp16")]; tensor v_3_begin_0 = const()[name = tensor("v_3_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_3_end_0 = const()[name = tensor("v_3_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_3_end_mask_0 = const()[name = tensor("v_3_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_3_squeeze_mask_0 = const()[name = tensor("v_3_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_3_cast_fp16 = slice_by_index(begin = v_3_begin_0, end = v_3_end_0, end_mask = v_3_end_mask_0, squeeze_mask = v_3_squeeze_mask_0, x = projected_7_cast_fp16)[name = tensor("v_3_cast_fp16")]; tensor var_750_shape_cast_fp16 = shape(x = q_5_cast_fp16)[name = tensor("op_750_shape_cast_fp16")]; tensor gather_18_axis_0 = const()[name = tensor("gather_18_axis_0"), val = tensor(0)]; tensor gather_18_batch_dims_0 = const()[name = tensor("gather_18_batch_dims_0"), val = tensor(0)]; tensor gather_18_validate_indices_0 = const()[name = tensor("gather_18_validate_indices_0"), val = tensor(false)]; tensor var_750_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_750_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_18_indices_0_to_uint16 = const()[name = tensor("gather_18_indices_0_to_uint16"), val = tensor(2)]; tensor var_750_shape_cast_fp16_to_uint16 = cast(dtype = var_750_shape_cast_fp16_to_uint16_dtype_0, x = var_750_shape_cast_fp16)[name = tensor("cast_389")]; tensor gather_18_cast_uint16 = gather(axis = gather_18_axis_0, batch_dims = gather_18_batch_dims_0, indices = gather_18_indices_0_to_uint16, validate_indices = gather_18_validate_indices_0, x = var_750_shape_cast_fp16_to_uint16)[name = tensor("gather_18_cast_uint16")]; tensor gather_18_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_18_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_12 = const()[name = tensor("const_12"), val = tensor(0)]; tensor const_13 = const()[name = tensor("const_13"), val = tensor(1)]; tensor gather_18_cast_uint16_to_int32 = cast(dtype = gather_18_cast_uint16_to_int32_dtype_0, x = gather_18_cast_uint16)[name = tensor("cast_388")]; tensor var_754 = range_1d(end = gather_18_cast_uint16_to_int32, start = const_12, step = const_13)[name = tensor("op_754")]; tensor var_753_to_fp16 = const()[name = tensor("op_753_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_754_promoted_to_fp16_dtype_0 = const()[name = tensor("op_754_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_754_to_fp16 = cast(dtype = var_754_promoted_to_fp16_dtype_0, x = var_754)[name = tensor("cast_387")]; tensor ts_5_cast_fp16 = add(x = var_753_to_fp16, y = var_754_to_fp16)[name = tensor("ts_5_cast_fp16")]; tensor var_757 = const()[name = tensor("op_757"), val = tensor([1, 1, -1, 1])]; tensor ts_7_cast_fp16 = reshape(shape = var_757, x = ts_5_cast_fp16)[name = tensor("ts_7_cast_fp16")]; tensor concat_15x = const()[name = tensor("concat_15x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_3_cast_fp16 = reshape(shape = concat_15x, x = q_5_cast_fp16)[name = tensor("q2_3_cast_fp16")]; tensor concat_16x = const()[name = tensor("concat_16x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_3_cast_fp16 = reshape(shape = concat_16x, x = k_5_cast_fp16)[name = tensor("k2_3_cast_fp16")]; tensor qr_3_begin_0 = const()[name = tensor("qr_3_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_3_end_0 = const()[name = tensor("qr_3_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_3_end_mask_0 = const()[name = tensor("qr_3_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_3_squeeze_mask_0 = const()[name = tensor("qr_3_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_3_cast_fp16 = slice_by_index(begin = qr_3_begin_0, end = qr_3_end_0, end_mask = qr_3_end_mask_0, squeeze_mask = qr_3_squeeze_mask_0, x = q2_3_cast_fp16)[name = tensor("qr_3_cast_fp16")]; tensor qi_3_begin_0 = const()[name = tensor("qi_3_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_3_end_0 = const()[name = tensor("qi_3_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_3_end_mask_0 = const()[name = tensor("qi_3_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_3_squeeze_mask_0 = const()[name = tensor("qi_3_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_3_cast_fp16 = slice_by_index(begin = qi_3_begin_0, end = qi_3_end_0, end_mask = qi_3_end_mask_0, squeeze_mask = qi_3_squeeze_mask_0, x = q2_3_cast_fp16)[name = tensor("qi_3_cast_fp16")]; tensor kr_3_begin_0 = const()[name = tensor("kr_3_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_3_end_0 = const()[name = tensor("kr_3_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_3_end_mask_0 = const()[name = tensor("kr_3_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_3_squeeze_mask_0 = const()[name = tensor("kr_3_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_3_cast_fp16 = slice_by_index(begin = kr_3_begin_0, end = kr_3_end_0, end_mask = kr_3_end_mask_0, squeeze_mask = kr_3_squeeze_mask_0, x = k2_3_cast_fp16)[name = tensor("kr_3_cast_fp16")]; tensor ki_3_begin_0 = const()[name = tensor("ki_3_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_3_end_0 = const()[name = tensor("ki_3_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_3_end_mask_0 = const()[name = tensor("ki_3_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_3_squeeze_mask_0 = const()[name = tensor("ki_3_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_3_cast_fp16 = slice_by_index(begin = ki_3_begin_0, end = ki_3_end_0, end_mask = ki_3_end_mask_0, squeeze_mask = ki_3_squeeze_mask_0, x = k2_3_cast_fp16)[name = tensor("ki_3_cast_fp16")]; tensor freqs_3_to_fp16 = const()[name = tensor("freqs_3_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3274624)))]; tensor var_770_cast_fp16 = mul(x = freqs_3_to_fp16, y = ts_7_cast_fp16)[name = tensor("op_770_cast_fp16")]; tensor rotr_3_cast_fp16 = cos(x = var_770_cast_fp16)[name = tensor("rotr_3_cast_fp16")]; tensor roti_3_cast_fp16 = sin(x = var_770_cast_fp16)[name = tensor("roti_3_cast_fp16")]; tensor var_774_cast_fp16 = mul(x = qr_3_cast_fp16, y = rotr_3_cast_fp16)[name = tensor("op_774_cast_fp16")]; tensor var_775_cast_fp16 = mul(x = qi_3_cast_fp16, y = roti_3_cast_fp16)[name = tensor("op_775_cast_fp16")]; tensor var_776_cast_fp16 = sub(x = var_774_cast_fp16, y = var_775_cast_fp16)[name = tensor("op_776_cast_fp16")]; tensor var_777_cast_fp16 = mul(x = qr_3_cast_fp16, y = roti_3_cast_fp16)[name = tensor("op_777_cast_fp16")]; tensor var_778_cast_fp16 = mul(x = qi_3_cast_fp16, y = rotr_3_cast_fp16)[name = tensor("op_778_cast_fp16")]; tensor var_779_cast_fp16 = add(x = var_777_cast_fp16, y = var_778_cast_fp16)[name = tensor("op_779_cast_fp16")]; tensor qo_3_axis_0 = const()[name = tensor("qo_3_axis_0"), val = tensor(-1)]; tensor qo_3_cast_fp16 = stack(axis = qo_3_axis_0, values = (var_776_cast_fp16, var_779_cast_fp16))[name = tensor("qo_3_cast_fp16")]; tensor var_782_cast_fp16 = mul(x = kr_3_cast_fp16, y = rotr_3_cast_fp16)[name = tensor("op_782_cast_fp16")]; tensor var_783_cast_fp16 = mul(x = ki_3_cast_fp16, y = roti_3_cast_fp16)[name = tensor("op_783_cast_fp16")]; tensor var_784_cast_fp16 = sub(x = var_782_cast_fp16, y = var_783_cast_fp16)[name = tensor("op_784_cast_fp16")]; tensor var_785_cast_fp16 = mul(x = kr_3_cast_fp16, y = roti_3_cast_fp16)[name = tensor("op_785_cast_fp16")]; tensor var_786_cast_fp16 = mul(x = ki_3_cast_fp16, y = rotr_3_cast_fp16)[name = tensor("op_786_cast_fp16")]; tensor var_787_cast_fp16 = add(x = var_785_cast_fp16, y = var_786_cast_fp16)[name = tensor("op_787_cast_fp16")]; tensor ko_3_axis_0 = const()[name = tensor("ko_3_axis_0"), val = tensor(-1)]; tensor ko_3_cast_fp16 = stack(axis = ko_3_axis_0, values = (var_784_cast_fp16, var_787_cast_fp16))[name = tensor("ko_3_cast_fp16")]; tensor concat_17x = const()[name = tensor("concat_17x"), val = tensor([1, 4, -1, 64])]; tensor q_7_cast_fp16 = reshape(shape = concat_17x, x = qo_3_cast_fp16)[name = tensor("q_7_cast_fp16")]; tensor concat_18x = const()[name = tensor("concat_18x"), val = tensor([1, 4, -1, 64])]; tensor k_7_cast_fp16 = reshape(shape = concat_18x, x = ko_3_cast_fp16)[name = tensor("k_7_cast_fp16")]; tensor const_14 = const()[name = tensor("const_14"), val = tensor(0)]; tensor const_15 = const()[name = tensor("const_15"), val = tensor(1)]; tensor gather_14_cast_uint16_to_int32 = cast(dtype = gather_14_cast_uint16_to_int32_dtype_0, x = gather_14_cast_uint16)[name = tensor("cast_390")]; tensor positions_3 = range_1d(end = gather_14_cast_uint16_to_int32, start = const_14, step = const_15)[name = tensor("positions_3")]; tensor concat_19x = const()[name = tensor("concat_19x"), val = tensor([1, 1, -1])]; tensor var_799 = reshape(shape = concat_19x, x = positions_3)[name = tensor("op_799")]; tensor valid_k_3 = less(x = var_799, y = var_664)[name = tensor("valid_k_3")]; tensor concat_20x = const()[name = tensor("concat_20x"), val = tensor([1, -1, 1])]; tensor var_804 = reshape(shape = concat_20x, x = positions_3)[name = tensor("op_804")]; tensor delta_3 = sub(x = var_804, y = var_799)[name = tensor("delta_3")]; tensor concat_22_axis_0 = const()[name = tensor("concat_22_axis_0"), val = tensor(0)]; tensor concat_22_interleave_0 = const()[name = tensor("concat_22_interleave_0"), val = tensor(false)]; tensor concat_22 = concat(axis = concat_22_axis_0, interleave = concat_22_interleave_0, values = (var_565, gather_14_cast_uint16_to_int32, gather_14_cast_uint16_to_int32))[name = tensor("concat_22")]; tensor fill_1_value_0_to_fp16 = const()[name = tensor("fill_1_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_1_cast_fp16 = fill(shape = concat_22, value = fill_1_value_0_to_fp16)[name = tensor("fill_1_cast_fp16")]; tensor attn_bias_9_dtype_0 = const()[name = tensor("attn_bias_9_dtype_0"), val = tensor("bool")]; tensor var_810 = greater_equal(x = delta_3, y = var_547)[name = tensor("op_810")]; tensor fill_1_cast_fp16_to_bool = cast(dtype = attn_bias_9_dtype_0, x = fill_1_cast_fp16)[name = tensor("cast_386")]; tensor attn_bias_11 = logical_and(x = fill_1_cast_fp16_to_bool, y = var_810)[name = tensor("attn_bias_11")]; tensor var_812 = less(x = delta_3, y = var_554)[name = tensor("op_812")]; tensor attn_bias_13 = logical_and(x = attn_bias_11, y = var_812)[name = tensor("attn_bias_13")]; tensor var_814 = logical_and(x = attn_bias_13, y = valid_k_3)[name = tensor("op_814")]; tensor var_816_axes_0 = const()[name = tensor("op_816_axes_0"), val = tensor([1])]; tensor var_816 = expand_dims(axes = var_816_axes_0, x = var_814)[name = tensor("op_816")]; tensor sub_1_x_0_to_fp16 = const()[name = tensor("sub_1_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_57_to_fp16_dtype_0 = const()[name = tensor("cast_57_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_816_to_fp16 = cast(dtype = cast_57_to_fp16_dtype_0, x = var_816)[name = tensor("cast_385")]; tensor sub_1_cast_fp16 = sub(x = sub_1_x_0_to_fp16, y = var_816_to_fp16)[name = tensor("sub_1_cast_fp16")]; tensor mul_2_x_0_to_fp16 = const()[name = tensor("mul_2_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_2_cast_fp16 = mul(x = mul_2_x_0_to_fp16, y = sub_1_cast_fp16)[name = tensor("mul_2_cast_fp16")]; tensor mul_3_y_0_to_fp16 = const()[name = tensor("mul_3_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_3_cast_fp16 = mul(x = q_7_cast_fp16, y = mul_3_y_0_to_fp16)[name = tensor("mul_3_cast_fp16")]; 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_cast_fp16 = matmul(transpose_x = matmul_1_transpose_x_0, transpose_y = matmul_1_transpose_y_0, x = mul_3_cast_fp16, y = k_7_cast_fp16)[name = tensor("matmul_1_cast_fp16")]; tensor add_1_cast_fp16 = add(x = matmul_1_cast_fp16, y = mul_2_cast_fp16)[name = tensor("add_1_cast_fp16")]; tensor add_1_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_1_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_1_axis_0 = const()[name = tensor("softmax_1_axis_0"), val = tensor(-1)]; tensor add_1_cast_fp16_to_fp32 = cast(dtype = add_1_cast_fp16_to_fp32_dtype_0, x = add_1_cast_fp16)[name = tensor("cast_384")]; tensor softmax_1 = softmax(axis = softmax_1_axis_0, x = add_1_cast_fp16_to_fp32)[name = tensor("softmax_1")]; tensor out_5_transpose_x_0 = const()[name = tensor("out_5_transpose_x_0"), val = tensor(false)]; tensor out_5_transpose_y_0 = const()[name = tensor("out_5_transpose_y_0"), val = tensor(false)]; tensor softmax_1_to_fp16_dtype_0 = const()[name = tensor("softmax_1_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_1_to_fp16 = cast(dtype = softmax_1_to_fp16_dtype_0, x = softmax_1)[name = tensor("cast_383")]; tensor out_5_cast_fp16 = matmul(transpose_x = out_5_transpose_x_0, transpose_y = out_5_transpose_y_0, x = softmax_1_to_fp16, y = v_3_cast_fp16)[name = tensor("out_5_cast_fp16")]; tensor concat_23x = const()[name = tensor("concat_23x"), val = tensor([1, -1])]; tensor var_822 = reshape(shape = concat_23x, x = positions_3)[name = tensor("op_822")]; tensor var_825 = less(x = var_822, y = var_687)[name = tensor("op_825")]; tensor concat_24x = const()[name = tensor("concat_24x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_3 = reshape(shape = concat_24x, x = var_825)[name = tensor("valid_q_3")]; tensor var_829_to_fp16 = const()[name = tensor("op_829_to_fp16"), val = tensor(0x0p+0)]; tensor out_7_cast_fp16 = select(a = out_5_cast_fp16, b = var_829_to_fp16, cond = valid_q_3)[name = tensor("out_7_cast_fp16")]; tensor var_831_perm_0 = const()[name = tensor("op_831_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_25x = const()[name = tensor("concat_25x"), val = tensor([1, -1, 256])]; tensor var_831_cast_fp16 = transpose(perm = var_831_perm_0, x = out_7_cast_fp16)[name = tensor("transpose_32")]; tensor input_81_cast_fp16 = reshape(shape = concat_25x, x = var_831_cast_fp16)[name = tensor("input_81_cast_fp16")]; tensor decoder_1_transformer_layers_1_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_1_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3274752)))]; tensor linear_6_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_1_transformer_layers_1_self_attn_out_proj_weight_to_fp16, x = input_81_cast_fp16)[name = tensor("linear_6_cast_fp16")]; tensor decoder_1_transformer_layers_1_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_1_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3405888)))]; tensor var_837_cast_fp16 = mul(x = decoder_1_transformer_layers_1_layer_scale_1_scale_to_fp16, y = linear_6_cast_fp16)[name = tensor("op_837_cast_fp16")]; tensor input_83_cast_fp16 = add(x = input_79_cast_fp16, y = var_837_cast_fp16)[name = tensor("input_83_cast_fp16")]; tensor input_85_axes_0 = const()[name = tensor("input_85_axes_0"), val = tensor([-1])]; tensor decoder_1_transformer_layers_1_norm2_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_1_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3406464)))]; tensor decoder_1_transformer_layers_1_norm2_bias_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_1_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3407040)))]; tensor input_85_cast_fp16 = layer_norm(axes = input_85_axes_0, beta = decoder_1_transformer_layers_1_norm2_bias_to_fp16, epsilon = var_561_to_fp16, gamma = decoder_1_transformer_layers_1_norm2_weight_to_fp16, x = input_83_cast_fp16)[name = tensor("input_85_cast_fp16")]; tensor decoder_1_transformer_layers_1_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_1_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3407616)))]; tensor linear_7_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_1_transformer_layers_1_ffn_0_weight_to_fp16, x = input_85_cast_fp16)[name = tensor("linear_7_cast_fp16")]; tensor input_89_mode_0 = const()[name = tensor("input_89_mode_0"), val = tensor("EXACT")]; tensor input_89_cast_fp16 = gelu(mode = input_89_mode_0, x = linear_7_cast_fp16)[name = tensor("input_89_cast_fp16")]; tensor decoder_1_transformer_layers_1_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_1_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(3931968)))]; tensor linear_8_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_1_transformer_layers_1_ffn_2_weight_to_fp16, x = input_89_cast_fp16)[name = tensor("linear_8_cast_fp16")]; tensor decoder_1_transformer_layers_1_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_1_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4456320)))]; tensor var_852_cast_fp16 = mul(x = decoder_1_transformer_layers_1_layer_scale_2_scale_to_fp16, y = linear_8_cast_fp16)[name = tensor("op_852_cast_fp16")]; tensor input_91_cast_fp16 = add(x = input_83_cast_fp16, y = var_852_cast_fp16)[name = tensor("input_91_cast_fp16")]; tensor x_19_axes_0 = const()[name = tensor("x_19_axes_0"), val = tensor([-1])]; tensor decoder_1_transformer_layers_2_norm1_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_2_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4456896)))]; tensor decoder_1_transformer_layers_2_norm1_bias_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_2_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4457472)))]; tensor x_19_cast_fp16 = layer_norm(axes = x_19_axes_0, beta = decoder_1_transformer_layers_2_norm1_bias_to_fp16, epsilon = var_561_to_fp16, gamma = decoder_1_transformer_layers_2_norm1_weight_to_fp16, x = input_91_cast_fp16)[name = tensor("x_19_cast_fp16")]; tensor var_867_shape_cast_fp16 = shape(x = x_19_cast_fp16)[name = tensor("op_867_shape_cast_fp16")]; tensor gather_24_axis_0 = const()[name = tensor("gather_24_axis_0"), val = tensor(0)]; tensor gather_24_batch_dims_0 = const()[name = tensor("gather_24_batch_dims_0"), val = tensor(0)]; tensor gather_24_validate_indices_0 = const()[name = tensor("gather_24_validate_indices_0"), val = tensor(false)]; tensor var_867_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_867_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_24_indices_0_to_uint16 = const()[name = tensor("gather_24_indices_0_to_uint16"), val = tensor(1)]; tensor var_867_shape_cast_fp16_to_uint16 = cast(dtype = var_867_shape_cast_fp16_to_uint16_dtype_0, x = var_867_shape_cast_fp16)[name = tensor("cast_382")]; tensor gather_24_cast_uint16 = gather(axis = gather_24_axis_0, batch_dims = gather_24_batch_dims_0, indices = gather_24_indices_0_to_uint16, validate_indices = gather_24_validate_indices_0, x = var_867_shape_cast_fp16_to_uint16)[name = tensor("gather_24_cast_uint16")]; tensor gather_24_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_24_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_1_transformer_layers_2_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_2_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4458048)))]; tensor linear_9_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_1_transformer_layers_2_self_attn_in_proj_weight_to_fp16, x = x_19_cast_fp16)[name = tensor("linear_9_cast_fp16")]; tensor concat_26x = const()[name = tensor("concat_26x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_874_cast_fp16 = reshape(shape = concat_26x, x = linear_9_cast_fp16)[name = tensor("op_874_cast_fp16")]; tensor var_875 = const()[name = tensor("op_875"), val = tensor([2, 0, 3, 1, 4])]; tensor q_9_begin_0 = const()[name = tensor("q_9_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_9_end_0 = const()[name = tensor("q_9_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_9_end_mask_0 = const()[name = tensor("q_9_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_9_squeeze_mask_0 = const()[name = tensor("q_9_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_11_cast_fp16 = transpose(perm = var_875, x = var_874_cast_fp16)[name = tensor("transpose_31")]; tensor q_9_cast_fp16 = slice_by_index(begin = q_9_begin_0, end = q_9_end_0, end_mask = q_9_end_mask_0, squeeze_mask = q_9_squeeze_mask_0, x = projected_11_cast_fp16)[name = tensor("q_9_cast_fp16")]; tensor k_9_begin_0 = const()[name = tensor("k_9_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_9_end_0 = const()[name = tensor("k_9_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_9_end_mask_0 = const()[name = tensor("k_9_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_9_squeeze_mask_0 = const()[name = tensor("k_9_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_9_cast_fp16 = slice_by_index(begin = k_9_begin_0, end = k_9_end_0, end_mask = k_9_end_mask_0, squeeze_mask = k_9_squeeze_mask_0, x = projected_11_cast_fp16)[name = tensor("k_9_cast_fp16")]; tensor v_5_begin_0 = const()[name = tensor("v_5_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_5_end_0 = const()[name = tensor("v_5_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_5_end_mask_0 = const()[name = tensor("v_5_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_5_squeeze_mask_0 = const()[name = tensor("v_5_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_5_cast_fp16 = slice_by_index(begin = v_5_begin_0, end = v_5_end_0, end_mask = v_5_end_mask_0, squeeze_mask = v_5_squeeze_mask_0, x = projected_11_cast_fp16)[name = tensor("v_5_cast_fp16")]; tensor var_887_shape_cast_fp16 = shape(x = q_9_cast_fp16)[name = tensor("op_887_shape_cast_fp16")]; tensor gather_28_axis_0 = const()[name = tensor("gather_28_axis_0"), val = tensor(0)]; tensor gather_28_batch_dims_0 = const()[name = tensor("gather_28_batch_dims_0"), val = tensor(0)]; tensor gather_28_validate_indices_0 = const()[name = tensor("gather_28_validate_indices_0"), val = tensor(false)]; tensor var_887_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_887_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_28_indices_0_to_uint16 = const()[name = tensor("gather_28_indices_0_to_uint16"), val = tensor(2)]; tensor var_887_shape_cast_fp16_to_uint16 = cast(dtype = var_887_shape_cast_fp16_to_uint16_dtype_0, x = var_887_shape_cast_fp16)[name = tensor("cast_380")]; tensor gather_28_cast_uint16 = gather(axis = gather_28_axis_0, batch_dims = gather_28_batch_dims_0, indices = gather_28_indices_0_to_uint16, validate_indices = gather_28_validate_indices_0, x = var_887_shape_cast_fp16_to_uint16)[name = tensor("gather_28_cast_uint16")]; tensor gather_28_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_28_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_21 = const()[name = tensor("const_21"), val = tensor(0)]; tensor const_22 = const()[name = tensor("const_22"), val = tensor(1)]; tensor gather_28_cast_uint16_to_int32 = cast(dtype = gather_28_cast_uint16_to_int32_dtype_0, x = gather_28_cast_uint16)[name = tensor("cast_379")]; tensor var_891 = range_1d(end = gather_28_cast_uint16_to_int32, start = const_21, step = const_22)[name = tensor("op_891")]; tensor var_890_to_fp16 = const()[name = tensor("op_890_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_891_promoted_to_fp16_dtype_0 = const()[name = tensor("op_891_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_891_to_fp16 = cast(dtype = var_891_promoted_to_fp16_dtype_0, x = var_891)[name = tensor("cast_378")]; tensor ts_9_cast_fp16 = add(x = var_890_to_fp16, y = var_891_to_fp16)[name = tensor("ts_9_cast_fp16")]; tensor var_894 = const()[name = tensor("op_894"), val = tensor([1, 1, -1, 1])]; tensor ts_11_cast_fp16 = reshape(shape = var_894, x = ts_9_cast_fp16)[name = tensor("ts_11_cast_fp16")]; tensor concat_27x = const()[name = tensor("concat_27x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_5_cast_fp16 = reshape(shape = concat_27x, x = q_9_cast_fp16)[name = tensor("q2_5_cast_fp16")]; tensor concat_28x = const()[name = tensor("concat_28x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_5_cast_fp16 = reshape(shape = concat_28x, x = k_9_cast_fp16)[name = tensor("k2_5_cast_fp16")]; tensor qr_5_begin_0 = const()[name = tensor("qr_5_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_5_end_0 = const()[name = tensor("qr_5_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_5_end_mask_0 = const()[name = tensor("qr_5_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_5_squeeze_mask_0 = const()[name = tensor("qr_5_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_5_cast_fp16 = slice_by_index(begin = qr_5_begin_0, end = qr_5_end_0, end_mask = qr_5_end_mask_0, squeeze_mask = qr_5_squeeze_mask_0, x = q2_5_cast_fp16)[name = tensor("qr_5_cast_fp16")]; tensor qi_5_begin_0 = const()[name = tensor("qi_5_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_5_end_0 = const()[name = tensor("qi_5_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_5_end_mask_0 = const()[name = tensor("qi_5_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_5_squeeze_mask_0 = const()[name = tensor("qi_5_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_5_cast_fp16 = slice_by_index(begin = qi_5_begin_0, end = qi_5_end_0, end_mask = qi_5_end_mask_0, squeeze_mask = qi_5_squeeze_mask_0, x = q2_5_cast_fp16)[name = tensor("qi_5_cast_fp16")]; tensor kr_5_begin_0 = const()[name = tensor("kr_5_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_5_end_0 = const()[name = tensor("kr_5_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_5_end_mask_0 = const()[name = tensor("kr_5_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_5_squeeze_mask_0 = const()[name = tensor("kr_5_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_5_cast_fp16 = slice_by_index(begin = kr_5_begin_0, end = kr_5_end_0, end_mask = kr_5_end_mask_0, squeeze_mask = kr_5_squeeze_mask_0, x = k2_5_cast_fp16)[name = tensor("kr_5_cast_fp16")]; tensor ki_5_begin_0 = const()[name = tensor("ki_5_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_5_end_0 = const()[name = tensor("ki_5_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_5_end_mask_0 = const()[name = tensor("ki_5_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_5_squeeze_mask_0 = const()[name = tensor("ki_5_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_5_cast_fp16 = slice_by_index(begin = ki_5_begin_0, end = ki_5_end_0, end_mask = ki_5_end_mask_0, squeeze_mask = ki_5_squeeze_mask_0, x = k2_5_cast_fp16)[name = tensor("ki_5_cast_fp16")]; tensor freqs_5_to_fp16 = const()[name = tensor("freqs_5_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4851328)))]; tensor var_907_cast_fp16 = mul(x = freqs_5_to_fp16, y = ts_11_cast_fp16)[name = tensor("op_907_cast_fp16")]; tensor rotr_5_cast_fp16 = cos(x = var_907_cast_fp16)[name = tensor("rotr_5_cast_fp16")]; tensor roti_5_cast_fp16 = sin(x = var_907_cast_fp16)[name = tensor("roti_5_cast_fp16")]; tensor var_911_cast_fp16 = mul(x = qr_5_cast_fp16, y = rotr_5_cast_fp16)[name = tensor("op_911_cast_fp16")]; tensor var_912_cast_fp16 = mul(x = qi_5_cast_fp16, y = roti_5_cast_fp16)[name = tensor("op_912_cast_fp16")]; tensor var_913_cast_fp16 = sub(x = var_911_cast_fp16, y = var_912_cast_fp16)[name = tensor("op_913_cast_fp16")]; tensor var_914_cast_fp16 = mul(x = qr_5_cast_fp16, y = roti_5_cast_fp16)[name = tensor("op_914_cast_fp16")]; tensor var_915_cast_fp16 = mul(x = qi_5_cast_fp16, y = rotr_5_cast_fp16)[name = tensor("op_915_cast_fp16")]; tensor var_916_cast_fp16 = add(x = var_914_cast_fp16, y = var_915_cast_fp16)[name = tensor("op_916_cast_fp16")]; tensor qo_5_axis_0 = const()[name = tensor("qo_5_axis_0"), val = tensor(-1)]; tensor qo_5_cast_fp16 = stack(axis = qo_5_axis_0, values = (var_913_cast_fp16, var_916_cast_fp16))[name = tensor("qo_5_cast_fp16")]; tensor var_919_cast_fp16 = mul(x = kr_5_cast_fp16, y = rotr_5_cast_fp16)[name = tensor("op_919_cast_fp16")]; tensor var_920_cast_fp16 = mul(x = ki_5_cast_fp16, y = roti_5_cast_fp16)[name = tensor("op_920_cast_fp16")]; tensor var_921_cast_fp16 = sub(x = var_919_cast_fp16, y = var_920_cast_fp16)[name = tensor("op_921_cast_fp16")]; tensor var_922_cast_fp16 = mul(x = kr_5_cast_fp16, y = roti_5_cast_fp16)[name = tensor("op_922_cast_fp16")]; tensor var_923_cast_fp16 = mul(x = ki_5_cast_fp16, y = rotr_5_cast_fp16)[name = tensor("op_923_cast_fp16")]; tensor var_924_cast_fp16 = add(x = var_922_cast_fp16, y = var_923_cast_fp16)[name = tensor("op_924_cast_fp16")]; tensor ko_5_axis_0 = const()[name = tensor("ko_5_axis_0"), val = tensor(-1)]; tensor ko_5_cast_fp16 = stack(axis = ko_5_axis_0, values = (var_921_cast_fp16, var_924_cast_fp16))[name = tensor("ko_5_cast_fp16")]; tensor concat_29x = const()[name = tensor("concat_29x"), val = tensor([1, 4, -1, 64])]; tensor q_11_cast_fp16 = reshape(shape = concat_29x, x = qo_5_cast_fp16)[name = tensor("q_11_cast_fp16")]; tensor concat_30x = const()[name = tensor("concat_30x"), val = tensor([1, 4, -1, 64])]; tensor k_11_cast_fp16 = reshape(shape = concat_30x, x = ko_5_cast_fp16)[name = tensor("k_11_cast_fp16")]; tensor const_23 = const()[name = tensor("const_23"), val = tensor(0)]; tensor const_24 = const()[name = tensor("const_24"), val = tensor(1)]; tensor gather_24_cast_uint16_to_int32 = cast(dtype = gather_24_cast_uint16_to_int32_dtype_0, x = gather_24_cast_uint16)[name = tensor("cast_381")]; tensor positions_5 = range_1d(end = gather_24_cast_uint16_to_int32, start = const_23, step = const_24)[name = tensor("positions_5")]; tensor concat_31x = const()[name = tensor("concat_31x"), val = tensor([1, 1, -1])]; tensor var_936 = reshape(shape = concat_31x, x = positions_5)[name = tensor("op_936")]; tensor valid_k_5 = less(x = var_936, y = var_664)[name = tensor("valid_k_5")]; tensor concat_32x = const()[name = tensor("concat_32x"), val = tensor([1, -1, 1])]; tensor var_941 = reshape(shape = concat_32x, x = positions_5)[name = tensor("op_941")]; tensor delta_5 = sub(x = var_941, y = var_936)[name = tensor("delta_5")]; tensor concat_34_axis_0 = const()[name = tensor("concat_34_axis_0"), val = tensor(0)]; tensor concat_34_interleave_0 = const()[name = tensor("concat_34_interleave_0"), val = tensor(false)]; tensor concat_34 = concat(axis = concat_34_axis_0, interleave = concat_34_interleave_0, values = (var_565, gather_24_cast_uint16_to_int32, gather_24_cast_uint16_to_int32))[name = tensor("concat_34")]; tensor fill_2_value_0_to_fp16 = const()[name = tensor("fill_2_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_2_cast_fp16 = fill(shape = concat_34, value = fill_2_value_0_to_fp16)[name = tensor("fill_2_cast_fp16")]; tensor attn_bias_17_dtype_0 = const()[name = tensor("attn_bias_17_dtype_0"), val = tensor("bool")]; tensor var_947 = greater_equal(x = delta_5, y = var_547)[name = tensor("op_947")]; tensor fill_2_cast_fp16_to_bool = cast(dtype = attn_bias_17_dtype_0, x = fill_2_cast_fp16)[name = tensor("cast_377")]; tensor attn_bias_19 = logical_and(x = fill_2_cast_fp16_to_bool, y = var_947)[name = tensor("attn_bias_19")]; tensor var_949 = less(x = delta_5, y = var_554)[name = tensor("op_949")]; tensor attn_bias_21 = logical_and(x = attn_bias_19, y = var_949)[name = tensor("attn_bias_21")]; tensor var_951 = logical_and(x = attn_bias_21, y = valid_k_5)[name = tensor("op_951")]; tensor var_953_axes_0 = const()[name = tensor("op_953_axes_0"), val = tensor([1])]; tensor var_953 = expand_dims(axes = var_953_axes_0, x = var_951)[name = tensor("op_953")]; tensor sub_2_x_0_to_fp16 = const()[name = tensor("sub_2_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_79_to_fp16_dtype_0 = const()[name = tensor("cast_79_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_953_to_fp16 = cast(dtype = cast_79_to_fp16_dtype_0, x = var_953)[name = tensor("cast_376")]; tensor sub_2_cast_fp16 = sub(x = sub_2_x_0_to_fp16, y = var_953_to_fp16)[name = tensor("sub_2_cast_fp16")]; tensor mul_4_x_0_to_fp16 = const()[name = tensor("mul_4_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_4_cast_fp16 = mul(x = mul_4_x_0_to_fp16, y = sub_2_cast_fp16)[name = tensor("mul_4_cast_fp16")]; tensor mul_5_y_0_to_fp16 = const()[name = tensor("mul_5_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_5_cast_fp16 = mul(x = q_11_cast_fp16, y = mul_5_y_0_to_fp16)[name = tensor("mul_5_cast_fp16")]; tensor matmul_2_transpose_y_0 = const()[name = tensor("matmul_2_transpose_y_0"), val = tensor(true)]; tensor matmul_2_transpose_x_0 = const()[name = tensor("matmul_2_transpose_x_0"), val = tensor(false)]; tensor matmul_2_cast_fp16 = matmul(transpose_x = matmul_2_transpose_x_0, transpose_y = matmul_2_transpose_y_0, x = mul_5_cast_fp16, y = k_11_cast_fp16)[name = tensor("matmul_2_cast_fp16")]; tensor add_2_cast_fp16 = add(x = matmul_2_cast_fp16, y = mul_4_cast_fp16)[name = tensor("add_2_cast_fp16")]; tensor add_2_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_2_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_2_axis_0 = const()[name = tensor("softmax_2_axis_0"), val = tensor(-1)]; tensor add_2_cast_fp16_to_fp32 = cast(dtype = add_2_cast_fp16_to_fp32_dtype_0, x = add_2_cast_fp16)[name = tensor("cast_375")]; tensor softmax_2 = softmax(axis = softmax_2_axis_0, x = add_2_cast_fp16_to_fp32)[name = tensor("softmax_2")]; tensor out_9_transpose_x_0 = const()[name = tensor("out_9_transpose_x_0"), val = tensor(false)]; tensor out_9_transpose_y_0 = const()[name = tensor("out_9_transpose_y_0"), val = tensor(false)]; tensor softmax_2_to_fp16_dtype_0 = const()[name = tensor("softmax_2_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_2_to_fp16 = cast(dtype = softmax_2_to_fp16_dtype_0, x = softmax_2)[name = tensor("cast_374")]; tensor out_9_cast_fp16 = matmul(transpose_x = out_9_transpose_x_0, transpose_y = out_9_transpose_y_0, x = softmax_2_to_fp16, y = v_5_cast_fp16)[name = tensor("out_9_cast_fp16")]; tensor concat_35x = const()[name = tensor("concat_35x"), val = tensor([1, -1])]; tensor var_959 = reshape(shape = concat_35x, x = positions_5)[name = tensor("op_959")]; tensor var_962 = less(x = var_959, y = var_687)[name = tensor("op_962")]; tensor concat_36x = const()[name = tensor("concat_36x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_5 = reshape(shape = concat_36x, x = var_962)[name = tensor("valid_q_5")]; tensor var_966_to_fp16 = const()[name = tensor("op_966_to_fp16"), val = tensor(0x0p+0)]; tensor out_11_cast_fp16 = select(a = out_9_cast_fp16, b = var_966_to_fp16, cond = valid_q_5)[name = tensor("out_11_cast_fp16")]; tensor var_968_perm_0 = const()[name = tensor("op_968_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_37x = const()[name = tensor("concat_37x"), val = tensor([1, -1, 256])]; tensor var_968_cast_fp16 = transpose(perm = var_968_perm_0, x = out_11_cast_fp16)[name = tensor("transpose_30")]; tensor input_93_cast_fp16 = reshape(shape = concat_37x, x = var_968_cast_fp16)[name = tensor("input_93_cast_fp16")]; tensor decoder_1_transformer_layers_2_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_2_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4851456)))]; tensor linear_10_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_1_transformer_layers_2_self_attn_out_proj_weight_to_fp16, x = input_93_cast_fp16)[name = tensor("linear_10_cast_fp16")]; tensor decoder_1_transformer_layers_2_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_2_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4982592)))]; tensor var_974_cast_fp16 = mul(x = decoder_1_transformer_layers_2_layer_scale_1_scale_to_fp16, y = linear_10_cast_fp16)[name = tensor("op_974_cast_fp16")]; tensor input_95_cast_fp16 = add(x = input_91_cast_fp16, y = var_974_cast_fp16)[name = tensor("input_95_cast_fp16")]; tensor input_97_axes_0 = const()[name = tensor("input_97_axes_0"), val = tensor([-1])]; tensor decoder_1_transformer_layers_2_norm2_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_2_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4983168)))]; tensor decoder_1_transformer_layers_2_norm2_bias_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_2_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4983744)))]; tensor input_97_cast_fp16 = layer_norm(axes = input_97_axes_0, beta = decoder_1_transformer_layers_2_norm2_bias_to_fp16, epsilon = var_561_to_fp16, gamma = decoder_1_transformer_layers_2_norm2_weight_to_fp16, x = input_95_cast_fp16)[name = tensor("input_97_cast_fp16")]; tensor decoder_1_transformer_layers_2_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_2_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(4984320)))]; tensor linear_11_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_1_transformer_layers_2_ffn_0_weight_to_fp16, x = input_97_cast_fp16)[name = tensor("linear_11_cast_fp16")]; tensor input_101_mode_0 = const()[name = tensor("input_101_mode_0"), val = tensor("EXACT")]; tensor input_101_cast_fp16 = gelu(mode = input_101_mode_0, x = linear_11_cast_fp16)[name = tensor("input_101_cast_fp16")]; tensor decoder_1_transformer_layers_2_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_2_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(5508672)))]; tensor linear_12_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_1_transformer_layers_2_ffn_2_weight_to_fp16, x = input_101_cast_fp16)[name = tensor("linear_12_cast_fp16")]; tensor decoder_1_transformer_layers_2_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_2_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6033024)))]; tensor var_989_cast_fp16 = mul(x = decoder_1_transformer_layers_2_layer_scale_2_scale_to_fp16, y = linear_12_cast_fp16)[name = tensor("op_989_cast_fp16")]; tensor input_103_cast_fp16 = add(x = input_95_cast_fp16, y = var_989_cast_fp16)[name = tensor("input_103_cast_fp16")]; tensor x_25_axes_0 = const()[name = tensor("x_25_axes_0"), val = tensor([-1])]; tensor decoder_1_transformer_layers_3_norm1_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_3_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6033600)))]; tensor decoder_1_transformer_layers_3_norm1_bias_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_3_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6034176)))]; tensor x_25_cast_fp16 = layer_norm(axes = x_25_axes_0, beta = decoder_1_transformer_layers_3_norm1_bias_to_fp16, epsilon = var_561_to_fp16, gamma = decoder_1_transformer_layers_3_norm1_weight_to_fp16, x = input_103_cast_fp16)[name = tensor("x_25_cast_fp16")]; tensor var_1004_shape_cast_fp16 = shape(x = x_25_cast_fp16)[name = tensor("op_1004_shape_cast_fp16")]; tensor gather_34_axis_0 = const()[name = tensor("gather_34_axis_0"), val = tensor(0)]; tensor gather_34_batch_dims_0 = const()[name = tensor("gather_34_batch_dims_0"), val = tensor(0)]; tensor gather_34_validate_indices_0 = const()[name = tensor("gather_34_validate_indices_0"), val = tensor(false)]; tensor var_1004_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1004_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_34_indices_0_to_uint16 = const()[name = tensor("gather_34_indices_0_to_uint16"), val = tensor(1)]; tensor var_1004_shape_cast_fp16_to_uint16 = cast(dtype = var_1004_shape_cast_fp16_to_uint16_dtype_0, x = var_1004_shape_cast_fp16)[name = tensor("cast_373")]; tensor gather_34_cast_uint16 = gather(axis = gather_34_axis_0, batch_dims = gather_34_batch_dims_0, indices = gather_34_indices_0_to_uint16, validate_indices = gather_34_validate_indices_0, x = var_1004_shape_cast_fp16_to_uint16)[name = tensor("gather_34_cast_uint16")]; tensor gather_34_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_34_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_1_transformer_layers_3_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_3_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6034752)))]; tensor linear_13_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_1_transformer_layers_3_self_attn_in_proj_weight_to_fp16, x = x_25_cast_fp16)[name = tensor("linear_13_cast_fp16")]; tensor concat_38x = const()[name = tensor("concat_38x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_1011_cast_fp16 = reshape(shape = concat_38x, x = linear_13_cast_fp16)[name = tensor("op_1011_cast_fp16")]; tensor var_1012 = const()[name = tensor("op_1012"), val = tensor([2, 0, 3, 1, 4])]; tensor q_13_begin_0 = const()[name = tensor("q_13_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_13_end_0 = const()[name = tensor("q_13_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_13_end_mask_0 = const()[name = tensor("q_13_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_13_squeeze_mask_0 = const()[name = tensor("q_13_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_15_cast_fp16 = transpose(perm = var_1012, x = var_1011_cast_fp16)[name = tensor("transpose_29")]; tensor q_13_cast_fp16 = slice_by_index(begin = q_13_begin_0, end = q_13_end_0, end_mask = q_13_end_mask_0, squeeze_mask = q_13_squeeze_mask_0, x = projected_15_cast_fp16)[name = tensor("q_13_cast_fp16")]; tensor k_13_begin_0 = const()[name = tensor("k_13_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_13_end_0 = const()[name = tensor("k_13_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_13_end_mask_0 = const()[name = tensor("k_13_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_13_squeeze_mask_0 = const()[name = tensor("k_13_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_13_cast_fp16 = slice_by_index(begin = k_13_begin_0, end = k_13_end_0, end_mask = k_13_end_mask_0, squeeze_mask = k_13_squeeze_mask_0, x = projected_15_cast_fp16)[name = tensor("k_13_cast_fp16")]; tensor v_7_begin_0 = const()[name = tensor("v_7_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_7_end_0 = const()[name = tensor("v_7_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_7_end_mask_0 = const()[name = tensor("v_7_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_7_squeeze_mask_0 = const()[name = tensor("v_7_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_7_cast_fp16 = slice_by_index(begin = v_7_begin_0, end = v_7_end_0, end_mask = v_7_end_mask_0, squeeze_mask = v_7_squeeze_mask_0, x = projected_15_cast_fp16)[name = tensor("v_7_cast_fp16")]; tensor var_1024_shape_cast_fp16 = shape(x = q_13_cast_fp16)[name = tensor("op_1024_shape_cast_fp16")]; tensor gather_38_axis_0 = const()[name = tensor("gather_38_axis_0"), val = tensor(0)]; tensor gather_38_batch_dims_0 = const()[name = tensor("gather_38_batch_dims_0"), val = tensor(0)]; tensor gather_38_validate_indices_0 = const()[name = tensor("gather_38_validate_indices_0"), val = tensor(false)]; tensor var_1024_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1024_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_38_indices_0_to_uint16 = const()[name = tensor("gather_38_indices_0_to_uint16"), val = tensor(2)]; tensor var_1024_shape_cast_fp16_to_uint16 = cast(dtype = var_1024_shape_cast_fp16_to_uint16_dtype_0, x = var_1024_shape_cast_fp16)[name = tensor("cast_371")]; tensor gather_38_cast_uint16 = gather(axis = gather_38_axis_0, batch_dims = gather_38_batch_dims_0, indices = gather_38_indices_0_to_uint16, validate_indices = gather_38_validate_indices_0, x = var_1024_shape_cast_fp16_to_uint16)[name = tensor("gather_38_cast_uint16")]; tensor gather_38_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_38_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_30 = const()[name = tensor("const_30"), val = tensor(0)]; tensor const_31 = const()[name = tensor("const_31"), val = tensor(1)]; tensor gather_38_cast_uint16_to_int32 = cast(dtype = gather_38_cast_uint16_to_int32_dtype_0, x = gather_38_cast_uint16)[name = tensor("cast_370")]; tensor var_1028 = range_1d(end = gather_38_cast_uint16_to_int32, start = const_30, step = const_31)[name = tensor("op_1028")]; tensor var_1027_to_fp16 = const()[name = tensor("op_1027_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_1028_promoted_to_fp16_dtype_0 = const()[name = tensor("op_1028_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1028_to_fp16 = cast(dtype = var_1028_promoted_to_fp16_dtype_0, x = var_1028)[name = tensor("cast_369")]; tensor ts_13_cast_fp16 = add(x = var_1027_to_fp16, y = var_1028_to_fp16)[name = tensor("ts_13_cast_fp16")]; tensor var_1031 = const()[name = tensor("op_1031"), val = tensor([1, 1, -1, 1])]; tensor ts_15_cast_fp16 = reshape(shape = var_1031, x = ts_13_cast_fp16)[name = tensor("ts_15_cast_fp16")]; tensor concat_39x = const()[name = tensor("concat_39x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_7_cast_fp16 = reshape(shape = concat_39x, x = q_13_cast_fp16)[name = tensor("q2_7_cast_fp16")]; tensor concat_40x = const()[name = tensor("concat_40x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_7_cast_fp16 = reshape(shape = concat_40x, x = k_13_cast_fp16)[name = tensor("k2_7_cast_fp16")]; tensor qr_7_begin_0 = const()[name = tensor("qr_7_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_7_end_0 = const()[name = tensor("qr_7_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_7_end_mask_0 = const()[name = tensor("qr_7_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_7_squeeze_mask_0 = const()[name = tensor("qr_7_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_7_cast_fp16 = slice_by_index(begin = qr_7_begin_0, end = qr_7_end_0, end_mask = qr_7_end_mask_0, squeeze_mask = qr_7_squeeze_mask_0, x = q2_7_cast_fp16)[name = tensor("qr_7_cast_fp16")]; tensor qi_7_begin_0 = const()[name = tensor("qi_7_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_7_end_0 = const()[name = tensor("qi_7_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_7_end_mask_0 = const()[name = tensor("qi_7_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_7_squeeze_mask_0 = const()[name = tensor("qi_7_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_7_cast_fp16 = slice_by_index(begin = qi_7_begin_0, end = qi_7_end_0, end_mask = qi_7_end_mask_0, squeeze_mask = qi_7_squeeze_mask_0, x = q2_7_cast_fp16)[name = tensor("qi_7_cast_fp16")]; tensor kr_7_begin_0 = const()[name = tensor("kr_7_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_7_end_0 = const()[name = tensor("kr_7_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_7_end_mask_0 = const()[name = tensor("kr_7_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_7_squeeze_mask_0 = const()[name = tensor("kr_7_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_7_cast_fp16 = slice_by_index(begin = kr_7_begin_0, end = kr_7_end_0, end_mask = kr_7_end_mask_0, squeeze_mask = kr_7_squeeze_mask_0, x = k2_7_cast_fp16)[name = tensor("kr_7_cast_fp16")]; tensor ki_7_begin_0 = const()[name = tensor("ki_7_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_7_end_0 = const()[name = tensor("ki_7_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_7_end_mask_0 = const()[name = tensor("ki_7_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_7_squeeze_mask_0 = const()[name = tensor("ki_7_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_7_cast_fp16 = slice_by_index(begin = ki_7_begin_0, end = ki_7_end_0, end_mask = ki_7_end_mask_0, squeeze_mask = ki_7_squeeze_mask_0, x = k2_7_cast_fp16)[name = tensor("ki_7_cast_fp16")]; tensor freqs_7_to_fp16 = const()[name = tensor("freqs_7_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6428032)))]; tensor var_1044_cast_fp16 = mul(x = freqs_7_to_fp16, y = ts_15_cast_fp16)[name = tensor("op_1044_cast_fp16")]; tensor rotr_7_cast_fp16 = cos(x = var_1044_cast_fp16)[name = tensor("rotr_7_cast_fp16")]; tensor roti_7_cast_fp16 = sin(x = var_1044_cast_fp16)[name = tensor("roti_7_cast_fp16")]; tensor var_1048_cast_fp16 = mul(x = qr_7_cast_fp16, y = rotr_7_cast_fp16)[name = tensor("op_1048_cast_fp16")]; tensor var_1049_cast_fp16 = mul(x = qi_7_cast_fp16, y = roti_7_cast_fp16)[name = tensor("op_1049_cast_fp16")]; tensor var_1050_cast_fp16 = sub(x = var_1048_cast_fp16, y = var_1049_cast_fp16)[name = tensor("op_1050_cast_fp16")]; tensor var_1051_cast_fp16 = mul(x = qr_7_cast_fp16, y = roti_7_cast_fp16)[name = tensor("op_1051_cast_fp16")]; tensor var_1052_cast_fp16 = mul(x = qi_7_cast_fp16, y = rotr_7_cast_fp16)[name = tensor("op_1052_cast_fp16")]; tensor var_1053_cast_fp16 = add(x = var_1051_cast_fp16, y = var_1052_cast_fp16)[name = tensor("op_1053_cast_fp16")]; tensor qo_7_axis_0 = const()[name = tensor("qo_7_axis_0"), val = tensor(-1)]; tensor qo_7_cast_fp16 = stack(axis = qo_7_axis_0, values = (var_1050_cast_fp16, var_1053_cast_fp16))[name = tensor("qo_7_cast_fp16")]; tensor var_1056_cast_fp16 = mul(x = kr_7_cast_fp16, y = rotr_7_cast_fp16)[name = tensor("op_1056_cast_fp16")]; tensor var_1057_cast_fp16 = mul(x = ki_7_cast_fp16, y = roti_7_cast_fp16)[name = tensor("op_1057_cast_fp16")]; tensor var_1058_cast_fp16 = sub(x = var_1056_cast_fp16, y = var_1057_cast_fp16)[name = tensor("op_1058_cast_fp16")]; tensor var_1059_cast_fp16 = mul(x = kr_7_cast_fp16, y = roti_7_cast_fp16)[name = tensor("op_1059_cast_fp16")]; tensor var_1060_cast_fp16 = mul(x = ki_7_cast_fp16, y = rotr_7_cast_fp16)[name = tensor("op_1060_cast_fp16")]; tensor var_1061_cast_fp16 = add(x = var_1059_cast_fp16, y = var_1060_cast_fp16)[name = tensor("op_1061_cast_fp16")]; tensor ko_7_axis_0 = const()[name = tensor("ko_7_axis_0"), val = tensor(-1)]; tensor ko_7_cast_fp16 = stack(axis = ko_7_axis_0, values = (var_1058_cast_fp16, var_1061_cast_fp16))[name = tensor("ko_7_cast_fp16")]; tensor concat_41x = const()[name = tensor("concat_41x"), val = tensor([1, 4, -1, 64])]; tensor q_15_cast_fp16 = reshape(shape = concat_41x, x = qo_7_cast_fp16)[name = tensor("q_15_cast_fp16")]; tensor concat_42x = const()[name = tensor("concat_42x"), val = tensor([1, 4, -1, 64])]; tensor k_15_cast_fp16 = reshape(shape = concat_42x, x = ko_7_cast_fp16)[name = tensor("k_15_cast_fp16")]; tensor const_32 = const()[name = tensor("const_32"), val = tensor(0)]; tensor const_33 = const()[name = tensor("const_33"), val = tensor(1)]; tensor gather_34_cast_uint16_to_int32 = cast(dtype = gather_34_cast_uint16_to_int32_dtype_0, x = gather_34_cast_uint16)[name = tensor("cast_372")]; tensor positions_7 = range_1d(end = gather_34_cast_uint16_to_int32, start = const_32, step = const_33)[name = tensor("positions_7")]; tensor concat_43x = const()[name = tensor("concat_43x"), val = tensor([1, 1, -1])]; tensor var_1073 = reshape(shape = concat_43x, x = positions_7)[name = tensor("op_1073")]; tensor valid_k_7 = less(x = var_1073, y = var_664)[name = tensor("valid_k_7")]; tensor concat_44x = const()[name = tensor("concat_44x"), val = tensor([1, -1, 1])]; tensor var_1078 = reshape(shape = concat_44x, x = positions_7)[name = tensor("op_1078")]; tensor delta_7 = sub(x = var_1078, y = var_1073)[name = tensor("delta_7")]; tensor concat_46_axis_0 = const()[name = tensor("concat_46_axis_0"), val = tensor(0)]; tensor concat_46_interleave_0 = const()[name = tensor("concat_46_interleave_0"), val = tensor(false)]; tensor concat_46 = concat(axis = concat_46_axis_0, interleave = concat_46_interleave_0, values = (var_565, gather_34_cast_uint16_to_int32, gather_34_cast_uint16_to_int32))[name = tensor("concat_46")]; tensor fill_3_value_0_to_fp16 = const()[name = tensor("fill_3_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_3_cast_fp16 = fill(shape = concat_46, value = fill_3_value_0_to_fp16)[name = tensor("fill_3_cast_fp16")]; tensor attn_bias_25_dtype_0 = const()[name = tensor("attn_bias_25_dtype_0"), val = tensor("bool")]; tensor var_1084 = greater_equal(x = delta_7, y = var_547)[name = tensor("op_1084")]; tensor fill_3_cast_fp16_to_bool = cast(dtype = attn_bias_25_dtype_0, x = fill_3_cast_fp16)[name = tensor("cast_368")]; tensor attn_bias_27 = logical_and(x = fill_3_cast_fp16_to_bool, y = var_1084)[name = tensor("attn_bias_27")]; tensor var_1086 = less(x = delta_7, y = var_554)[name = tensor("op_1086")]; tensor attn_bias_29 = logical_and(x = attn_bias_27, y = var_1086)[name = tensor("attn_bias_29")]; tensor var_1088 = logical_and(x = attn_bias_29, y = valid_k_7)[name = tensor("op_1088")]; tensor var_1090_axes_0 = const()[name = tensor("op_1090_axes_0"), val = tensor([1])]; tensor var_1090 = expand_dims(axes = var_1090_axes_0, x = var_1088)[name = tensor("op_1090")]; tensor sub_3_x_0_to_fp16 = const()[name = tensor("sub_3_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_101_to_fp16_dtype_0 = const()[name = tensor("cast_101_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1090_to_fp16 = cast(dtype = cast_101_to_fp16_dtype_0, x = var_1090)[name = tensor("cast_367")]; tensor sub_3_cast_fp16 = sub(x = sub_3_x_0_to_fp16, y = var_1090_to_fp16)[name = tensor("sub_3_cast_fp16")]; tensor mul_6_x_0_to_fp16 = const()[name = tensor("mul_6_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_6_cast_fp16 = mul(x = mul_6_x_0_to_fp16, y = sub_3_cast_fp16)[name = tensor("mul_6_cast_fp16")]; tensor mul_7_y_0_to_fp16 = const()[name = tensor("mul_7_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_7_cast_fp16 = mul(x = q_15_cast_fp16, y = mul_7_y_0_to_fp16)[name = tensor("mul_7_cast_fp16")]; tensor matmul_3_transpose_y_0 = const()[name = tensor("matmul_3_transpose_y_0"), val = tensor(true)]; tensor matmul_3_transpose_x_0 = const()[name = tensor("matmul_3_transpose_x_0"), val = tensor(false)]; tensor matmul_3_cast_fp16 = matmul(transpose_x = matmul_3_transpose_x_0, transpose_y = matmul_3_transpose_y_0, x = mul_7_cast_fp16, y = k_15_cast_fp16)[name = tensor("matmul_3_cast_fp16")]; tensor add_3_cast_fp16 = add(x = matmul_3_cast_fp16, y = mul_6_cast_fp16)[name = tensor("add_3_cast_fp16")]; tensor add_3_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_3_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_3_axis_0 = const()[name = tensor("softmax_3_axis_0"), val = tensor(-1)]; tensor add_3_cast_fp16_to_fp32 = cast(dtype = add_3_cast_fp16_to_fp32_dtype_0, x = add_3_cast_fp16)[name = tensor("cast_366")]; tensor softmax_3 = softmax(axis = softmax_3_axis_0, x = add_3_cast_fp16_to_fp32)[name = tensor("softmax_3")]; tensor out_13_transpose_x_0 = const()[name = tensor("out_13_transpose_x_0"), val = tensor(false)]; tensor out_13_transpose_y_0 = const()[name = tensor("out_13_transpose_y_0"), val = tensor(false)]; tensor softmax_3_to_fp16_dtype_0 = const()[name = tensor("softmax_3_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_3_to_fp16 = cast(dtype = softmax_3_to_fp16_dtype_0, x = softmax_3)[name = tensor("cast_365")]; tensor out_13_cast_fp16 = matmul(transpose_x = out_13_transpose_x_0, transpose_y = out_13_transpose_y_0, x = softmax_3_to_fp16, y = v_7_cast_fp16)[name = tensor("out_13_cast_fp16")]; tensor concat_47x = const()[name = tensor("concat_47x"), val = tensor([1, -1])]; tensor var_1096 = reshape(shape = concat_47x, x = positions_7)[name = tensor("op_1096")]; tensor var_1099 = less(x = var_1096, y = var_687)[name = tensor("op_1099")]; tensor concat_48x = const()[name = tensor("concat_48x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_7 = reshape(shape = concat_48x, x = var_1099)[name = tensor("valid_q_7")]; tensor var_1103_to_fp16 = const()[name = tensor("op_1103_to_fp16"), val = tensor(0x0p+0)]; tensor out_15_cast_fp16 = select(a = out_13_cast_fp16, b = var_1103_to_fp16, cond = valid_q_7)[name = tensor("out_15_cast_fp16")]; tensor var_1105_perm_0 = const()[name = tensor("op_1105_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_49x = const()[name = tensor("concat_49x"), val = tensor([1, -1, 256])]; tensor var_1105_cast_fp16 = transpose(perm = var_1105_perm_0, x = out_15_cast_fp16)[name = tensor("transpose_28")]; tensor input_105_cast_fp16 = reshape(shape = concat_49x, x = var_1105_cast_fp16)[name = tensor("input_105_cast_fp16")]; tensor decoder_1_transformer_layers_3_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_3_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6428160)))]; tensor linear_14_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_1_transformer_layers_3_self_attn_out_proj_weight_to_fp16, x = input_105_cast_fp16)[name = tensor("linear_14_cast_fp16")]; tensor decoder_1_transformer_layers_3_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_3_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6559296)))]; tensor var_1111_cast_fp16 = mul(x = decoder_1_transformer_layers_3_layer_scale_1_scale_to_fp16, y = linear_14_cast_fp16)[name = tensor("op_1111_cast_fp16")]; tensor input_107_cast_fp16 = add(x = input_103_cast_fp16, y = var_1111_cast_fp16)[name = tensor("input_107_cast_fp16")]; tensor input_109_axes_0 = const()[name = tensor("input_109_axes_0"), val = tensor([-1])]; tensor decoder_1_transformer_layers_3_norm2_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_3_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6559872)))]; tensor decoder_1_transformer_layers_3_norm2_bias_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_3_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6560448)))]; tensor input_109_cast_fp16 = layer_norm(axes = input_109_axes_0, beta = decoder_1_transformer_layers_3_norm2_bias_to_fp16, epsilon = var_561_to_fp16, gamma = decoder_1_transformer_layers_3_norm2_weight_to_fp16, x = input_107_cast_fp16)[name = tensor("input_109_cast_fp16")]; tensor decoder_1_transformer_layers_3_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_3_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(6561024)))]; tensor linear_15_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_1_transformer_layers_3_ffn_0_weight_to_fp16, x = input_109_cast_fp16)[name = tensor("linear_15_cast_fp16")]; tensor input_113_mode_0 = const()[name = tensor("input_113_mode_0"), val = tensor("EXACT")]; tensor input_113_cast_fp16 = gelu(mode = input_113_mode_0, x = linear_15_cast_fp16)[name = tensor("input_113_cast_fp16")]; tensor decoder_1_transformer_layers_3_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_3_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7085376)))]; tensor linear_16_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_1_transformer_layers_3_ffn_2_weight_to_fp16, x = input_113_cast_fp16)[name = tensor("linear_16_cast_fp16")]; tensor decoder_1_transformer_layers_3_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_1_transformer_layers_3_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7609728)))]; tensor var_1126_cast_fp16 = mul(x = decoder_1_transformer_layers_3_layer_scale_2_scale_to_fp16, y = linear_16_cast_fp16)[name = tensor("op_1126_cast_fp16")]; tensor input_115_cast_fp16 = add(x = input_107_cast_fp16, y = var_1126_cast_fp16)[name = tensor("input_115_cast_fp16")]; tensor decoder_1_output_proj_weight_to_fp16 = const()[name = tensor("decoder_1_output_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(7610304)))]; tensor linear_17_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_1_output_proj_weight_to_fp16, x = input_115_cast_fp16)[name = tensor("linear_17_cast_fp16")]; tensor x_31_perm_0 = const()[name = tensor("x_31_perm_0"), val = tensor([0, 2, 1])]; tensor concat_50x = const()[name = tensor("concat_50x"), val = tensor([1, 384, 2, -1])]; tensor x_31_cast_fp16 = transpose(perm = x_31_perm_0, x = linear_17_cast_fp16)[name = tensor("transpose_27")]; tensor var_1145_cast_fp16 = reshape(shape = concat_50x, x = x_31_cast_fp16)[name = tensor("op_1145_cast_fp16")]; tensor var_1146 = const()[name = tensor("op_1146"), val = tensor([0, 1, 3, 2])]; tensor concat_51x = const()[name = tensor("concat_51x"), val = tensor([1, 384, -1])]; tensor var_1147_cast_fp16 = transpose(perm = var_1146, x = var_1145_cast_fp16)[name = tensor("transpose_26")]; tensor x_33_cast_fp16 = reshape(shape = concat_51x, x = var_1147_cast_fp16)[name = tensor("x_33_cast_fp16")]; tensor var_1153 = const()[name = tensor("op_1153"), val = tensor(2)]; tensor input_lengths_5 = mul(x = input_lengths_3, y = var_1153)[name = tensor("input_lengths_5")]; tensor var_1159 = const()[name = tensor("op_1159"), val = tensor(0)]; tensor var_1166 = const()[name = tensor("op_1166"), val = tensor(800)]; tensor var_1177 = const()[name = tensor("op_1177"), val = tensor(1)]; tensor input_117_perm_0 = const()[name = tensor("input_117_perm_0"), val = tensor([0, 2, 1])]; tensor decoder_3_input_proj_weight_to_fp16 = const()[name = tensor("decoder_3_input_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8003584)))]; tensor input_117_cast_fp16 = transpose(perm = input_117_perm_0, x = x_33_cast_fp16)[name = tensor("transpose_25")]; tensor linear_18_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_3_input_proj_weight_to_fp16, x = input_117_cast_fp16)[name = tensor("linear_18_cast_fp16")]; tensor x_37_axes_0 = const()[name = tensor("x_37_axes_0"), val = tensor([-1])]; tensor decoder_3_transformer_layers_0_norm1_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_0_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8200256)))]; tensor decoder_3_transformer_layers_0_norm1_bias_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_0_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8200832)))]; tensor var_1173_to_fp16 = const()[name = tensor("op_1173_to_fp16"), val = tensor(0x1.5p-17)]; tensor x_37_cast_fp16 = layer_norm(axes = x_37_axes_0, beta = decoder_3_transformer_layers_0_norm1_bias_to_fp16, epsilon = var_1173_to_fp16, gamma = decoder_3_transformer_layers_0_norm1_weight_to_fp16, x = linear_18_cast_fp16)[name = tensor("x_37_cast_fp16")]; tensor var_1201_shape_cast_fp16 = shape(x = x_37_cast_fp16)[name = tensor("op_1201_shape_cast_fp16")]; tensor gather_47_axis_0 = const()[name = tensor("gather_47_axis_0"), val = tensor(0)]; tensor gather_47_batch_dims_0 = const()[name = tensor("gather_47_batch_dims_0"), val = tensor(0)]; tensor gather_47_validate_indices_0 = const()[name = tensor("gather_47_validate_indices_0"), val = tensor(false)]; tensor var_1201_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1201_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_47_indices_0_to_uint16 = const()[name = tensor("gather_47_indices_0_to_uint16"), val = tensor(1)]; tensor var_1201_shape_cast_fp16_to_uint16 = cast(dtype = var_1201_shape_cast_fp16_to_uint16_dtype_0, x = var_1201_shape_cast_fp16)[name = tensor("cast_364")]; tensor gather_47_cast_uint16 = gather(axis = gather_47_axis_0, batch_dims = gather_47_batch_dims_0, indices = gather_47_indices_0_to_uint16, validate_indices = gather_47_validate_indices_0, x = var_1201_shape_cast_fp16_to_uint16)[name = tensor("gather_47_cast_uint16")]; tensor gather_47_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_47_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_3_transformer_layers_0_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_0_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8201408)))]; tensor linear_19_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_3_transformer_layers_0_self_attn_in_proj_weight_to_fp16, x = x_37_cast_fp16)[name = tensor("linear_19_cast_fp16")]; tensor concat_52x = const()[name = tensor("concat_52x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_1208_cast_fp16 = reshape(shape = concat_52x, x = linear_19_cast_fp16)[name = tensor("op_1208_cast_fp16")]; tensor var_1209 = const()[name = tensor("op_1209"), val = tensor([2, 0, 3, 1, 4])]; tensor q_17_begin_0 = const()[name = tensor("q_17_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_17_end_0 = const()[name = tensor("q_17_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_17_end_mask_0 = const()[name = tensor("q_17_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_17_squeeze_mask_0 = const()[name = tensor("q_17_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_19_cast_fp16 = transpose(perm = var_1209, x = var_1208_cast_fp16)[name = tensor("transpose_24")]; tensor q_17_cast_fp16 = slice_by_index(begin = q_17_begin_0, end = q_17_end_0, end_mask = q_17_end_mask_0, squeeze_mask = q_17_squeeze_mask_0, x = projected_19_cast_fp16)[name = tensor("q_17_cast_fp16")]; tensor k_17_begin_0 = const()[name = tensor("k_17_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_17_end_0 = const()[name = tensor("k_17_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_17_end_mask_0 = const()[name = tensor("k_17_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_17_squeeze_mask_0 = const()[name = tensor("k_17_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_17_cast_fp16 = slice_by_index(begin = k_17_begin_0, end = k_17_end_0, end_mask = k_17_end_mask_0, squeeze_mask = k_17_squeeze_mask_0, x = projected_19_cast_fp16)[name = tensor("k_17_cast_fp16")]; tensor v_9_begin_0 = const()[name = tensor("v_9_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_9_end_0 = const()[name = tensor("v_9_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_9_end_mask_0 = const()[name = tensor("v_9_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_9_squeeze_mask_0 = const()[name = tensor("v_9_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_9_cast_fp16 = slice_by_index(begin = v_9_begin_0, end = v_9_end_0, end_mask = v_9_end_mask_0, squeeze_mask = v_9_squeeze_mask_0, x = projected_19_cast_fp16)[name = tensor("v_9_cast_fp16")]; tensor var_1221_shape_cast_fp16 = shape(x = q_17_cast_fp16)[name = tensor("op_1221_shape_cast_fp16")]; tensor gather_51_axis_0 = const()[name = tensor("gather_51_axis_0"), val = tensor(0)]; tensor gather_51_batch_dims_0 = const()[name = tensor("gather_51_batch_dims_0"), val = tensor(0)]; tensor gather_51_validate_indices_0 = const()[name = tensor("gather_51_validate_indices_0"), val = tensor(false)]; tensor var_1221_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1221_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_51_indices_0_to_uint16 = const()[name = tensor("gather_51_indices_0_to_uint16"), val = tensor(2)]; tensor var_1221_shape_cast_fp16_to_uint16 = cast(dtype = var_1221_shape_cast_fp16_to_uint16_dtype_0, x = var_1221_shape_cast_fp16)[name = tensor("cast_362")]; tensor gather_51_cast_uint16 = gather(axis = gather_51_axis_0, batch_dims = gather_51_batch_dims_0, indices = gather_51_indices_0_to_uint16, validate_indices = gather_51_validate_indices_0, x = var_1221_shape_cast_fp16_to_uint16)[name = tensor("gather_51_cast_uint16")]; tensor gather_51_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_51_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_39 = const()[name = tensor("const_39"), val = tensor(0)]; tensor const_40 = const()[name = tensor("const_40"), val = tensor(1)]; tensor gather_51_cast_uint16_to_int32 = cast(dtype = gather_51_cast_uint16_to_int32_dtype_0, x = gather_51_cast_uint16)[name = tensor("cast_361")]; tensor var_1225 = range_1d(end = gather_51_cast_uint16_to_int32, start = const_39, step = const_40)[name = tensor("op_1225")]; tensor var_1224_to_fp16 = const()[name = tensor("op_1224_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_1225_promoted_to_fp16_dtype_0 = const()[name = tensor("op_1225_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1225_to_fp16 = cast(dtype = var_1225_promoted_to_fp16_dtype_0, x = var_1225)[name = tensor("cast_360")]; tensor ts_17_cast_fp16 = add(x = var_1224_to_fp16, y = var_1225_to_fp16)[name = tensor("ts_17_cast_fp16")]; tensor var_1228 = const()[name = tensor("op_1228"), val = tensor([1, 1, -1, 1])]; tensor ts_19_cast_fp16 = reshape(shape = var_1228, x = ts_17_cast_fp16)[name = tensor("ts_19_cast_fp16")]; tensor concat_53x = const()[name = tensor("concat_53x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_9_cast_fp16 = reshape(shape = concat_53x, x = q_17_cast_fp16)[name = tensor("q2_9_cast_fp16")]; tensor concat_54x = const()[name = tensor("concat_54x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_9_cast_fp16 = reshape(shape = concat_54x, x = k_17_cast_fp16)[name = tensor("k2_9_cast_fp16")]; tensor qr_9_begin_0 = const()[name = tensor("qr_9_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_9_end_0 = const()[name = tensor("qr_9_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_9_end_mask_0 = const()[name = tensor("qr_9_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_9_squeeze_mask_0 = const()[name = tensor("qr_9_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_9_cast_fp16 = slice_by_index(begin = qr_9_begin_0, end = qr_9_end_0, end_mask = qr_9_end_mask_0, squeeze_mask = qr_9_squeeze_mask_0, x = q2_9_cast_fp16)[name = tensor("qr_9_cast_fp16")]; tensor qi_9_begin_0 = const()[name = tensor("qi_9_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_9_end_0 = const()[name = tensor("qi_9_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_9_end_mask_0 = const()[name = tensor("qi_9_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_9_squeeze_mask_0 = const()[name = tensor("qi_9_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_9_cast_fp16 = slice_by_index(begin = qi_9_begin_0, end = qi_9_end_0, end_mask = qi_9_end_mask_0, squeeze_mask = qi_9_squeeze_mask_0, x = q2_9_cast_fp16)[name = tensor("qi_9_cast_fp16")]; tensor kr_9_begin_0 = const()[name = tensor("kr_9_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_9_end_0 = const()[name = tensor("kr_9_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_9_end_mask_0 = const()[name = tensor("kr_9_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_9_squeeze_mask_0 = const()[name = tensor("kr_9_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_9_cast_fp16 = slice_by_index(begin = kr_9_begin_0, end = kr_9_end_0, end_mask = kr_9_end_mask_0, squeeze_mask = kr_9_squeeze_mask_0, x = k2_9_cast_fp16)[name = tensor("kr_9_cast_fp16")]; tensor ki_9_begin_0 = const()[name = tensor("ki_9_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_9_end_0 = const()[name = tensor("ki_9_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_9_end_mask_0 = const()[name = tensor("ki_9_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_9_squeeze_mask_0 = const()[name = tensor("ki_9_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_9_cast_fp16 = slice_by_index(begin = ki_9_begin_0, end = ki_9_end_0, end_mask = ki_9_end_mask_0, squeeze_mask = ki_9_squeeze_mask_0, x = k2_9_cast_fp16)[name = tensor("ki_9_cast_fp16")]; tensor freqs_9_to_fp16 = const()[name = tensor("freqs_9_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8594688)))]; tensor var_1241_cast_fp16 = mul(x = freqs_9_to_fp16, y = ts_19_cast_fp16)[name = tensor("op_1241_cast_fp16")]; tensor rotr_9_cast_fp16 = cos(x = var_1241_cast_fp16)[name = tensor("rotr_9_cast_fp16")]; tensor roti_9_cast_fp16 = sin(x = var_1241_cast_fp16)[name = tensor("roti_9_cast_fp16")]; tensor var_1245_cast_fp16 = mul(x = qr_9_cast_fp16, y = rotr_9_cast_fp16)[name = tensor("op_1245_cast_fp16")]; tensor var_1246_cast_fp16 = mul(x = qi_9_cast_fp16, y = roti_9_cast_fp16)[name = tensor("op_1246_cast_fp16")]; tensor var_1247_cast_fp16 = sub(x = var_1245_cast_fp16, y = var_1246_cast_fp16)[name = tensor("op_1247_cast_fp16")]; tensor var_1248_cast_fp16 = mul(x = qr_9_cast_fp16, y = roti_9_cast_fp16)[name = tensor("op_1248_cast_fp16")]; tensor var_1249_cast_fp16 = mul(x = qi_9_cast_fp16, y = rotr_9_cast_fp16)[name = tensor("op_1249_cast_fp16")]; tensor var_1250_cast_fp16 = add(x = var_1248_cast_fp16, y = var_1249_cast_fp16)[name = tensor("op_1250_cast_fp16")]; tensor qo_9_axis_0 = const()[name = tensor("qo_9_axis_0"), val = tensor(-1)]; tensor qo_9_cast_fp16 = stack(axis = qo_9_axis_0, values = (var_1247_cast_fp16, var_1250_cast_fp16))[name = tensor("qo_9_cast_fp16")]; tensor var_1253_cast_fp16 = mul(x = kr_9_cast_fp16, y = rotr_9_cast_fp16)[name = tensor("op_1253_cast_fp16")]; tensor var_1254_cast_fp16 = mul(x = ki_9_cast_fp16, y = roti_9_cast_fp16)[name = tensor("op_1254_cast_fp16")]; tensor var_1255_cast_fp16 = sub(x = var_1253_cast_fp16, y = var_1254_cast_fp16)[name = tensor("op_1255_cast_fp16")]; tensor var_1256_cast_fp16 = mul(x = kr_9_cast_fp16, y = roti_9_cast_fp16)[name = tensor("op_1256_cast_fp16")]; tensor var_1257_cast_fp16 = mul(x = ki_9_cast_fp16, y = rotr_9_cast_fp16)[name = tensor("op_1257_cast_fp16")]; tensor var_1258_cast_fp16 = add(x = var_1256_cast_fp16, y = var_1257_cast_fp16)[name = tensor("op_1258_cast_fp16")]; tensor ko_9_axis_0 = const()[name = tensor("ko_9_axis_0"), val = tensor(-1)]; tensor ko_9_cast_fp16 = stack(axis = ko_9_axis_0, values = (var_1255_cast_fp16, var_1258_cast_fp16))[name = tensor("ko_9_cast_fp16")]; tensor concat_55x = const()[name = tensor("concat_55x"), val = tensor([1, 4, -1, 64])]; tensor q_19_cast_fp16 = reshape(shape = concat_55x, x = qo_9_cast_fp16)[name = tensor("q_19_cast_fp16")]; tensor concat_56x = const()[name = tensor("concat_56x"), val = tensor([1, 4, -1, 64])]; tensor k_19_cast_fp16 = reshape(shape = concat_56x, x = ko_9_cast_fp16)[name = tensor("k_19_cast_fp16")]; tensor const_41 = const()[name = tensor("const_41"), val = tensor(0)]; tensor const_42 = const()[name = tensor("const_42"), val = tensor(1)]; tensor gather_47_cast_uint16_to_int32 = cast(dtype = gather_47_cast_uint16_to_int32_dtype_0, x = gather_47_cast_uint16)[name = tensor("cast_363")]; tensor positions_9 = range_1d(end = gather_47_cast_uint16_to_int32, start = const_41, step = const_42)[name = tensor("positions_9")]; tensor concat_57x = const()[name = tensor("concat_57x"), val = tensor([1, 1, -1])]; tensor var_1270 = reshape(shape = concat_57x, x = positions_9)[name = tensor("op_1270")]; tensor var_1271 = const()[name = tensor("op_1271"), val = tensor([-1, 1, 1])]; tensor var_1272 = reshape(shape = var_1271, x = input_lengths_5)[name = tensor("op_1272")]; tensor valid_k_9 = less(x = var_1270, y = var_1272)[name = tensor("valid_k_9")]; tensor concat_58x = const()[name = tensor("concat_58x"), val = tensor([1, -1, 1])]; tensor var_1275 = reshape(shape = concat_58x, x = positions_9)[name = tensor("op_1275")]; tensor delta_9 = sub(x = var_1275, y = var_1270)[name = tensor("delta_9")]; tensor concat_60_axis_0 = const()[name = tensor("concat_60_axis_0"), val = tensor(0)]; tensor concat_60_interleave_0 = const()[name = tensor("concat_60_interleave_0"), val = tensor(false)]; tensor concat_60 = concat(axis = concat_60_axis_0, interleave = concat_60_interleave_0, values = (var_1177, gather_47_cast_uint16_to_int32, gather_47_cast_uint16_to_int32))[name = tensor("concat_60")]; tensor fill_4_value_0_to_fp16 = const()[name = tensor("fill_4_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_4_cast_fp16 = fill(shape = concat_60, value = fill_4_value_0_to_fp16)[name = tensor("fill_4_cast_fp16")]; tensor attn_bias_33_dtype_0 = const()[name = tensor("attn_bias_33_dtype_0"), val = tensor("bool")]; tensor var_1281 = greater_equal(x = delta_9, y = var_1159)[name = tensor("op_1281")]; tensor fill_4_cast_fp16_to_bool = cast(dtype = attn_bias_33_dtype_0, x = fill_4_cast_fp16)[name = tensor("cast_359")]; tensor attn_bias_35 = logical_and(x = fill_4_cast_fp16_to_bool, y = var_1281)[name = tensor("attn_bias_35")]; tensor var_1283 = less(x = delta_9, y = var_1166)[name = tensor("op_1283")]; tensor attn_bias_37 = logical_and(x = attn_bias_35, y = var_1283)[name = tensor("attn_bias_37")]; tensor var_1285 = logical_and(x = attn_bias_37, y = valid_k_9)[name = tensor("op_1285")]; tensor var_1287_axes_0 = const()[name = tensor("op_1287_axes_0"), val = tensor([1])]; tensor var_1287 = expand_dims(axes = var_1287_axes_0, x = var_1285)[name = tensor("op_1287")]; tensor sub_4_x_0_to_fp16 = const()[name = tensor("sub_4_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_125_to_fp16_dtype_0 = const()[name = tensor("cast_125_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1287_to_fp16 = cast(dtype = cast_125_to_fp16_dtype_0, x = var_1287)[name = tensor("cast_358")]; tensor sub_4_cast_fp16 = sub(x = sub_4_x_0_to_fp16, y = var_1287_to_fp16)[name = tensor("sub_4_cast_fp16")]; tensor mul_8_x_0_to_fp16 = const()[name = tensor("mul_8_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_8_cast_fp16 = mul(x = mul_8_x_0_to_fp16, y = sub_4_cast_fp16)[name = tensor("mul_8_cast_fp16")]; tensor mul_9_y_0_to_fp16 = const()[name = tensor("mul_9_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_9_cast_fp16 = mul(x = q_19_cast_fp16, y = mul_9_y_0_to_fp16)[name = tensor("mul_9_cast_fp16")]; tensor matmul_4_transpose_y_0 = const()[name = tensor("matmul_4_transpose_y_0"), val = tensor(true)]; tensor matmul_4_transpose_x_0 = const()[name = tensor("matmul_4_transpose_x_0"), val = tensor(false)]; tensor matmul_4_cast_fp16 = matmul(transpose_x = matmul_4_transpose_x_0, transpose_y = matmul_4_transpose_y_0, x = mul_9_cast_fp16, y = k_19_cast_fp16)[name = tensor("matmul_4_cast_fp16")]; tensor add_4_cast_fp16 = add(x = matmul_4_cast_fp16, y = mul_8_cast_fp16)[name = tensor("add_4_cast_fp16")]; tensor add_4_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_4_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_4_axis_0 = const()[name = tensor("softmax_4_axis_0"), val = tensor(-1)]; tensor add_4_cast_fp16_to_fp32 = cast(dtype = add_4_cast_fp16_to_fp32_dtype_0, x = add_4_cast_fp16)[name = tensor("cast_357")]; tensor softmax_4 = softmax(axis = softmax_4_axis_0, x = add_4_cast_fp16_to_fp32)[name = tensor("softmax_4")]; tensor out_17_transpose_x_0 = const()[name = tensor("out_17_transpose_x_0"), val = tensor(false)]; tensor out_17_transpose_y_0 = const()[name = tensor("out_17_transpose_y_0"), val = tensor(false)]; tensor softmax_4_to_fp16_dtype_0 = const()[name = tensor("softmax_4_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_4_to_fp16 = cast(dtype = softmax_4_to_fp16_dtype_0, x = softmax_4)[name = tensor("cast_356")]; tensor out_17_cast_fp16 = matmul(transpose_x = out_17_transpose_x_0, transpose_y = out_17_transpose_y_0, x = softmax_4_to_fp16, y = v_9_cast_fp16)[name = tensor("out_17_cast_fp16")]; tensor concat_61x = const()[name = tensor("concat_61x"), val = tensor([1, -1])]; tensor var_1293 = reshape(shape = concat_61x, x = positions_9)[name = tensor("op_1293")]; tensor var_1294 = const()[name = tensor("op_1294"), val = tensor([-1, 1])]; tensor var_1295 = reshape(shape = var_1294, x = input_lengths_5)[name = tensor("op_1295")]; tensor var_1296 = less(x = var_1293, y = var_1295)[name = tensor("op_1296")]; tensor concat_62x = const()[name = tensor("concat_62x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_9 = reshape(shape = concat_62x, x = var_1296)[name = tensor("valid_q_9")]; tensor var_1300_to_fp16 = const()[name = tensor("op_1300_to_fp16"), val = tensor(0x0p+0)]; tensor out_19_cast_fp16 = select(a = out_17_cast_fp16, b = var_1300_to_fp16, cond = valid_q_9)[name = tensor("out_19_cast_fp16")]; tensor var_1302_perm_0 = const()[name = tensor("op_1302_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_63x = const()[name = tensor("concat_63x"), val = tensor([1, -1, 256])]; tensor var_1302_cast_fp16 = transpose(perm = var_1302_perm_0, x = out_19_cast_fp16)[name = tensor("transpose_23")]; tensor input_119_cast_fp16 = reshape(shape = concat_63x, x = var_1302_cast_fp16)[name = tensor("input_119_cast_fp16")]; tensor decoder_3_transformer_layers_0_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_0_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8594816)))]; tensor linear_20_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_3_transformer_layers_0_self_attn_out_proj_weight_to_fp16, x = input_119_cast_fp16)[name = tensor("linear_20_cast_fp16")]; tensor decoder_3_transformer_layers_0_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_0_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8725952)))]; tensor var_1308_cast_fp16 = mul(x = decoder_3_transformer_layers_0_layer_scale_1_scale_to_fp16, y = linear_20_cast_fp16)[name = tensor("op_1308_cast_fp16")]; tensor input_121_cast_fp16 = add(x = linear_18_cast_fp16, y = var_1308_cast_fp16)[name = tensor("input_121_cast_fp16")]; tensor input_123_axes_0 = const()[name = tensor("input_123_axes_0"), val = tensor([-1])]; tensor decoder_3_transformer_layers_0_norm2_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_0_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8726528)))]; tensor decoder_3_transformer_layers_0_norm2_bias_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_0_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8727104)))]; tensor input_123_cast_fp16 = layer_norm(axes = input_123_axes_0, beta = decoder_3_transformer_layers_0_norm2_bias_to_fp16, epsilon = var_1173_to_fp16, gamma = decoder_3_transformer_layers_0_norm2_weight_to_fp16, x = input_121_cast_fp16)[name = tensor("input_123_cast_fp16")]; tensor decoder_3_transformer_layers_0_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_0_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(8727680)))]; tensor linear_21_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_3_transformer_layers_0_ffn_0_weight_to_fp16, x = input_123_cast_fp16)[name = tensor("linear_21_cast_fp16")]; tensor input_127_mode_0 = const()[name = tensor("input_127_mode_0"), val = tensor("EXACT")]; tensor input_127_cast_fp16 = gelu(mode = input_127_mode_0, x = linear_21_cast_fp16)[name = tensor("input_127_cast_fp16")]; tensor decoder_3_transformer_layers_0_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_0_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9252032)))]; tensor linear_22_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_3_transformer_layers_0_ffn_2_weight_to_fp16, x = input_127_cast_fp16)[name = tensor("linear_22_cast_fp16")]; tensor decoder_3_transformer_layers_0_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_0_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9776384)))]; tensor var_1323_cast_fp16 = mul(x = decoder_3_transformer_layers_0_layer_scale_2_scale_to_fp16, y = linear_22_cast_fp16)[name = tensor("op_1323_cast_fp16")]; tensor input_129_cast_fp16 = add(x = input_121_cast_fp16, y = var_1323_cast_fp16)[name = tensor("input_129_cast_fp16")]; tensor x_43_axes_0 = const()[name = tensor("x_43_axes_0"), val = tensor([-1])]; tensor decoder_3_transformer_layers_1_norm1_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_1_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9776960)))]; tensor decoder_3_transformer_layers_1_norm1_bias_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_1_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9777536)))]; tensor x_43_cast_fp16 = layer_norm(axes = x_43_axes_0, beta = decoder_3_transformer_layers_1_norm1_bias_to_fp16, epsilon = var_1173_to_fp16, gamma = decoder_3_transformer_layers_1_norm1_weight_to_fp16, x = input_129_cast_fp16)[name = tensor("x_43_cast_fp16")]; tensor var_1338_shape_cast_fp16 = shape(x = x_43_cast_fp16)[name = tensor("op_1338_shape_cast_fp16")]; tensor gather_57_axis_0 = const()[name = tensor("gather_57_axis_0"), val = tensor(0)]; tensor gather_57_batch_dims_0 = const()[name = tensor("gather_57_batch_dims_0"), val = tensor(0)]; tensor gather_57_validate_indices_0 = const()[name = tensor("gather_57_validate_indices_0"), val = tensor(false)]; tensor var_1338_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1338_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_57_indices_0_to_uint16 = const()[name = tensor("gather_57_indices_0_to_uint16"), val = tensor(1)]; tensor var_1338_shape_cast_fp16_to_uint16 = cast(dtype = var_1338_shape_cast_fp16_to_uint16_dtype_0, x = var_1338_shape_cast_fp16)[name = tensor("cast_355")]; tensor gather_57_cast_uint16 = gather(axis = gather_57_axis_0, batch_dims = gather_57_batch_dims_0, indices = gather_57_indices_0_to_uint16, validate_indices = gather_57_validate_indices_0, x = var_1338_shape_cast_fp16_to_uint16)[name = tensor("gather_57_cast_uint16")]; tensor gather_57_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_57_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_3_transformer_layers_1_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_1_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(9778112)))]; tensor linear_23_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_3_transformer_layers_1_self_attn_in_proj_weight_to_fp16, x = x_43_cast_fp16)[name = tensor("linear_23_cast_fp16")]; tensor concat_64x = const()[name = tensor("concat_64x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_1345_cast_fp16 = reshape(shape = concat_64x, x = linear_23_cast_fp16)[name = tensor("op_1345_cast_fp16")]; tensor var_1346 = const()[name = tensor("op_1346"), val = tensor([2, 0, 3, 1, 4])]; tensor q_21_begin_0 = const()[name = tensor("q_21_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_21_end_0 = const()[name = tensor("q_21_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_21_end_mask_0 = const()[name = tensor("q_21_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_21_squeeze_mask_0 = const()[name = tensor("q_21_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_23_cast_fp16 = transpose(perm = var_1346, x = var_1345_cast_fp16)[name = tensor("transpose_22")]; tensor q_21_cast_fp16 = slice_by_index(begin = q_21_begin_0, end = q_21_end_0, end_mask = q_21_end_mask_0, squeeze_mask = q_21_squeeze_mask_0, x = projected_23_cast_fp16)[name = tensor("q_21_cast_fp16")]; tensor k_21_begin_0 = const()[name = tensor("k_21_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_21_end_0 = const()[name = tensor("k_21_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_21_end_mask_0 = const()[name = tensor("k_21_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_21_squeeze_mask_0 = const()[name = tensor("k_21_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_21_cast_fp16 = slice_by_index(begin = k_21_begin_0, end = k_21_end_0, end_mask = k_21_end_mask_0, squeeze_mask = k_21_squeeze_mask_0, x = projected_23_cast_fp16)[name = tensor("k_21_cast_fp16")]; tensor v_11_begin_0 = const()[name = tensor("v_11_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_11_end_0 = const()[name = tensor("v_11_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_11_end_mask_0 = const()[name = tensor("v_11_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_11_squeeze_mask_0 = const()[name = tensor("v_11_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_11_cast_fp16 = 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 = projected_23_cast_fp16)[name = tensor("v_11_cast_fp16")]; tensor var_1358_shape_cast_fp16 = shape(x = q_21_cast_fp16)[name = tensor("op_1358_shape_cast_fp16")]; tensor gather_61_axis_0 = const()[name = tensor("gather_61_axis_0"), val = tensor(0)]; tensor gather_61_batch_dims_0 = const()[name = tensor("gather_61_batch_dims_0"), val = tensor(0)]; tensor gather_61_validate_indices_0 = const()[name = tensor("gather_61_validate_indices_0"), val = tensor(false)]; tensor var_1358_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1358_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_61_indices_0_to_uint16 = const()[name = tensor("gather_61_indices_0_to_uint16"), val = tensor(2)]; tensor var_1358_shape_cast_fp16_to_uint16 = cast(dtype = var_1358_shape_cast_fp16_to_uint16_dtype_0, x = var_1358_shape_cast_fp16)[name = tensor("cast_353")]; tensor gather_61_cast_uint16 = gather(axis = gather_61_axis_0, batch_dims = gather_61_batch_dims_0, indices = gather_61_indices_0_to_uint16, validate_indices = gather_61_validate_indices_0, x = var_1358_shape_cast_fp16_to_uint16)[name = tensor("gather_61_cast_uint16")]; tensor gather_61_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_61_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_48 = const()[name = tensor("const_48"), val = tensor(0)]; tensor const_49 = const()[name = tensor("const_49"), val = tensor(1)]; tensor gather_61_cast_uint16_to_int32 = cast(dtype = gather_61_cast_uint16_to_int32_dtype_0, x = gather_61_cast_uint16)[name = tensor("cast_352")]; tensor var_1362 = range_1d(end = gather_61_cast_uint16_to_int32, start = const_48, step = const_49)[name = tensor("op_1362")]; tensor var_1361_to_fp16 = const()[name = tensor("op_1361_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_1362_promoted_to_fp16_dtype_0 = const()[name = tensor("op_1362_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1362_to_fp16 = cast(dtype = var_1362_promoted_to_fp16_dtype_0, x = var_1362)[name = tensor("cast_351")]; tensor ts_21_cast_fp16 = add(x = var_1361_to_fp16, y = var_1362_to_fp16)[name = tensor("ts_21_cast_fp16")]; tensor var_1365 = const()[name = tensor("op_1365"), val = tensor([1, 1, -1, 1])]; tensor ts_23_cast_fp16 = reshape(shape = var_1365, x = ts_21_cast_fp16)[name = tensor("ts_23_cast_fp16")]; tensor concat_65x = const()[name = tensor("concat_65x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_11_cast_fp16 = reshape(shape = concat_65x, x = q_21_cast_fp16)[name = tensor("q2_11_cast_fp16")]; tensor concat_66x = const()[name = tensor("concat_66x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_11_cast_fp16 = reshape(shape = concat_66x, x = k_21_cast_fp16)[name = tensor("k2_11_cast_fp16")]; tensor qr_11_begin_0 = const()[name = tensor("qr_11_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_11_end_0 = const()[name = tensor("qr_11_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_11_end_mask_0 = const()[name = tensor("qr_11_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_11_squeeze_mask_0 = const()[name = tensor("qr_11_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_11_cast_fp16 = slice_by_index(begin = qr_11_begin_0, end = qr_11_end_0, end_mask = qr_11_end_mask_0, squeeze_mask = qr_11_squeeze_mask_0, x = q2_11_cast_fp16)[name = tensor("qr_11_cast_fp16")]; tensor qi_11_begin_0 = const()[name = tensor("qi_11_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_11_end_0 = const()[name = tensor("qi_11_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_11_end_mask_0 = const()[name = tensor("qi_11_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_11_squeeze_mask_0 = const()[name = tensor("qi_11_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_11_cast_fp16 = slice_by_index(begin = qi_11_begin_0, end = qi_11_end_0, end_mask = qi_11_end_mask_0, squeeze_mask = qi_11_squeeze_mask_0, x = q2_11_cast_fp16)[name = tensor("qi_11_cast_fp16")]; tensor kr_11_begin_0 = const()[name = tensor("kr_11_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_11_end_0 = const()[name = tensor("kr_11_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_11_end_mask_0 = const()[name = tensor("kr_11_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_11_squeeze_mask_0 = const()[name = tensor("kr_11_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_11_cast_fp16 = slice_by_index(begin = kr_11_begin_0, end = kr_11_end_0, end_mask = kr_11_end_mask_0, squeeze_mask = kr_11_squeeze_mask_0, x = k2_11_cast_fp16)[name = tensor("kr_11_cast_fp16")]; tensor ki_11_begin_0 = const()[name = tensor("ki_11_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_11_end_0 = const()[name = tensor("ki_11_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_11_end_mask_0 = const()[name = tensor("ki_11_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_11_squeeze_mask_0 = const()[name = tensor("ki_11_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_11_cast_fp16 = slice_by_index(begin = ki_11_begin_0, end = ki_11_end_0, end_mask = ki_11_end_mask_0, squeeze_mask = ki_11_squeeze_mask_0, x = k2_11_cast_fp16)[name = tensor("ki_11_cast_fp16")]; tensor freqs_11_to_fp16 = const()[name = tensor("freqs_11_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10171392)))]; tensor var_1378_cast_fp16 = mul(x = freqs_11_to_fp16, y = ts_23_cast_fp16)[name = tensor("op_1378_cast_fp16")]; tensor rotr_11_cast_fp16 = cos(x = var_1378_cast_fp16)[name = tensor("rotr_11_cast_fp16")]; tensor roti_11_cast_fp16 = sin(x = var_1378_cast_fp16)[name = tensor("roti_11_cast_fp16")]; tensor var_1382_cast_fp16 = mul(x = qr_11_cast_fp16, y = rotr_11_cast_fp16)[name = tensor("op_1382_cast_fp16")]; tensor var_1383_cast_fp16 = mul(x = qi_11_cast_fp16, y = roti_11_cast_fp16)[name = tensor("op_1383_cast_fp16")]; tensor var_1384_cast_fp16 = sub(x = var_1382_cast_fp16, y = var_1383_cast_fp16)[name = tensor("op_1384_cast_fp16")]; tensor var_1385_cast_fp16 = mul(x = qr_11_cast_fp16, y = roti_11_cast_fp16)[name = tensor("op_1385_cast_fp16")]; tensor var_1386_cast_fp16 = mul(x = qi_11_cast_fp16, y = rotr_11_cast_fp16)[name = tensor("op_1386_cast_fp16")]; tensor var_1387_cast_fp16 = add(x = var_1385_cast_fp16, y = var_1386_cast_fp16)[name = tensor("op_1387_cast_fp16")]; tensor qo_11_axis_0 = const()[name = tensor("qo_11_axis_0"), val = tensor(-1)]; tensor qo_11_cast_fp16 = stack(axis = qo_11_axis_0, values = (var_1384_cast_fp16, var_1387_cast_fp16))[name = tensor("qo_11_cast_fp16")]; tensor var_1390_cast_fp16 = mul(x = kr_11_cast_fp16, y = rotr_11_cast_fp16)[name = tensor("op_1390_cast_fp16")]; tensor var_1391_cast_fp16 = mul(x = ki_11_cast_fp16, y = roti_11_cast_fp16)[name = tensor("op_1391_cast_fp16")]; tensor var_1392_cast_fp16 = sub(x = var_1390_cast_fp16, y = var_1391_cast_fp16)[name = tensor("op_1392_cast_fp16")]; tensor var_1393_cast_fp16 = mul(x = kr_11_cast_fp16, y = roti_11_cast_fp16)[name = tensor("op_1393_cast_fp16")]; tensor var_1394_cast_fp16 = mul(x = ki_11_cast_fp16, y = rotr_11_cast_fp16)[name = tensor("op_1394_cast_fp16")]; tensor var_1395_cast_fp16 = add(x = var_1393_cast_fp16, y = var_1394_cast_fp16)[name = tensor("op_1395_cast_fp16")]; tensor ko_11_axis_0 = const()[name = tensor("ko_11_axis_0"), val = tensor(-1)]; tensor ko_11_cast_fp16 = stack(axis = ko_11_axis_0, values = (var_1392_cast_fp16, var_1395_cast_fp16))[name = tensor("ko_11_cast_fp16")]; tensor concat_67x = const()[name = tensor("concat_67x"), val = tensor([1, 4, -1, 64])]; tensor q_23_cast_fp16 = reshape(shape = concat_67x, x = qo_11_cast_fp16)[name = tensor("q_23_cast_fp16")]; tensor concat_68x = const()[name = tensor("concat_68x"), val = tensor([1, 4, -1, 64])]; tensor k_23_cast_fp16 = reshape(shape = concat_68x, x = ko_11_cast_fp16)[name = tensor("k_23_cast_fp16")]; tensor const_50 = const()[name = tensor("const_50"), val = tensor(0)]; tensor const_51 = const()[name = tensor("const_51"), val = tensor(1)]; tensor gather_57_cast_uint16_to_int32 = cast(dtype = gather_57_cast_uint16_to_int32_dtype_0, x = gather_57_cast_uint16)[name = tensor("cast_354")]; tensor positions_11 = range_1d(end = gather_57_cast_uint16_to_int32, start = const_50, step = const_51)[name = tensor("positions_11")]; tensor concat_69x = const()[name = tensor("concat_69x"), val = tensor([1, 1, -1])]; tensor var_1407 = reshape(shape = concat_69x, x = positions_11)[name = tensor("op_1407")]; tensor valid_k_11 = less(x = var_1407, y = var_1272)[name = tensor("valid_k_11")]; tensor concat_70x = const()[name = tensor("concat_70x"), val = tensor([1, -1, 1])]; tensor var_1412 = reshape(shape = concat_70x, x = positions_11)[name = tensor("op_1412")]; tensor delta_11 = sub(x = var_1412, y = var_1407)[name = tensor("delta_11")]; tensor concat_72_axis_0 = const()[name = tensor("concat_72_axis_0"), val = tensor(0)]; tensor concat_72_interleave_0 = const()[name = tensor("concat_72_interleave_0"), val = tensor(false)]; tensor concat_72 = concat(axis = concat_72_axis_0, interleave = concat_72_interleave_0, values = (var_1177, gather_57_cast_uint16_to_int32, gather_57_cast_uint16_to_int32))[name = tensor("concat_72")]; tensor fill_5_value_0_to_fp16 = const()[name = tensor("fill_5_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_5_cast_fp16 = fill(shape = concat_72, value = fill_5_value_0_to_fp16)[name = tensor("fill_5_cast_fp16")]; tensor attn_bias_41_dtype_0 = const()[name = tensor("attn_bias_41_dtype_0"), val = tensor("bool")]; tensor var_1418 = greater_equal(x = delta_11, y = var_1159)[name = tensor("op_1418")]; tensor fill_5_cast_fp16_to_bool = cast(dtype = attn_bias_41_dtype_0, x = fill_5_cast_fp16)[name = tensor("cast_350")]; tensor attn_bias_43 = logical_and(x = fill_5_cast_fp16_to_bool, y = var_1418)[name = tensor("attn_bias_43")]; tensor var_1420 = less(x = delta_11, y = var_1166)[name = tensor("op_1420")]; tensor attn_bias_45 = logical_and(x = attn_bias_43, y = var_1420)[name = tensor("attn_bias_45")]; tensor var_1422 = logical_and(x = attn_bias_45, y = valid_k_11)[name = tensor("op_1422")]; tensor var_1424_axes_0 = const()[name = tensor("op_1424_axes_0"), val = tensor([1])]; tensor var_1424 = expand_dims(axes = var_1424_axes_0, x = var_1422)[name = tensor("op_1424")]; tensor sub_5_x_0_to_fp16 = const()[name = tensor("sub_5_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_147_to_fp16_dtype_0 = const()[name = tensor("cast_147_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1424_to_fp16 = cast(dtype = cast_147_to_fp16_dtype_0, x = var_1424)[name = tensor("cast_349")]; tensor sub_5_cast_fp16 = sub(x = sub_5_x_0_to_fp16, y = var_1424_to_fp16)[name = tensor("sub_5_cast_fp16")]; tensor mul_10_x_0_to_fp16 = const()[name = tensor("mul_10_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_10_cast_fp16 = mul(x = mul_10_x_0_to_fp16, y = sub_5_cast_fp16)[name = tensor("mul_10_cast_fp16")]; tensor mul_11_y_0_to_fp16 = const()[name = tensor("mul_11_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_11_cast_fp16 = mul(x = q_23_cast_fp16, y = mul_11_y_0_to_fp16)[name = tensor("mul_11_cast_fp16")]; tensor matmul_5_transpose_y_0 = const()[name = tensor("matmul_5_transpose_y_0"), val = tensor(true)]; tensor matmul_5_transpose_x_0 = const()[name = tensor("matmul_5_transpose_x_0"), val = tensor(false)]; tensor matmul_5_cast_fp16 = matmul(transpose_x = matmul_5_transpose_x_0, transpose_y = matmul_5_transpose_y_0, x = mul_11_cast_fp16, y = k_23_cast_fp16)[name = tensor("matmul_5_cast_fp16")]; tensor add_5_cast_fp16 = add(x = matmul_5_cast_fp16, y = mul_10_cast_fp16)[name = tensor("add_5_cast_fp16")]; tensor add_5_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_5_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_5_axis_0 = const()[name = tensor("softmax_5_axis_0"), val = tensor(-1)]; tensor add_5_cast_fp16_to_fp32 = cast(dtype = add_5_cast_fp16_to_fp32_dtype_0, x = add_5_cast_fp16)[name = tensor("cast_348")]; tensor softmax_5 = softmax(axis = softmax_5_axis_0, x = add_5_cast_fp16_to_fp32)[name = tensor("softmax_5")]; tensor out_21_transpose_x_0 = const()[name = tensor("out_21_transpose_x_0"), val = tensor(false)]; tensor out_21_transpose_y_0 = const()[name = tensor("out_21_transpose_y_0"), val = tensor(false)]; tensor softmax_5_to_fp16_dtype_0 = const()[name = tensor("softmax_5_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_5_to_fp16 = cast(dtype = softmax_5_to_fp16_dtype_0, x = softmax_5)[name = tensor("cast_347")]; tensor out_21_cast_fp16 = matmul(transpose_x = out_21_transpose_x_0, transpose_y = out_21_transpose_y_0, x = softmax_5_to_fp16, y = v_11_cast_fp16)[name = tensor("out_21_cast_fp16")]; tensor concat_73x = const()[name = tensor("concat_73x"), val = tensor([1, -1])]; tensor var_1430 = reshape(shape = concat_73x, x = positions_11)[name = tensor("op_1430")]; tensor var_1433 = less(x = var_1430, y = var_1295)[name = tensor("op_1433")]; tensor concat_74x = const()[name = tensor("concat_74x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_11 = reshape(shape = concat_74x, x = var_1433)[name = tensor("valid_q_11")]; tensor var_1437_to_fp16 = const()[name = tensor("op_1437_to_fp16"), val = tensor(0x0p+0)]; tensor out_23_cast_fp16 = select(a = out_21_cast_fp16, b = var_1437_to_fp16, cond = valid_q_11)[name = tensor("out_23_cast_fp16")]; tensor var_1439_perm_0 = const()[name = tensor("op_1439_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_75x = const()[name = tensor("concat_75x"), val = tensor([1, -1, 256])]; tensor var_1439_cast_fp16 = transpose(perm = var_1439_perm_0, x = out_23_cast_fp16)[name = tensor("transpose_21")]; tensor input_131_cast_fp16 = reshape(shape = concat_75x, x = var_1439_cast_fp16)[name = tensor("input_131_cast_fp16")]; tensor decoder_3_transformer_layers_1_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_1_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10171520)))]; tensor linear_24_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_3_transformer_layers_1_self_attn_out_proj_weight_to_fp16, x = input_131_cast_fp16)[name = tensor("linear_24_cast_fp16")]; tensor decoder_3_transformer_layers_1_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_1_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10302656)))]; tensor var_1445_cast_fp16 = mul(x = decoder_3_transformer_layers_1_layer_scale_1_scale_to_fp16, y = linear_24_cast_fp16)[name = tensor("op_1445_cast_fp16")]; tensor input_133_cast_fp16 = add(x = input_129_cast_fp16, y = var_1445_cast_fp16)[name = tensor("input_133_cast_fp16")]; tensor input_135_axes_0 = const()[name = tensor("input_135_axes_0"), val = tensor([-1])]; tensor decoder_3_transformer_layers_1_norm2_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_1_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10303232)))]; tensor decoder_3_transformer_layers_1_norm2_bias_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_1_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10303808)))]; tensor input_135_cast_fp16 = layer_norm(axes = input_135_axes_0, beta = decoder_3_transformer_layers_1_norm2_bias_to_fp16, epsilon = var_1173_to_fp16, gamma = decoder_3_transformer_layers_1_norm2_weight_to_fp16, x = input_133_cast_fp16)[name = tensor("input_135_cast_fp16")]; tensor decoder_3_transformer_layers_1_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_1_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10304384)))]; tensor linear_25_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_3_transformer_layers_1_ffn_0_weight_to_fp16, x = input_135_cast_fp16)[name = tensor("linear_25_cast_fp16")]; tensor input_139_mode_0 = const()[name = tensor("input_139_mode_0"), val = tensor("EXACT")]; tensor input_139_cast_fp16 = gelu(mode = input_139_mode_0, x = linear_25_cast_fp16)[name = tensor("input_139_cast_fp16")]; tensor decoder_3_transformer_layers_1_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_1_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(10828736)))]; tensor linear_26_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_3_transformer_layers_1_ffn_2_weight_to_fp16, x = input_139_cast_fp16)[name = tensor("linear_26_cast_fp16")]; tensor decoder_3_transformer_layers_1_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_3_transformer_layers_1_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11353088)))]; tensor var_1460_cast_fp16 = mul(x = decoder_3_transformer_layers_1_layer_scale_2_scale_to_fp16, y = linear_26_cast_fp16)[name = tensor("op_1460_cast_fp16")]; tensor input_141_cast_fp16 = add(x = input_133_cast_fp16, y = var_1460_cast_fp16)[name = tensor("input_141_cast_fp16")]; tensor decoder_3_output_proj_weight_to_fp16 = const()[name = tensor("decoder_3_output_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11353664)))]; tensor linear_27_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_3_output_proj_weight_to_fp16, x = input_141_cast_fp16)[name = tensor("linear_27_cast_fp16")]; tensor x_49_perm_0 = const()[name = tensor("x_49_perm_0"), val = tensor([0, 2, 1])]; tensor concat_76x = const()[name = tensor("concat_76x"), val = tensor([1, 384, 2, -1])]; tensor x_49_cast_fp16 = transpose(perm = x_49_perm_0, x = linear_27_cast_fp16)[name = tensor("transpose_20")]; tensor var_1479_cast_fp16 = reshape(shape = concat_76x, x = x_49_cast_fp16)[name = tensor("op_1479_cast_fp16")]; tensor var_1480 = const()[name = tensor("op_1480"), val = tensor([0, 1, 3, 2])]; tensor concat_77x = const()[name = tensor("concat_77x"), val = tensor([1, 384, -1])]; tensor var_1481_cast_fp16 = transpose(perm = var_1480, x = var_1479_cast_fp16)[name = tensor("transpose_19")]; tensor x_51_cast_fp16 = reshape(shape = concat_77x, x = var_1481_cast_fp16)[name = tensor("x_51_cast_fp16")]; tensor var_1487 = const()[name = tensor("op_1487"), val = tensor(2)]; tensor input_lengths_7 = mul(x = input_lengths_5, y = var_1487)[name = tensor("input_lengths_7")]; tensor var_1493 = const()[name = tensor("op_1493"), val = tensor(0)]; tensor var_1500 = const()[name = tensor("op_1500"), val = tensor(1200)]; tensor var_1511 = const()[name = tensor("op_1511"), val = tensor(1)]; tensor input_143_perm_0 = const()[name = tensor("input_143_perm_0"), val = tensor([0, 2, 1])]; tensor decoder_5_input_proj_weight_to_fp16 = const()[name = tensor("decoder_5_input_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11746944)))]; tensor input_143_cast_fp16 = transpose(perm = input_143_perm_0, x = x_51_cast_fp16)[name = tensor("transpose_18")]; tensor linear_28_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_5_input_proj_weight_to_fp16, x = input_143_cast_fp16)[name = tensor("linear_28_cast_fp16")]; tensor x_55_axes_0 = const()[name = tensor("x_55_axes_0"), val = tensor([-1])]; tensor decoder_5_transformer_layers_0_norm1_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_0_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11943616)))]; tensor decoder_5_transformer_layers_0_norm1_bias_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_0_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11944192)))]; tensor var_1507_to_fp16 = const()[name = tensor("op_1507_to_fp16"), val = tensor(0x1.5p-17)]; tensor x_55_cast_fp16 = layer_norm(axes = x_55_axes_0, beta = decoder_5_transformer_layers_0_norm1_bias_to_fp16, epsilon = var_1507_to_fp16, gamma = decoder_5_transformer_layers_0_norm1_weight_to_fp16, x = linear_28_cast_fp16)[name = tensor("x_55_cast_fp16")]; tensor var_1535_shape_cast_fp16 = shape(x = x_55_cast_fp16)[name = tensor("op_1535_shape_cast_fp16")]; tensor gather_70_axis_0 = const()[name = tensor("gather_70_axis_0"), val = tensor(0)]; tensor gather_70_batch_dims_0 = const()[name = tensor("gather_70_batch_dims_0"), val = tensor(0)]; tensor gather_70_validate_indices_0 = const()[name = tensor("gather_70_validate_indices_0"), val = tensor(false)]; tensor var_1535_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1535_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_70_indices_0_to_uint16 = const()[name = tensor("gather_70_indices_0_to_uint16"), val = tensor(1)]; tensor var_1535_shape_cast_fp16_to_uint16 = cast(dtype = var_1535_shape_cast_fp16_to_uint16_dtype_0, x = var_1535_shape_cast_fp16)[name = tensor("cast_346")]; tensor gather_70_cast_uint16 = gather(axis = gather_70_axis_0, batch_dims = gather_70_batch_dims_0, indices = gather_70_indices_0_to_uint16, validate_indices = gather_70_validate_indices_0, x = var_1535_shape_cast_fp16_to_uint16)[name = tensor("gather_70_cast_uint16")]; tensor gather_70_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_70_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_5_transformer_layers_0_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_0_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(11944768)))]; tensor linear_29_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_5_transformer_layers_0_self_attn_in_proj_weight_to_fp16, x = x_55_cast_fp16)[name = tensor("linear_29_cast_fp16")]; tensor concat_78x = const()[name = tensor("concat_78x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_1542_cast_fp16 = reshape(shape = concat_78x, x = linear_29_cast_fp16)[name = tensor("op_1542_cast_fp16")]; tensor var_1543 = const()[name = tensor("op_1543"), val = tensor([2, 0, 3, 1, 4])]; tensor q_25_begin_0 = const()[name = tensor("q_25_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_25_end_0 = const()[name = tensor("q_25_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_25_end_mask_0 = const()[name = tensor("q_25_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_25_squeeze_mask_0 = const()[name = tensor("q_25_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_27_cast_fp16 = transpose(perm = var_1543, x = var_1542_cast_fp16)[name = tensor("transpose_17")]; tensor q_25_cast_fp16 = slice_by_index(begin = q_25_begin_0, end = q_25_end_0, end_mask = q_25_end_mask_0, squeeze_mask = q_25_squeeze_mask_0, x = projected_27_cast_fp16)[name = tensor("q_25_cast_fp16")]; tensor k_25_begin_0 = const()[name = tensor("k_25_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_25_end_0 = const()[name = tensor("k_25_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_25_end_mask_0 = const()[name = tensor("k_25_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_25_squeeze_mask_0 = const()[name = tensor("k_25_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_25_cast_fp16 = slice_by_index(begin = k_25_begin_0, end = k_25_end_0, end_mask = k_25_end_mask_0, squeeze_mask = k_25_squeeze_mask_0, x = projected_27_cast_fp16)[name = tensor("k_25_cast_fp16")]; tensor v_13_begin_0 = const()[name = tensor("v_13_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_13_end_0 = const()[name = tensor("v_13_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_13_end_mask_0 = const()[name = tensor("v_13_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_13_squeeze_mask_0 = const()[name = tensor("v_13_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_13_cast_fp16 = slice_by_index(begin = v_13_begin_0, end = v_13_end_0, end_mask = v_13_end_mask_0, squeeze_mask = v_13_squeeze_mask_0, x = projected_27_cast_fp16)[name = tensor("v_13_cast_fp16")]; tensor var_1555_shape_cast_fp16 = shape(x = q_25_cast_fp16)[name = tensor("op_1555_shape_cast_fp16")]; tensor gather_74_axis_0 = const()[name = tensor("gather_74_axis_0"), val = tensor(0)]; tensor gather_74_batch_dims_0 = const()[name = tensor("gather_74_batch_dims_0"), val = tensor(0)]; tensor gather_74_validate_indices_0 = const()[name = tensor("gather_74_validate_indices_0"), val = tensor(false)]; tensor var_1555_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1555_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_74_indices_0_to_uint16 = const()[name = tensor("gather_74_indices_0_to_uint16"), val = tensor(2)]; tensor var_1555_shape_cast_fp16_to_uint16 = cast(dtype = var_1555_shape_cast_fp16_to_uint16_dtype_0, x = var_1555_shape_cast_fp16)[name = tensor("cast_344")]; tensor gather_74_cast_uint16 = gather(axis = gather_74_axis_0, batch_dims = gather_74_batch_dims_0, indices = gather_74_indices_0_to_uint16, validate_indices = gather_74_validate_indices_0, x = var_1555_shape_cast_fp16_to_uint16)[name = tensor("gather_74_cast_uint16")]; tensor gather_74_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_74_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_57 = const()[name = tensor("const_57"), val = tensor(0)]; tensor const_58 = const()[name = tensor("const_58"), val = tensor(1)]; tensor gather_74_cast_uint16_to_int32 = cast(dtype = gather_74_cast_uint16_to_int32_dtype_0, x = gather_74_cast_uint16)[name = tensor("cast_343")]; tensor var_1559 = range_1d(end = gather_74_cast_uint16_to_int32, start = const_57, step = const_58)[name = tensor("op_1559")]; tensor var_1558_to_fp16 = const()[name = tensor("op_1558_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_1559_promoted_to_fp16_dtype_0 = const()[name = tensor("op_1559_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1559_to_fp16 = cast(dtype = var_1559_promoted_to_fp16_dtype_0, x = var_1559)[name = tensor("cast_342")]; tensor ts_25_cast_fp16 = add(x = var_1558_to_fp16, y = var_1559_to_fp16)[name = tensor("ts_25_cast_fp16")]; tensor var_1562 = const()[name = tensor("op_1562"), val = tensor([1, 1, -1, 1])]; tensor ts_27_cast_fp16 = reshape(shape = var_1562, x = ts_25_cast_fp16)[name = tensor("ts_27_cast_fp16")]; tensor concat_79x = const()[name = tensor("concat_79x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_13_cast_fp16 = reshape(shape = concat_79x, x = q_25_cast_fp16)[name = tensor("q2_13_cast_fp16")]; tensor concat_80x = const()[name = tensor("concat_80x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_13_cast_fp16 = reshape(shape = concat_80x, x = k_25_cast_fp16)[name = tensor("k2_13_cast_fp16")]; tensor qr_13_begin_0 = const()[name = tensor("qr_13_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_13_end_0 = const()[name = tensor("qr_13_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_13_end_mask_0 = const()[name = tensor("qr_13_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_13_squeeze_mask_0 = const()[name = tensor("qr_13_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_13_cast_fp16 = slice_by_index(begin = qr_13_begin_0, end = qr_13_end_0, end_mask = qr_13_end_mask_0, squeeze_mask = qr_13_squeeze_mask_0, x = q2_13_cast_fp16)[name = tensor("qr_13_cast_fp16")]; tensor qi_13_begin_0 = const()[name = tensor("qi_13_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_13_end_0 = const()[name = tensor("qi_13_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_13_end_mask_0 = const()[name = tensor("qi_13_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_13_squeeze_mask_0 = const()[name = tensor("qi_13_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_13_cast_fp16 = slice_by_index(begin = qi_13_begin_0, end = qi_13_end_0, end_mask = qi_13_end_mask_0, squeeze_mask = qi_13_squeeze_mask_0, x = q2_13_cast_fp16)[name = tensor("qi_13_cast_fp16")]; tensor kr_13_begin_0 = const()[name = tensor("kr_13_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_13_end_0 = const()[name = tensor("kr_13_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_13_end_mask_0 = const()[name = tensor("kr_13_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_13_squeeze_mask_0 = const()[name = tensor("kr_13_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_13_cast_fp16 = slice_by_index(begin = kr_13_begin_0, end = kr_13_end_0, end_mask = kr_13_end_mask_0, squeeze_mask = kr_13_squeeze_mask_0, x = k2_13_cast_fp16)[name = tensor("kr_13_cast_fp16")]; tensor ki_13_begin_0 = const()[name = tensor("ki_13_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_13_end_0 = const()[name = tensor("ki_13_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_13_end_mask_0 = const()[name = tensor("ki_13_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_13_squeeze_mask_0 = const()[name = tensor("ki_13_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_13_cast_fp16 = slice_by_index(begin = ki_13_begin_0, end = ki_13_end_0, end_mask = ki_13_end_mask_0, squeeze_mask = ki_13_squeeze_mask_0, x = k2_13_cast_fp16)[name = tensor("ki_13_cast_fp16")]; tensor freqs_13_to_fp16 = const()[name = tensor("freqs_13_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(12338048)))]; tensor var_1575_cast_fp16 = mul(x = freqs_13_to_fp16, y = ts_27_cast_fp16)[name = tensor("op_1575_cast_fp16")]; tensor rotr_13_cast_fp16 = cos(x = var_1575_cast_fp16)[name = tensor("rotr_13_cast_fp16")]; tensor roti_13_cast_fp16 = sin(x = var_1575_cast_fp16)[name = tensor("roti_13_cast_fp16")]; tensor var_1579_cast_fp16 = mul(x = qr_13_cast_fp16, y = rotr_13_cast_fp16)[name = tensor("op_1579_cast_fp16")]; tensor var_1580_cast_fp16 = mul(x = qi_13_cast_fp16, y = roti_13_cast_fp16)[name = tensor("op_1580_cast_fp16")]; tensor var_1581_cast_fp16 = sub(x = var_1579_cast_fp16, y = var_1580_cast_fp16)[name = tensor("op_1581_cast_fp16")]; tensor var_1582_cast_fp16 = mul(x = qr_13_cast_fp16, y = roti_13_cast_fp16)[name = tensor("op_1582_cast_fp16")]; tensor var_1583_cast_fp16 = mul(x = qi_13_cast_fp16, y = rotr_13_cast_fp16)[name = tensor("op_1583_cast_fp16")]; tensor var_1584_cast_fp16 = add(x = var_1582_cast_fp16, y = var_1583_cast_fp16)[name = tensor("op_1584_cast_fp16")]; tensor qo_13_axis_0 = const()[name = tensor("qo_13_axis_0"), val = tensor(-1)]; tensor qo_13_cast_fp16 = stack(axis = qo_13_axis_0, values = (var_1581_cast_fp16, var_1584_cast_fp16))[name = tensor("qo_13_cast_fp16")]; tensor var_1587_cast_fp16 = mul(x = kr_13_cast_fp16, y = rotr_13_cast_fp16)[name = tensor("op_1587_cast_fp16")]; tensor var_1588_cast_fp16 = mul(x = ki_13_cast_fp16, y = roti_13_cast_fp16)[name = tensor("op_1588_cast_fp16")]; tensor var_1589_cast_fp16 = sub(x = var_1587_cast_fp16, y = var_1588_cast_fp16)[name = tensor("op_1589_cast_fp16")]; tensor var_1590_cast_fp16 = mul(x = kr_13_cast_fp16, y = roti_13_cast_fp16)[name = tensor("op_1590_cast_fp16")]; tensor var_1591_cast_fp16 = mul(x = ki_13_cast_fp16, y = rotr_13_cast_fp16)[name = tensor("op_1591_cast_fp16")]; tensor var_1592_cast_fp16 = add(x = var_1590_cast_fp16, y = var_1591_cast_fp16)[name = tensor("op_1592_cast_fp16")]; tensor ko_13_axis_0 = const()[name = tensor("ko_13_axis_0"), val = tensor(-1)]; tensor ko_13_cast_fp16 = stack(axis = ko_13_axis_0, values = (var_1589_cast_fp16, var_1592_cast_fp16))[name = tensor("ko_13_cast_fp16")]; tensor concat_81x = const()[name = tensor("concat_81x"), val = tensor([1, 4, -1, 64])]; tensor q_27_cast_fp16 = reshape(shape = concat_81x, x = qo_13_cast_fp16)[name = tensor("q_27_cast_fp16")]; tensor concat_82x = const()[name = tensor("concat_82x"), val = tensor([1, 4, -1, 64])]; tensor k_27_cast_fp16 = reshape(shape = concat_82x, x = ko_13_cast_fp16)[name = tensor("k_27_cast_fp16")]; tensor const_59 = const()[name = tensor("const_59"), val = tensor(0)]; tensor const_60 = const()[name = tensor("const_60"), val = tensor(1)]; tensor gather_70_cast_uint16_to_int32 = cast(dtype = gather_70_cast_uint16_to_int32_dtype_0, x = gather_70_cast_uint16)[name = tensor("cast_345")]; tensor positions_13 = range_1d(end = gather_70_cast_uint16_to_int32, start = const_59, step = const_60)[name = tensor("positions_13")]; tensor concat_83x = const()[name = tensor("concat_83x"), val = tensor([1, 1, -1])]; tensor var_1604 = reshape(shape = concat_83x, x = positions_13)[name = tensor("op_1604")]; tensor var_1605 = const()[name = tensor("op_1605"), val = tensor([-1, 1, 1])]; tensor var_1606 = reshape(shape = var_1605, x = input_lengths_7)[name = tensor("op_1606")]; tensor valid_k_13 = less(x = var_1604, y = var_1606)[name = tensor("valid_k_13")]; tensor concat_84x = const()[name = tensor("concat_84x"), val = tensor([1, -1, 1])]; tensor var_1609 = reshape(shape = concat_84x, x = positions_13)[name = tensor("op_1609")]; tensor delta_13 = sub(x = var_1609, y = var_1604)[name = tensor("delta_13")]; tensor concat_86_axis_0 = const()[name = tensor("concat_86_axis_0"), val = tensor(0)]; tensor concat_86_interleave_0 = const()[name = tensor("concat_86_interleave_0"), val = tensor(false)]; tensor concat_86 = concat(axis = concat_86_axis_0, interleave = concat_86_interleave_0, values = (var_1511, gather_70_cast_uint16_to_int32, gather_70_cast_uint16_to_int32))[name = tensor("concat_86")]; tensor fill_6_value_0_to_fp16 = const()[name = tensor("fill_6_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_6_cast_fp16 = fill(shape = concat_86, value = fill_6_value_0_to_fp16)[name = tensor("fill_6_cast_fp16")]; tensor attn_bias_49_dtype_0 = const()[name = tensor("attn_bias_49_dtype_0"), val = tensor("bool")]; tensor var_1615 = greater_equal(x = delta_13, y = var_1493)[name = tensor("op_1615")]; tensor fill_6_cast_fp16_to_bool = cast(dtype = attn_bias_49_dtype_0, x = fill_6_cast_fp16)[name = tensor("cast_341")]; tensor attn_bias_51 = logical_and(x = fill_6_cast_fp16_to_bool, y = var_1615)[name = tensor("attn_bias_51")]; tensor var_1617 = less(x = delta_13, y = var_1500)[name = tensor("op_1617")]; tensor attn_bias_53 = logical_and(x = attn_bias_51, y = var_1617)[name = tensor("attn_bias_53")]; tensor var_1619 = logical_and(x = attn_bias_53, y = valid_k_13)[name = tensor("op_1619")]; tensor var_1621_axes_0 = const()[name = tensor("op_1621_axes_0"), val = tensor([1])]; tensor var_1621 = expand_dims(axes = var_1621_axes_0, x = var_1619)[name = tensor("op_1621")]; tensor sub_6_x_0_to_fp16 = const()[name = tensor("sub_6_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_171_to_fp16_dtype_0 = const()[name = tensor("cast_171_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1621_to_fp16 = cast(dtype = cast_171_to_fp16_dtype_0, x = var_1621)[name = tensor("cast_340")]; tensor sub_6_cast_fp16 = sub(x = sub_6_x_0_to_fp16, y = var_1621_to_fp16)[name = tensor("sub_6_cast_fp16")]; tensor mul_12_x_0_to_fp16 = const()[name = tensor("mul_12_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_12_cast_fp16 = mul(x = mul_12_x_0_to_fp16, y = sub_6_cast_fp16)[name = tensor("mul_12_cast_fp16")]; tensor mul_13_y_0_to_fp16 = const()[name = tensor("mul_13_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_13_cast_fp16 = mul(x = q_27_cast_fp16, y = mul_13_y_0_to_fp16)[name = tensor("mul_13_cast_fp16")]; tensor matmul_6_transpose_y_0 = const()[name = tensor("matmul_6_transpose_y_0"), val = tensor(true)]; tensor matmul_6_transpose_x_0 = const()[name = tensor("matmul_6_transpose_x_0"), val = tensor(false)]; tensor matmul_6_cast_fp16 = matmul(transpose_x = matmul_6_transpose_x_0, transpose_y = matmul_6_transpose_y_0, x = mul_13_cast_fp16, y = k_27_cast_fp16)[name = tensor("matmul_6_cast_fp16")]; tensor add_6_cast_fp16 = add(x = matmul_6_cast_fp16, y = mul_12_cast_fp16)[name = tensor("add_6_cast_fp16")]; tensor add_6_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_6_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_6_axis_0 = const()[name = tensor("softmax_6_axis_0"), val = tensor(-1)]; tensor add_6_cast_fp16_to_fp32 = cast(dtype = add_6_cast_fp16_to_fp32_dtype_0, x = add_6_cast_fp16)[name = tensor("cast_339")]; tensor softmax_6 = softmax(axis = softmax_6_axis_0, x = add_6_cast_fp16_to_fp32)[name = tensor("softmax_6")]; tensor out_25_transpose_x_0 = const()[name = tensor("out_25_transpose_x_0"), val = tensor(false)]; tensor out_25_transpose_y_0 = const()[name = tensor("out_25_transpose_y_0"), val = tensor(false)]; tensor softmax_6_to_fp16_dtype_0 = const()[name = tensor("softmax_6_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_6_to_fp16 = cast(dtype = softmax_6_to_fp16_dtype_0, x = softmax_6)[name = tensor("cast_338")]; tensor out_25_cast_fp16 = matmul(transpose_x = out_25_transpose_x_0, transpose_y = out_25_transpose_y_0, x = softmax_6_to_fp16, y = v_13_cast_fp16)[name = tensor("out_25_cast_fp16")]; tensor concat_87x = const()[name = tensor("concat_87x"), val = tensor([1, -1])]; tensor var_1627 = reshape(shape = concat_87x, x = positions_13)[name = tensor("op_1627")]; tensor var_1628 = const()[name = tensor("op_1628"), val = tensor([-1, 1])]; tensor var_1629 = reshape(shape = var_1628, x = input_lengths_7)[name = tensor("op_1629")]; tensor var_1630 = less(x = var_1627, y = var_1629)[name = tensor("op_1630")]; tensor concat_88x = const()[name = tensor("concat_88x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_13 = reshape(shape = concat_88x, x = var_1630)[name = tensor("valid_q_13")]; tensor var_1634_to_fp16 = const()[name = tensor("op_1634_to_fp16"), val = tensor(0x0p+0)]; tensor out_27_cast_fp16 = select(a = out_25_cast_fp16, b = var_1634_to_fp16, cond = valid_q_13)[name = tensor("out_27_cast_fp16")]; tensor var_1636_perm_0 = const()[name = tensor("op_1636_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_89x = const()[name = tensor("concat_89x"), val = tensor([1, -1, 256])]; tensor var_1636_cast_fp16 = transpose(perm = var_1636_perm_0, x = out_27_cast_fp16)[name = tensor("transpose_16")]; tensor input_145_cast_fp16 = reshape(shape = concat_89x, x = var_1636_cast_fp16)[name = tensor("input_145_cast_fp16")]; tensor decoder_5_transformer_layers_0_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_0_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(12338176)))]; tensor linear_30_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_5_transformer_layers_0_self_attn_out_proj_weight_to_fp16, x = input_145_cast_fp16)[name = tensor("linear_30_cast_fp16")]; tensor decoder_5_transformer_layers_0_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_0_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(12469312)))]; tensor var_1642_cast_fp16 = mul(x = decoder_5_transformer_layers_0_layer_scale_1_scale_to_fp16, y = linear_30_cast_fp16)[name = tensor("op_1642_cast_fp16")]; tensor input_147_cast_fp16 = add(x = linear_28_cast_fp16, y = var_1642_cast_fp16)[name = tensor("input_147_cast_fp16")]; tensor input_149_axes_0 = const()[name = tensor("input_149_axes_0"), val = tensor([-1])]; tensor decoder_5_transformer_layers_0_norm2_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_0_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(12469888)))]; tensor decoder_5_transformer_layers_0_norm2_bias_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_0_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(12470464)))]; tensor input_149_cast_fp16 = layer_norm(axes = input_149_axes_0, beta = decoder_5_transformer_layers_0_norm2_bias_to_fp16, epsilon = var_1507_to_fp16, gamma = decoder_5_transformer_layers_0_norm2_weight_to_fp16, x = input_147_cast_fp16)[name = tensor("input_149_cast_fp16")]; tensor decoder_5_transformer_layers_0_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_0_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(12471040)))]; tensor linear_31_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_5_transformer_layers_0_ffn_0_weight_to_fp16, x = input_149_cast_fp16)[name = tensor("linear_31_cast_fp16")]; tensor input_153_mode_0 = const()[name = tensor("input_153_mode_0"), val = tensor("EXACT")]; tensor input_153_cast_fp16 = gelu(mode = input_153_mode_0, x = linear_31_cast_fp16)[name = tensor("input_153_cast_fp16")]; tensor decoder_5_transformer_layers_0_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_0_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(12995392)))]; tensor linear_32_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_5_transformer_layers_0_ffn_2_weight_to_fp16, x = input_153_cast_fp16)[name = tensor("linear_32_cast_fp16")]; tensor decoder_5_transformer_layers_0_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_0_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(13519744)))]; tensor var_1657_cast_fp16 = mul(x = decoder_5_transformer_layers_0_layer_scale_2_scale_to_fp16, y = linear_32_cast_fp16)[name = tensor("op_1657_cast_fp16")]; tensor input_155_cast_fp16 = add(x = input_147_cast_fp16, y = var_1657_cast_fp16)[name = tensor("input_155_cast_fp16")]; tensor x_61_axes_0 = const()[name = tensor("x_61_axes_0"), val = tensor([-1])]; tensor decoder_5_transformer_layers_1_norm1_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_1_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(13520320)))]; tensor decoder_5_transformer_layers_1_norm1_bias_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_1_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(13520896)))]; tensor x_61_cast_fp16 = layer_norm(axes = x_61_axes_0, beta = decoder_5_transformer_layers_1_norm1_bias_to_fp16, epsilon = var_1507_to_fp16, gamma = decoder_5_transformer_layers_1_norm1_weight_to_fp16, x = input_155_cast_fp16)[name = tensor("x_61_cast_fp16")]; tensor var_1672_shape_cast_fp16 = shape(x = x_61_cast_fp16)[name = tensor("op_1672_shape_cast_fp16")]; tensor gather_80_axis_0 = const()[name = tensor("gather_80_axis_0"), val = tensor(0)]; tensor gather_80_batch_dims_0 = const()[name = tensor("gather_80_batch_dims_0"), val = tensor(0)]; tensor gather_80_validate_indices_0 = const()[name = tensor("gather_80_validate_indices_0"), val = tensor(false)]; tensor var_1672_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1672_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_80_indices_0_to_uint16 = const()[name = tensor("gather_80_indices_0_to_uint16"), val = tensor(1)]; tensor var_1672_shape_cast_fp16_to_uint16 = cast(dtype = var_1672_shape_cast_fp16_to_uint16_dtype_0, x = var_1672_shape_cast_fp16)[name = tensor("cast_337")]; tensor gather_80_cast_uint16 = gather(axis = gather_80_axis_0, batch_dims = gather_80_batch_dims_0, indices = gather_80_indices_0_to_uint16, validate_indices = gather_80_validate_indices_0, x = var_1672_shape_cast_fp16_to_uint16)[name = tensor("gather_80_cast_uint16")]; tensor gather_80_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_80_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_5_transformer_layers_1_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_1_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(13521472)))]; tensor linear_33_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_5_transformer_layers_1_self_attn_in_proj_weight_to_fp16, x = x_61_cast_fp16)[name = tensor("linear_33_cast_fp16")]; tensor concat_90x = const()[name = tensor("concat_90x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_1679_cast_fp16 = reshape(shape = concat_90x, x = linear_33_cast_fp16)[name = tensor("op_1679_cast_fp16")]; tensor var_1680 = const()[name = tensor("op_1680"), val = tensor([2, 0, 3, 1, 4])]; tensor q_29_begin_0 = const()[name = tensor("q_29_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_29_end_0 = const()[name = tensor("q_29_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_29_end_mask_0 = const()[name = tensor("q_29_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_29_squeeze_mask_0 = const()[name = tensor("q_29_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_31_cast_fp16 = transpose(perm = var_1680, x = var_1679_cast_fp16)[name = tensor("transpose_15")]; tensor q_29_cast_fp16 = slice_by_index(begin = q_29_begin_0, end = q_29_end_0, end_mask = q_29_end_mask_0, squeeze_mask = q_29_squeeze_mask_0, x = projected_31_cast_fp16)[name = tensor("q_29_cast_fp16")]; tensor k_29_begin_0 = const()[name = tensor("k_29_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_29_end_0 = const()[name = tensor("k_29_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_29_end_mask_0 = const()[name = tensor("k_29_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_29_squeeze_mask_0 = const()[name = tensor("k_29_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_29_cast_fp16 = slice_by_index(begin = k_29_begin_0, end = k_29_end_0, end_mask = k_29_end_mask_0, squeeze_mask = k_29_squeeze_mask_0, x = projected_31_cast_fp16)[name = tensor("k_29_cast_fp16")]; tensor v_15_begin_0 = const()[name = tensor("v_15_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_15_end_0 = const()[name = tensor("v_15_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_15_end_mask_0 = const()[name = tensor("v_15_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_15_squeeze_mask_0 = const()[name = tensor("v_15_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_15_cast_fp16 = slice_by_index(begin = v_15_begin_0, end = v_15_end_0, end_mask = v_15_end_mask_0, squeeze_mask = v_15_squeeze_mask_0, x = projected_31_cast_fp16)[name = tensor("v_15_cast_fp16")]; tensor var_1692_shape_cast_fp16 = shape(x = q_29_cast_fp16)[name = tensor("op_1692_shape_cast_fp16")]; tensor gather_84_axis_0 = const()[name = tensor("gather_84_axis_0"), val = tensor(0)]; tensor gather_84_batch_dims_0 = const()[name = tensor("gather_84_batch_dims_0"), val = tensor(0)]; tensor gather_84_validate_indices_0 = const()[name = tensor("gather_84_validate_indices_0"), val = tensor(false)]; tensor var_1692_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1692_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_84_indices_0_to_uint16 = const()[name = tensor("gather_84_indices_0_to_uint16"), val = tensor(2)]; tensor var_1692_shape_cast_fp16_to_uint16 = cast(dtype = var_1692_shape_cast_fp16_to_uint16_dtype_0, x = var_1692_shape_cast_fp16)[name = tensor("cast_335")]; tensor gather_84_cast_uint16 = gather(axis = gather_84_axis_0, batch_dims = gather_84_batch_dims_0, indices = gather_84_indices_0_to_uint16, validate_indices = gather_84_validate_indices_0, x = var_1692_shape_cast_fp16_to_uint16)[name = tensor("gather_84_cast_uint16")]; tensor gather_84_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_84_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_66 = const()[name = tensor("const_66"), val = tensor(0)]; tensor const_67 = const()[name = tensor("const_67"), val = tensor(1)]; tensor gather_84_cast_uint16_to_int32 = cast(dtype = gather_84_cast_uint16_to_int32_dtype_0, x = gather_84_cast_uint16)[name = tensor("cast_334")]; tensor var_1696 = range_1d(end = gather_84_cast_uint16_to_int32, start = const_66, step = const_67)[name = tensor("op_1696")]; tensor var_1695_to_fp16 = const()[name = tensor("op_1695_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_1696_promoted_to_fp16_dtype_0 = const()[name = tensor("op_1696_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1696_to_fp16 = cast(dtype = var_1696_promoted_to_fp16_dtype_0, x = var_1696)[name = tensor("cast_333")]; tensor ts_29_cast_fp16 = add(x = var_1695_to_fp16, y = var_1696_to_fp16)[name = tensor("ts_29_cast_fp16")]; tensor var_1699 = const()[name = tensor("op_1699"), val = tensor([1, 1, -1, 1])]; tensor ts_31_cast_fp16 = reshape(shape = var_1699, x = ts_29_cast_fp16)[name = tensor("ts_31_cast_fp16")]; tensor concat_91x = const()[name = tensor("concat_91x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_15_cast_fp16 = reshape(shape = concat_91x, x = q_29_cast_fp16)[name = tensor("q2_15_cast_fp16")]; tensor concat_92x = const()[name = tensor("concat_92x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_15_cast_fp16 = reshape(shape = concat_92x, x = k_29_cast_fp16)[name = tensor("k2_15_cast_fp16")]; tensor qr_15_begin_0 = const()[name = tensor("qr_15_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_15_end_0 = const()[name = tensor("qr_15_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_15_end_mask_0 = const()[name = tensor("qr_15_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_15_squeeze_mask_0 = const()[name = tensor("qr_15_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_15_cast_fp16 = slice_by_index(begin = qr_15_begin_0, end = qr_15_end_0, end_mask = qr_15_end_mask_0, squeeze_mask = qr_15_squeeze_mask_0, x = q2_15_cast_fp16)[name = tensor("qr_15_cast_fp16")]; tensor qi_15_begin_0 = const()[name = tensor("qi_15_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_15_end_0 = const()[name = tensor("qi_15_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_15_end_mask_0 = const()[name = tensor("qi_15_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_15_squeeze_mask_0 = const()[name = tensor("qi_15_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_15_cast_fp16 = slice_by_index(begin = qi_15_begin_0, end = qi_15_end_0, end_mask = qi_15_end_mask_0, squeeze_mask = qi_15_squeeze_mask_0, x = q2_15_cast_fp16)[name = tensor("qi_15_cast_fp16")]; tensor kr_15_begin_0 = const()[name = tensor("kr_15_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_15_end_0 = const()[name = tensor("kr_15_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_15_end_mask_0 = const()[name = tensor("kr_15_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_15_squeeze_mask_0 = const()[name = tensor("kr_15_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_15_cast_fp16 = slice_by_index(begin = kr_15_begin_0, end = kr_15_end_0, end_mask = kr_15_end_mask_0, squeeze_mask = kr_15_squeeze_mask_0, x = k2_15_cast_fp16)[name = tensor("kr_15_cast_fp16")]; tensor ki_15_begin_0 = const()[name = tensor("ki_15_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_15_end_0 = const()[name = tensor("ki_15_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_15_end_mask_0 = const()[name = tensor("ki_15_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_15_squeeze_mask_0 = const()[name = tensor("ki_15_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_15_cast_fp16 = slice_by_index(begin = ki_15_begin_0, end = ki_15_end_0, end_mask = ki_15_end_mask_0, squeeze_mask = ki_15_squeeze_mask_0, x = k2_15_cast_fp16)[name = tensor("ki_15_cast_fp16")]; tensor freqs_15_to_fp16 = const()[name = tensor("freqs_15_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(13914752)))]; tensor var_1712_cast_fp16 = mul(x = freqs_15_to_fp16, y = ts_31_cast_fp16)[name = tensor("op_1712_cast_fp16")]; tensor rotr_15_cast_fp16 = cos(x = var_1712_cast_fp16)[name = tensor("rotr_15_cast_fp16")]; tensor roti_15_cast_fp16 = sin(x = var_1712_cast_fp16)[name = tensor("roti_15_cast_fp16")]; tensor var_1716_cast_fp16 = mul(x = qr_15_cast_fp16, y = rotr_15_cast_fp16)[name = tensor("op_1716_cast_fp16")]; tensor var_1717_cast_fp16 = mul(x = qi_15_cast_fp16, y = roti_15_cast_fp16)[name = tensor("op_1717_cast_fp16")]; tensor var_1718_cast_fp16 = sub(x = var_1716_cast_fp16, y = var_1717_cast_fp16)[name = tensor("op_1718_cast_fp16")]; tensor var_1719_cast_fp16 = mul(x = qr_15_cast_fp16, y = roti_15_cast_fp16)[name = tensor("op_1719_cast_fp16")]; tensor var_1720_cast_fp16 = mul(x = qi_15_cast_fp16, y = rotr_15_cast_fp16)[name = tensor("op_1720_cast_fp16")]; tensor var_1721_cast_fp16 = add(x = var_1719_cast_fp16, y = var_1720_cast_fp16)[name = tensor("op_1721_cast_fp16")]; tensor qo_15_axis_0 = const()[name = tensor("qo_15_axis_0"), val = tensor(-1)]; tensor qo_15_cast_fp16 = stack(axis = qo_15_axis_0, values = (var_1718_cast_fp16, var_1721_cast_fp16))[name = tensor("qo_15_cast_fp16")]; tensor var_1724_cast_fp16 = mul(x = kr_15_cast_fp16, y = rotr_15_cast_fp16)[name = tensor("op_1724_cast_fp16")]; tensor var_1725_cast_fp16 = mul(x = ki_15_cast_fp16, y = roti_15_cast_fp16)[name = tensor("op_1725_cast_fp16")]; tensor var_1726_cast_fp16 = sub(x = var_1724_cast_fp16, y = var_1725_cast_fp16)[name = tensor("op_1726_cast_fp16")]; tensor var_1727_cast_fp16 = mul(x = kr_15_cast_fp16, y = roti_15_cast_fp16)[name = tensor("op_1727_cast_fp16")]; tensor var_1728_cast_fp16 = mul(x = ki_15_cast_fp16, y = rotr_15_cast_fp16)[name = tensor("op_1728_cast_fp16")]; tensor var_1729_cast_fp16 = add(x = var_1727_cast_fp16, y = var_1728_cast_fp16)[name = tensor("op_1729_cast_fp16")]; tensor ko_15_axis_0 = const()[name = tensor("ko_15_axis_0"), val = tensor(-1)]; tensor ko_15_cast_fp16 = stack(axis = ko_15_axis_0, values = (var_1726_cast_fp16, var_1729_cast_fp16))[name = tensor("ko_15_cast_fp16")]; tensor concat_93x = const()[name = tensor("concat_93x"), val = tensor([1, 4, -1, 64])]; tensor q_31_cast_fp16 = reshape(shape = concat_93x, x = qo_15_cast_fp16)[name = tensor("q_31_cast_fp16")]; tensor concat_94x = const()[name = tensor("concat_94x"), val = tensor([1, 4, -1, 64])]; tensor k_31_cast_fp16 = reshape(shape = concat_94x, x = ko_15_cast_fp16)[name = tensor("k_31_cast_fp16")]; tensor const_68 = const()[name = tensor("const_68"), val = tensor(0)]; tensor const_69 = const()[name = tensor("const_69"), val = tensor(1)]; tensor gather_80_cast_uint16_to_int32 = cast(dtype = gather_80_cast_uint16_to_int32_dtype_0, x = gather_80_cast_uint16)[name = tensor("cast_336")]; tensor positions_15 = range_1d(end = gather_80_cast_uint16_to_int32, start = const_68, step = const_69)[name = tensor("positions_15")]; tensor concat_95x = const()[name = tensor("concat_95x"), val = tensor([1, 1, -1])]; tensor var_1741 = reshape(shape = concat_95x, x = positions_15)[name = tensor("op_1741")]; tensor valid_k_15 = less(x = var_1741, y = var_1606)[name = tensor("valid_k_15")]; tensor concat_96x = const()[name = tensor("concat_96x"), val = tensor([1, -1, 1])]; tensor var_1746 = reshape(shape = concat_96x, x = positions_15)[name = tensor("op_1746")]; tensor delta_15 = sub(x = var_1746, y = var_1741)[name = tensor("delta_15")]; tensor concat_98_axis_0 = const()[name = tensor("concat_98_axis_0"), val = tensor(0)]; tensor concat_98_interleave_0 = const()[name = tensor("concat_98_interleave_0"), val = tensor(false)]; tensor concat_98 = concat(axis = concat_98_axis_0, interleave = concat_98_interleave_0, values = (var_1511, gather_80_cast_uint16_to_int32, gather_80_cast_uint16_to_int32))[name = tensor("concat_98")]; tensor fill_7_value_0_to_fp16 = const()[name = tensor("fill_7_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_7_cast_fp16 = fill(shape = concat_98, value = fill_7_value_0_to_fp16)[name = tensor("fill_7_cast_fp16")]; tensor attn_bias_57_dtype_0 = const()[name = tensor("attn_bias_57_dtype_0"), val = tensor("bool")]; tensor var_1752 = greater_equal(x = delta_15, y = var_1493)[name = tensor("op_1752")]; tensor fill_7_cast_fp16_to_bool = cast(dtype = attn_bias_57_dtype_0, x = fill_7_cast_fp16)[name = tensor("cast_332")]; tensor attn_bias_59 = logical_and(x = fill_7_cast_fp16_to_bool, y = var_1752)[name = tensor("attn_bias_59")]; tensor var_1754 = less(x = delta_15, y = var_1500)[name = tensor("op_1754")]; tensor attn_bias_61 = logical_and(x = attn_bias_59, y = var_1754)[name = tensor("attn_bias_61")]; tensor var_1756 = logical_and(x = attn_bias_61, y = valid_k_15)[name = tensor("op_1756")]; tensor var_1758_axes_0 = const()[name = tensor("op_1758_axes_0"), val = tensor([1])]; tensor var_1758 = expand_dims(axes = var_1758_axes_0, x = var_1756)[name = tensor("op_1758")]; tensor sub_7_x_0_to_fp16 = const()[name = tensor("sub_7_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_193_to_fp16_dtype_0 = const()[name = tensor("cast_193_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1758_to_fp16 = cast(dtype = cast_193_to_fp16_dtype_0, x = var_1758)[name = tensor("cast_331")]; tensor sub_7_cast_fp16 = sub(x = sub_7_x_0_to_fp16, y = var_1758_to_fp16)[name = tensor("sub_7_cast_fp16")]; tensor mul_14_x_0_to_fp16 = const()[name = tensor("mul_14_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_14_cast_fp16 = mul(x = mul_14_x_0_to_fp16, y = sub_7_cast_fp16)[name = tensor("mul_14_cast_fp16")]; tensor mul_15_y_0_to_fp16 = const()[name = tensor("mul_15_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_15_cast_fp16 = mul(x = q_31_cast_fp16, y = mul_15_y_0_to_fp16)[name = tensor("mul_15_cast_fp16")]; tensor matmul_7_transpose_y_0 = const()[name = tensor("matmul_7_transpose_y_0"), val = tensor(true)]; tensor matmul_7_transpose_x_0 = const()[name = tensor("matmul_7_transpose_x_0"), val = tensor(false)]; tensor matmul_7_cast_fp16 = matmul(transpose_x = matmul_7_transpose_x_0, transpose_y = matmul_7_transpose_y_0, x = mul_15_cast_fp16, y = k_31_cast_fp16)[name = tensor("matmul_7_cast_fp16")]; tensor add_7_cast_fp16 = add(x = matmul_7_cast_fp16, y = mul_14_cast_fp16)[name = tensor("add_7_cast_fp16")]; tensor add_7_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_7_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_7_axis_0 = const()[name = tensor("softmax_7_axis_0"), val = tensor(-1)]; tensor add_7_cast_fp16_to_fp32 = cast(dtype = add_7_cast_fp16_to_fp32_dtype_0, x = add_7_cast_fp16)[name = tensor("cast_330")]; tensor softmax_7 = softmax(axis = softmax_7_axis_0, x = add_7_cast_fp16_to_fp32)[name = tensor("softmax_7")]; tensor out_29_transpose_x_0 = const()[name = tensor("out_29_transpose_x_0"), val = tensor(false)]; tensor out_29_transpose_y_0 = const()[name = tensor("out_29_transpose_y_0"), val = tensor(false)]; tensor softmax_7_to_fp16_dtype_0 = const()[name = tensor("softmax_7_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_7_to_fp16 = cast(dtype = softmax_7_to_fp16_dtype_0, x = softmax_7)[name = tensor("cast_329")]; tensor out_29_cast_fp16 = matmul(transpose_x = out_29_transpose_x_0, transpose_y = out_29_transpose_y_0, x = softmax_7_to_fp16, y = v_15_cast_fp16)[name = tensor("out_29_cast_fp16")]; tensor concat_99x = const()[name = tensor("concat_99x"), val = tensor([1, -1])]; tensor var_1764 = reshape(shape = concat_99x, x = positions_15)[name = tensor("op_1764")]; tensor var_1767 = less(x = var_1764, y = var_1629)[name = tensor("op_1767")]; tensor concat_100x = const()[name = tensor("concat_100x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_15 = reshape(shape = concat_100x, x = var_1767)[name = tensor("valid_q_15")]; tensor var_1771_to_fp16 = const()[name = tensor("op_1771_to_fp16"), val = tensor(0x0p+0)]; tensor out_31_cast_fp16 = select(a = out_29_cast_fp16, b = var_1771_to_fp16, cond = valid_q_15)[name = tensor("out_31_cast_fp16")]; tensor var_1773_perm_0 = const()[name = tensor("op_1773_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_101x = const()[name = tensor("concat_101x"), val = tensor([1, -1, 256])]; tensor var_1773_cast_fp16 = transpose(perm = var_1773_perm_0, x = out_31_cast_fp16)[name = tensor("transpose_14")]; tensor input_157_cast_fp16 = reshape(shape = concat_101x, x = var_1773_cast_fp16)[name = tensor("input_157_cast_fp16")]; tensor decoder_5_transformer_layers_1_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_1_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(13914880)))]; tensor linear_34_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_5_transformer_layers_1_self_attn_out_proj_weight_to_fp16, x = input_157_cast_fp16)[name = tensor("linear_34_cast_fp16")]; tensor decoder_5_transformer_layers_1_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_1_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14046016)))]; tensor var_1779_cast_fp16 = mul(x = decoder_5_transformer_layers_1_layer_scale_1_scale_to_fp16, y = linear_34_cast_fp16)[name = tensor("op_1779_cast_fp16")]; tensor input_159_cast_fp16 = add(x = input_155_cast_fp16, y = var_1779_cast_fp16)[name = tensor("input_159_cast_fp16")]; tensor input_161_axes_0 = const()[name = tensor("input_161_axes_0"), val = tensor([-1])]; tensor decoder_5_transformer_layers_1_norm2_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_1_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14046592)))]; tensor decoder_5_transformer_layers_1_norm2_bias_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_1_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14047168)))]; tensor input_161_cast_fp16 = layer_norm(axes = input_161_axes_0, beta = decoder_5_transformer_layers_1_norm2_bias_to_fp16, epsilon = var_1507_to_fp16, gamma = decoder_5_transformer_layers_1_norm2_weight_to_fp16, x = input_159_cast_fp16)[name = tensor("input_161_cast_fp16")]; tensor decoder_5_transformer_layers_1_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_1_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14047744)))]; tensor linear_35_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_5_transformer_layers_1_ffn_0_weight_to_fp16, x = input_161_cast_fp16)[name = tensor("linear_35_cast_fp16")]; tensor input_165_mode_0 = const()[name = tensor("input_165_mode_0"), val = tensor("EXACT")]; tensor input_165_cast_fp16 = gelu(mode = input_165_mode_0, x = linear_35_cast_fp16)[name = tensor("input_165_cast_fp16")]; tensor decoder_5_transformer_layers_1_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_1_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(14572096)))]; tensor linear_36_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_5_transformer_layers_1_ffn_2_weight_to_fp16, x = input_165_cast_fp16)[name = tensor("linear_36_cast_fp16")]; tensor decoder_5_transformer_layers_1_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_5_transformer_layers_1_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15096448)))]; tensor var_1794_cast_fp16 = mul(x = decoder_5_transformer_layers_1_layer_scale_2_scale_to_fp16, y = linear_36_cast_fp16)[name = tensor("op_1794_cast_fp16")]; tensor input_167_cast_fp16 = add(x = input_159_cast_fp16, y = var_1794_cast_fp16)[name = tensor("input_167_cast_fp16")]; tensor decoder_5_output_proj_weight_to_fp16 = const()[name = tensor("decoder_5_output_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15097024)))]; tensor linear_37_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_5_output_proj_weight_to_fp16, x = input_167_cast_fp16)[name = tensor("linear_37_cast_fp16")]; tensor x_67_perm_0 = const()[name = tensor("x_67_perm_0"), val = tensor([0, 2, 1])]; tensor concat_102x = const()[name = tensor("concat_102x"), val = tensor([1, 384, 2, -1])]; tensor x_67_cast_fp16 = transpose(perm = x_67_perm_0, x = linear_37_cast_fp16)[name = tensor("transpose_13")]; tensor var_1813_cast_fp16 = reshape(shape = concat_102x, x = x_67_cast_fp16)[name = tensor("op_1813_cast_fp16")]; tensor var_1814 = const()[name = tensor("op_1814"), val = tensor([0, 1, 3, 2])]; tensor concat_103x = const()[name = tensor("concat_103x"), val = tensor([1, 384, -1])]; tensor var_1815_cast_fp16 = transpose(perm = var_1814, x = var_1813_cast_fp16)[name = tensor("transpose_12")]; tensor x_69_cast_fp16 = reshape(shape = concat_103x, x = var_1815_cast_fp16)[name = tensor("x_69_cast_fp16")]; tensor var_1821 = const()[name = tensor("op_1821"), val = tensor(2)]; tensor input_lengths = mul(x = input_lengths_7, y = var_1821)[name = tensor("input_lengths")]; tensor var_1827 = const()[name = tensor("op_1827"), val = tensor(0)]; tensor var_1834 = const()[name = tensor("op_1834"), val = tensor(1600)]; tensor var_1845 = const()[name = tensor("op_1845"), val = tensor(1)]; tensor input_169_perm_0 = const()[name = tensor("input_169_perm_0"), val = tensor([0, 2, 1])]; tensor decoder_7_input_proj_weight_to_fp16 = const()[name = tensor("decoder_7_input_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15490304)))]; tensor input_169_cast_fp16 = transpose(perm = input_169_perm_0, x = x_69_cast_fp16)[name = tensor("transpose_11")]; tensor linear_38_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_7_input_proj_weight_to_fp16, x = input_169_cast_fp16)[name = tensor("linear_38_cast_fp16")]; tensor x_73_axes_0 = const()[name = tensor("x_73_axes_0"), val = tensor([-1])]; tensor decoder_7_transformer_layers_0_norm1_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_0_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15686976)))]; tensor decoder_7_transformer_layers_0_norm1_bias_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_0_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15687552)))]; tensor var_1841_to_fp16 = const()[name = tensor("op_1841_to_fp16"), val = tensor(0x1.5p-17)]; tensor x_73_cast_fp16 = layer_norm(axes = x_73_axes_0, beta = decoder_7_transformer_layers_0_norm1_bias_to_fp16, epsilon = var_1841_to_fp16, gamma = decoder_7_transformer_layers_0_norm1_weight_to_fp16, x = linear_38_cast_fp16)[name = tensor("x_73_cast_fp16")]; tensor var_1873_shape_cast_fp16 = shape(x = x_73_cast_fp16)[name = tensor("op_1873_shape_cast_fp16")]; tensor gather_93_axis_0 = const()[name = tensor("gather_93_axis_0"), val = tensor(0)]; tensor gather_93_batch_dims_0 = const()[name = tensor("gather_93_batch_dims_0"), val = tensor(0)]; tensor gather_93_validate_indices_0 = const()[name = tensor("gather_93_validate_indices_0"), val = tensor(false)]; tensor var_1873_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1873_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_93_indices_0_to_uint16 = const()[name = tensor("gather_93_indices_0_to_uint16"), val = tensor(1)]; tensor var_1873_shape_cast_fp16_to_uint16 = cast(dtype = var_1873_shape_cast_fp16_to_uint16_dtype_0, x = var_1873_shape_cast_fp16)[name = tensor("cast_328")]; tensor gather_93_cast_uint16 = gather(axis = gather_93_axis_0, batch_dims = gather_93_batch_dims_0, indices = gather_93_indices_0_to_uint16, validate_indices = gather_93_validate_indices_0, x = var_1873_shape_cast_fp16_to_uint16)[name = tensor("gather_93_cast_uint16")]; tensor gather_93_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_93_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_7_transformer_layers_0_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_0_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(15688128)))]; tensor linear_39_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_7_transformer_layers_0_self_attn_in_proj_weight_to_fp16, x = x_73_cast_fp16)[name = tensor("linear_39_cast_fp16")]; tensor concat_104x = const()[name = tensor("concat_104x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_1880_cast_fp16 = reshape(shape = concat_104x, x = linear_39_cast_fp16)[name = tensor("op_1880_cast_fp16")]; tensor var_1881 = const()[name = tensor("op_1881"), val = tensor([2, 0, 3, 1, 4])]; tensor q_33_begin_0 = const()[name = tensor("q_33_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_33_end_0 = const()[name = tensor("q_33_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_33_end_mask_0 = const()[name = tensor("q_33_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_33_squeeze_mask_0 = const()[name = tensor("q_33_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_35_cast_fp16 = transpose(perm = var_1881, x = var_1880_cast_fp16)[name = tensor("transpose_10")]; tensor q_33_cast_fp16 = slice_by_index(begin = q_33_begin_0, end = q_33_end_0, end_mask = q_33_end_mask_0, squeeze_mask = q_33_squeeze_mask_0, x = projected_35_cast_fp16)[name = tensor("q_33_cast_fp16")]; tensor k_33_begin_0 = const()[name = tensor("k_33_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_33_end_0 = const()[name = tensor("k_33_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_33_end_mask_0 = const()[name = tensor("k_33_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_33_squeeze_mask_0 = const()[name = tensor("k_33_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_33_cast_fp16 = slice_by_index(begin = k_33_begin_0, end = k_33_end_0, end_mask = k_33_end_mask_0, squeeze_mask = k_33_squeeze_mask_0, x = projected_35_cast_fp16)[name = tensor("k_33_cast_fp16")]; tensor v_17_begin_0 = const()[name = tensor("v_17_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_17_end_0 = const()[name = tensor("v_17_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_17_end_mask_0 = const()[name = tensor("v_17_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_17_squeeze_mask_0 = const()[name = tensor("v_17_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_17_cast_fp16 = slice_by_index(begin = v_17_begin_0, end = v_17_end_0, end_mask = v_17_end_mask_0, squeeze_mask = v_17_squeeze_mask_0, x = projected_35_cast_fp16)[name = tensor("v_17_cast_fp16")]; tensor var_1893_shape_cast_fp16 = shape(x = q_33_cast_fp16)[name = tensor("op_1893_shape_cast_fp16")]; tensor gather_97_axis_0 = const()[name = tensor("gather_97_axis_0"), val = tensor(0)]; tensor gather_97_batch_dims_0 = const()[name = tensor("gather_97_batch_dims_0"), val = tensor(0)]; tensor gather_97_validate_indices_0 = const()[name = tensor("gather_97_validate_indices_0"), val = tensor(false)]; tensor var_1893_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_1893_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_97_indices_0_to_uint16 = const()[name = tensor("gather_97_indices_0_to_uint16"), val = tensor(2)]; tensor var_1893_shape_cast_fp16_to_uint16 = cast(dtype = var_1893_shape_cast_fp16_to_uint16_dtype_0, x = var_1893_shape_cast_fp16)[name = tensor("cast_326")]; tensor gather_97_cast_uint16 = gather(axis = gather_97_axis_0, batch_dims = gather_97_batch_dims_0, indices = gather_97_indices_0_to_uint16, validate_indices = gather_97_validate_indices_0, x = var_1893_shape_cast_fp16_to_uint16)[name = tensor("gather_97_cast_uint16")]; tensor gather_97_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_97_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_75 = const()[name = tensor("const_75"), val = tensor(0)]; tensor const_76 = const()[name = tensor("const_76"), val = tensor(1)]; tensor gather_97_cast_uint16_to_int32 = cast(dtype = gather_97_cast_uint16_to_int32_dtype_0, x = gather_97_cast_uint16)[name = tensor("cast_325")]; tensor var_1897 = range_1d(end = gather_97_cast_uint16_to_int32, start = const_75, step = const_76)[name = tensor("op_1897")]; tensor var_1896_to_fp16 = const()[name = tensor("op_1896_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_1897_promoted_to_fp16_dtype_0 = const()[name = tensor("op_1897_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1897_to_fp16 = cast(dtype = var_1897_promoted_to_fp16_dtype_0, x = var_1897)[name = tensor("cast_324")]; tensor ts_33_cast_fp16 = add(x = var_1896_to_fp16, y = var_1897_to_fp16)[name = tensor("ts_33_cast_fp16")]; tensor var_1900 = const()[name = tensor("op_1900"), val = tensor([1, 1, -1, 1])]; tensor ts_35_cast_fp16 = reshape(shape = var_1900, x = ts_33_cast_fp16)[name = tensor("ts_35_cast_fp16")]; tensor concat_105x = const()[name = tensor("concat_105x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_17_cast_fp16 = reshape(shape = concat_105x, x = q_33_cast_fp16)[name = tensor("q2_17_cast_fp16")]; tensor concat_106x = const()[name = tensor("concat_106x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_17_cast_fp16 = reshape(shape = concat_106x, x = k_33_cast_fp16)[name = tensor("k2_17_cast_fp16")]; tensor qr_17_begin_0 = const()[name = tensor("qr_17_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_17_end_0 = const()[name = tensor("qr_17_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_17_end_mask_0 = const()[name = tensor("qr_17_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_17_squeeze_mask_0 = const()[name = tensor("qr_17_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_17_cast_fp16 = slice_by_index(begin = qr_17_begin_0, end = qr_17_end_0, end_mask = qr_17_end_mask_0, squeeze_mask = qr_17_squeeze_mask_0, x = q2_17_cast_fp16)[name = tensor("qr_17_cast_fp16")]; tensor qi_17_begin_0 = const()[name = tensor("qi_17_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_17_end_0 = const()[name = tensor("qi_17_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_17_end_mask_0 = const()[name = tensor("qi_17_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_17_squeeze_mask_0 = const()[name = tensor("qi_17_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_17_cast_fp16 = slice_by_index(begin = qi_17_begin_0, end = qi_17_end_0, end_mask = qi_17_end_mask_0, squeeze_mask = qi_17_squeeze_mask_0, x = q2_17_cast_fp16)[name = tensor("qi_17_cast_fp16")]; tensor kr_17_begin_0 = const()[name = tensor("kr_17_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_17_end_0 = const()[name = tensor("kr_17_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_17_end_mask_0 = const()[name = tensor("kr_17_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_17_squeeze_mask_0 = const()[name = tensor("kr_17_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_17_cast_fp16 = slice_by_index(begin = kr_17_begin_0, end = kr_17_end_0, end_mask = kr_17_end_mask_0, squeeze_mask = kr_17_squeeze_mask_0, x = k2_17_cast_fp16)[name = tensor("kr_17_cast_fp16")]; tensor ki_17_begin_0 = const()[name = tensor("ki_17_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_17_end_0 = const()[name = tensor("ki_17_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_17_end_mask_0 = const()[name = tensor("ki_17_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_17_squeeze_mask_0 = const()[name = tensor("ki_17_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_17_cast_fp16 = slice_by_index(begin = ki_17_begin_0, end = ki_17_end_0, end_mask = ki_17_end_mask_0, squeeze_mask = ki_17_squeeze_mask_0, x = k2_17_cast_fp16)[name = tensor("ki_17_cast_fp16")]; tensor freqs_17_to_fp16 = const()[name = tensor("freqs_17_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(16081408)))]; tensor var_1913_cast_fp16 = mul(x = freqs_17_to_fp16, y = ts_35_cast_fp16)[name = tensor("op_1913_cast_fp16")]; tensor rotr_17_cast_fp16 = cos(x = var_1913_cast_fp16)[name = tensor("rotr_17_cast_fp16")]; tensor roti_17_cast_fp16 = sin(x = var_1913_cast_fp16)[name = tensor("roti_17_cast_fp16")]; tensor var_1917_cast_fp16 = mul(x = qr_17_cast_fp16, y = rotr_17_cast_fp16)[name = tensor("op_1917_cast_fp16")]; tensor var_1918_cast_fp16 = mul(x = qi_17_cast_fp16, y = roti_17_cast_fp16)[name = tensor("op_1918_cast_fp16")]; tensor var_1919_cast_fp16 = sub(x = var_1917_cast_fp16, y = var_1918_cast_fp16)[name = tensor("op_1919_cast_fp16")]; tensor var_1920_cast_fp16 = mul(x = qr_17_cast_fp16, y = roti_17_cast_fp16)[name = tensor("op_1920_cast_fp16")]; tensor var_1921_cast_fp16 = mul(x = qi_17_cast_fp16, y = rotr_17_cast_fp16)[name = tensor("op_1921_cast_fp16")]; tensor var_1922_cast_fp16 = add(x = var_1920_cast_fp16, y = var_1921_cast_fp16)[name = tensor("op_1922_cast_fp16")]; tensor qo_17_axis_0 = const()[name = tensor("qo_17_axis_0"), val = tensor(-1)]; tensor qo_17_cast_fp16 = stack(axis = qo_17_axis_0, values = (var_1919_cast_fp16, var_1922_cast_fp16))[name = tensor("qo_17_cast_fp16")]; tensor var_1925_cast_fp16 = mul(x = kr_17_cast_fp16, y = rotr_17_cast_fp16)[name = tensor("op_1925_cast_fp16")]; tensor var_1926_cast_fp16 = mul(x = ki_17_cast_fp16, y = roti_17_cast_fp16)[name = tensor("op_1926_cast_fp16")]; tensor var_1927_cast_fp16 = sub(x = var_1925_cast_fp16, y = var_1926_cast_fp16)[name = tensor("op_1927_cast_fp16")]; tensor var_1928_cast_fp16 = mul(x = kr_17_cast_fp16, y = roti_17_cast_fp16)[name = tensor("op_1928_cast_fp16")]; tensor var_1929_cast_fp16 = mul(x = ki_17_cast_fp16, y = rotr_17_cast_fp16)[name = tensor("op_1929_cast_fp16")]; tensor var_1930_cast_fp16 = add(x = var_1928_cast_fp16, y = var_1929_cast_fp16)[name = tensor("op_1930_cast_fp16")]; tensor ko_17_axis_0 = const()[name = tensor("ko_17_axis_0"), val = tensor(-1)]; tensor ko_17_cast_fp16 = stack(axis = ko_17_axis_0, values = (var_1927_cast_fp16, var_1930_cast_fp16))[name = tensor("ko_17_cast_fp16")]; tensor concat_107x = const()[name = tensor("concat_107x"), val = tensor([1, 4, -1, 64])]; tensor q_35_cast_fp16 = reshape(shape = concat_107x, x = qo_17_cast_fp16)[name = tensor("q_35_cast_fp16")]; tensor concat_108x = const()[name = tensor("concat_108x"), val = tensor([1, 4, -1, 64])]; tensor k_35_cast_fp16 = reshape(shape = concat_108x, x = ko_17_cast_fp16)[name = tensor("k_35_cast_fp16")]; tensor const_77 = const()[name = tensor("const_77"), val = tensor(0)]; tensor const_78 = const()[name = tensor("const_78"), val = tensor(1)]; tensor gather_93_cast_uint16_to_int32 = cast(dtype = gather_93_cast_uint16_to_int32_dtype_0, x = gather_93_cast_uint16)[name = tensor("cast_327")]; tensor positions_17 = range_1d(end = gather_93_cast_uint16_to_int32, start = const_77, step = const_78)[name = tensor("positions_17")]; tensor concat_109x = const()[name = tensor("concat_109x"), val = tensor([1, 1, -1])]; tensor var_1942 = reshape(shape = concat_109x, x = positions_17)[name = tensor("op_1942")]; tensor var_1943 = const()[name = tensor("op_1943"), val = tensor([-1, 1, 1])]; tensor var_1944 = reshape(shape = var_1943, x = input_lengths)[name = tensor("op_1944")]; tensor valid_k_17 = less(x = var_1942, y = var_1944)[name = tensor("valid_k_17")]; tensor concat_110x = const()[name = tensor("concat_110x"), val = tensor([1, -1, 1])]; tensor var_1947 = reshape(shape = concat_110x, x = positions_17)[name = tensor("op_1947")]; tensor delta_17 = sub(x = var_1947, y = var_1942)[name = tensor("delta_17")]; tensor concat_112_axis_0 = const()[name = tensor("concat_112_axis_0"), val = tensor(0)]; tensor concat_112_interleave_0 = const()[name = tensor("concat_112_interleave_0"), val = tensor(false)]; tensor concat_112 = concat(axis = concat_112_axis_0, interleave = concat_112_interleave_0, values = (var_1845, gather_93_cast_uint16_to_int32, gather_93_cast_uint16_to_int32))[name = tensor("concat_112")]; tensor fill_8_value_0_to_fp16 = const()[name = tensor("fill_8_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_8_cast_fp16 = fill(shape = concat_112, value = fill_8_value_0_to_fp16)[name = tensor("fill_8_cast_fp16")]; tensor attn_bias_65_dtype_0 = const()[name = tensor("attn_bias_65_dtype_0"), val = tensor("bool")]; tensor var_1953 = greater_equal(x = delta_17, y = var_1827)[name = tensor("op_1953")]; tensor fill_8_cast_fp16_to_bool = cast(dtype = attn_bias_65_dtype_0, x = fill_8_cast_fp16)[name = tensor("cast_323")]; tensor attn_bias_67 = logical_and(x = fill_8_cast_fp16_to_bool, y = var_1953)[name = tensor("attn_bias_67")]; tensor var_1955 = less(x = delta_17, y = var_1834)[name = tensor("op_1955")]; tensor attn_bias_69 = logical_and(x = attn_bias_67, y = var_1955)[name = tensor("attn_bias_69")]; tensor var_1957 = logical_and(x = attn_bias_69, y = valid_k_17)[name = tensor("op_1957")]; tensor var_1959_axes_0 = const()[name = tensor("op_1959_axes_0"), val = tensor([1])]; tensor var_1959 = expand_dims(axes = var_1959_axes_0, x = var_1957)[name = tensor("op_1959")]; tensor sub_8_x_0_to_fp16 = const()[name = tensor("sub_8_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_217_to_fp16_dtype_0 = const()[name = tensor("cast_217_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_1959_to_fp16 = cast(dtype = cast_217_to_fp16_dtype_0, x = var_1959)[name = tensor("cast_322")]; tensor sub_8_cast_fp16 = sub(x = sub_8_x_0_to_fp16, y = var_1959_to_fp16)[name = tensor("sub_8_cast_fp16")]; tensor mul_16_x_0_to_fp16 = const()[name = tensor("mul_16_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_16_cast_fp16 = mul(x = mul_16_x_0_to_fp16, y = sub_8_cast_fp16)[name = tensor("mul_16_cast_fp16")]; tensor mul_17_y_0_to_fp16 = const()[name = tensor("mul_17_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_17_cast_fp16 = mul(x = q_35_cast_fp16, y = mul_17_y_0_to_fp16)[name = tensor("mul_17_cast_fp16")]; tensor matmul_8_transpose_y_0 = const()[name = tensor("matmul_8_transpose_y_0"), val = tensor(true)]; tensor matmul_8_transpose_x_0 = const()[name = tensor("matmul_8_transpose_x_0"), val = tensor(false)]; tensor matmul_8_cast_fp16 = matmul(transpose_x = matmul_8_transpose_x_0, transpose_y = matmul_8_transpose_y_0, x = mul_17_cast_fp16, y = k_35_cast_fp16)[name = tensor("matmul_8_cast_fp16")]; tensor add_8_cast_fp16 = add(x = matmul_8_cast_fp16, y = mul_16_cast_fp16)[name = tensor("add_8_cast_fp16")]; tensor add_8_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_8_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_8_axis_0 = const()[name = tensor("softmax_8_axis_0"), val = tensor(-1)]; tensor add_8_cast_fp16_to_fp32 = cast(dtype = add_8_cast_fp16_to_fp32_dtype_0, x = add_8_cast_fp16)[name = tensor("cast_321")]; tensor softmax_8 = softmax(axis = softmax_8_axis_0, x = add_8_cast_fp16_to_fp32)[name = tensor("softmax_8")]; tensor out_33_transpose_x_0 = const()[name = tensor("out_33_transpose_x_0"), val = tensor(false)]; tensor out_33_transpose_y_0 = const()[name = tensor("out_33_transpose_y_0"), val = tensor(false)]; tensor softmax_8_to_fp16_dtype_0 = const()[name = tensor("softmax_8_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_8_to_fp16 = cast(dtype = softmax_8_to_fp16_dtype_0, x = softmax_8)[name = tensor("cast_320")]; tensor out_33_cast_fp16 = matmul(transpose_x = out_33_transpose_x_0, transpose_y = out_33_transpose_y_0, x = softmax_8_to_fp16, y = v_17_cast_fp16)[name = tensor("out_33_cast_fp16")]; tensor concat_113x = const()[name = tensor("concat_113x"), val = tensor([1, -1])]; tensor var_1965 = reshape(shape = concat_113x, x = positions_17)[name = tensor("op_1965")]; tensor var_1966 = const()[name = tensor("op_1966"), val = tensor([-1, 1])]; tensor var_1967 = reshape(shape = var_1966, x = input_lengths)[name = tensor("op_1967")]; tensor var_1968 = less(x = var_1965, y = var_1967)[name = tensor("op_1968")]; tensor concat_114x = const()[name = tensor("concat_114x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_17 = reshape(shape = concat_114x, x = var_1968)[name = tensor("valid_q_17")]; tensor var_1972_to_fp16 = const()[name = tensor("op_1972_to_fp16"), val = tensor(0x0p+0)]; tensor out_35_cast_fp16 = select(a = out_33_cast_fp16, b = var_1972_to_fp16, cond = valid_q_17)[name = tensor("out_35_cast_fp16")]; tensor var_1974_perm_0 = const()[name = tensor("op_1974_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_115x = const()[name = tensor("concat_115x"), val = tensor([1, -1, 256])]; tensor var_1974_cast_fp16 = transpose(perm = var_1974_perm_0, x = out_35_cast_fp16)[name = tensor("transpose_9")]; tensor input_171_cast_fp16 = reshape(shape = concat_115x, x = var_1974_cast_fp16)[name = tensor("input_171_cast_fp16")]; tensor decoder_7_transformer_layers_0_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_0_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(16081536)))]; tensor linear_40_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_7_transformer_layers_0_self_attn_out_proj_weight_to_fp16, x = input_171_cast_fp16)[name = tensor("linear_40_cast_fp16")]; tensor decoder_7_transformer_layers_0_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_0_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(16212672)))]; tensor var_1980_cast_fp16 = mul(x = decoder_7_transformer_layers_0_layer_scale_1_scale_to_fp16, y = linear_40_cast_fp16)[name = tensor("op_1980_cast_fp16")]; tensor input_173_cast_fp16 = add(x = linear_38_cast_fp16, y = var_1980_cast_fp16)[name = tensor("input_173_cast_fp16")]; tensor input_175_axes_0 = const()[name = tensor("input_175_axes_0"), val = tensor([-1])]; tensor decoder_7_transformer_layers_0_norm2_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_0_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(16213248)))]; tensor decoder_7_transformer_layers_0_norm2_bias_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_0_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(16213824)))]; tensor input_175_cast_fp16 = layer_norm(axes = input_175_axes_0, beta = decoder_7_transformer_layers_0_norm2_bias_to_fp16, epsilon = var_1841_to_fp16, gamma = decoder_7_transformer_layers_0_norm2_weight_to_fp16, x = input_173_cast_fp16)[name = tensor("input_175_cast_fp16")]; tensor decoder_7_transformer_layers_0_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_0_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(16214400)))]; tensor linear_41_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_7_transformer_layers_0_ffn_0_weight_to_fp16, x = input_175_cast_fp16)[name = tensor("linear_41_cast_fp16")]; tensor input_179_mode_0 = const()[name = tensor("input_179_mode_0"), val = tensor("EXACT")]; tensor input_179_cast_fp16 = gelu(mode = input_179_mode_0, x = linear_41_cast_fp16)[name = tensor("input_179_cast_fp16")]; tensor decoder_7_transformer_layers_0_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_0_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(16738752)))]; tensor linear_42_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_7_transformer_layers_0_ffn_2_weight_to_fp16, x = input_179_cast_fp16)[name = tensor("linear_42_cast_fp16")]; tensor decoder_7_transformer_layers_0_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_0_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(17263104)))]; tensor var_1995_cast_fp16 = mul(x = decoder_7_transformer_layers_0_layer_scale_2_scale_to_fp16, y = linear_42_cast_fp16)[name = tensor("op_1995_cast_fp16")]; tensor input_181_cast_fp16 = add(x = input_173_cast_fp16, y = var_1995_cast_fp16)[name = tensor("input_181_cast_fp16")]; tensor x_79_axes_0 = const()[name = tensor("x_79_axes_0"), val = tensor([-1])]; tensor decoder_7_transformer_layers_1_norm1_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_1_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(17263680)))]; tensor decoder_7_transformer_layers_1_norm1_bias_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_1_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(17264256)))]; tensor x_79_cast_fp16 = layer_norm(axes = x_79_axes_0, beta = decoder_7_transformer_layers_1_norm1_bias_to_fp16, epsilon = var_1841_to_fp16, gamma = decoder_7_transformer_layers_1_norm1_weight_to_fp16, x = input_181_cast_fp16)[name = tensor("x_79_cast_fp16")]; tensor var_2010_shape_cast_fp16 = shape(x = x_79_cast_fp16)[name = tensor("op_2010_shape_cast_fp16")]; tensor gather_103_axis_0 = const()[name = tensor("gather_103_axis_0"), val = tensor(0)]; tensor gather_103_batch_dims_0 = const()[name = tensor("gather_103_batch_dims_0"), val = tensor(0)]; tensor gather_103_validate_indices_0 = const()[name = tensor("gather_103_validate_indices_0"), val = tensor(false)]; tensor var_2010_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_2010_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_103_indices_0_to_uint16 = const()[name = tensor("gather_103_indices_0_to_uint16"), val = tensor(1)]; tensor var_2010_shape_cast_fp16_to_uint16 = cast(dtype = var_2010_shape_cast_fp16_to_uint16_dtype_0, x = var_2010_shape_cast_fp16)[name = tensor("cast_319")]; tensor gather_103_cast_uint16 = gather(axis = gather_103_axis_0, batch_dims = gather_103_batch_dims_0, indices = gather_103_indices_0_to_uint16, validate_indices = gather_103_validate_indices_0, x = var_2010_shape_cast_fp16_to_uint16)[name = tensor("gather_103_cast_uint16")]; tensor gather_103_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_103_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_7_transformer_layers_1_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_1_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(17264832)))]; tensor linear_43_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_7_transformer_layers_1_self_attn_in_proj_weight_to_fp16, x = x_79_cast_fp16)[name = tensor("linear_43_cast_fp16")]; tensor concat_116x = const()[name = tensor("concat_116x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_2017_cast_fp16 = reshape(shape = concat_116x, x = linear_43_cast_fp16)[name = tensor("op_2017_cast_fp16")]; tensor var_2018 = const()[name = tensor("op_2018"), val = tensor([2, 0, 3, 1, 4])]; tensor q_37_begin_0 = const()[name = tensor("q_37_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_37_end_0 = const()[name = tensor("q_37_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_37_end_mask_0 = const()[name = tensor("q_37_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_37_squeeze_mask_0 = const()[name = tensor("q_37_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_39_cast_fp16 = transpose(perm = var_2018, x = var_2017_cast_fp16)[name = tensor("transpose_8")]; tensor q_37_cast_fp16 = slice_by_index(begin = q_37_begin_0, end = q_37_end_0, end_mask = q_37_end_mask_0, squeeze_mask = q_37_squeeze_mask_0, x = projected_39_cast_fp16)[name = tensor("q_37_cast_fp16")]; tensor k_37_begin_0 = const()[name = tensor("k_37_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_37_end_0 = const()[name = tensor("k_37_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_37_end_mask_0 = const()[name = tensor("k_37_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_37_squeeze_mask_0 = const()[name = tensor("k_37_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_37_cast_fp16 = slice_by_index(begin = k_37_begin_0, end = k_37_end_0, end_mask = k_37_end_mask_0, squeeze_mask = k_37_squeeze_mask_0, x = projected_39_cast_fp16)[name = tensor("k_37_cast_fp16")]; tensor v_19_begin_0 = const()[name = tensor("v_19_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_19_end_0 = const()[name = tensor("v_19_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_19_end_mask_0 = const()[name = tensor("v_19_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_19_squeeze_mask_0 = const()[name = tensor("v_19_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_19_cast_fp16 = 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 = projected_39_cast_fp16)[name = tensor("v_19_cast_fp16")]; tensor var_2030_shape_cast_fp16 = shape(x = q_37_cast_fp16)[name = tensor("op_2030_shape_cast_fp16")]; tensor gather_107_axis_0 = const()[name = tensor("gather_107_axis_0"), val = tensor(0)]; tensor gather_107_batch_dims_0 = const()[name = tensor("gather_107_batch_dims_0"), val = tensor(0)]; tensor gather_107_validate_indices_0 = const()[name = tensor("gather_107_validate_indices_0"), val = tensor(false)]; tensor var_2030_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_2030_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_107_indices_0_to_uint16 = const()[name = tensor("gather_107_indices_0_to_uint16"), val = tensor(2)]; tensor var_2030_shape_cast_fp16_to_uint16 = cast(dtype = var_2030_shape_cast_fp16_to_uint16_dtype_0, x = var_2030_shape_cast_fp16)[name = tensor("cast_317")]; tensor gather_107_cast_uint16 = gather(axis = gather_107_axis_0, batch_dims = gather_107_batch_dims_0, indices = gather_107_indices_0_to_uint16, validate_indices = gather_107_validate_indices_0, x = var_2030_shape_cast_fp16_to_uint16)[name = tensor("gather_107_cast_uint16")]; tensor gather_107_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_107_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_84 = const()[name = tensor("const_84"), val = tensor(0)]; tensor const_85 = const()[name = tensor("const_85"), val = tensor(1)]; tensor gather_107_cast_uint16_to_int32 = cast(dtype = gather_107_cast_uint16_to_int32_dtype_0, x = gather_107_cast_uint16)[name = tensor("cast_316")]; tensor var_2034 = range_1d(end = gather_107_cast_uint16_to_int32, start = const_84, step = const_85)[name = tensor("op_2034")]; tensor var_2033_to_fp16 = const()[name = tensor("op_2033_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_2034_promoted_to_fp16_dtype_0 = const()[name = tensor("op_2034_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_2034_to_fp16 = cast(dtype = var_2034_promoted_to_fp16_dtype_0, x = var_2034)[name = tensor("cast_315")]; tensor ts_37_cast_fp16 = add(x = var_2033_to_fp16, y = var_2034_to_fp16)[name = tensor("ts_37_cast_fp16")]; tensor var_2037 = const()[name = tensor("op_2037"), val = tensor([1, 1, -1, 1])]; tensor ts_39_cast_fp16 = reshape(shape = var_2037, x = ts_37_cast_fp16)[name = tensor("ts_39_cast_fp16")]; tensor concat_117x = const()[name = tensor("concat_117x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_19_cast_fp16 = reshape(shape = concat_117x, x = q_37_cast_fp16)[name = tensor("q2_19_cast_fp16")]; tensor concat_118x = const()[name = tensor("concat_118x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_19_cast_fp16 = reshape(shape = concat_118x, x = k_37_cast_fp16)[name = tensor("k2_19_cast_fp16")]; tensor qr_19_begin_0 = const()[name = tensor("qr_19_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_19_end_0 = const()[name = tensor("qr_19_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_19_end_mask_0 = const()[name = tensor("qr_19_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_19_squeeze_mask_0 = const()[name = tensor("qr_19_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_19_cast_fp16 = slice_by_index(begin = qr_19_begin_0, end = qr_19_end_0, end_mask = qr_19_end_mask_0, squeeze_mask = qr_19_squeeze_mask_0, x = q2_19_cast_fp16)[name = tensor("qr_19_cast_fp16")]; tensor qi_19_begin_0 = const()[name = tensor("qi_19_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_19_end_0 = const()[name = tensor("qi_19_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_19_end_mask_0 = const()[name = tensor("qi_19_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_19_squeeze_mask_0 = const()[name = tensor("qi_19_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_19_cast_fp16 = slice_by_index(begin = qi_19_begin_0, end = qi_19_end_0, end_mask = qi_19_end_mask_0, squeeze_mask = qi_19_squeeze_mask_0, x = q2_19_cast_fp16)[name = tensor("qi_19_cast_fp16")]; tensor kr_19_begin_0 = const()[name = tensor("kr_19_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_19_end_0 = const()[name = tensor("kr_19_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_19_end_mask_0 = const()[name = tensor("kr_19_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_19_squeeze_mask_0 = const()[name = tensor("kr_19_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_19_cast_fp16 = slice_by_index(begin = kr_19_begin_0, end = kr_19_end_0, end_mask = kr_19_end_mask_0, squeeze_mask = kr_19_squeeze_mask_0, x = k2_19_cast_fp16)[name = tensor("kr_19_cast_fp16")]; tensor ki_19_begin_0 = const()[name = tensor("ki_19_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_19_end_0 = const()[name = tensor("ki_19_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_19_end_mask_0 = const()[name = tensor("ki_19_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_19_squeeze_mask_0 = const()[name = tensor("ki_19_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_19_cast_fp16 = slice_by_index(begin = ki_19_begin_0, end = ki_19_end_0, end_mask = ki_19_end_mask_0, squeeze_mask = ki_19_squeeze_mask_0, x = k2_19_cast_fp16)[name = tensor("ki_19_cast_fp16")]; tensor freqs_19_to_fp16 = const()[name = tensor("freqs_19_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(17658112)))]; tensor var_2050_cast_fp16 = mul(x = freqs_19_to_fp16, y = ts_39_cast_fp16)[name = tensor("op_2050_cast_fp16")]; tensor rotr_19_cast_fp16 = cos(x = var_2050_cast_fp16)[name = tensor("rotr_19_cast_fp16")]; tensor roti_19_cast_fp16 = sin(x = var_2050_cast_fp16)[name = tensor("roti_19_cast_fp16")]; tensor var_2054_cast_fp16 = mul(x = qr_19_cast_fp16, y = rotr_19_cast_fp16)[name = tensor("op_2054_cast_fp16")]; tensor var_2055_cast_fp16 = mul(x = qi_19_cast_fp16, y = roti_19_cast_fp16)[name = tensor("op_2055_cast_fp16")]; tensor var_2056_cast_fp16 = sub(x = var_2054_cast_fp16, y = var_2055_cast_fp16)[name = tensor("op_2056_cast_fp16")]; tensor var_2057_cast_fp16 = mul(x = qr_19_cast_fp16, y = roti_19_cast_fp16)[name = tensor("op_2057_cast_fp16")]; tensor var_2058_cast_fp16 = mul(x = qi_19_cast_fp16, y = rotr_19_cast_fp16)[name = tensor("op_2058_cast_fp16")]; tensor var_2059_cast_fp16 = add(x = var_2057_cast_fp16, y = var_2058_cast_fp16)[name = tensor("op_2059_cast_fp16")]; tensor qo_19_axis_0 = const()[name = tensor("qo_19_axis_0"), val = tensor(-1)]; tensor qo_19_cast_fp16 = stack(axis = qo_19_axis_0, values = (var_2056_cast_fp16, var_2059_cast_fp16))[name = tensor("qo_19_cast_fp16")]; tensor var_2062_cast_fp16 = mul(x = kr_19_cast_fp16, y = rotr_19_cast_fp16)[name = tensor("op_2062_cast_fp16")]; tensor var_2063_cast_fp16 = mul(x = ki_19_cast_fp16, y = roti_19_cast_fp16)[name = tensor("op_2063_cast_fp16")]; tensor var_2064_cast_fp16 = sub(x = var_2062_cast_fp16, y = var_2063_cast_fp16)[name = tensor("op_2064_cast_fp16")]; tensor var_2065_cast_fp16 = mul(x = kr_19_cast_fp16, y = roti_19_cast_fp16)[name = tensor("op_2065_cast_fp16")]; tensor var_2066_cast_fp16 = mul(x = ki_19_cast_fp16, y = rotr_19_cast_fp16)[name = tensor("op_2066_cast_fp16")]; tensor var_2067_cast_fp16 = add(x = var_2065_cast_fp16, y = var_2066_cast_fp16)[name = tensor("op_2067_cast_fp16")]; tensor ko_19_axis_0 = const()[name = tensor("ko_19_axis_0"), val = tensor(-1)]; tensor ko_19_cast_fp16 = stack(axis = ko_19_axis_0, values = (var_2064_cast_fp16, var_2067_cast_fp16))[name = tensor("ko_19_cast_fp16")]; tensor concat_119x = const()[name = tensor("concat_119x"), val = tensor([1, 4, -1, 64])]; tensor q_39_cast_fp16 = reshape(shape = concat_119x, x = qo_19_cast_fp16)[name = tensor("q_39_cast_fp16")]; tensor concat_120x = const()[name = tensor("concat_120x"), val = tensor([1, 4, -1, 64])]; tensor k_39_cast_fp16 = reshape(shape = concat_120x, x = ko_19_cast_fp16)[name = tensor("k_39_cast_fp16")]; tensor const_86 = const()[name = tensor("const_86"), val = tensor(0)]; tensor const_87 = const()[name = tensor("const_87"), val = tensor(1)]; tensor gather_103_cast_uint16_to_int32 = cast(dtype = gather_103_cast_uint16_to_int32_dtype_0, x = gather_103_cast_uint16)[name = tensor("cast_318")]; tensor positions_19 = range_1d(end = gather_103_cast_uint16_to_int32, start = const_86, step = const_87)[name = tensor("positions_19")]; tensor concat_121x = const()[name = tensor("concat_121x"), val = tensor([1, 1, -1])]; tensor var_2079 = reshape(shape = concat_121x, x = positions_19)[name = tensor("op_2079")]; tensor valid_k_19 = less(x = var_2079, y = var_1944)[name = tensor("valid_k_19")]; tensor concat_122x = const()[name = tensor("concat_122x"), val = tensor([1, -1, 1])]; tensor var_2084 = reshape(shape = concat_122x, x = positions_19)[name = tensor("op_2084")]; tensor delta_19 = sub(x = var_2084, y = var_2079)[name = tensor("delta_19")]; tensor concat_124_axis_0 = const()[name = tensor("concat_124_axis_0"), val = tensor(0)]; tensor concat_124_interleave_0 = const()[name = tensor("concat_124_interleave_0"), val = tensor(false)]; tensor concat_124 = concat(axis = concat_124_axis_0, interleave = concat_124_interleave_0, values = (var_1845, gather_103_cast_uint16_to_int32, gather_103_cast_uint16_to_int32))[name = tensor("concat_124")]; tensor fill_9_value_0_to_fp16 = const()[name = tensor("fill_9_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_9_cast_fp16 = fill(shape = concat_124, value = fill_9_value_0_to_fp16)[name = tensor("fill_9_cast_fp16")]; tensor attn_bias_73_dtype_0 = const()[name = tensor("attn_bias_73_dtype_0"), val = tensor("bool")]; tensor var_2090 = greater_equal(x = delta_19, y = var_1827)[name = tensor("op_2090")]; tensor fill_9_cast_fp16_to_bool = cast(dtype = attn_bias_73_dtype_0, x = fill_9_cast_fp16)[name = tensor("cast_314")]; tensor attn_bias_75 = logical_and(x = fill_9_cast_fp16_to_bool, y = var_2090)[name = tensor("attn_bias_75")]; tensor var_2092 = less(x = delta_19, y = var_1834)[name = tensor("op_2092")]; tensor attn_bias_77 = logical_and(x = attn_bias_75, y = var_2092)[name = tensor("attn_bias_77")]; tensor var_2094 = logical_and(x = attn_bias_77, y = valid_k_19)[name = tensor("op_2094")]; tensor var_2096_axes_0 = const()[name = tensor("op_2096_axes_0"), val = tensor([1])]; tensor var_2096 = expand_dims(axes = var_2096_axes_0, x = var_2094)[name = tensor("op_2096")]; tensor sub_9_x_0_to_fp16 = const()[name = tensor("sub_9_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_239_to_fp16_dtype_0 = const()[name = tensor("cast_239_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_2096_to_fp16 = cast(dtype = cast_239_to_fp16_dtype_0, x = var_2096)[name = tensor("cast_313")]; tensor sub_9_cast_fp16 = sub(x = sub_9_x_0_to_fp16, y = var_2096_to_fp16)[name = tensor("sub_9_cast_fp16")]; tensor mul_18_x_0_to_fp16 = const()[name = tensor("mul_18_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_18_cast_fp16 = mul(x = mul_18_x_0_to_fp16, y = sub_9_cast_fp16)[name = tensor("mul_18_cast_fp16")]; tensor mul_19_y_0_to_fp16 = const()[name = tensor("mul_19_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_19_cast_fp16 = mul(x = q_39_cast_fp16, y = mul_19_y_0_to_fp16)[name = tensor("mul_19_cast_fp16")]; tensor matmul_9_transpose_y_0 = const()[name = tensor("matmul_9_transpose_y_0"), val = tensor(true)]; tensor matmul_9_transpose_x_0 = const()[name = tensor("matmul_9_transpose_x_0"), val = tensor(false)]; tensor matmul_9_cast_fp16 = matmul(transpose_x = matmul_9_transpose_x_0, transpose_y = matmul_9_transpose_y_0, x = mul_19_cast_fp16, y = k_39_cast_fp16)[name = tensor("matmul_9_cast_fp16")]; tensor add_9_cast_fp16 = add(x = matmul_9_cast_fp16, y = mul_18_cast_fp16)[name = tensor("add_9_cast_fp16")]; tensor add_9_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_9_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_9_axis_0 = const()[name = tensor("softmax_9_axis_0"), val = tensor(-1)]; tensor add_9_cast_fp16_to_fp32 = cast(dtype = add_9_cast_fp16_to_fp32_dtype_0, x = add_9_cast_fp16)[name = tensor("cast_312")]; tensor softmax_9 = softmax(axis = softmax_9_axis_0, x = add_9_cast_fp16_to_fp32)[name = tensor("softmax_9")]; tensor out_37_transpose_x_0 = const()[name = tensor("out_37_transpose_x_0"), val = tensor(false)]; tensor out_37_transpose_y_0 = const()[name = tensor("out_37_transpose_y_0"), val = tensor(false)]; tensor softmax_9_to_fp16_dtype_0 = const()[name = tensor("softmax_9_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_9_to_fp16 = cast(dtype = softmax_9_to_fp16_dtype_0, x = softmax_9)[name = tensor("cast_311")]; tensor out_37_cast_fp16 = matmul(transpose_x = out_37_transpose_x_0, transpose_y = out_37_transpose_y_0, x = softmax_9_to_fp16, y = v_19_cast_fp16)[name = tensor("out_37_cast_fp16")]; tensor concat_125x = const()[name = tensor("concat_125x"), val = tensor([1, -1])]; tensor var_2102 = reshape(shape = concat_125x, x = positions_19)[name = tensor("op_2102")]; tensor var_2105 = less(x = var_2102, y = var_1967)[name = tensor("op_2105")]; tensor concat_126x = const()[name = tensor("concat_126x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_19 = reshape(shape = concat_126x, x = var_2105)[name = tensor("valid_q_19")]; tensor var_2109_to_fp16 = const()[name = tensor("op_2109_to_fp16"), val = tensor(0x0p+0)]; tensor out_39_cast_fp16 = select(a = out_37_cast_fp16, b = var_2109_to_fp16, cond = valid_q_19)[name = tensor("out_39_cast_fp16")]; tensor var_2111_perm_0 = const()[name = tensor("op_2111_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_127x = const()[name = tensor("concat_127x"), val = tensor([1, -1, 256])]; tensor var_2111_cast_fp16 = transpose(perm = var_2111_perm_0, x = out_39_cast_fp16)[name = tensor("transpose_7")]; tensor input_183_cast_fp16 = reshape(shape = concat_127x, x = var_2111_cast_fp16)[name = tensor("input_183_cast_fp16")]; tensor decoder_7_transformer_layers_1_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_1_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(17658240)))]; tensor linear_44_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_7_transformer_layers_1_self_attn_out_proj_weight_to_fp16, x = input_183_cast_fp16)[name = tensor("linear_44_cast_fp16")]; tensor decoder_7_transformer_layers_1_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_1_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(17789376)))]; tensor var_2117_cast_fp16 = mul(x = decoder_7_transformer_layers_1_layer_scale_1_scale_to_fp16, y = linear_44_cast_fp16)[name = tensor("op_2117_cast_fp16")]; tensor input_185_cast_fp16 = add(x = input_181_cast_fp16, y = var_2117_cast_fp16)[name = tensor("input_185_cast_fp16")]; tensor input_187_axes_0 = const()[name = tensor("input_187_axes_0"), val = tensor([-1])]; tensor decoder_7_transformer_layers_1_norm2_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_1_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(17789952)))]; tensor decoder_7_transformer_layers_1_norm2_bias_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_1_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(17790528)))]; tensor input_187_cast_fp16 = layer_norm(axes = input_187_axes_0, beta = decoder_7_transformer_layers_1_norm2_bias_to_fp16, epsilon = var_1841_to_fp16, gamma = decoder_7_transformer_layers_1_norm2_weight_to_fp16, x = input_185_cast_fp16)[name = tensor("input_187_cast_fp16")]; tensor decoder_7_transformer_layers_1_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_1_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(17791104)))]; tensor linear_45_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_7_transformer_layers_1_ffn_0_weight_to_fp16, x = input_187_cast_fp16)[name = tensor("linear_45_cast_fp16")]; tensor input_191_mode_0 = const()[name = tensor("input_191_mode_0"), val = tensor("EXACT")]; tensor input_191_cast_fp16 = gelu(mode = input_191_mode_0, x = linear_45_cast_fp16)[name = tensor("input_191_cast_fp16")]; tensor decoder_7_transformer_layers_1_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_1_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18315456)))]; tensor linear_46_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_7_transformer_layers_1_ffn_2_weight_to_fp16, x = input_191_cast_fp16)[name = tensor("linear_46_cast_fp16")]; tensor decoder_7_transformer_layers_1_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_1_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18839808)))]; tensor var_2132_cast_fp16 = mul(x = decoder_7_transformer_layers_1_layer_scale_2_scale_to_fp16, y = linear_46_cast_fp16)[name = tensor("op_2132_cast_fp16")]; tensor input_193_cast_fp16 = add(x = input_185_cast_fp16, y = var_2132_cast_fp16)[name = tensor("input_193_cast_fp16")]; tensor x_85_axes_0 = const()[name = tensor("x_85_axes_0"), val = tensor([-1])]; tensor decoder_7_transformer_layers_2_norm1_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_2_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18840384)))]; tensor decoder_7_transformer_layers_2_norm1_bias_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_2_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18840960)))]; tensor x_85_cast_fp16 = layer_norm(axes = x_85_axes_0, beta = decoder_7_transformer_layers_2_norm1_bias_to_fp16, epsilon = var_1841_to_fp16, gamma = decoder_7_transformer_layers_2_norm1_weight_to_fp16, x = input_193_cast_fp16)[name = tensor("x_85_cast_fp16")]; tensor var_2147_shape_cast_fp16 = shape(x = x_85_cast_fp16)[name = tensor("op_2147_shape_cast_fp16")]; tensor gather_113_axis_0 = const()[name = tensor("gather_113_axis_0"), val = tensor(0)]; tensor gather_113_batch_dims_0 = const()[name = tensor("gather_113_batch_dims_0"), val = tensor(0)]; tensor gather_113_validate_indices_0 = const()[name = tensor("gather_113_validate_indices_0"), val = tensor(false)]; tensor var_2147_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_2147_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_113_indices_0_to_uint16 = const()[name = tensor("gather_113_indices_0_to_uint16"), val = tensor(1)]; tensor var_2147_shape_cast_fp16_to_uint16 = cast(dtype = var_2147_shape_cast_fp16_to_uint16_dtype_0, x = var_2147_shape_cast_fp16)[name = tensor("cast_310")]; tensor gather_113_cast_uint16 = gather(axis = gather_113_axis_0, batch_dims = gather_113_batch_dims_0, indices = gather_113_indices_0_to_uint16, validate_indices = gather_113_validate_indices_0, x = var_2147_shape_cast_fp16_to_uint16)[name = tensor("gather_113_cast_uint16")]; tensor gather_113_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_113_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_7_transformer_layers_2_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_2_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(18841536)))]; tensor linear_47_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_7_transformer_layers_2_self_attn_in_proj_weight_to_fp16, x = x_85_cast_fp16)[name = tensor("linear_47_cast_fp16")]; tensor concat_128x = const()[name = tensor("concat_128x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_2154_cast_fp16 = reshape(shape = concat_128x, x = linear_47_cast_fp16)[name = tensor("op_2154_cast_fp16")]; tensor var_2155 = const()[name = tensor("op_2155"), val = tensor([2, 0, 3, 1, 4])]; tensor q_41_begin_0 = const()[name = tensor("q_41_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_41_end_0 = const()[name = tensor("q_41_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_41_end_mask_0 = const()[name = tensor("q_41_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_41_squeeze_mask_0 = const()[name = tensor("q_41_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_43_cast_fp16 = transpose(perm = var_2155, x = var_2154_cast_fp16)[name = tensor("transpose_6")]; tensor q_41_cast_fp16 = slice_by_index(begin = q_41_begin_0, end = q_41_end_0, end_mask = q_41_end_mask_0, squeeze_mask = q_41_squeeze_mask_0, x = projected_43_cast_fp16)[name = tensor("q_41_cast_fp16")]; tensor k_41_begin_0 = const()[name = tensor("k_41_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_41_end_0 = const()[name = tensor("k_41_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_41_end_mask_0 = const()[name = tensor("k_41_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_41_squeeze_mask_0 = const()[name = tensor("k_41_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_41_cast_fp16 = slice_by_index(begin = k_41_begin_0, end = k_41_end_0, end_mask = k_41_end_mask_0, squeeze_mask = k_41_squeeze_mask_0, x = projected_43_cast_fp16)[name = tensor("k_41_cast_fp16")]; tensor v_21_begin_0 = const()[name = tensor("v_21_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_21_end_0 = const()[name = tensor("v_21_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_21_end_mask_0 = const()[name = tensor("v_21_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_21_squeeze_mask_0 = const()[name = tensor("v_21_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_21_cast_fp16 = slice_by_index(begin = v_21_begin_0, end = v_21_end_0, end_mask = v_21_end_mask_0, squeeze_mask = v_21_squeeze_mask_0, x = projected_43_cast_fp16)[name = tensor("v_21_cast_fp16")]; tensor var_2167_shape_cast_fp16 = shape(x = q_41_cast_fp16)[name = tensor("op_2167_shape_cast_fp16")]; tensor gather_117_axis_0 = const()[name = tensor("gather_117_axis_0"), val = tensor(0)]; tensor gather_117_batch_dims_0 = const()[name = tensor("gather_117_batch_dims_0"), val = tensor(0)]; tensor gather_117_validate_indices_0 = const()[name = tensor("gather_117_validate_indices_0"), val = tensor(false)]; tensor var_2167_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_2167_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_117_indices_0_to_uint16 = const()[name = tensor("gather_117_indices_0_to_uint16"), val = tensor(2)]; tensor var_2167_shape_cast_fp16_to_uint16 = cast(dtype = var_2167_shape_cast_fp16_to_uint16_dtype_0, x = var_2167_shape_cast_fp16)[name = tensor("cast_308")]; tensor gather_117_cast_uint16 = gather(axis = gather_117_axis_0, batch_dims = gather_117_batch_dims_0, indices = gather_117_indices_0_to_uint16, validate_indices = gather_117_validate_indices_0, x = var_2167_shape_cast_fp16_to_uint16)[name = tensor("gather_117_cast_uint16")]; tensor gather_117_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_117_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_93 = const()[name = tensor("const_93"), val = tensor(0)]; tensor const_94 = const()[name = tensor("const_94"), val = tensor(1)]; tensor gather_117_cast_uint16_to_int32 = cast(dtype = gather_117_cast_uint16_to_int32_dtype_0, x = gather_117_cast_uint16)[name = tensor("cast_307")]; tensor var_2171 = range_1d(end = gather_117_cast_uint16_to_int32, start = const_93, step = const_94)[name = tensor("op_2171")]; tensor var_2170_to_fp16 = const()[name = tensor("op_2170_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_2171_promoted_to_fp16_dtype_0 = const()[name = tensor("op_2171_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_2171_to_fp16 = cast(dtype = var_2171_promoted_to_fp16_dtype_0, x = var_2171)[name = tensor("cast_306")]; tensor ts_41_cast_fp16 = add(x = var_2170_to_fp16, y = var_2171_to_fp16)[name = tensor("ts_41_cast_fp16")]; tensor var_2174 = const()[name = tensor("op_2174"), val = tensor([1, 1, -1, 1])]; tensor ts_43_cast_fp16 = reshape(shape = var_2174, x = ts_41_cast_fp16)[name = tensor("ts_43_cast_fp16")]; tensor concat_129x = const()[name = tensor("concat_129x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_21_cast_fp16 = reshape(shape = concat_129x, x = q_41_cast_fp16)[name = tensor("q2_21_cast_fp16")]; tensor concat_130x = const()[name = tensor("concat_130x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_21_cast_fp16 = reshape(shape = concat_130x, x = k_41_cast_fp16)[name = tensor("k2_21_cast_fp16")]; tensor qr_21_begin_0 = const()[name = tensor("qr_21_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_21_end_0 = const()[name = tensor("qr_21_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_21_end_mask_0 = const()[name = tensor("qr_21_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_21_squeeze_mask_0 = const()[name = tensor("qr_21_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_21_cast_fp16 = slice_by_index(begin = qr_21_begin_0, end = qr_21_end_0, end_mask = qr_21_end_mask_0, squeeze_mask = qr_21_squeeze_mask_0, x = q2_21_cast_fp16)[name = tensor("qr_21_cast_fp16")]; tensor qi_21_begin_0 = const()[name = tensor("qi_21_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_21_end_0 = const()[name = tensor("qi_21_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_21_end_mask_0 = const()[name = tensor("qi_21_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_21_squeeze_mask_0 = const()[name = tensor("qi_21_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_21_cast_fp16 = slice_by_index(begin = qi_21_begin_0, end = qi_21_end_0, end_mask = qi_21_end_mask_0, squeeze_mask = qi_21_squeeze_mask_0, x = q2_21_cast_fp16)[name = tensor("qi_21_cast_fp16")]; tensor kr_21_begin_0 = const()[name = tensor("kr_21_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_21_end_0 = const()[name = tensor("kr_21_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_21_end_mask_0 = const()[name = tensor("kr_21_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_21_squeeze_mask_0 = const()[name = tensor("kr_21_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_21_cast_fp16 = slice_by_index(begin = kr_21_begin_0, end = kr_21_end_0, end_mask = kr_21_end_mask_0, squeeze_mask = kr_21_squeeze_mask_0, x = k2_21_cast_fp16)[name = tensor("kr_21_cast_fp16")]; tensor ki_21_begin_0 = const()[name = tensor("ki_21_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_21_end_0 = const()[name = tensor("ki_21_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_21_end_mask_0 = const()[name = tensor("ki_21_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_21_squeeze_mask_0 = const()[name = tensor("ki_21_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_21_cast_fp16 = slice_by_index(begin = ki_21_begin_0, end = ki_21_end_0, end_mask = ki_21_end_mask_0, squeeze_mask = ki_21_squeeze_mask_0, x = k2_21_cast_fp16)[name = tensor("ki_21_cast_fp16")]; tensor freqs_21_to_fp16 = const()[name = tensor("freqs_21_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(19234816)))]; tensor var_2187_cast_fp16 = mul(x = freqs_21_to_fp16, y = ts_43_cast_fp16)[name = tensor("op_2187_cast_fp16")]; tensor rotr_21_cast_fp16 = cos(x = var_2187_cast_fp16)[name = tensor("rotr_21_cast_fp16")]; tensor roti_21_cast_fp16 = sin(x = var_2187_cast_fp16)[name = tensor("roti_21_cast_fp16")]; tensor var_2191_cast_fp16 = mul(x = qr_21_cast_fp16, y = rotr_21_cast_fp16)[name = tensor("op_2191_cast_fp16")]; tensor var_2192_cast_fp16 = mul(x = qi_21_cast_fp16, y = roti_21_cast_fp16)[name = tensor("op_2192_cast_fp16")]; tensor var_2193_cast_fp16 = sub(x = var_2191_cast_fp16, y = var_2192_cast_fp16)[name = tensor("op_2193_cast_fp16")]; tensor var_2194_cast_fp16 = mul(x = qr_21_cast_fp16, y = roti_21_cast_fp16)[name = tensor("op_2194_cast_fp16")]; tensor var_2195_cast_fp16 = mul(x = qi_21_cast_fp16, y = rotr_21_cast_fp16)[name = tensor("op_2195_cast_fp16")]; tensor var_2196_cast_fp16 = add(x = var_2194_cast_fp16, y = var_2195_cast_fp16)[name = tensor("op_2196_cast_fp16")]; tensor qo_21_axis_0 = const()[name = tensor("qo_21_axis_0"), val = tensor(-1)]; tensor qo_21_cast_fp16 = stack(axis = qo_21_axis_0, values = (var_2193_cast_fp16, var_2196_cast_fp16))[name = tensor("qo_21_cast_fp16")]; tensor var_2199_cast_fp16 = mul(x = kr_21_cast_fp16, y = rotr_21_cast_fp16)[name = tensor("op_2199_cast_fp16")]; tensor var_2200_cast_fp16 = mul(x = ki_21_cast_fp16, y = roti_21_cast_fp16)[name = tensor("op_2200_cast_fp16")]; tensor var_2201_cast_fp16 = sub(x = var_2199_cast_fp16, y = var_2200_cast_fp16)[name = tensor("op_2201_cast_fp16")]; tensor var_2202_cast_fp16 = mul(x = kr_21_cast_fp16, y = roti_21_cast_fp16)[name = tensor("op_2202_cast_fp16")]; tensor var_2203_cast_fp16 = mul(x = ki_21_cast_fp16, y = rotr_21_cast_fp16)[name = tensor("op_2203_cast_fp16")]; tensor var_2204_cast_fp16 = add(x = var_2202_cast_fp16, y = var_2203_cast_fp16)[name = tensor("op_2204_cast_fp16")]; tensor ko_21_axis_0 = const()[name = tensor("ko_21_axis_0"), val = tensor(-1)]; tensor ko_21_cast_fp16 = stack(axis = ko_21_axis_0, values = (var_2201_cast_fp16, var_2204_cast_fp16))[name = tensor("ko_21_cast_fp16")]; tensor concat_131x = const()[name = tensor("concat_131x"), val = tensor([1, 4, -1, 64])]; tensor q_43_cast_fp16 = reshape(shape = concat_131x, x = qo_21_cast_fp16)[name = tensor("q_43_cast_fp16")]; tensor concat_132x = const()[name = tensor("concat_132x"), val = tensor([1, 4, -1, 64])]; tensor k_43_cast_fp16 = reshape(shape = concat_132x, x = ko_21_cast_fp16)[name = tensor("k_43_cast_fp16")]; tensor const_95 = const()[name = tensor("const_95"), val = tensor(0)]; tensor const_96 = const()[name = tensor("const_96"), val = tensor(1)]; tensor gather_113_cast_uint16_to_int32 = cast(dtype = gather_113_cast_uint16_to_int32_dtype_0, x = gather_113_cast_uint16)[name = tensor("cast_309")]; tensor positions_21 = range_1d(end = gather_113_cast_uint16_to_int32, start = const_95, step = const_96)[name = tensor("positions_21")]; tensor concat_133x = const()[name = tensor("concat_133x"), val = tensor([1, 1, -1])]; tensor var_2216 = reshape(shape = concat_133x, x = positions_21)[name = tensor("op_2216")]; tensor valid_k_21 = less(x = var_2216, y = var_1944)[name = tensor("valid_k_21")]; tensor concat_134x = const()[name = tensor("concat_134x"), val = tensor([1, -1, 1])]; tensor var_2221 = reshape(shape = concat_134x, x = positions_21)[name = tensor("op_2221")]; tensor delta_21 = sub(x = var_2221, y = var_2216)[name = tensor("delta_21")]; tensor concat_136_axis_0 = const()[name = tensor("concat_136_axis_0"), val = tensor(0)]; tensor concat_136_interleave_0 = const()[name = tensor("concat_136_interleave_0"), val = tensor(false)]; tensor concat_136 = concat(axis = concat_136_axis_0, interleave = concat_136_interleave_0, values = (var_1845, gather_113_cast_uint16_to_int32, gather_113_cast_uint16_to_int32))[name = tensor("concat_136")]; tensor fill_10_value_0_to_fp16 = const()[name = tensor("fill_10_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_10_cast_fp16 = fill(shape = concat_136, value = fill_10_value_0_to_fp16)[name = tensor("fill_10_cast_fp16")]; tensor attn_bias_81_dtype_0 = const()[name = tensor("attn_bias_81_dtype_0"), val = tensor("bool")]; tensor var_2227 = greater_equal(x = delta_21, y = var_1827)[name = tensor("op_2227")]; tensor fill_10_cast_fp16_to_bool = cast(dtype = attn_bias_81_dtype_0, x = fill_10_cast_fp16)[name = tensor("cast_305")]; tensor attn_bias_83 = logical_and(x = fill_10_cast_fp16_to_bool, y = var_2227)[name = tensor("attn_bias_83")]; tensor var_2229 = less(x = delta_21, y = var_1834)[name = tensor("op_2229")]; tensor attn_bias_85 = logical_and(x = attn_bias_83, y = var_2229)[name = tensor("attn_bias_85")]; tensor var_2231 = logical_and(x = attn_bias_85, y = valid_k_21)[name = tensor("op_2231")]; tensor var_2233_axes_0 = const()[name = tensor("op_2233_axes_0"), val = tensor([1])]; tensor var_2233 = expand_dims(axes = var_2233_axes_0, x = var_2231)[name = tensor("op_2233")]; tensor sub_10_x_0_to_fp16 = const()[name = tensor("sub_10_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_261_to_fp16_dtype_0 = const()[name = tensor("cast_261_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_2233_to_fp16 = cast(dtype = cast_261_to_fp16_dtype_0, x = var_2233)[name = tensor("cast_304")]; tensor sub_10_cast_fp16 = sub(x = sub_10_x_0_to_fp16, y = var_2233_to_fp16)[name = tensor("sub_10_cast_fp16")]; tensor mul_20_x_0_to_fp16 = const()[name = tensor("mul_20_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_20_cast_fp16 = mul(x = mul_20_x_0_to_fp16, y = sub_10_cast_fp16)[name = tensor("mul_20_cast_fp16")]; tensor mul_21_y_0_to_fp16 = const()[name = tensor("mul_21_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_21_cast_fp16 = mul(x = q_43_cast_fp16, y = mul_21_y_0_to_fp16)[name = tensor("mul_21_cast_fp16")]; tensor matmul_10_transpose_y_0 = const()[name = tensor("matmul_10_transpose_y_0"), val = tensor(true)]; tensor matmul_10_transpose_x_0 = const()[name = tensor("matmul_10_transpose_x_0"), val = tensor(false)]; tensor matmul_10_cast_fp16 = matmul(transpose_x = matmul_10_transpose_x_0, transpose_y = matmul_10_transpose_y_0, x = mul_21_cast_fp16, y = k_43_cast_fp16)[name = tensor("matmul_10_cast_fp16")]; tensor add_10_cast_fp16 = add(x = matmul_10_cast_fp16, y = mul_20_cast_fp16)[name = tensor("add_10_cast_fp16")]; tensor add_10_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_10_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_10_axis_0 = const()[name = tensor("softmax_10_axis_0"), val = tensor(-1)]; tensor add_10_cast_fp16_to_fp32 = cast(dtype = add_10_cast_fp16_to_fp32_dtype_0, x = add_10_cast_fp16)[name = tensor("cast_303")]; tensor softmax_10 = softmax(axis = softmax_10_axis_0, x = add_10_cast_fp16_to_fp32)[name = tensor("softmax_10")]; tensor out_41_transpose_x_0 = const()[name = tensor("out_41_transpose_x_0"), val = tensor(false)]; tensor out_41_transpose_y_0 = const()[name = tensor("out_41_transpose_y_0"), val = tensor(false)]; tensor softmax_10_to_fp16_dtype_0 = const()[name = tensor("softmax_10_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_10_to_fp16 = cast(dtype = softmax_10_to_fp16_dtype_0, x = softmax_10)[name = tensor("cast_302")]; tensor out_41_cast_fp16 = matmul(transpose_x = out_41_transpose_x_0, transpose_y = out_41_transpose_y_0, x = softmax_10_to_fp16, y = v_21_cast_fp16)[name = tensor("out_41_cast_fp16")]; tensor concat_137x = const()[name = tensor("concat_137x"), val = tensor([1, -1])]; tensor var_2239 = reshape(shape = concat_137x, x = positions_21)[name = tensor("op_2239")]; tensor var_2242 = less(x = var_2239, y = var_1967)[name = tensor("op_2242")]; tensor concat_138x = const()[name = tensor("concat_138x"), val = tensor([1, 1, -1, 1])]; tensor valid_q_21 = reshape(shape = concat_138x, x = var_2242)[name = tensor("valid_q_21")]; tensor var_2246_to_fp16 = const()[name = tensor("op_2246_to_fp16"), val = tensor(0x0p+0)]; tensor out_43_cast_fp16 = select(a = out_41_cast_fp16, b = var_2246_to_fp16, cond = valid_q_21)[name = tensor("out_43_cast_fp16")]; tensor var_2248_perm_0 = const()[name = tensor("op_2248_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_139x = const()[name = tensor("concat_139x"), val = tensor([1, -1, 256])]; tensor var_2248_cast_fp16 = transpose(perm = var_2248_perm_0, x = out_43_cast_fp16)[name = tensor("transpose_5")]; tensor input_195_cast_fp16 = reshape(shape = concat_139x, x = var_2248_cast_fp16)[name = tensor("input_195_cast_fp16")]; tensor decoder_7_transformer_layers_2_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_2_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(19234944)))]; tensor linear_48_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_7_transformer_layers_2_self_attn_out_proj_weight_to_fp16, x = input_195_cast_fp16)[name = tensor("linear_48_cast_fp16")]; tensor decoder_7_transformer_layers_2_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_2_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(19366080)))]; tensor var_2254_cast_fp16 = mul(x = decoder_7_transformer_layers_2_layer_scale_1_scale_to_fp16, y = linear_48_cast_fp16)[name = tensor("op_2254_cast_fp16")]; tensor input_197_cast_fp16 = add(x = input_193_cast_fp16, y = var_2254_cast_fp16)[name = tensor("input_197_cast_fp16")]; tensor input_199_axes_0 = const()[name = tensor("input_199_axes_0"), val = tensor([-1])]; tensor decoder_7_transformer_layers_2_norm2_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_2_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(19366656)))]; tensor decoder_7_transformer_layers_2_norm2_bias_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_2_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(19367232)))]; tensor input_199_cast_fp16 = layer_norm(axes = input_199_axes_0, beta = decoder_7_transformer_layers_2_norm2_bias_to_fp16, epsilon = var_1841_to_fp16, gamma = decoder_7_transformer_layers_2_norm2_weight_to_fp16, x = input_197_cast_fp16)[name = tensor("input_199_cast_fp16")]; tensor decoder_7_transformer_layers_2_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_2_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(19367808)))]; tensor linear_49_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_7_transformer_layers_2_ffn_0_weight_to_fp16, x = input_199_cast_fp16)[name = tensor("linear_49_cast_fp16")]; tensor input_203_mode_0 = const()[name = tensor("input_203_mode_0"), val = tensor("EXACT")]; tensor input_203_cast_fp16 = gelu(mode = input_203_mode_0, x = linear_49_cast_fp16)[name = tensor("input_203_cast_fp16")]; tensor decoder_7_transformer_layers_2_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_2_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(19892160)))]; tensor linear_50_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_7_transformer_layers_2_ffn_2_weight_to_fp16, x = input_203_cast_fp16)[name = tensor("linear_50_cast_fp16")]; tensor decoder_7_transformer_layers_2_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_2_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20416512)))]; tensor var_2269_cast_fp16 = mul(x = decoder_7_transformer_layers_2_layer_scale_2_scale_to_fp16, y = linear_50_cast_fp16)[name = tensor("op_2269_cast_fp16")]; tensor input_205_cast_fp16 = add(x = input_197_cast_fp16, y = var_2269_cast_fp16)[name = tensor("input_205_cast_fp16")]; tensor x_91_axes_0 = const()[name = tensor("x_91_axes_0"), val = tensor([-1])]; tensor decoder_7_transformer_layers_3_norm1_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_3_norm1_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20417088)))]; tensor decoder_7_transformer_layers_3_norm1_bias_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_3_norm1_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20417664)))]; tensor x_91_cast_fp16 = layer_norm(axes = x_91_axes_0, beta = decoder_7_transformer_layers_3_norm1_bias_to_fp16, epsilon = var_1841_to_fp16, gamma = decoder_7_transformer_layers_3_norm1_weight_to_fp16, x = input_205_cast_fp16)[name = tensor("x_91_cast_fp16")]; tensor var_2284_shape_cast_fp16 = shape(x = x_91_cast_fp16)[name = tensor("op_2284_shape_cast_fp16")]; tensor gather_123_axis_0 = const()[name = tensor("gather_123_axis_0"), val = tensor(0)]; tensor gather_123_batch_dims_0 = const()[name = tensor("gather_123_batch_dims_0"), val = tensor(0)]; tensor gather_123_validate_indices_0 = const()[name = tensor("gather_123_validate_indices_0"), val = tensor(false)]; tensor var_2284_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_2284_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_123_indices_0_to_uint16 = const()[name = tensor("gather_123_indices_0_to_uint16"), val = tensor(1)]; tensor var_2284_shape_cast_fp16_to_uint16 = cast(dtype = var_2284_shape_cast_fp16_to_uint16_dtype_0, x = var_2284_shape_cast_fp16)[name = tensor("cast_301")]; tensor gather_123_cast_uint16 = gather(axis = gather_123_axis_0, batch_dims = gather_123_batch_dims_0, indices = gather_123_indices_0_to_uint16, validate_indices = gather_123_validate_indices_0, x = var_2284_shape_cast_fp16_to_uint16)[name = tensor("gather_123_cast_uint16")]; tensor gather_123_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_123_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor decoder_7_transformer_layers_3_self_attn_in_proj_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_3_self_attn_in_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20418240)))]; tensor linear_51_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = decoder_7_transformer_layers_3_self_attn_in_proj_weight_to_fp16, x = x_91_cast_fp16)[name = tensor("linear_51_cast_fp16")]; tensor concat_140x = const()[name = tensor("concat_140x"), val = tensor([1, -1, 3, 4, 64])]; tensor var_2291_cast_fp16 = reshape(shape = concat_140x, x = linear_51_cast_fp16)[name = tensor("op_2291_cast_fp16")]; tensor var_2292 = const()[name = tensor("op_2292"), val = tensor([2, 0, 3, 1, 4])]; tensor q_45_begin_0 = const()[name = tensor("q_45_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor q_45_end_0 = const()[name = tensor("q_45_end_0"), val = tensor([1, 1, 4, 0, 64])]; tensor q_45_end_mask_0 = const()[name = tensor("q_45_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor q_45_squeeze_mask_0 = const()[name = tensor("q_45_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor projected_cast_fp16 = transpose(perm = var_2292, x = var_2291_cast_fp16)[name = tensor("transpose_4")]; tensor q_45_cast_fp16 = slice_by_index(begin = q_45_begin_0, end = q_45_end_0, end_mask = q_45_end_mask_0, squeeze_mask = q_45_squeeze_mask_0, x = projected_cast_fp16)[name = tensor("q_45_cast_fp16")]; tensor k_45_begin_0 = const()[name = tensor("k_45_begin_0"), val = tensor([1, 0, 0, 0, 0])]; tensor k_45_end_0 = const()[name = tensor("k_45_end_0"), val = tensor([2, 1, 4, 0, 64])]; tensor k_45_end_mask_0 = const()[name = tensor("k_45_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor k_45_squeeze_mask_0 = const()[name = tensor("k_45_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor k_45_cast_fp16 = slice_by_index(begin = k_45_begin_0, end = k_45_end_0, end_mask = k_45_end_mask_0, squeeze_mask = k_45_squeeze_mask_0, x = projected_cast_fp16)[name = tensor("k_45_cast_fp16")]; tensor v_begin_0 = const()[name = tensor("v_begin_0"), val = tensor([2, 0, 0, 0, 0])]; tensor v_end_0 = const()[name = tensor("v_end_0"), val = tensor([3, 1, 4, 0, 64])]; tensor v_end_mask_0 = const()[name = tensor("v_end_mask_0"), val = tensor([false, true, true, true, true])]; tensor v_squeeze_mask_0 = const()[name = tensor("v_squeeze_mask_0"), val = tensor([true, false, false, false, false])]; tensor v_cast_fp16 = slice_by_index(begin = v_begin_0, end = v_end_0, end_mask = v_end_mask_0, squeeze_mask = v_squeeze_mask_0, x = projected_cast_fp16)[name = tensor("v_cast_fp16")]; tensor var_2304_shape_cast_fp16 = shape(x = q_45_cast_fp16)[name = tensor("op_2304_shape_cast_fp16")]; tensor gather_127_axis_0 = const()[name = tensor("gather_127_axis_0"), val = tensor(0)]; tensor gather_127_batch_dims_0 = const()[name = tensor("gather_127_batch_dims_0"), val = tensor(0)]; tensor gather_127_validate_indices_0 = const()[name = tensor("gather_127_validate_indices_0"), val = tensor(false)]; tensor var_2304_shape_cast_fp16_to_uint16_dtype_0 = const()[name = tensor("op_2304_shape_cast_fp16_to_uint16_dtype_0"), val = tensor("uint16")]; tensor gather_127_indices_0_to_uint16 = const()[name = tensor("gather_127_indices_0_to_uint16"), val = tensor(2)]; tensor var_2304_shape_cast_fp16_to_uint16 = cast(dtype = var_2304_shape_cast_fp16_to_uint16_dtype_0, x = var_2304_shape_cast_fp16)[name = tensor("cast_299")]; tensor gather_127_cast_uint16 = gather(axis = gather_127_axis_0, batch_dims = gather_127_batch_dims_0, indices = gather_127_indices_0_to_uint16, validate_indices = gather_127_validate_indices_0, x = var_2304_shape_cast_fp16_to_uint16)[name = tensor("gather_127_cast_uint16")]; tensor gather_127_cast_uint16_to_int32_dtype_0 = const()[name = tensor("gather_127_cast_uint16_to_int32_dtype_0"), val = tensor("int32")]; tensor const_102 = const()[name = tensor("const_102"), val = tensor(0)]; tensor const_103 = const()[name = tensor("const_103"), val = tensor(1)]; tensor gather_127_cast_uint16_to_int32 = cast(dtype = gather_127_cast_uint16_to_int32_dtype_0, x = gather_127_cast_uint16)[name = tensor("cast_298")]; tensor var_2308 = range_1d(end = gather_127_cast_uint16_to_int32, start = const_102, step = const_103)[name = tensor("op_2308")]; tensor var_2307_to_fp16 = const()[name = tensor("op_2307_to_fp16"), val = tensor([[0x0p+0]])]; tensor var_2308_promoted_to_fp16_dtype_0 = const()[name = tensor("op_2308_promoted_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_2308_to_fp16 = cast(dtype = var_2308_promoted_to_fp16_dtype_0, x = var_2308)[name = tensor("cast_297")]; tensor ts_45_cast_fp16 = add(x = var_2307_to_fp16, y = var_2308_to_fp16)[name = tensor("ts_45_cast_fp16")]; tensor var_2311 = const()[name = tensor("op_2311"), val = tensor([1, 1, -1, 1])]; tensor ts_cast_fp16 = reshape(shape = var_2311, x = ts_45_cast_fp16)[name = tensor("ts_cast_fp16")]; tensor concat_141x = const()[name = tensor("concat_141x"), val = tensor([1, 4, -1, 32, 2])]; tensor q2_cast_fp16 = reshape(shape = concat_141x, x = q_45_cast_fp16)[name = tensor("q2_cast_fp16")]; tensor concat_142x = const()[name = tensor("concat_142x"), val = tensor([1, 4, -1, 32, 2])]; tensor k2_cast_fp16 = reshape(shape = concat_142x, x = k_45_cast_fp16)[name = tensor("k2_cast_fp16")]; tensor qr_begin_0 = const()[name = tensor("qr_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor qr_end_0 = const()[name = tensor("qr_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor qr_end_mask_0 = const()[name = tensor("qr_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qr_squeeze_mask_0 = const()[name = tensor("qr_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qr_cast_fp16 = slice_by_index(begin = qr_begin_0, end = qr_end_0, end_mask = qr_end_mask_0, squeeze_mask = qr_squeeze_mask_0, x = q2_cast_fp16)[name = tensor("qr_cast_fp16")]; tensor qi_begin_0 = const()[name = tensor("qi_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor qi_end_0 = const()[name = tensor("qi_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor qi_end_mask_0 = const()[name = tensor("qi_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor qi_squeeze_mask_0 = const()[name = tensor("qi_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor qi_cast_fp16 = slice_by_index(begin = qi_begin_0, end = qi_end_0, end_mask = qi_end_mask_0, squeeze_mask = qi_squeeze_mask_0, x = q2_cast_fp16)[name = tensor("qi_cast_fp16")]; tensor kr_begin_0 = const()[name = tensor("kr_begin_0"), val = tensor([0, 0, 0, 0, 0])]; tensor kr_end_0 = const()[name = tensor("kr_end_0"), val = tensor([1, 4, 0, 32, 1])]; tensor kr_end_mask_0 = const()[name = tensor("kr_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor kr_squeeze_mask_0 = const()[name = tensor("kr_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor kr_cast_fp16 = slice_by_index(begin = kr_begin_0, end = kr_end_0, end_mask = kr_end_mask_0, squeeze_mask = kr_squeeze_mask_0, x = k2_cast_fp16)[name = tensor("kr_cast_fp16")]; tensor ki_begin_0 = const()[name = tensor("ki_begin_0"), val = tensor([0, 0, 0, 0, 1])]; tensor ki_end_0 = const()[name = tensor("ki_end_0"), val = tensor([1, 4, 0, 32, 2])]; tensor ki_end_mask_0 = const()[name = tensor("ki_end_mask_0"), val = tensor([true, true, true, true, false])]; tensor ki_squeeze_mask_0 = const()[name = tensor("ki_squeeze_mask_0"), val = tensor([false, false, false, false, true])]; tensor ki_cast_fp16 = slice_by_index(begin = ki_begin_0, end = ki_end_0, end_mask = ki_end_mask_0, squeeze_mask = ki_squeeze_mask_0, x = k2_cast_fp16)[name = tensor("ki_cast_fp16")]; tensor freqs_to_fp16 = const()[name = tensor("freqs_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20811520)))]; tensor var_2324_cast_fp16 = mul(x = freqs_to_fp16, y = ts_cast_fp16)[name = tensor("op_2324_cast_fp16")]; tensor rotr_cast_fp16 = cos(x = var_2324_cast_fp16)[name = tensor("rotr_cast_fp16")]; tensor roti_cast_fp16 = sin(x = var_2324_cast_fp16)[name = tensor("roti_cast_fp16")]; tensor var_2328_cast_fp16 = mul(x = qr_cast_fp16, y = rotr_cast_fp16)[name = tensor("op_2328_cast_fp16")]; tensor var_2329_cast_fp16 = mul(x = qi_cast_fp16, y = roti_cast_fp16)[name = tensor("op_2329_cast_fp16")]; tensor var_2330_cast_fp16 = sub(x = var_2328_cast_fp16, y = var_2329_cast_fp16)[name = tensor("op_2330_cast_fp16")]; tensor var_2331_cast_fp16 = mul(x = qr_cast_fp16, y = roti_cast_fp16)[name = tensor("op_2331_cast_fp16")]; tensor var_2332_cast_fp16 = mul(x = qi_cast_fp16, y = rotr_cast_fp16)[name = tensor("op_2332_cast_fp16")]; tensor var_2333_cast_fp16 = add(x = var_2331_cast_fp16, y = var_2332_cast_fp16)[name = tensor("op_2333_cast_fp16")]; tensor qo_axis_0 = const()[name = tensor("qo_axis_0"), val = tensor(-1)]; tensor qo_cast_fp16 = stack(axis = qo_axis_0, values = (var_2330_cast_fp16, var_2333_cast_fp16))[name = tensor("qo_cast_fp16")]; tensor var_2336_cast_fp16 = mul(x = kr_cast_fp16, y = rotr_cast_fp16)[name = tensor("op_2336_cast_fp16")]; tensor var_2337_cast_fp16 = mul(x = ki_cast_fp16, y = roti_cast_fp16)[name = tensor("op_2337_cast_fp16")]; tensor var_2338_cast_fp16 = sub(x = var_2336_cast_fp16, y = var_2337_cast_fp16)[name = tensor("op_2338_cast_fp16")]; tensor var_2339_cast_fp16 = mul(x = kr_cast_fp16, y = roti_cast_fp16)[name = tensor("op_2339_cast_fp16")]; tensor var_2340_cast_fp16 = mul(x = ki_cast_fp16, y = rotr_cast_fp16)[name = tensor("op_2340_cast_fp16")]; tensor var_2341_cast_fp16 = add(x = var_2339_cast_fp16, y = var_2340_cast_fp16)[name = tensor("op_2341_cast_fp16")]; tensor ko_axis_0 = const()[name = tensor("ko_axis_0"), val = tensor(-1)]; tensor ko_cast_fp16 = stack(axis = ko_axis_0, values = (var_2338_cast_fp16, var_2341_cast_fp16))[name = tensor("ko_cast_fp16")]; tensor concat_143x = const()[name = tensor("concat_143x"), val = tensor([1, 4, -1, 64])]; tensor q_cast_fp16 = reshape(shape = concat_143x, x = qo_cast_fp16)[name = tensor("q_cast_fp16")]; tensor concat_144x = const()[name = tensor("concat_144x"), val = tensor([1, 4, -1, 64])]; tensor k_cast_fp16 = reshape(shape = concat_144x, x = ko_cast_fp16)[name = tensor("k_cast_fp16")]; tensor const_104 = const()[name = tensor("const_104"), val = tensor(0)]; tensor const_105 = const()[name = tensor("const_105"), val = tensor(1)]; tensor gather_123_cast_uint16_to_int32 = cast(dtype = gather_123_cast_uint16_to_int32_dtype_0, x = gather_123_cast_uint16)[name = tensor("cast_300")]; tensor positions = range_1d(end = gather_123_cast_uint16_to_int32, start = const_104, step = const_105)[name = tensor("positions")]; tensor concat_145x = const()[name = tensor("concat_145x"), val = tensor([1, 1, -1])]; tensor var_2353 = reshape(shape = concat_145x, x = positions)[name = tensor("op_2353")]; tensor valid_k = less(x = var_2353, y = var_1944)[name = tensor("valid_k")]; tensor concat_146x = const()[name = tensor("concat_146x"), val = tensor([1, -1, 1])]; tensor var_2358 = reshape(shape = concat_146x, x = positions)[name = tensor("op_2358")]; tensor delta = sub(x = var_2358, y = var_2353)[name = tensor("delta")]; tensor concat_148_axis_0 = const()[name = tensor("concat_148_axis_0"), val = tensor(0)]; tensor concat_148_interleave_0 = const()[name = tensor("concat_148_interleave_0"), val = tensor(false)]; tensor concat_148 = concat(axis = concat_148_axis_0, interleave = concat_148_interleave_0, values = (var_1845, gather_123_cast_uint16_to_int32, gather_123_cast_uint16_to_int32))[name = tensor("concat_148")]; tensor fill_11_value_0_to_fp16 = const()[name = tensor("fill_11_value_0_to_fp16"), val = tensor(0x1p+0)]; tensor fill_11_cast_fp16 = fill(shape = concat_148, value = fill_11_value_0_to_fp16)[name = tensor("fill_11_cast_fp16")]; tensor attn_bias_89_dtype_0 = const()[name = tensor("attn_bias_89_dtype_0"), val = tensor("bool")]; tensor var_2364 = greater_equal(x = delta, y = var_1827)[name = tensor("op_2364")]; tensor fill_11_cast_fp16_to_bool = cast(dtype = attn_bias_89_dtype_0, x = fill_11_cast_fp16)[name = tensor("cast_296")]; tensor attn_bias_91 = logical_and(x = fill_11_cast_fp16_to_bool, y = var_2364)[name = tensor("attn_bias_91")]; tensor var_2366 = less(x = delta, y = var_1834)[name = tensor("op_2366")]; tensor attn_bias_93 = logical_and(x = attn_bias_91, y = var_2366)[name = tensor("attn_bias_93")]; tensor var_2368 = logical_and(x = attn_bias_93, y = valid_k)[name = tensor("op_2368")]; tensor var_2370_axes_0 = const()[name = tensor("op_2370_axes_0"), val = tensor([1])]; tensor var_2370 = expand_dims(axes = var_2370_axes_0, x = var_2368)[name = tensor("op_2370")]; tensor sub_11_x_0_to_fp16 = const()[name = tensor("sub_11_x_0_to_fp16"), val = tensor(0x1p+0)]; tensor cast_283_to_fp16_dtype_0 = const()[name = tensor("cast_283_to_fp16_dtype_0"), val = tensor("fp16")]; tensor var_2370_to_fp16 = cast(dtype = cast_283_to_fp16_dtype_0, x = var_2370)[name = tensor("cast_295")]; tensor sub_11_cast_fp16 = sub(x = sub_11_x_0_to_fp16, y = var_2370_to_fp16)[name = tensor("sub_11_cast_fp16")]; tensor mul_22_x_0_to_fp16 = const()[name = tensor("mul_22_x_0_to_fp16"), val = tensor(-0x1.d4cp+14)]; tensor mul_22_cast_fp16 = mul(x = mul_22_x_0_to_fp16, y = sub_11_cast_fp16)[name = tensor("mul_22_cast_fp16")]; tensor mul_23_y_0_to_fp16 = const()[name = tensor("mul_23_y_0_to_fp16"), val = tensor(0x1p-3)]; tensor mul_23_cast_fp16 = mul(x = q_cast_fp16, y = mul_23_y_0_to_fp16)[name = tensor("mul_23_cast_fp16")]; tensor matmul_11_transpose_y_0 = const()[name = tensor("matmul_11_transpose_y_0"), val = tensor(true)]; tensor matmul_11_transpose_x_0 = const()[name = tensor("matmul_11_transpose_x_0"), val = tensor(false)]; tensor matmul_11_cast_fp16 = matmul(transpose_x = matmul_11_transpose_x_0, transpose_y = matmul_11_transpose_y_0, x = mul_23_cast_fp16, y = k_cast_fp16)[name = tensor("matmul_11_cast_fp16")]; tensor add_11_cast_fp16 = add(x = matmul_11_cast_fp16, y = mul_22_cast_fp16)[name = tensor("add_11_cast_fp16")]; tensor add_11_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("add_11_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor softmax_11_axis_0 = const()[name = tensor("softmax_11_axis_0"), val = tensor(-1)]; tensor add_11_cast_fp16_to_fp32 = cast(dtype = add_11_cast_fp16_to_fp32_dtype_0, x = add_11_cast_fp16)[name = tensor("cast_294")]; tensor softmax_11 = softmax(axis = softmax_11_axis_0, x = add_11_cast_fp16_to_fp32)[name = tensor("softmax_11")]; tensor out_45_transpose_x_0 = const()[name = tensor("out_45_transpose_x_0"), val = tensor(false)]; tensor out_45_transpose_y_0 = const()[name = tensor("out_45_transpose_y_0"), val = tensor(false)]; tensor softmax_11_to_fp16_dtype_0 = const()[name = tensor("softmax_11_to_fp16_dtype_0"), val = tensor("fp16")]; tensor softmax_11_to_fp16 = cast(dtype = softmax_11_to_fp16_dtype_0, x = softmax_11)[name = tensor("cast_293")]; tensor out_45_cast_fp16 = matmul(transpose_x = out_45_transpose_x_0, transpose_y = out_45_transpose_y_0, x = softmax_11_to_fp16, y = v_cast_fp16)[name = tensor("out_45_cast_fp16")]; tensor concat_149x = const()[name = tensor("concat_149x"), val = tensor([1, -1])]; tensor var_2376 = reshape(shape = concat_149x, x = positions)[name = tensor("op_2376")]; tensor var_2379 = less(x = var_2376, y = var_1967)[name = tensor("op_2379")]; tensor concat_150x = const()[name = tensor("concat_150x"), val = tensor([1, 1, -1, 1])]; tensor valid_q = reshape(shape = concat_150x, x = var_2379)[name = tensor("valid_q")]; tensor var_2383_to_fp16 = const()[name = tensor("op_2383_to_fp16"), val = tensor(0x0p+0)]; tensor out_cast_fp16 = select(a = out_45_cast_fp16, b = var_2383_to_fp16, cond = valid_q)[name = tensor("out_cast_fp16")]; tensor var_2385_perm_0 = const()[name = tensor("op_2385_perm_0"), val = tensor([0, 2, 1, 3])]; tensor concat_151x = const()[name = tensor("concat_151x"), val = tensor([1, -1, 256])]; tensor var_2385_cast_fp16 = transpose(perm = var_2385_perm_0, x = out_cast_fp16)[name = tensor("transpose_3")]; tensor input_207_cast_fp16 = reshape(shape = concat_151x, x = var_2385_cast_fp16)[name = tensor("input_207_cast_fp16")]; tensor decoder_7_transformer_layers_3_self_attn_out_proj_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_3_self_attn_out_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20811648)))]; tensor linear_52_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_7_transformer_layers_3_self_attn_out_proj_weight_to_fp16, x = input_207_cast_fp16)[name = tensor("linear_52_cast_fp16")]; tensor decoder_7_transformer_layers_3_layer_scale_1_scale_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_3_layer_scale_1_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20942784)))]; tensor var_2391_cast_fp16 = mul(x = decoder_7_transformer_layers_3_layer_scale_1_scale_to_fp16, y = linear_52_cast_fp16)[name = tensor("op_2391_cast_fp16")]; tensor input_209_cast_fp16 = add(x = input_205_cast_fp16, y = var_2391_cast_fp16)[name = tensor("input_209_cast_fp16")]; tensor input_211_axes_0 = const()[name = tensor("input_211_axes_0"), val = tensor([-1])]; tensor decoder_7_transformer_layers_3_norm2_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_3_norm2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20943360)))]; tensor decoder_7_transformer_layers_3_norm2_bias_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_3_norm2_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20943936)))]; tensor input_211_cast_fp16 = layer_norm(axes = input_211_axes_0, beta = decoder_7_transformer_layers_3_norm2_bias_to_fp16, epsilon = var_1841_to_fp16, gamma = decoder_7_transformer_layers_3_norm2_weight_to_fp16, x = input_209_cast_fp16)[name = tensor("input_211_cast_fp16")]; tensor decoder_7_transformer_layers_3_ffn_0_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_3_ffn_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(20944512)))]; tensor linear_53_cast_fp16 = linear(bias = linear_3_bias_0_to_fp16, weight = decoder_7_transformer_layers_3_ffn_0_weight_to_fp16, x = input_211_cast_fp16)[name = tensor("linear_53_cast_fp16")]; tensor input_215_mode_0 = const()[name = tensor("input_215_mode_0"), val = tensor("EXACT")]; tensor input_215_cast_fp16 = gelu(mode = input_215_mode_0, x = linear_53_cast_fp16)[name = tensor("input_215_cast_fp16")]; tensor decoder_7_transformer_layers_3_ffn_2_weight_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_3_ffn_2_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21468864)))]; tensor linear_54_cast_fp16 = linear(bias = linear_0_bias_0_to_fp16, weight = decoder_7_transformer_layers_3_ffn_2_weight_to_fp16, x = input_215_cast_fp16)[name = tensor("linear_54_cast_fp16")]; tensor decoder_7_transformer_layers_3_layer_scale_2_scale_to_fp16 = const()[name = tensor("decoder_7_transformer_layers_3_layer_scale_2_scale_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21993216)))]; tensor var_2406_cast_fp16 = mul(x = decoder_7_transformer_layers_3_layer_scale_2_scale_to_fp16, y = linear_54_cast_fp16)[name = tensor("op_2406_cast_fp16")]; tensor input_cast_fp16 = add(x = input_209_cast_fp16, y = var_2406_cast_fp16)[name = tensor("input_cast_fp16")]; tensor decoder_7_output_proj_weight_to_fp16 = const()[name = tensor("decoder_7_output_proj_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(21993792)))]; tensor linear_55_bias_0_to_fp16 = const()[name = tensor("linear_55_bias_0_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(22116736)))]; tensor linear_55_cast_fp16 = linear(bias = linear_55_bias_0_to_fp16, weight = decoder_7_output_proj_weight_to_fp16, x = input_cast_fp16)[name = tensor("linear_55_cast_fp16")]; tensor x_97_perm_0 = const()[name = tensor("x_97_perm_0"), val = tensor([0, 2, 1])]; tensor concat_152x = const()[name = tensor("concat_152x"), val = tensor([1, 1, 240, -1])]; tensor x_97_cast_fp16 = transpose(perm = x_97_perm_0, x = linear_55_cast_fp16)[name = tensor("transpose_2")]; tensor var_2426_cast_fp16 = reshape(shape = concat_152x, x = x_97_cast_fp16)[name = tensor("op_2426_cast_fp16")]; tensor var_2427 = const()[name = tensor("op_2427"), val = tensor([0, 1, 3, 2])]; tensor concat_153x = const()[name = tensor("concat_153x"), val = tensor([1, 1, -1])]; tensor var_2428_cast_fp16 = transpose(perm = var_2427, x = var_2426_cast_fp16)[name = tensor("transpose_1")]; tensor x_cast_fp16 = reshape(shape = concat_153x, x = var_2428_cast_fp16)[name = tensor("x_cast_fp16")]; tensor var_2435_axes_0 = const()[name = tensor("op_2435_axes_0"), val = tensor([1])]; tensor var_2435_cast_fp16 = squeeze(axes = var_2435_axes_0, x = x_cast_fp16)[name = tensor("op_2435_cast_fp16")]; tensor var_2442 = const()[name = tensor("op_2442"), val = tensor([1, -1, 2])]; tensor var_2443_cast_fp16 = reshape(shape = var_2442, x = var_2435_cast_fp16)[name = tensor("op_2443_cast_fp16")]; tensor var_2446_perm_0 = const()[name = tensor("op_2446_perm_0"), val = tensor([0, 2, 1])]; tensor var_2446_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("op_2446_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor var_2446_cast_fp16 = transpose(perm = var_2446_perm_0, x = var_2443_cast_fp16)[name = tensor("transpose_0")]; tensor audio = cast(dtype = var_2446_cast_fp16_to_fp32_dtype_0, x = var_2446_cast_fp16)[name = tensor("cast_292")]; } -> (audio); }