program(1.0) [buildInfo = dict, tensor>({{"coremlc-component-MIL", "3520.4.1"}, {"coremlc-version", "3520.5.1"}, {"coremltools-component-torch", "2.7.0"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0b1"}})] { func main(tensor encoder_output) { tensor var_4 = const()[name = tensor("op_4"), val = tensor(-1)]; tensor var_18_pad_type_0 = const()[name = tensor("op_18_pad_type_0"), val = tensor("valid")]; tensor var_18_strides_0 = const()[name = tensor("op_18_strides_0"), val = tensor([1])]; tensor var_18_pad_0 = const()[name = tensor("op_18_pad_0"), val = tensor([0, 0])]; tensor var_18_dilations_0 = const()[name = tensor("op_18_dilations_0"), val = tensor([1])]; tensor var_18_groups_0 = const()[name = tensor("op_18_groups_0"), val = tensor(1)]; tensor encoder_output_to_fp16_dtype_0 = const()[name = tensor("encoder_output_to_fp16_dtype_0"), val = tensor("fp16")]; tensor module_decoder_layers_0_weight_to_fp16 = const()[name = tensor("module_decoder_layers_0_weight_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(64)))]; tensor module_decoder_layers_0_bias_to_fp16 = const()[name = tensor("module_decoder_layers_0_bias_to_fp16"), val = tensor(BLOBFILE(path = tensor("@model_path/weights/weight.bin"), offset = tensor(1049728)))]; tensor encoder_output_to_fp16 = cast(dtype = encoder_output_to_fp16_dtype_0, x = encoder_output)[name = tensor("cast_1")]; tensor var_18_cast_fp16 = conv(bias = module_decoder_layers_0_bias_to_fp16, dilations = var_18_dilations_0, groups = var_18_groups_0, pad = var_18_pad_0, pad_type = var_18_pad_type_0, strides = var_18_strides_0, weight = module_decoder_layers_0_weight_to_fp16, x = encoder_output_to_fp16)[name = tensor("op_18_cast_fp16")]; tensor input_perm_0 = const()[name = tensor("input_perm_0"), val = tensor([0, 2, 1])]; tensor input_cast_fp16 = transpose(perm = input_perm_0, x = var_18_cast_fp16)[name = tensor("transpose_0")]; tensor out_objects_softmax_cast_fp16 = softmax(axis = var_4, x = input_cast_fp16)[name = tensor("out_objects_softmax_cast_fp16")]; tensor out_objects_epsilon_0 = const()[name = tensor("out_objects_epsilon_0"), val = tensor(0x1p-149)]; tensor out_objects_cast_fp16 = log(epsilon = out_objects_epsilon_0, x = out_objects_softmax_cast_fp16)[name = tensor("out_objects_cast_fp16")]; tensor out_objects_cast_fp16_to_fp32_dtype_0 = const()[name = tensor("out_objects_cast_fp16_to_fp32_dtype_0"), val = tensor("fp32")]; tensor ctc_logits = cast(dtype = out_objects_cast_fp16_to_fp32_dtype_0, x = out_objects_cast_fp16)[name = tensor("cast_0")]; } -> (ctc_logits); }