Searched refs:to_extent_tensor (Results 1 – 9 of 9) sorted by relevance
155 // Lower `to_extent_tensor` to `std.tensor_cast`156 // Fold to_extent_tensor when already on tensor.157 // CHECK-LABEL: @to_extent_tensor159 func @to_extent_tensor(%arg: tensor<?xindex>) -> tensor<3xindex> {160 // CHECK-NOT: to_extent_tensor162 %casted = shape.to_extent_tensor %arg : tensor<?xindex> -> tensor<3xindex>
17 // CHECK: [[LHSSHAPEEXTENTS:%.*]] = shape.to_extent_tensor [[LHSBCASTSHAPE]]
446 // CHECK: %[[ARG0_EXTENTS:.+]] = shape.to_extent_tensor %[[ARG0_SHAPE]]457 // CHECK: %[[ARG0_EXTENTS:.+]] = shape.to_extent_tensor %[[ARG0_SHAPE]]468 // CHECK: %[[ARG0_EXTENTS:.+]] = shape.to_extent_tensor %[[ARG0_SHAPE]]479 // CHECK: %[[ARG0_EXTENTS:.+]] = shape.to_extent_tensor %[[ARG0_SHAPE]]527 // CHECK-DAG: [[EXT:%.+]] = shape.to_extent_tensor [[SHP]]1770 // CHECK: %[[RESULT_EXTENTS:.+]] = shape.to_extent_tensor %[[RESULT_SHAPE]]1785 // CHECK: %[[RESULT_EXTENTS:.+]] = shape.to_extent_tensor %[[RESULT_SHAPE]]1842 // CHECK: %[[RESULT_EXTENTS:.+]] = shape.to_extent_tensor %[[RESULT_SHAPE]]1926 // CHECK: [[TENSOR:%.+]] = shape.to_extent_tensor [[SHAPE]]1937 // CHECK: [[TENSOR:%.+]] = shape.to_extent_tensor [[SHAPE]][all …]
152 %0 = shape.to_extent_tensor %cs : !shape.shape -> tensor<2xindex>853 // Fold to_extent_tensor when already on tensor.856 // CHECK-NOT: to_extent_tensor857 %casted = shape.to_extent_tensor %arg : tensor<?xindex> -> tensor<?xindex>
152 %0 = shape.to_extent_tensor %arg : !shape.shape -> tensor<3xindex>
80 %2 = shape.to_extent_tensor %1 : !shape.shape -> tensor<1xindex>84 // CANON-NEXT: %[[e:.*]] = shape.to_extent_tensor %[[s]] : tensor<?xindex> -> tensor<1xindex>
228 def Shape_ToExtentTensorOp : Shape_Op<"to_extent_tensor", [NoSideEffect]> {
437 %1 = shape.to_extent_tensor %0 : !shape.shape -> tensor<1xindex>
5574 auto to_extent_tensor = rewriter.create<shape::ToExtentTensorOp>( in matchAndRewrite() local5580 to_extent_tensor); in matchAndRewrite()