@@ -1052,7 +1052,7 @@ func.func @max_pool2d_nan_ignore(%arg0: tensor<1x6x34x62xf32>) -> (tensor<1x4x32
10521052func.func @conv2d_local_bound_true (%input: tensor <1 x49 x42 x27 xf32 >, %weights: tensor <28 x3 x3 x27 xf32 >, %bias: tensor <1 xf32 >) -> () {
10531053 %input_zp = " tosa.const" () <{values = dense <0.0 > : tensor <1 xf32 >}> : () -> tensor <1 xf32 >
10541054 %weight_zp = " tosa.const" () <{values = dense <0.0 > : tensor <1 xf32 >}> : () -> tensor <1 xf32 >
1055- // CHECK: linalg.conv_2d_nhwc_fhwc {dilations = dense<[2, 1]> : tensor<2xi64>, tosa.local_bound = true, strides = dense<1> : tensor<2xi64>}
1055+ // CHECK: linalg.conv_2d_nhwc_fhwc {dilations = dense<[2, 1]> : tensor<2xi64>, strides = dense<1> : tensor<2xi64>, tosa.local_bound = true }
10561056 %0 = tosa.conv2d %input , %weights , %bias , %input_zp , %weight_zp {acc_type = f32 , local_bound = true , pad = array<i64 : 0 , 0 , 0 , 0 >, stride = array<i64 : 1 , 1 >, dilation = array<i64 : 2 , 1 >} : (tensor <1 x49 x42 x27 xf32 >, tensor <28 x3 x3 x27 xf32 >, tensor <1 xf32 >, tensor <1 xf32 >, tensor <1 xf32 >) -> tensor <1 x45 x40 x28 xf32 >
10571057 return
10581058}
@@ -1074,7 +1074,7 @@ func.func @conv2d_local_bound_false(%input: tensor<1x49x42x27xf32>, %weights: te
10741074func.func @conv3d_local_bound_true (%input: tensor <1 x49 x48 x47 x27 xf32 >, %weights: tensor <28 x3 x4 x5 x27 xf32 >, %bias: tensor <1 xf32 >) -> () {
10751075 %input_zp = " tosa.const" () <{values = dense <0.0 > : tensor <1 xf32 >}> : () -> tensor <1 xf32 >
10761076 %weight_zp = " tosa.const" () <{values = dense <0.0 > : tensor <1 xf32 >}> : () -> tensor <1 xf32 >
1077- // CHECK: linalg.conv_3d_ndhwc_dhwcf {dilations = dense<1> : tensor<3xi64>, tosa.local_bound = true, strides = dense<1> : tensor<3xi64>}
1077+ // CHECK: linalg.conv_3d_ndhwc_dhwcf {dilations = dense<1> : tensor<3xi64>, strides = dense<1> : tensor<3xi64>, tosa.local_bound = true }
10781078 %0 = tosa.conv3d %input , %weights , %bias , %input_zp , %weight_zp {acc_type = f32 , local_bound = true , pad = array<i64 : 0 , 0 , 0 , 0 , 0 , 0 >, stride = array<i64 : 1 , 1 , 1 >, dilation = array<i64 : 1 , 1 , 1 >} : (tensor <1 x49 x48 x47 x27 xf32 >, tensor <28 x3 x4 x5 x27 xf32 >, tensor <1 xf32 >, tensor <1 xf32 >, tensor <1 xf32 >) -> tensor <1 x47 x45 x43 x28 xf32 >
10791079 return
10801080}
@@ -1096,7 +1096,7 @@ func.func @conv3d_local_bound_false(%input: tensor<1x49x48x47x27xf32>, %weights:
10961096func.func @depthwise_conv_local_bound_true (%arg0 : tensor <1 x7 x5 x3 xf32 >, %arg1 : tensor <3 x1 x3 x11 xf32 >, %arg2 : tensor <1 xf32 >) -> () {
10971097 %input_zp = " tosa.const" () <{values = dense <0.0 > : tensor <1 xf32 >}> : () -> tensor <1 xf32 >
10981098 %weight_zp = " tosa.const" () <{values = dense <0.0 > : tensor <1 xf32 >}> : () -> tensor <1 xf32 >
1099- // CHECK: linalg.depthwise_conv_2d_nhwc_hwcm {dilations = dense<1> : tensor<2xi64>, tosa.local_bound = true, strides = dense<1> : tensor<2xi64>}
1099+ // CHECK: linalg.depthwise_conv_2d_nhwc_hwcm {dilations = dense<1> : tensor<2xi64>, strides = dense<1> : tensor<2xi64>, tosa.local_bound = true }
11001100 %2 = tosa.depthwise_conv2d %arg0 , %arg1 , %arg2 , %input_zp , %weight_zp {acc_type = f32 , local_bound = true , pad = array<i64 : 0 , 0 , 0 , 0 >, stride = array<i64 : 1 , 1 >, dilation = array<i64 : 1 , 1 > } : (tensor <1 x7 x5 x3 xf32 >, tensor <3 x1 x3 x11 xf32 >, tensor <1 xf32 >, tensor <1 xf32 >, tensor <1 xf32 >) -> tensor <1 x5 x5 x33 xf32 >
11011101 return
11021102}
0 commit comments