diff options
| author | Smit Hinsu <hinsu@google.com> | 2019-10-25 10:31:26 -0700 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-10-25 10:31:56 -0700 |
| commit | cde337cfdeecc495692caf42aa8fdd0bf89e7c50 (patch) | |
| tree | 45d30f93974769702b46d6bd9f3f7c2259375e4c /mlir/test/lib/TestDialect | |
| parent | b69e8ee049ff1ba1e1ecb90acaa197cf869d9aa3 (diff) | |
| download | bcm5719-llvm-cde337cfdeecc495692caf42aa8fdd0bf89e7c50.tar.gz bcm5719-llvm-cde337cfdeecc495692caf42aa8fdd0bf89e7c50.zip | |
Define AnyRankedTensor Type in TableGen
PiperOrigin-RevId: 276714649
Diffstat (limited to 'mlir/test/lib/TestDialect')
| -rw-r--r-- | mlir/test/lib/TestDialect/TestOps.td | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/mlir/test/lib/TestDialect/TestOps.td b/mlir/test/lib/TestDialect/TestOps.td index b8443ac7a8d..1ce13c79468 100644 --- a/mlir/test/lib/TestDialect/TestOps.td +++ b/mlir/test/lib/TestDialect/TestOps.td @@ -72,6 +72,10 @@ def NDTensorOfOp : TEST_Op<"nd_tensor_of"> { ); } +def RankedTensorOp : TEST_Op<"ranked_tensor_op"> { + let arguments = (ins AnyRankedTensor:$input); +} + def MultiTensorRankOf : TEST_Op<"multi_tensor_rank_of"> { let arguments = (ins TensorRankOf<[I8, I32, F32], [0, 1]>:$arg0 |
