@@ -170,8 +170,8 @@ func.func @test_pad_f32_non_constant_padval(%arg0: tensor<20x16x44x32xf32>, %arg
170170 return %2 : tensor <24 x22 x52 x42 xf32 >
171171// CHECK-LABEL: func.func @test_pad_f32_non_constant_padval
172172// CHECK-SAME: ([[PARAM_0_:%.+]]: tensor<20x16x44x32xf32>, [[PARAM_1_:%.+]]: tensor<f32>) -> tensor<24x22x52x42xf32> {
173- // CHECK: [[VAR_0_:%.+]] = " tosa.const"() < {value = dense<[0, 4, 1, 5, 2, 6, 3, 7]> : tensor<8xi64>}> : () -> tensor<8xi64 >
174- // CHECK: [[VAR_1_:%.+]] = tosa.pad [[PARAM_0_]], [[VAR_0_]], [[PARAM_1_]] : (tensor<20x16x44x32xf32>, tensor<8xi64 >, tensor<f32>) -> tensor<24x22x52x42xf32>
173+ // CHECK: [[VAR_0_:%.+]] = tosa.const_shape {value = dense<[0, 4, 1, 5, 2, 6, 3, 7]> : tensor<8xindex>} : () -> !tosa.shape<8 >
174+ // CHECK: [[VAR_1_:%.+]] = tosa.pad [[PARAM_0_]], [[VAR_0_]], [[PARAM_1_]] : (tensor<20x16x44x32xf32>, !tosa.shape<8 >, tensor<f32>) -> tensor<24x22x52x42xf32>
175175// CHECK: return [[VAR_1_]] : tensor<24x22x52x42xf32>
176176}
177177
@@ -184,9 +184,9 @@ func.func @test_pad_f32_non_constant_1Dpadval(%arg0: tensor<20x16x44x32xf32>, %a
184184 return %2 : tensor <24 x22 x52 x42 xf32 >
185185// CHECK-LABEL: func.func @test_pad_f32_non_constant_1Dpadval
186186// CHECK-SAME: ([[PARAM_0_:%.+]]: tensor<20x16x44x32xf32>, [[PARAM_1_:%.+]]: tensor<1xf32>) -> tensor<24x22x52x42xf32> {
187- // CHECK-DAG: [[VAR_0_:%.+]] = " tosa.const"() < {value = dense<[0, 4, 1, 5, 2, 6, 3, 7]> : tensor<8xi64>}> : () -> tensor<8xi64 >
187+ // CHECK-DAG: [[VAR_0_:%.+]] = tosa.const_shape {value = dense<[0, 4, 1, 5, 2, 6, 3, 7]> : tensor<8xindex>} : () -> !tosa.shape<8 >
188188// CHECK-DAG: [[VAL_1_:%.+]] = tosa.reshape [[PARAM_1_]] {new_shape = array<i64>} : (tensor<1xf32>) -> tensor<f32>
189- // CHECK: [[VAR_2_:%.+]] = tosa.pad [[PARAM_0_]], [[VAR_0_]], [[VAL_1_]] : (tensor<20x16x44x32xf32>, tensor<8xi64 >, tensor<f32>) -> tensor<24x22x52x42xf32>
189+ // CHECK: [[VAR_2_:%.+]] = tosa.pad [[PARAM_0_]], [[VAR_0_]], [[VAL_1_]] : (tensor<20x16x44x32xf32>, !tosa.shape<8 >, tensor<f32>) -> tensor<24x22x52x42xf32>
190190// CHECK: return [[VAR_2_]] : tensor<24x22x52x42xf32>
191191}
192192
@@ -199,8 +199,8 @@ func.func @test_pad_i64_non_constant_padval(%arg0: tensor<20x16x44x32xi64>, %arg
199199 return %2 : tensor <24 x22 x52 x42 xi64 >
200200// CHECK-LABEL: func.func @test_pad_i64_non_constant_padval
201201// CHECK-SAME: ([[PARAM_0_:%.+]]: tensor<20x16x44x32xi64>, [[PARAM_1_:%.+]]: tensor<i64>) -> tensor<24x22x52x42xi64> {
202- // CHECK: [[VAR_0_:%.+]] = " tosa.const"() < {value = dense<[0, 4, 1, 5, 2, 6, 3, 7]> : tensor<8xi64>}> : () -> tensor<8xi64 >
203- // CHECK: [[VAR_1_:%.+]] = tosa.pad [[PARAM_0_]], [[VAR_0_]], [[PARAM_1_]] : (tensor<20x16x44x32xi64>, tensor<8xi64 >, tensor<i64>) -> tensor<24x22x52x42xi64>
202+ // CHECK: [[VAR_0_:%.+]] = tosa.const_shape {value = dense<[0, 4, 1, 5, 2, 6, 3, 7]> : tensor<8xindex>} : () -> !tosa.shape<8 >
203+ // CHECK: [[VAR_1_:%.+]] = tosa.pad [[PARAM_0_]], [[VAR_0_]], [[PARAM_1_]] : (tensor<20x16x44x32xi64>, !tosa.shape<8 >, tensor<i64>) -> tensor<24x22x52x42xi64>
204204// CHECK: return [[VAR_1_]] : tensor<24x22x52x42xi64>
205205}
206206
@@ -212,7 +212,7 @@ func.func @test_pad_f16_non_constant_padval(%arg0: tensor<20x16x44x32xf16>, %arg
212212 return %2 : tensor <24 x22 x52 x42 xf16 >
213213// CHECK-LABEL: func.func @test_pad_f16_non_constant_padval
214214// CHECK-SAME: ([[PARAM_0_:%.+]]: tensor<20x16x44x32xf16>, [[PARAM_1_:%.+]]: tensor<f16>) -> tensor<24x22x52x42xf16> {
215- // CHECK: [[VAR_0_:%.+]] = " tosa.const"() < {value = dense<[0, 4, 1, 5, 2, 6, 3, 7]> : tensor<8xi64>}> : () -> tensor<8xi64 >
216- // CHECK: [[VAR_1_:%.+]] = tosa.pad [[PARAM_0_]], [[VAR_0_]], [[PARAM_1_]] : (tensor<20x16x44x32xf16>, tensor<8xi64 >, tensor<f16>) -> tensor<24x22x52x42xf16>
215+ // CHECK: [[VAR_0_:%.+]] = tosa.const_shape {value = dense<[0, 4, 1, 5, 2, 6, 3, 7]> : tensor<8xindex>} : () -> !tosa.shape<8 >
216+ // CHECK: [[VAR_1_:%.+]] = tosa.pad [[PARAM_0_]], [[VAR_0_]], [[PARAM_1_]] : (tensor<20x16x44x32xf16>, !tosa.shape<8 >, tensor<f16>) -> tensor<24x22x52x42xf16>
217217// CHECK: return [[VAR_1_]] : tensor<24x22x52x42xf16>
218218}
0 commit comments