22
33#map = affine_map <() -> ()>
44
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 > {
66 %0 = tensor.empty () : tensor <f32 >
77 %1 = linalg.generic {index ing_maps = [#map , #map , #map ], iterator_types = []}
88 ins (%arg1 , %arg2 : tensor <f32 >, tensor <f32 >)
@@ -21,7 +21,7 @@ func.func @detensor_simple(%arg1: tensor<f32>, %arg2: tensor<f32>) -> tensor<f32
2121// CHECK: %[[new_tensor_res:.*]] = tensor.from_elements %[[detensored_res]]
2222// CHECK: return %[[new_tensor_res]]
2323
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 > {
2525 %0 = tensor.empty () : tensor <f32 >
2626 %1 = linalg.generic {index ing_maps = [#map , #map , #map ], iterator_types = []}
2727 ins (%arg1 , %arg2 : tensor <f32 >, tensor <f32 >)
@@ -61,7 +61,7 @@ func.func @detensor_op_sequence(%arg1: tensor<f32>, %arg2: tensor<f32>) -> tenso
6161// CHECK: %[[new_tensor_res:.*]] = tensor.from_elements %[[detensored_res3]]
6262// CHECK: return %[[new_tensor_res]]
6363
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 > {
6565 %0 = tensor.empty () : tensor <f32 >
6666 %1 = linalg.generic {index ing_maps = [#map , #map , #map ], iterator_types = []}
6767 ins (%arg1 , %arg2 : tensor <f32 >, tensor <f32 >)
@@ -82,7 +82,7 @@ func.func @detensor_multiple_ops(%arg1: tensor<f32>, %arg2: tensor<f32>) -> tens
8282// CHECK: %[[new_tensor_res:.*]] = tensor.from_elements %[[detensored_res2]]
8383// CHECK: return %[[new_tensor_res]]
8484
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 > {
8686 %0 = tensor.empty () : tensor <f32 >
8787 %1 = linalg.generic {index ing_maps = [#map , #map , #map ], iterator_types = []}
8888 ins (%arg1 , %arg2 : tensor <f32 >, tensor <f32 >)
0 commit comments