| program(1.3) |
| [buildInfo = dict<string, string>({{"coremlc-component-MIL", "3405.2.1"}, {"coremlc-version", "3405.2.1"}, {"coremltools-component-milinternal", ""}, {"coremltools-version", "9.0"}})] |
| { |
| func main<ios18>(tensor<fp16, [1, 896, 1]> input_x) { |
| string var_14_pad_type_0 = const()[name = string("op_14_pad_type_0"), val = string("valid")]; |
| tensor<int32, [1]> var_14_strides_0 = const()[name = string("op_14_strides_0"), val = tensor<int32, [1]>([1])]; |
| tensor<int32, [2]> var_14_pad_0 = const()[name = string("op_14_pad_0"), val = tensor<int32, [2]>([0, 0])]; |
| tensor<int32, [1]> var_14_dilations_0 = const()[name = string("op_14_dilations_0"), val = tensor<int32, [1]>([1])]; |
| int32 var_14_groups_0 = const()[name = string("op_14_groups_0"), val = int32(1)]; |
| tensor<fp16, [896, 896, 1]> var_8_to_fp16 = const()[name = string("op_8_to_fp16"), val = tensor<fp16, [896, 896, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))]; |
| tensor<fp16, [896]> fc1_bias_to_fp16 = const()[name = string("fc1_bias_to_fp16"), val = tensor<fp16, [896]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1605760)))]; |
| tensor<fp16, [1, 896, 1]> var_14_cast_fp16 = conv(bias = fc1_bias_to_fp16, dilations = var_14_dilations_0, groups = var_14_groups_0, pad = var_14_pad_0, pad_type = var_14_pad_type_0, strides = var_14_strides_0, weight = var_8_to_fp16, x = input_x)[name = string("op_14_cast_fp16")]; |
| tensor<fp16, [1, 896, 1]> x_cast_fp16 = relu(x = var_14_cast_fp16)[name = string("x_cast_fp16")]; |
| string var_26_pad_type_0 = const()[name = string("op_26_pad_type_0"), val = string("valid")]; |
| tensor<int32, [1]> var_26_strides_0 = const()[name = string("op_26_strides_0"), val = tensor<int32, [1]>([1])]; |
| tensor<int32, [2]> var_26_pad_0 = const()[name = string("op_26_pad_0"), val = tensor<int32, [2]>([0, 0])]; |
| tensor<int32, [1]> var_26_dilations_0 = const()[name = string("op_26_dilations_0"), val = tensor<int32, [1]>([1])]; |
| int32 var_26_groups_0 = const()[name = string("op_26_groups_0"), val = int32(1)]; |
| tensor<fp16, [1, 896, 1]> var_20_to_fp16 = const()[name = string("op_20_to_fp16"), val = tensor<fp16, [1, 896, 1]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1607616)))]; |
| tensor<fp16, [1]> fc2_bias_to_fp16 = const()[name = string("fc2_bias_to_fp16"), val = tensor<fp16, [1]>([-0x1.36p-5])]; |
| tensor<fp16, [1, 1, 1]> output = conv(bias = fc2_bias_to_fp16, dilations = var_26_dilations_0, groups = var_26_groups_0, pad = var_26_pad_0, pad_type = var_26_pad_type_0, strides = var_26_strides_0, weight = var_20_to_fp16, x = x_cast_fp16)[name = string("op_26_cast_fp16")]; |
| } -> (output); |
| } |