Lines Matching refs:xindex

16 …ECK-SAME:                                       %[[DYNAMIC_EXTENT:.*]]: index) -> tensor<?xindex> {
17 // CHECK:           %[[MEMREF:.*]] = alloc(%[[DYNAMIC_EXTENT]]) : memref<?xindex>
23 // CHECK:             store %[[ELEM]], %[[MEMREF]][%[[I]]] : memref<?xindex>
26 // CHECK:           %[[RET:.*]] = tensor_load %[[MEMREF]] : memref<?xindex>
27 // CHECK:           return %[[RET]] : tensor<?xindex>
29 func @dynamic_tensor_from_elements(%arg: tensor<*xf32>, %rank: index) -> tensor<?xindex> {
34   } : tensor<?xindex>
35   return %result : tensor<?xindex>
42 …                                             %[[DYNAMIC_EXTENT:.*]]: index) -> tensor<16x?xindex> {
43 // CHECK:           %[[MEMREF:.*]] = alloc(%[[DYNAMIC_EXTENT]]) : memref<16x?xindex>
49 // CHECK:             store %[[VAL_7]], %[[MEMREF]][%[[I]], %[[J]]] : memref<16x?xindex>
52 // CHECK:           %[[RET:.*]] = tensor_load %[[MEMREF]] : memref<16x?xindex>
53 // CHECK:           return %[[RET]] : tensor<16x?xindex>
55 func @dynamic_tensor_from_elements_static_and_dynamic(%arg0: index) -> tensor<16x?xindex> {
60   } : tensor<16x?xindex>
61   return %result : tensor<16x?xindex>
91 // CHECK-SAME:                      %[[TENSOR:.*]]: tensor<?xindex>) -> tensor<2xindex> {
93 // CHECK:           %[[CASTED:.*]] = memref_cast %[[MEMREF]] : memref<?xindex> to memref<2xindex>
95 // CHECK:           return %[[RET]] : tensor<2xindex>
96 func @tensor_cast(%arg0: tensor<?xindex>) -> tensor<2xindex> {
97   %0 = tensor_cast %arg0 : tensor<?xindex> to tensor<2xindex>
98   return %0 : tensor<2xindex>
125 // CHECK-SAME:                               %[[ELEM1:.*]]: index) -> tensor<2xindex> {
132 // CHECK:           return %[[RET]] : tensor<2xindex>
133 func @tensor_from_elements(%arg0: index, %arg1: index) -> tensor<2xindex> {
134   %0 = tensor_from_elements %arg0, %arg1 : tensor<2xindex>
135   return %0 : tensor<2xindex>
141 …@recursively_convert_cloned_regions(%arg0: tensor<?xindex>, %arg1: index, %arg2: i1) -> tensor<?xi…
148       %50 = extract_element %arg0[%iv] : tensor<?xindex>
152   } : tensor<?xindex>
153   return %tensor : tensor<?xindex>