2
2
3
3
#map = affine_map <() -> ()>
4
4
5
- func.func @detensor_simple (%arg1: tensor <f32 >, %arg2: tensor <f32 >) -> tensor <f32 > attributes { iree.module.export } {
5
+ func.func @detensor_simple (%arg1: tensor <f32 >, %arg2: tensor <f32 >) -> tensor <f32 > {
6
6
%0 = tensor.empty () : tensor <f32 >
7
7
%1 = linalg.generic {index ing_maps = [#map , #map , #map ], iterator_types = []}
8
8
ins (%arg1 , %arg2 : tensor <f32 >, tensor <f32 >)
@@ -21,7 +21,7 @@ func.func @detensor_simple(%arg1: tensor<f32>, %arg2: tensor<f32>) -> tensor<f32
21
21
// CHECK: %[[new_tensor_res:.*]] = tensor.from_elements %[[detensored_res]]
22
22
// CHECK: return %[[new_tensor_res]]
23
23
24
- func.func @detensor_op_sequence (%arg1: tensor <f32 >, %arg2: tensor <f32 >) -> tensor <f32 > attributes { iree.module.export } {
24
+ func.func @detensor_op_sequence (%arg1: tensor <f32 >, %arg2: tensor <f32 >) -> tensor <f32 > {
25
25
%0 = tensor.empty () : tensor <f32 >
26
26
%1 = linalg.generic {index ing_maps = [#map , #map , #map ], iterator_types = []}
27
27
ins (%arg1 , %arg2 : tensor <f32 >, tensor <f32 >)
@@ -61,7 +61,7 @@ func.func @detensor_op_sequence(%arg1: tensor<f32>, %arg2: tensor<f32>) -> tenso
61
61
// CHECK: %[[new_tensor_res:.*]] = tensor.from_elements %[[detensored_res3]]
62
62
// CHECK: return %[[new_tensor_res]]
63
63
64
- func.func @detensor_multiple_ops (%arg1: tensor <f32 >, %arg2: tensor <f32 >) -> tensor <f32 > attributes { iree.module.export } {
64
+ func.func @detensor_multiple_ops (%arg1: tensor <f32 >, %arg2: tensor <f32 >) -> tensor <f32 > {
65
65
%0 = tensor.empty () : tensor <f32 >
66
66
%1 = linalg.generic {index ing_maps = [#map , #map , #map ], iterator_types = []}
67
67
ins (%arg1 , %arg2 : tensor <f32 >, tensor <f32 >)
@@ -82,7 +82,7 @@ func.func @detensor_multiple_ops(%arg1: tensor<f32>, %arg2: tensor<f32>) -> tens
82
82
// CHECK: %[[new_tensor_res:.*]] = tensor.from_elements %[[detensored_res2]]
83
83
// CHECK: return %[[new_tensor_res]]
84
84
85
- func.func @detensor_foreign_op (%arg1: tensor <f32 >, %arg2: tensor <f32 >) -> tensor <f32 > attributes { iree.module.export } {
85
+ func.func @detensor_foreign_op (%arg1: tensor <f32 >, %arg2: tensor <f32 >) -> tensor <f32 > {
86
86
%0 = tensor.empty () : tensor <f32 >
87
87
%1 = linalg.generic {index ing_maps = [#map , #map , #map ], iterator_types = []}
88
88
ins (%arg1 , %arg2 : tensor <f32 >, tensor <f32 >)
0 commit comments