Lines Matching full:tensor
4 …CK-SAME: (%[[INPUT:.+]]: tensor<1x4x4x6xf64>, %[[FILTER:.+]]: tensor<8x2x2x6xf64>, %[[INIT:.+]]: t…
5 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xf64>
6 …[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xf64>) outs(%[[NEWF]] : tensor…
7 …tensor<2xi64>, strides = dense<2> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]] : tensor<1x4x4x6…
8 // CHECK: return %[[CONV]] : tensor<1x2x2x8xf64>
9 … @conv_2d_nhwc_fhwc_f64(%input: tensor<1x4x4x6xf64>, %filter: tensor<8x2x2x6xf64>, %init: tensor<1…
10 %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
11 strides = dense<2> : tensor<2xi64>}
12 ins (%input, %filter: tensor<1x4x4x6xf64>, tensor<8x2x2x6xf64>)
13 outs (%init: tensor<1x2x2x8xf64>) -> tensor<1x2x2x8xf64>
14 return %0 : tensor<1x2x2x8xf64>
18 …CK-SAME: (%[[INPUT:.+]]: tensor<1x4x4x6xf32>, %[[FILTER:.+]]: tensor<8x2x2x6xf32>, %[[INIT:.+]]: t…
19 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xf32>
20 …[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xf32>) outs(%[[NEWF]] : tensor…
21 …tensor<2xi64>, strides = dense<2> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]] : tensor<1x4x4x6…
22 // CHECK: return %[[CONV]] : tensor<1x2x2x8xf32>
23 … @conv_2d_nhwc_fhwc_f32(%input: tensor<1x4x4x6xf32>, %filter: tensor<8x2x2x6xf32>, %init: tensor<1…
24 %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
25 strides = dense<2> : tensor<2xi64>}
26 ins (%input, %filter: tensor<1x4x4x6xf32>, tensor<8x2x2x6xf32>)
27 outs (%init: tensor<1x2x2x8xf32>) -> tensor<1x2x2x8xf32>
28 return %0 : tensor<1x2x2x8xf32>
32 …CK-SAME: (%[[INPUT:.+]]: tensor<1x4x4x6xf16>, %[[FILTER:.+]]: tensor<8x2x2x6xf16>, %[[INIT:.+]]: t…
33 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xf16>
34 …[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xf16>) outs(%[[NEWF]] : tensor…
35 …tensor<2xi64>, strides = dense<2> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]] : tensor<1x4x4x6…
36 // CHECK: return %[[CONV]] : tensor<1x2x2x8xf16>
37 … @conv_2d_nhwc_fhwc_f16(%input: tensor<1x4x4x6xf16>, %filter: tensor<8x2x2x6xf16>, %init: tensor<1…
38 %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
39 strides = dense<2> : tensor<2xi64>}
40 ins (%input, %filter: tensor<1x4x4x6xf16>, tensor<8x2x2x6xf16>)
41 outs (%init: tensor<1x2x2x8xf16>) -> tensor<1x2x2x8xf16>
42 return %0 : tensor<1x2x2x8xf16>
46 …K-SAME: (%[[INPUT:.+]]: tensor<1x4x4x6xbf16>, %[[FILTER:.+]]: tensor<8x2x2x6xbf16>, %[[INIT:.+]]: …
47 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xbf16>
48 …[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xbf16>) outs(%[[NEWF]] : tenso…
49 …tensor<2xi64>, strides = dense<2> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]] : tensor<1x4x4x6…
50 // CHECK: return %[[CONV]] : tensor<1x2x2x8xbf16>
51 …@conv_2d_nhwc_fhwc_b16(%input: tensor<1x4x4x6xbf16>, %filter: tensor<8x2x2x6xbf16>, %init: tensor<…
52 %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
53 strides = dense<2> : tensor<2xi64>}
54 ins (%input, %filter: tensor<1x4x4x6xbf16>, tensor<8x2x2x6xbf16>)
55 outs (%init: tensor<1x2x2x8xbf16>) -> tensor<1x2x2x8xbf16>
56 return %0 : tensor<1x2x2x8xbf16>
60 …CK-SAME: (%[[INPUT:.+]]: tensor<1x4x4x6xi64>, %[[FILTER:.+]]: tensor<8x2x2x6xi64>, %[[INIT:.+]]: t…
61 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xi64>
62 …[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xi64>) outs(%[[NEWF]] : tensor…
63 …tensor<2xi64>, strides = dense<2> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]] : tensor<1x4x4x6…
64 // CHECK: return %[[CONV]] : tensor<1x2x2x8xi64>
65 … @conv_2d_nhwc_fhwc_i64(%input: tensor<1x4x4x6xi64>, %filter: tensor<8x2x2x6xi64>, %init: tensor<1…
66 %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
67 strides = dense<2> : tensor<2xi64>}
68 ins (%input, %filter: tensor<1x4x4x6xi64>, tensor<8x2x2x6xi64>)
69 outs (%init: tensor<1x2x2x8xi64>) -> tensor<1x2x2x8xi64>
70 return %0 : tensor<1x2x2x8xi64>
74 …CK-SAME: (%[[INPUT:.+]]: tensor<1x4x4x6xi32>, %[[FILTER:.+]]: tensor<8x2x2x6xi32>, %[[INIT:.+]]: t…
75 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xi32>
76 …[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xi32>) outs(%[[NEWF]] : tensor…
77 …tensor<2xi64>, strides = dense<2> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]] : tensor<1x4x4x6…
78 // CHECK: return %[[CONV]] : tensor<1x2x2x8xi32>
79 … @conv_2d_nhwc_fhwc_i32(%input: tensor<1x4x4x6xi32>, %filter: tensor<8x2x2x6xi32>, %init: tensor<1…
80 %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
81 strides = dense<2> : tensor<2xi64>}
82 ins (%input, %filter: tensor<1x4x4x6xi32>, tensor<8x2x2x6xi32>)
83 outs (%init: tensor<1x2x2x8xi32>) -> tensor<1x2x2x8xi32>
84 return %0 : tensor<1x2x2x8xi32>
88 …CK-SAME: (%[[INPUT:.+]]: tensor<1x4x4x6xi16>, %[[FILTER:.+]]: tensor<8x2x2x6xi16>, %[[INIT:.+]]: t…
89 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xi16>
90 …[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xi16>) outs(%[[NEWF]] : tensor…
91 …tensor<2xi64>, strides = dense<2> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]] : tensor<1x4x4x6…
92 // CHECK: return %[[CONV]] : tensor<1x2x2x8xi16>
93 … @conv_2d_nhwc_fhwc_i16(%input: tensor<1x4x4x6xi16>, %filter: tensor<8x2x2x6xi16>, %init: tensor<1…
94 %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
95 strides = dense<2> : tensor<2xi64>}
96 ins (%input, %filter: tensor<1x4x4x6xi16>, tensor<8x2x2x6xi16>)
97 outs (%init: tensor<1x2x2x8xi16>) -> tensor<1x2x2x8xi16>
98 return %0 : tensor<1x2x2x8xi16>
102 …HECK-SAME: (%[[INPUT:.+]]: tensor<1x4x4x6xi8>, %[[FILTER:.+]]: tensor<8x2x2x6xi8>, %[[INIT:.+]]: t…
103 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xi8>
104 …%[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xi8>) outs(%[[NEWF]] : tensor…
105 …tensor<2xi64>, strides = dense<2> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]] : tensor<1x4x4x6…
106 // CHECK: return %[[CONV]] : tensor<1x2x2x8xi8>
107 …unc @conv_2d_nhwc_fhwc_i8(%input: tensor<1x4x4x6xi8>, %filter: tensor<8x2x2x6xi8>, %init: tensor<1…
108 %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
109 strides = dense<2> : tensor<2xi64>}
110 ins (%input, %filter: tensor<1x4x4x6xi8>, tensor<8x2x2x6xi8>)
111 outs (%init: tensor<1x2x2x8xi8>) -> tensor<1x2x2x8xi8>
112 return %0 : tensor<1x2x2x8xi8>
116 …UT:.+]]: tensor<1x4x4x6xf32>, %[[FILTER:.+]]: tensor<8x2x2x6xf32>, %[[INIT:.+]]: tensor<1x2x2x8xf3…
117 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xf32>
118 …[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xf32>) outs(%[[NEWF]] : tensor…
119 …tensor<2xi64>, strides = dense<2> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]], %[[A]], %[[B]] …
120 // CHECK: return %[[CONV]] : tensor<1x2x2x8xf32>
121 …2d_nhwc_fhwc_q(%input: tensor<1x4x4x6xf32>, %filter: tensor<8x2x2x6xf32>, %init: tensor<1x2x2x8xf3…
122 %0 = linalg.conv_2d_nhwc_fhwc_q {dilations = dense<1> : tensor<2xi64>,
123 strides = dense<2> : tensor<2xi64>}
124 ins (%input, %filter, %a, %b: tensor<1x4x4x6xf32>, tensor<8x2x2x6xf32>, i32, i32)
125 outs (%init: tensor<1x2x2x8xf32>) -> tensor<1x2x2x8xf32>
126 return %0 : tensor<1x2x2x8xf32>
130 …CK-SAME: (%[[INPUT:.+]]: tensor<1x4x4x6xf32>, %[[FILTER:.+]]: tensor<8x2x2x6xf32>, %[[INIT:.+]]: t…
131 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xf32>
132 …[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xf32>) outs(%[[NEWF]] : tensor…
133 …tensor<2xi64>, strides = dense<1> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]] : tensor<1x4x4x6…
134 // CHECK: return %[[CONV]] : tensor<1x3x3x8xf32>
135 …wc_fhwc_f32_unit_stride(%input: tensor<1x4x4x6xf32>, %filter: tensor<8x2x2x6xf32>, %init: tensor<1…
136 %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
137 strides = dense<1> : tensor<2xi64>}
138 ins (%input, %filter: tensor<1x4x4x6xf32>, tensor<8x2x2x6xf32>)
139 outs (%init: tensor<1x3x3x8xf32>) -> tensor<1x3x3x8xf32>
140 return %0 : tensor<1x3x3x8xf32>
144 …CK-SAME: (%[[INPUT:.+]]: tensor<1x4x4x6xf32>, %[[FILTER:.+]]: tensor<8x2x2x6xf32>, %[[INIT:.+]]: t…
145 // CHECK-DAG: %[[NEWF:.+]] = tensor.empty() : tensor<2x2x6x8xf32>
146 …[[TRANSPOSE:.+]] = linalg.transpose ins(%[[FILTER]] : tensor<8x2x2x6xf32>) outs(%[[NEWF]] : tensor…
147 …tensor<2xi64>, strides = dense<1> : tensor<2xi64>} ins(%[[INPUT]], %[[TRANSPOSE]] : tensor<1x4x4x6…
148 // CHECK: return %[[CONV]] : tensor<1x2x2x8xf32>
149 …wc_fhwc_f32_2_dialation(%input: tensor<1x4x4x6xf32>, %filter: tensor<8x2x2x6xf32>, %init: tensor<1…
150 %0 = linalg.conv_2d_nhwc_fhwc {dilations = dense<2> : tensor<2xi64>,
151 strides = dense<1> : tensor<2xi64>}
152 ins (%input, %filter: tensor<1x4x4x6xf32>, tensor<8x2x2x6xf32>)
153 outs (%init: tensor<1x2x2x8xf32>) -> tensor<1x2x2x8xf32>
154 return %0 : tensor<1x2x2x8xf32>
161 …K: linalg.conv_2d_nhwc_hwcf {dilations = dense<1> : tensor<2xi64>, strides = dense<2> : tensor<…
164 linalg.conv_2d_nhwc_fhwc {dilations = dense<1> : tensor<2xi64>,
165 strides = dense<2> : tensor<2xi64>}