Lines Matching full:index
4 func.func @unknown_op() -> index {
7 %1 = "test.reify_bound"(%0) {dim = 0} : (tensor<?x?xf32>) -> (index)
8 return %1 : index
14 // CHECK: %[[c10:.*]] = arith.constant 10 : index
16 func.func @cast(%t: tensor<10xf32>) -> index {
18 %1 = "test.reify_bound"(%0) {dim = 0} : (tensor<?xf32>) -> (index)
19 return %1 : index
24 func.func @cast_unranked(%t: tensor<*xf32>) -> index {
27 %1 = "test.reify_bound"(%0) {dim = 0} : (tensor<?xf32>) -> (index)
28 return %1 : index
38 func.func @dim(%t: tensor<?xf32>) -> index {
39 %c0 = arith.constant 0 : index
41 %1 = "test.reify_bound"(%0) : (index) -> (index)
42 return %1 : index
48 func.func @dim_all_positive(%t: tensor<?xf32>, %x: index) {
49 %c0 = arith.constant 0 : index
52 "test.compare"(%0, %c0) {cmp = "GE" } : (index, index) -> ()
59 // CHECK-SAME: %[[sz:.*]]: index
60 // CHECK: %[[c6:.*]] = arith.constant 6 : index
62 func.func @empty(%sz: index) -> (index, index) {
64 %1 = "test.reify_bound"(%0) {dim = 0} : (tensor<6x?xf32>) -> (index)
65 %2 = "test.reify_bound"(%0) {dim = 1} : (tensor<6x?xf32>) -> (index)
66 return %1, %2 : index, index
72 // CHECK-SAME: %[[t:.*]]: tensor<?xf32>, %[[sz:.*]]: index
74 func.func @extract_slice_dynamic(%t: tensor<?xf32>, %sz: index) -> index {
76 %1 = "test.reify_bound"(%0) {dim = 0} : (tensor<?xf32>) -> (index)
77 return %1 : index
84 // CHECK: %[[c5:.*]] = arith.constant 5 : index
86 func.func @extract_slice_static(%t: tensor<?xf32>) -> index {
88 %1 = "test.reify_bound"(%0) {dim = 0} : (tensor<5xf32>) -> (index)
89 return %1 : index
94 func.func @extract_slice_dynamic_constant(%t: tensor<?xf32>, %sz: index) -> index {
97 %1 = "test.reify_bound"(%0) {dim = 0, constant} : (tensor<?xf32>) -> (index)
98 return %1 : index
105 // CHECK: %[[c5:.*]] = arith.constant 5 : index
107 func.func @extract_slice_static_constant(%t: tensor<?xf32>) -> index {
109 %1 = "test.reify_bound"(%0) {dim = 0, constant} : (tensor<5xf32>) -> (index)
110 return %1 : index
116 // CHECK-SAME: %[[t:.*]]: tensor<?x?xf32>, %[[sz:.*]]: index
118 func.func @extract_slice_rank_reduce(%t: tensor<?x?xf32>, %sz: index) -> index {
120 %1 = "test.reify_bound"(%0) {dim = 0} : (tensor<?xf32>) -> (index)
121 return %1 : index
128 // CHECK: %[[c0:.*]] = arith.constant 0 : index
131 func.func @insert(%t: tensor<?xf32>, %f: f32, %pos: index) -> index {
133 %1 = "test.reify_bound"(%0) {dim = 0} : (tensor<?xf32>) -> (index)
134 return %1 : index
142 // CHECK-SAME: %[[t:.*]]: tensor<?x7xf32>, %[[a:.*]]: index, %[[b:.*]]: index
143 // CHECK: %[[c0:.*]] = arith.constant 0 : index
148 func.func @pad(%t: tensor<?x7xf32>, %a: index, %b: index) -> (index, index) {
151 ^bb0(%arg1: index, %arg2: index):
154 %1 = "test.reify_bound"(%0) {dim = 0} : (tensor<?x?xf32>) -> (index)
155 %2 = "test.reify_bound"(%0) {dim = 1} : (tensor<?x?xf32>) -> (index)
156 return %1, %2 : index, index
163 // CHECK: %[[c1:.*]] = arith.constant 1 : index
165 func.func @rank(%t: tensor<5xf32>) -> index {
167 %1 = "test.reify_bound"(%0) : (index) -> (index)
168 return %1 : index
174 %c0 = arith.constant 0 : index
178 "test.compare"(%dim0, %dim1) : (index, index) -> ()
185 %c0 = arith.constant 0 : index
186 %c1 = arith.constant 1 : index
188 %val = arith.addi %dim0, %c1 : index
190 "test.compare"(%dim0, %val) : (index, index) -> ()
197 %c0 = arith.constant 0 : index
198 %c5 = arith.constant 5 : index
201 "test.compare"(%dim0, %c5) : (index, index) -> ()
208 %c0 = arith.constant 0 : index
209 %c1 = arith.constant 1 : index
213 "test.compare"(%dim0, %dim1) : (index, index) -> ()