Lines Matching full:pad
7 func.func @static_data_only(%arg0 : tensor<4x5xf32>, %pad : f32)
9 %0 = tensor.pad %arg0 low[0, 0] high[7, 8] {
11 tensor.yield %pad : f32
20 // CHECK-SAME: %[[ARG0:.*]]: tensor<4x5xf32>, %[[PAD:.*]]: f32
21 // CHECK-NOT: tensor.pad
24 // CHECK: tensor.yield %[[PAD]]
26 func.func @static_high_pad_only(%arg0 : tensor<4x5xf32>, %pad : f32)
28 %0 = tensor.pad %arg0 low[0, 0] high[7, 8] {
30 tensor.yield %pad : f32
39 // CHECK-SAME: %[[ARG0:.*]]: tensor<4x5xf32>, %[[PAD:.*]]: f32
40 // CHECK-NOT: tensor.pad
43 // CHECK: tensor.yield %[[PAD]]
45 func.func @static_low_pad_only(%arg0 : tensor<4x5xf32>, %pad : f32)
47 %0 = tensor.pad %arg0 low[3, 7] high[7, 8] {
49 tensor.yield %pad : f32
58 // CHECK-SAME: %[[ARG0:.*]]: tensor<4x5xf32>, %[[PAD:.*]]: f32
59 // CHECK-NOT: tensor.pad
62 // CHECK: tensor.yield %[[PAD]]
64 func.func @static_low_pad_only_2(%arg0 : tensor<4x5xf32>, %pad : f32)
66 %0 = tensor.pad %arg0 low[3, 7] high[7, 8] {
68 tensor.yield %pad : f32
77 // CHECK-SAME: %[[ARG0:.*]]: tensor<4x5xf32>, %[[PAD:.*]]: f32
78 // CHECK-NOT: tensor.pad
80 // CHECK: %[[RESULT:.*]] = tensor.pad %[[SUBTENSOR]] low[0, 0] high[1, 3]
81 // CHECK: tensor.yield %[[PAD]]
83 func.func @static_mixed_data_high_pad(%arg0 : tensor<4x5xf32>, %pad : f32)
85 %0 = tensor.pad %arg0 low[0, 0] high[7, 8] {
87 tensor.yield %pad : f32
96 // CHECK-SAME: %[[ARG0:.*]]: tensor<4x5xf32>, %[[PAD:.*]]: f32
97 // CHECK-NOT: tensor.pad
99 // CHECK: %[[RESULT:.*]] = tensor.pad %[[SUBTENSOR]] low[1, 3] high[0, 0]
100 // CHECK: tensor.yield %[[PAD]]
102 func.func @static_mixed_data_low_pad(%arg0 : tensor<4x5xf32>, %pad : f32)
104 %0 = tensor.pad %arg0 low[3, 7] high[7, 8] {
106 tensor.yield %pad : f32
115 // CHECK-SAME: %[[ARG0:.*]]: tensor<4x5xf32>, %[[PAD:.*]]: f32
116 // CHECK-NOT: tensor.pad
117 // CHECK: %[[RESULT:.*]] = tensor.pad %[[ARG0]] low[1, 1] high[2, 3]
118 // CHECK: tensor.yield %[[PAD]]
120 func.func @static_mixed_data_low_high_pad(%arg0 : tensor<4x5xf32>, %pad : f32)
122 %0 = tensor.pad %arg0 low[2, 3] high[7, 8] {
124 tensor.yield %pad : f32
134 // CHECK-NOT: tensor.pad
142 // CHECK: %[[PADTENSOR:.*]] = tensor.pad %[[SUBTENSOR]] low[0, 0] high[%{{.*}}, 3]
146 func.func @dynamic_high_pad(%arg0 : tensor<?x5xf32>, %h1: index, %pad : f32) -> tensor<3x4xf32> {
147 %0 = tensor.pad %arg0 low[0, 0] high[%h1, 8] {
149 tensor.yield %pad : f32
159 // CHECK-NOT: tensor.pad
167 // CHECK: %[[PADTENSOR:.*]] = tensor.pad %[[SUBTENSOR]] low[0, 0] high[%{{.*}}, 3]
171 func.func @dynamic_extract_size(%arg0 : tensor<?x5xf32>, %s1: index, %pad : f32) -> tensor<?x4xf32>…
172 %0 = tensor.pad %arg0 low[0, 0] high[7, 8] {
174 tensor.yield %pad : f32
187 // CHECK: tensor.pad %[[SLICE]] low[0, 0]
188 func.func @dynamic_zero_low_padding(%arg0 : tensor<?x?xf32>, %pad : f32,
192 %0 = tensor.pad %arg0 low[0, 0] high[7, 8] {
194 tensor.yield %pad : f32
207 // CHECK: tensor.pad %[[SLICE]] low[%{{.*}}, %{{.*}}] high[0, 0]
208 func.func @dynamic_zero_high_padding(%arg0 : tensor<?x?xf32>, %pad : f32,
212 %0 = tensor.pad %arg0 low[7, 8] high[0, 0] {
214 tensor.yield %pad : f32