program(1.3) [buildInfo = dict({{"coremlc-component-MIL", "3510.2.1"}, {"coremlc-version", "3500.32.1"}, {"coremltools-component-torch", "2.9.1"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})] { func main(tensor logmel_data) { string var_28_pad_type_0 = const()[name = string("op_28_pad_type_0"), val = string("custom")]; tensor var_28_pad_0 = const()[name = string("op_28_pad_0"), val = tensor([1, 1])]; tensor var_28_strides_0 = const()[name = string("op_28_strides_0"), val = tensor([1])]; tensor var_28_dilations_0 = const()[name = string("op_28_dilations_0"), val = tensor([1])]; int32 var_28_groups_0 = const()[name = string("op_28_groups_0"), val = int32(1)]; tensor const_0_to_fp16 = const()[name = string("const_0_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))]; tensor const_1_to_fp16 = const()[name = string("const_1_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(184448)))]; tensor var_28_cast_fp16 = conv(bias = const_1_to_fp16, dilations = var_28_dilations_0, groups = var_28_groups_0, pad = var_28_pad_0, pad_type = var_28_pad_type_0, strides = var_28_strides_0, weight = const_0_to_fp16, x = logmel_data)[name = string("op_28_cast_fp16")]; string input_1_mode_0 = const()[name = string("input_1_mode_0"), val = string("EXACT")]; tensor input_1_cast_fp16 = gelu(mode = input_1_mode_0, x = var_28_cast_fp16)[name = string("input_1_cast_fp16")]; string var_46_pad_type_0 = const()[name = string("op_46_pad_type_0"), val = string("custom")]; tensor var_46_pad_0 = const()[name = string("op_46_pad_0"), val = tensor([1, 1])]; tensor var_46_strides_0 = const()[name = string("op_46_strides_0"), val = tensor([2])]; tensor var_46_dilations_0 = const()[name = string("op_46_dilations_0"), val = tensor([1])]; int32 var_46_groups_0 = const()[name = string("op_46_groups_0"), val = int32(1)]; tensor const_2_to_fp16 = const()[name = string("const_2_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(185280)))]; tensor const_3_to_fp16 = const()[name = string("const_3_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1070080)))]; tensor var_46_cast_fp16 = conv(bias = const_3_to_fp16, dilations = var_46_dilations_0, groups = var_46_groups_0, pad = var_46_pad_0, pad_type = var_46_pad_type_0, strides = var_46_strides_0, weight = const_2_to_fp16, x = input_1_cast_fp16)[name = string("op_46_cast_fp16")]; string x_3_mode_0 = const()[name = string("x_3_mode_0"), val = string("EXACT")]; tensor x_3_cast_fp16 = gelu(mode = x_3_mode_0, x = var_46_cast_fp16)[name = string("x_3_cast_fp16")]; tensor var_52 = const()[name = string("op_52"), val = tensor([0, 2, 1])]; tensor positional_embedding_to_fp16 = const()[name = string("positional_embedding_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1070912)))]; tensor x_5_cast_fp16 = transpose(perm = var_52, x = x_3_cast_fp16)[name = string("transpose_52")]; tensor var_55_cast_fp16 = add(x = x_5_cast_fp16, y = positional_embedding_to_fp16)[name = string("op_55_cast_fp16")]; tensor var_82_axes_0 = const()[name = string("op_82_axes_0"), val = tensor([-1])]; tensor blocks_0_attn_ln_weight_to_fp16 = const()[name = string("blocks_0_attn_ln_weight_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2222976)))]; tensor blocks_0_attn_ln_bias_to_fp16 = const()[name = string("blocks_0_attn_ln_bias_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2223808)))]; fp16 var_72_to_fp16 = const()[name = string("op_72_to_fp16"), val = fp16(0x1.5p-17)]; tensor var_82_cast_fp16 = layer_norm(axes = var_82_axes_0, beta = blocks_0_attn_ln_bias_to_fp16, epsilon = var_72_to_fp16, gamma = blocks_0_attn_ln_weight_to_fp16, x = var_55_cast_fp16)[name = string("op_82_cast_fp16")]; tensor const_4_to_fp16 = const()[name = string("const_4_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2224640)))]; tensor const_5_to_fp16 = const()[name = string("const_5_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2519616)))]; tensor linear_0_cast_fp16 = linear(bias = const_5_to_fp16, weight = const_4_to_fp16, x = var_82_cast_fp16)[name = string("linear_0_cast_fp16")]; tensor const_6_to_fp16 = const()[name = string("const_6_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2520448)))]; tensor linear_1_bias_0_to_fp16 = const()[name = string("linear_1_bias_0_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2815424)))]; tensor linear_1_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = const_6_to_fp16, x = var_82_cast_fp16)[name = string("linear_1_cast_fp16")]; tensor const_7_to_fp16 = const()[name = string("const_7_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2816256)))]; tensor const_8_to_fp16 = const()[name = string("const_8_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(3111232)))]; tensor linear_2_cast_fp16 = linear(bias = const_8_to_fp16, weight = const_7_to_fp16, x = var_82_cast_fp16)[name = string("linear_2_cast_fp16")]; tensor var_106 = const()[name = string("op_106"), val = tensor([1, 1500, 6, -1])]; tensor var_107_cast_fp16 = reshape(shape = var_106, x = linear_0_cast_fp16)[name = string("op_107_cast_fp16")]; tensor var_112 = const()[name = string("op_112"), val = tensor([1, 1500, 6, -1])]; tensor var_113_cast_fp16 = reshape(shape = var_112, x = linear_1_cast_fp16)[name = string("op_113_cast_fp16")]; tensor var_118 = const()[name = string("op_118"), val = tensor([1, 1500, 6, -1])]; tensor var_119_cast_fp16 = reshape(shape = var_118, x = linear_2_cast_fp16)[name = string("op_119_cast_fp16")]; tensor transpose_24_perm_0 = const()[name = string("transpose_24_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_25_perm_0 = const()[name = string("transpose_25_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_26_perm_0 = const()[name = string("transpose_26_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_26 = transpose(perm = transpose_26_perm_0, x = var_119_cast_fp16)[name = string("transpose_49")]; tensor transpose_25 = transpose(perm = transpose_25_perm_0, x = var_113_cast_fp16)[name = string("transpose_50")]; tensor transpose_24 = transpose(perm = transpose_24_perm_0, x = var_107_cast_fp16)[name = string("transpose_51")]; tensor a_1_cast_fp16 = scaled_dot_product_attention(key = transpose_25, query = transpose_24, value = transpose_26)[name = string("a_1_cast_fp16")]; tensor var_123 = const()[name = string("op_123"), val = tensor([0, 2, 1, 3])]; tensor concat_0 = const()[name = string("concat_0"), val = tensor([1, 1500, 384])]; tensor var_124_cast_fp16 = transpose(perm = var_123, x = a_1_cast_fp16)[name = string("transpose_48")]; tensor x_11_cast_fp16 = reshape(shape = concat_0, x = var_124_cast_fp16)[name = string("x_11_cast_fp16")]; tensor const_15_to_fp16 = const()[name = string("const_15_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(3112064)))]; tensor const_16_to_fp16 = const()[name = string("const_16_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(3407040)))]; tensor linear_3_cast_fp16 = linear(bias = const_16_to_fp16, weight = const_15_to_fp16, x = x_11_cast_fp16)[name = string("linear_3_cast_fp16")]; tensor x_13_cast_fp16 = add(x = var_55_cast_fp16, y = linear_3_cast_fp16)[name = string("x_13_cast_fp16")]; tensor var_136_axes_0 = const()[name = string("op_136_axes_0"), val = tensor([-1])]; tensor blocks_0_mlp_ln_weight_to_fp16 = const()[name = string("blocks_0_mlp_ln_weight_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(3407872)))]; tensor blocks_0_mlp_ln_bias_to_fp16 = const()[name = string("blocks_0_mlp_ln_bias_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(3408704)))]; tensor var_136_cast_fp16 = layer_norm(axes = var_136_axes_0, beta = blocks_0_mlp_ln_bias_to_fp16, epsilon = var_72_to_fp16, gamma = blocks_0_mlp_ln_weight_to_fp16, x = x_13_cast_fp16)[name = string("op_136_cast_fp16")]; tensor const_17_to_fp16 = const()[name = string("const_17_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(3409536)))]; tensor const_18_to_fp16 = const()[name = string("const_18_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4589248)))]; tensor linear_4_cast_fp16 = linear(bias = const_18_to_fp16, weight = const_17_to_fp16, x = var_136_cast_fp16)[name = string("linear_4_cast_fp16")]; string x_17_mode_0 = const()[name = string("x_17_mode_0"), val = string("EXACT")]; tensor x_17_cast_fp16 = gelu(mode = x_17_mode_0, x = linear_4_cast_fp16)[name = string("x_17_cast_fp16")]; tensor const_19_to_fp16 = const()[name = string("const_19_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(4592384)))]; tensor const_20_to_fp16 = const()[name = string("const_20_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(5772096)))]; tensor linear_5_cast_fp16 = linear(bias = const_20_to_fp16, weight = const_19_to_fp16, x = x_17_cast_fp16)[name = string("linear_5_cast_fp16")]; tensor x_19_cast_fp16 = add(x = x_13_cast_fp16, y = linear_5_cast_fp16)[name = string("x_19_cast_fp16")]; tensor var_176_axes_0 = const()[name = string("op_176_axes_0"), val = tensor([-1])]; tensor blocks_1_attn_ln_weight_to_fp16 = const()[name = string("blocks_1_attn_ln_weight_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(5772928)))]; tensor blocks_1_attn_ln_bias_to_fp16 = const()[name = string("blocks_1_attn_ln_bias_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(5773760)))]; fp16 var_166_to_fp16 = const()[name = string("op_166_to_fp16"), val = fp16(0x1.5p-17)]; tensor var_176_cast_fp16 = layer_norm(axes = var_176_axes_0, beta = blocks_1_attn_ln_bias_to_fp16, epsilon = var_166_to_fp16, gamma = blocks_1_attn_ln_weight_to_fp16, x = x_19_cast_fp16)[name = string("op_176_cast_fp16")]; tensor const_21_to_fp16 = const()[name = string("const_21_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(5774592)))]; tensor const_22_to_fp16 = const()[name = string("const_22_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6069568)))]; tensor linear_6_cast_fp16 = linear(bias = const_22_to_fp16, weight = const_21_to_fp16, x = var_176_cast_fp16)[name = string("linear_6_cast_fp16")]; tensor const_23_to_fp16 = const()[name = string("const_23_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6070400)))]; tensor linear_7_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = const_23_to_fp16, x = var_176_cast_fp16)[name = string("linear_7_cast_fp16")]; tensor const_24_to_fp16 = const()[name = string("const_24_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6365376)))]; tensor const_25_to_fp16 = const()[name = string("const_25_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6660352)))]; tensor linear_8_cast_fp16 = linear(bias = const_25_to_fp16, weight = const_24_to_fp16, x = var_176_cast_fp16)[name = string("linear_8_cast_fp16")]; tensor var_200 = const()[name = string("op_200"), val = tensor([1, 1500, 6, -1])]; tensor var_201_cast_fp16 = reshape(shape = var_200, x = linear_6_cast_fp16)[name = string("op_201_cast_fp16")]; tensor var_206 = const()[name = string("op_206"), val = tensor([1, 1500, 6, -1])]; tensor var_207_cast_fp16 = reshape(shape = var_206, x = linear_7_cast_fp16)[name = string("op_207_cast_fp16")]; tensor var_212 = const()[name = string("op_212"), val = tensor([1, 1500, 6, -1])]; tensor var_213_cast_fp16 = reshape(shape = var_212, x = linear_8_cast_fp16)[name = string("op_213_cast_fp16")]; tensor transpose_27_perm_0 = const()[name = string("transpose_27_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_28_perm_0 = const()[name = string("transpose_28_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_29_perm_0 = const()[name = string("transpose_29_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_29 = transpose(perm = transpose_29_perm_0, x = var_213_cast_fp16)[name = string("transpose_45")]; tensor transpose_28 = transpose(perm = transpose_28_perm_0, x = var_207_cast_fp16)[name = string("transpose_46")]; tensor transpose_27 = transpose(perm = transpose_27_perm_0, x = var_201_cast_fp16)[name = string("transpose_47")]; tensor a_3_cast_fp16 = scaled_dot_product_attention(key = transpose_28, query = transpose_27, value = transpose_29)[name = string("a_3_cast_fp16")]; tensor var_217 = const()[name = string("op_217"), val = tensor([0, 2, 1, 3])]; tensor concat_1 = const()[name = string("concat_1"), val = tensor([1, 1500, 384])]; tensor var_218_cast_fp16 = transpose(perm = var_217, x = a_3_cast_fp16)[name = string("transpose_44")]; tensor x_23_cast_fp16 = reshape(shape = concat_1, x = var_218_cast_fp16)[name = string("x_23_cast_fp16")]; tensor const_32_to_fp16 = const()[name = string("const_32_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6661184)))]; tensor const_33_to_fp16 = const()[name = string("const_33_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6956160)))]; tensor linear_9_cast_fp16 = linear(bias = const_33_to_fp16, weight = const_32_to_fp16, x = x_23_cast_fp16)[name = string("linear_9_cast_fp16")]; tensor x_25_cast_fp16 = add(x = x_19_cast_fp16, y = linear_9_cast_fp16)[name = string("x_25_cast_fp16")]; tensor var_230_axes_0 = const()[name = string("op_230_axes_0"), val = tensor([-1])]; tensor blocks_1_mlp_ln_weight_to_fp16 = const()[name = string("blocks_1_mlp_ln_weight_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6956992)))]; tensor blocks_1_mlp_ln_bias_to_fp16 = const()[name = string("blocks_1_mlp_ln_bias_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6957824)))]; tensor var_230_cast_fp16 = layer_norm(axes = var_230_axes_0, beta = blocks_1_mlp_ln_bias_to_fp16, epsilon = var_166_to_fp16, gamma = blocks_1_mlp_ln_weight_to_fp16, x = x_25_cast_fp16)[name = string("op_230_cast_fp16")]; tensor const_34_to_fp16 = const()[name = string("const_34_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(6958656)))]; tensor const_35_to_fp16 = const()[name = string("const_35_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8138368)))]; tensor linear_10_cast_fp16 = linear(bias = const_35_to_fp16, weight = const_34_to_fp16, x = var_230_cast_fp16)[name = string("linear_10_cast_fp16")]; string x_29_mode_0 = const()[name = string("x_29_mode_0"), val = string("EXACT")]; tensor x_29_cast_fp16 = gelu(mode = x_29_mode_0, x = linear_10_cast_fp16)[name = string("x_29_cast_fp16")]; tensor const_36_to_fp16 = const()[name = string("const_36_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8141504)))]; tensor const_37_to_fp16 = const()[name = string("const_37_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9321216)))]; tensor linear_11_cast_fp16 = linear(bias = const_37_to_fp16, weight = const_36_to_fp16, x = x_29_cast_fp16)[name = string("linear_11_cast_fp16")]; tensor x_31_cast_fp16 = add(x = x_25_cast_fp16, y = linear_11_cast_fp16)[name = string("x_31_cast_fp16")]; tensor var_270_axes_0 = const()[name = string("op_270_axes_0"), val = tensor([-1])]; tensor blocks_2_attn_ln_weight_to_fp16 = const()[name = string("blocks_2_attn_ln_weight_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9322048)))]; tensor blocks_2_attn_ln_bias_to_fp16 = const()[name = string("blocks_2_attn_ln_bias_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9322880)))]; fp16 var_260_to_fp16 = const()[name = string("op_260_to_fp16"), val = fp16(0x1.5p-17)]; tensor var_270_cast_fp16 = layer_norm(axes = var_270_axes_0, beta = blocks_2_attn_ln_bias_to_fp16, epsilon = var_260_to_fp16, gamma = blocks_2_attn_ln_weight_to_fp16, x = x_31_cast_fp16)[name = string("op_270_cast_fp16")]; tensor const_38_to_fp16 = const()[name = string("const_38_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9323712)))]; tensor const_39_to_fp16 = const()[name = string("const_39_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9618688)))]; tensor linear_12_cast_fp16 = linear(bias = const_39_to_fp16, weight = const_38_to_fp16, x = var_270_cast_fp16)[name = string("linear_12_cast_fp16")]; tensor const_40_to_fp16 = const()[name = string("const_40_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9619520)))]; tensor linear_13_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = const_40_to_fp16, x = var_270_cast_fp16)[name = string("linear_13_cast_fp16")]; tensor const_41_to_fp16 = const()[name = string("const_41_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9914496)))]; tensor const_42_to_fp16 = const()[name = string("const_42_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10209472)))]; tensor linear_14_cast_fp16 = linear(bias = const_42_to_fp16, weight = const_41_to_fp16, x = var_270_cast_fp16)[name = string("linear_14_cast_fp16")]; tensor var_294 = const()[name = string("op_294"), val = tensor([1, 1500, 6, -1])]; tensor var_295_cast_fp16 = reshape(shape = var_294, x = linear_12_cast_fp16)[name = string("op_295_cast_fp16")]; tensor var_300 = const()[name = string("op_300"), val = tensor([1, 1500, 6, -1])]; tensor var_301_cast_fp16 = reshape(shape = var_300, x = linear_13_cast_fp16)[name = string("op_301_cast_fp16")]; tensor var_306 = const()[name = string("op_306"), val = tensor([1, 1500, 6, -1])]; tensor var_307_cast_fp16 = reshape(shape = var_306, x = linear_14_cast_fp16)[name = string("op_307_cast_fp16")]; tensor transpose_30_perm_0 = const()[name = string("transpose_30_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_31_perm_0 = const()[name = string("transpose_31_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_32_perm_0 = const()[name = string("transpose_32_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_32 = transpose(perm = transpose_32_perm_0, x = var_307_cast_fp16)[name = string("transpose_41")]; tensor transpose_31 = transpose(perm = transpose_31_perm_0, x = var_301_cast_fp16)[name = string("transpose_42")]; tensor transpose_30 = transpose(perm = transpose_30_perm_0, x = var_295_cast_fp16)[name = string("transpose_43")]; tensor a_5_cast_fp16 = scaled_dot_product_attention(key = transpose_31, query = transpose_30, value = transpose_32)[name = string("a_5_cast_fp16")]; tensor var_311 = const()[name = string("op_311"), val = tensor([0, 2, 1, 3])]; tensor concat_2 = const()[name = string("concat_2"), val = tensor([1, 1500, 384])]; tensor var_312_cast_fp16 = transpose(perm = var_311, x = a_5_cast_fp16)[name = string("transpose_40")]; tensor x_35_cast_fp16 = reshape(shape = concat_2, x = var_312_cast_fp16)[name = string("x_35_cast_fp16")]; tensor const_49_to_fp16 = const()[name = string("const_49_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10210304)))]; tensor const_50_to_fp16 = const()[name = string("const_50_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10505280)))]; tensor linear_15_cast_fp16 = linear(bias = const_50_to_fp16, weight = const_49_to_fp16, x = x_35_cast_fp16)[name = string("linear_15_cast_fp16")]; tensor x_37_cast_fp16 = add(x = x_31_cast_fp16, y = linear_15_cast_fp16)[name = string("x_37_cast_fp16")]; tensor var_324_axes_0 = const()[name = string("op_324_axes_0"), val = tensor([-1])]; tensor blocks_2_mlp_ln_weight_to_fp16 = const()[name = string("blocks_2_mlp_ln_weight_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10506112)))]; tensor blocks_2_mlp_ln_bias_to_fp16 = const()[name = string("blocks_2_mlp_ln_bias_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10506944)))]; tensor var_324_cast_fp16 = layer_norm(axes = var_324_axes_0, beta = blocks_2_mlp_ln_bias_to_fp16, epsilon = var_260_to_fp16, gamma = blocks_2_mlp_ln_weight_to_fp16, x = x_37_cast_fp16)[name = string("op_324_cast_fp16")]; tensor const_51_to_fp16 = const()[name = string("const_51_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10507776)))]; tensor const_52_to_fp16 = const()[name = string("const_52_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(11687488)))]; tensor linear_16_cast_fp16 = linear(bias = const_52_to_fp16, weight = const_51_to_fp16, x = var_324_cast_fp16)[name = string("linear_16_cast_fp16")]; string x_41_mode_0 = const()[name = string("x_41_mode_0"), val = string("EXACT")]; tensor x_41_cast_fp16 = gelu(mode = x_41_mode_0, x = linear_16_cast_fp16)[name = string("x_41_cast_fp16")]; tensor const_53_to_fp16 = const()[name = string("const_53_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(11690624)))]; tensor const_54_to_fp16 = const()[name = string("const_54_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(12870336)))]; tensor linear_17_cast_fp16 = linear(bias = const_54_to_fp16, weight = const_53_to_fp16, x = x_41_cast_fp16)[name = string("linear_17_cast_fp16")]; tensor x_43_cast_fp16 = add(x = x_37_cast_fp16, y = linear_17_cast_fp16)[name = string("x_43_cast_fp16")]; tensor var_364_axes_0 = const()[name = string("op_364_axes_0"), val = tensor([-1])]; tensor blocks_3_attn_ln_weight_to_fp16 = const()[name = string("blocks_3_attn_ln_weight_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(12871168)))]; tensor blocks_3_attn_ln_bias_to_fp16 = const()[name = string("blocks_3_attn_ln_bias_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(12872000)))]; fp16 var_354_to_fp16 = const()[name = string("op_354_to_fp16"), val = fp16(0x1.5p-17)]; tensor var_364_cast_fp16 = layer_norm(axes = var_364_axes_0, beta = blocks_3_attn_ln_bias_to_fp16, epsilon = var_354_to_fp16, gamma = blocks_3_attn_ln_weight_to_fp16, x = x_43_cast_fp16)[name = string("op_364_cast_fp16")]; tensor const_55_to_fp16 = const()[name = string("const_55_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(12872832)))]; tensor const_56_to_fp16 = const()[name = string("const_56_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13167808)))]; tensor linear_18_cast_fp16 = linear(bias = const_56_to_fp16, weight = const_55_to_fp16, x = var_364_cast_fp16)[name = string("linear_18_cast_fp16")]; tensor const_57_to_fp16 = const()[name = string("const_57_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13168640)))]; tensor linear_19_cast_fp16 = linear(bias = linear_1_bias_0_to_fp16, weight = const_57_to_fp16, x = var_364_cast_fp16)[name = string("linear_19_cast_fp16")]; tensor const_58_to_fp16 = const()[name = string("const_58_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13463616)))]; tensor const_59_to_fp16 = const()[name = string("const_59_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13758592)))]; tensor linear_20_cast_fp16 = linear(bias = const_59_to_fp16, weight = const_58_to_fp16, x = var_364_cast_fp16)[name = string("linear_20_cast_fp16")]; tensor var_388 = const()[name = string("op_388"), val = tensor([1, 1500, 6, -1])]; tensor var_389_cast_fp16 = reshape(shape = var_388, x = linear_18_cast_fp16)[name = string("op_389_cast_fp16")]; tensor var_394 = const()[name = string("op_394"), val = tensor([1, 1500, 6, -1])]; tensor var_395_cast_fp16 = reshape(shape = var_394, x = linear_19_cast_fp16)[name = string("op_395_cast_fp16")]; tensor var_400 = const()[name = string("op_400"), val = tensor([1, 1500, 6, -1])]; tensor var_401_cast_fp16 = reshape(shape = var_400, x = linear_20_cast_fp16)[name = string("op_401_cast_fp16")]; tensor transpose_33_perm_0 = const()[name = string("transpose_33_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_34_perm_0 = const()[name = string("transpose_34_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_35_perm_0 = const()[name = string("transpose_35_perm_0"), val = tensor([0, 2, 1, 3])]; tensor transpose_35 = transpose(perm = transpose_35_perm_0, x = var_401_cast_fp16)[name = string("transpose_37")]; tensor transpose_34 = transpose(perm = transpose_34_perm_0, x = var_395_cast_fp16)[name = string("transpose_38")]; tensor transpose_33 = transpose(perm = transpose_33_perm_0, x = var_389_cast_fp16)[name = string("transpose_39")]; tensor a_cast_fp16 = scaled_dot_product_attention(key = transpose_34, query = transpose_33, value = transpose_35)[name = string("a_cast_fp16")]; tensor var_405 = const()[name = string("op_405"), val = tensor([0, 2, 1, 3])]; tensor concat_3 = const()[name = string("concat_3"), val = tensor([1, 1500, 384])]; tensor var_406_cast_fp16 = transpose(perm = var_405, x = a_cast_fp16)[name = string("transpose_36")]; tensor x_47_cast_fp16 = reshape(shape = concat_3, x = var_406_cast_fp16)[name = string("x_47_cast_fp16")]; tensor const_66_to_fp16 = const()[name = string("const_66_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13759424)))]; tensor const_67_to_fp16 = const()[name = string("const_67_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14054400)))]; tensor linear_21_cast_fp16 = linear(bias = const_67_to_fp16, weight = const_66_to_fp16, x = x_47_cast_fp16)[name = string("linear_21_cast_fp16")]; tensor x_49_cast_fp16 = add(x = x_43_cast_fp16, y = linear_21_cast_fp16)[name = string("x_49_cast_fp16")]; tensor var_418_axes_0 = const()[name = string("op_418_axes_0"), val = tensor([-1])]; tensor blocks_3_mlp_ln_weight_to_fp16 = const()[name = string("blocks_3_mlp_ln_weight_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14055232)))]; tensor blocks_3_mlp_ln_bias_to_fp16 = const()[name = string("blocks_3_mlp_ln_bias_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14056064)))]; tensor var_418_cast_fp16 = layer_norm(axes = var_418_axes_0, beta = blocks_3_mlp_ln_bias_to_fp16, epsilon = var_354_to_fp16, gamma = blocks_3_mlp_ln_weight_to_fp16, x = x_49_cast_fp16)[name = string("op_418_cast_fp16")]; tensor const_68_to_fp16 = const()[name = string("const_68_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14056896)))]; tensor const_69_to_fp16 = const()[name = string("const_69_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(15236608)))]; tensor linear_22_cast_fp16 = linear(bias = const_69_to_fp16, weight = const_68_to_fp16, x = var_418_cast_fp16)[name = string("linear_22_cast_fp16")]; string x_53_mode_0 = const()[name = string("x_53_mode_0"), val = string("EXACT")]; tensor x_53_cast_fp16 = gelu(mode = x_53_mode_0, x = linear_22_cast_fp16)[name = string("x_53_cast_fp16")]; tensor const_70_to_fp16 = const()[name = string("const_70_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(15239744)))]; tensor const_71_to_fp16 = const()[name = string("const_71_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(16419456)))]; tensor linear_23_cast_fp16 = linear(bias = const_71_to_fp16, weight = const_70_to_fp16, x = x_53_cast_fp16)[name = string("linear_23_cast_fp16")]; tensor x_cast_fp16 = add(x = x_49_cast_fp16, y = linear_23_cast_fp16)[name = string("x_cast_fp16")]; tensor var_447_axes_0 = const()[name = string("op_447_axes_0"), val = tensor([-1])]; tensor ln_post_weight_to_fp16 = const()[name = string("ln_post_weight_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(16420288)))]; tensor ln_post_bias_to_fp16 = const()[name = string("ln_post_bias_to_fp16"), val = tensor(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(16421120)))]; fp16 var_438_to_fp16 = const()[name = string("op_438_to_fp16"), val = fp16(0x1.5p-17)]; tensor output = layer_norm(axes = var_447_axes_0, beta = ln_post_bias_to_fp16, epsilon = var_438_to_fp16, gamma = ln_post_weight_to_fp16, x = x_cast_fp16)[name = string("op_447_cast_fp16")]; } -> (output); }