File size: 1,156 Bytes
c65455a |
1 2 3 4 5 6 7 8 9 |
program(1.0)
[buildInfo = dict<tensor<string, []>, tensor<string, []>>({{"coremlc-component-MIL", "3304.6.1"}, {"coremlc-version", "3304.7.1"}, {"coremltools-component-torch", "2.1.1"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "7.2"}})]
{
func main<ios17>(tensor<fp16, [?, 384]> files, tensor<fp16, [1, 384]> query) [FlexibleShapeInformation = tuple<tuple<tensor<string, []>, dict<tensor<string, []>, tensor<int32, [?]>>>, tuple<tensor<string, []>, dict<tensor<string, []>, list<tensor<int32, [2]>, ?>>>>((("DefaultShapes", {{"files", [1, 384]}}), ("RangeDims", {{"files", [[1, 65536], [384, 384]]}})))] {
tensor<bool, []> matmul_0_transpose_x_1 = const()[name = tensor<string, []>("matmul_0_transpose_x_1"), val = tensor<bool, []>(false)];
tensor<bool, []> matmul_0_transpose_y_1 = const()[name = tensor<string, []>("matmul_0_transpose_y_1"), val = tensor<bool, []>(true)];
tensor<fp16, [1, ?]> outputs = matmul(transpose_x = matmul_0_transpose_x_1, transpose_y = matmul_0_transpose_y_1, x = query, y = files)[name = tensor<string, []>("matmul_0_cast_fp16")];
} -> (outputs);
} |