diff options
| author | Rob Suderman <suderman@google.com> | 2019-10-16 16:58:42 -0700 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-10-16 16:59:08 -0700 |
| commit | a245023c1ceff40aa565104e27c2a9cc23b5ef46 (patch) | |
| tree | 2b86074ae901daa0d548116855fdfd20364c13aa /mlir/test/lib/TestDialect | |
| parent | ab79c25d646ed7ef214b19042d49f15425c49818 (diff) | |
| download | bcm5719-llvm-a245023c1ceff40aa565104e27c2a9cc23b5ef46.tar.gz bcm5719-llvm-a245023c1ceff40aa565104e27c2a9cc23b5ef46.zip | |
Add ComplexType to TableGen with Tensor support
Create a ComplexType for table gen references. Include an AnyComplex type
to check whether the resulting tensor can be complex. Expand tensors to
allow complex types.
PiperOrigin-RevId: 275144804
Diffstat (limited to 'mlir/test/lib/TestDialect')
| -rw-r--r-- | mlir/test/lib/TestDialect/TestOps.td | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/mlir/test/lib/TestDialect/TestOps.td b/mlir/test/lib/TestDialect/TestOps.td index 73769e72a4a..bc1d13ce0a7 100644 --- a/mlir/test/lib/TestDialect/TestOps.td +++ b/mlir/test/lib/TestDialect/TestOps.td @@ -35,6 +35,15 @@ class TEST_Op<string mnemonic, list<OpTrait> traits = []> : // Test Types //===----------------------------------------------------------------------===// +def ComplexF64 : Complex<F64>; +def ComplexOp : TEST_Op<"complex_f64"> { + let results = (outs ComplexF64); +} + +def ComplexTensorOp : TEST_Op<"complex_f64_tensor"> { + let results = (outs TensorOf<[ComplexF64]>); +} + def AnyShaped: ShapedContainerType<[AnyType], IsShapedTypePred, "shaped">; def TupleOp : TEST_Op<"tuple_32_bit"> { |
