Searched refs:tensor_cast (Results 1 – 17 of 17) sorted by relevance
90 // CHECK-LABEL: func @tensor_cast(96 func @tensor_cast(%arg0: tensor<?xindex>) -> tensor<2xindex> {97 %0 = tensor_cast %arg0 : tensor<?xindex> to tensor<2xindex>108 %0 = tensor_cast %arg0 : tensor<*xf32> to tensor<2xf32>119 %0 = tensor_cast %arg0 : tensor<2xf32> to tensor<*xf32>
98 // CHECK: %[[RESULT:.*]] = tensor_cast %[[TENSOR3]] : tensor<3xindex> to tensor<?xindex>111 // CHECK: %[[RESULT:.*]] = tensor_cast %[[TENSOR]] : tensor<0xindex> to tensor<?xindex>155 // Lower `to_extent_tensor` to `std.tensor_cast`161 // CHECK: %[[RES:.*]] = tensor_cast %[[ARG]] : tensor<?xindex> to tensor<3xindex319 // CHECK: %[[ERASED_LHS:.*]] = tensor_cast %[[LHS]] : tensor<?xindex> to tensor<?xindex>320 // CHECK: %[[ERASED_RHS:.*]] = tensor_cast %[[RHS]] : tensor<?xindex> to tensor<?xindex>359 // CHECK: %[[ERASED_LHS:.*]] = tensor_cast %[[LHS]] : tensor<2xindex> to tensor<?xindex>360 // CHECK: %[[ERASED_RHS:.*]] = tensor_cast %[[RHS]] : tensor<3xindex> to tensor<?xindex>403 // CHECK: %[[RANK_ERASED_LHS:.*]] = tensor_cast %[[LHS]] : tensor<3xindex> to tensor<?xin…404 // CHECK: %[[RANK_ERASED_RHS:.*]] = tensor_cast %[[RHS]] : tensor<?xindex> to tensor<?xin…[all …]
18 %unranked_at_position_0 = tensor_cast %inserted_at_position_0 : tensor<2xf32> to tensor<*xf32>25 %unranked_at_position_1 = tensor_cast %inserted_at_position_1 : tensor<2xf32> to tensor<*xf32>
11 %addf_unranked = tensor_cast %addf : tensor<3xf32> to tensor<*xf32>
12 %unranked = tensor_cast %inserted : tensor<2xf32> to tensor<*xf32>
26 %unranked = tensor_cast %D : tensor<2x4xf32> to tensor<*xf32>
22 %unranked = tensor_cast %0 : tensor<4xf32> to tensor<*xf32>
667 %0 = tensor_cast %arg0 : tensor<*xi32> to tensor<*xi32>670 // CHECK-NEXT: %0 = tensor_cast %arg0 : tensor<*xi32> to tensor<2xi32>672 %2 = tensor_cast %0 : tensor<*xi32> to tensor<2xi32>676 %4 = tensor_cast %2 : tensor<2xi32> to tensor<2xi32>1124 // CHECK: tensor_cast %{{.*}} : tensor<3x?x5x7x?xindex> to tensor<3x?x?x7x?xindex>1133 // CHECK-NEXT: %[[RES:.*]] = tensor_cast %[[IN]] : tensor<*xi32> to tensor<4x8xi32>1134 %0 = tensor_cast %input : tensor<*xi32> to tensor<4x?xi32>1135 %1 = tensor_cast %0 : tensor<4x?xi32> to tensor<4x8xi32>1145 %0 = tensor_cast %input : tensor<4xi32> to tensor<?xi32>1146 %1 = tensor_cast %0 : tensor<?xi32> to tensor<4xi32>[all …]
71 // CHECK: %0 = tensor_cast %arg0 : tensor<*xf32> to tensor<?x?xf32>72 // CHECK-NEXT: %1 = tensor_cast %arg0 : tensor<*xf32> to tensor<4x?xf32>73 %0 = tensor_cast %arg0 : tensor<*xf32> to tensor<?x?xf32>74 %1 = tensor_cast %arg0 : tensor<*xf32> to tensor<4x?xf32>
320 // CHECK-LABEL: func @tensor_cast(321 func @tensor_cast(%a : tensor<3x4xf32>, %b : tensor<4x?xf32>, %c : tensor<3x?xf32>)324 %ta = tensor_cast %a : tensor<3x4xf32> to tensor<?x?xf32>325 %tb = tensor_cast %b : tensor<4x?xf32> to tensor<?x?xf32>326 %tc = tensor_cast %c : tensor<3x?xf32> to tensor<?x?xf32>333 %1 = tensor_cast %0 : tensor<?x?xf32> to tensor<3x?xf32>
114 if (auto tensor_cast = dyn_cast<tensor::CastOp>(definingOp)) { in runOnFunction() local115 auto alias = tensor_cast.source(); in runOnFunction()
709 // CHECK-LABEL: func @tensor_cast(%arg0710 func @tensor_cast(%arg0: tensor<*xf32>, %arg1 : tensor<4x4xf32>, %arg2: tensor<?x?xf32>) {711 // CHECK: %0 = tensor_cast %arg0 : tensor<*xf32> to tensor<?x?xf32>712 %0 = tensor_cast %arg0 : tensor<*xf32> to tensor<?x?xf32>714 // CHECK: %1 = tensor_cast %arg1 : tensor<4x4xf32> to tensor<*xf32>715 %1 = tensor_cast %arg1 : tensor<4x4xf32> to tensor<*xf32>717 // CHECK: %2 = tensor_cast %arg2 : tensor<?x?xf32> to tensor<4x?xf32>718 %2 = tensor_cast %arg2 : tensor<?x?xf32> to tensor<4x?xf32>720 // CHECK: %3 = tensor_cast %2 : tensor<4x?xf32> to tensor<?x?xf32>721 %3 = tensor_cast %2 : tensor<4x?xf32> to tensor<?x?xf32>
43 syn keyword mlirOps splat store select sqrt subf subi subview tanh tensor_cast
434 auto tensor_cast = builder_.create<mlir::tensor::CastOp>( in RewriteFunc() local437 tensor_cast.getResult()); in RewriteFunc()
875 // Fold tensor_cast of a const_shape to const_shape878 // CHECK-NOT: tensor_cast880 %1 = tensor_cast %0 : tensor<?xindex> to tensor<1xindex>
3455 def TensorCastOp : CastOp<"tensor_cast"> {3461 operation ::= ssa-id `=` `std.tensor_cast` ssa-use `:` type `to` type3474 %2 = "std.tensor_cast"(%1) : (tensor<*xf32>) -> tensor<?x?xf32>3475 %2 = tensor_cast %1 : tensor<*xf32> to tensor<?x?xf32>3478 %3 = "std.tensor_cast"(%2) : (tensor<?x?xf32>) -> tensor<4x?xf32>3481 %4 = "std.tensor_cast"(%3) : (tensor<4x?xf32>) -> tensor<?x?xf32>3482 %5 = "std.tensor_cast"(%4) : (tensor<?x?xf32>) -> tensor<*xf32>3494 /// The result of a tensor_cast is always a tensor.
788 // CHECK-LABEL: func @tensor_cast(%arg0: tensor<1xi32>) -> tensor<1xi32>789 func @tensor_cast(%arg0: tensor<1xi32>) -> tensor<*xi32> {