Searched refs:from_extents (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/include/mlir/Dialect/Shape/IR/ |
| D | ShapeOps.td | 151 def Shape_FromExtentsOp : Shape_Op<"from_extents", [NoSideEffect]> {
159 %s0 = shape.from_extents %a, %b
161 %s1 = shape.from_extents
|
| /external/llvm-project/mlir/test/Dialect/Shape/ |
| D | canonicalize.mlir | 165 %ret = shape.from_extents %e0, %e1, %e2
175 %ret = shape.from_extents %e0, %arg0
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5571 auto from_extents = rewriter.create<shape::FromExtentsOp>( in matchAndRewrite() local
5577 from_extents); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 2563 // CHECK-DAG: [[FROMEXTENTS:%.+]] = shape.from_extents [[GETEXTENT0]], [[CST1]], [[GETEXTENT1]]
2584 …// CHECK-DAG: [[FROMEXTENTS:%.+]] = shape.from_extents [[GETEXTENT0]], [[CST1]], [[GETEXTENT1]], […
|