Removing test name

tenstorrent · Dec 27, 2024 · c8f0156 · c8f0156
1 parent 435c787
commit c8f0156
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/ttmlir/Silicon/TTNN/simple_mean.mlir b/test/ttmlir/Silicon/TTNN/simple_mean.mlir
@@ -37,7 +37,7 @@ module {
     return %1 : tensor<128x1xf32>
   }
 
-  func.func public @mean_into_reshape_dim_array(%arg0: tensor<1x1x49x2048xf32>) -> (tensor<1x2048x1x1xf32> {ttir.name = "AvgPool2d.output_avg_pool2d_0"}) {
+  func.func public @mean_into_reshape_dim_array(%arg0: tensor<1x1x49x2048xf32>) -> tensor<1x2048x1x1xf32> {
     // CHECK: "ttnn.mean"
     // CHECK-SAME: {dim = [-2 : i32], keep_dim = true}
     // CHECK: "ttnn.reshape"
@@ -48,7 +48,7 @@ module {
     return %4 : tensor<1x2048x1x1xf32>
   }
 
-  func.func public @mean_into_reshape_dim_scalar(%arg0: tensor<1x1x49x2048xf32>) -> (tensor<1x2048x1x1xf32> {ttir.name = "AvgPool2d.output_avg_pool2d_0"}) {
+  func.func public @mean_into_reshape_dim_scalar(%arg0: tensor<1x1x49x2048xf32>) -> tensor<1x2048x1x1xf32> {
     // CHECK: "ttnn.mean"
     // CHECK-SAME: {dim = -2 : si32, keep_dim = true}
     // CHECK: "ttnn.reshape"