diff options
| author | Uday Bondhugula <uday@polymagelabs.com> | 2019-12-14 11:21:52 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-14 11:22:24 -0800 |
| commit | 97af93227283e9252d7e497bd08ea2b78ece8f92 (patch) | |
| tree | eddd067e7cb18063ab052fce2312f4ff0ea549a2 /mlir/include | |
| parent | 2d22b1e04e7f75ecc38247fc2a5cd18058374cc0 (diff) | |
| download | bcm5719-llvm-97af93227283e9252d7e497bd08ea2b78ece8f92.tar.gz bcm5719-llvm-97af93227283e9252d7e497bd08ea2b78ece8f92.zip | |
Splat op doc - fix misformat / update tablegen op desc. comment
- bring op description comment in sync with the doc
- fix misformat in doc
Signed-off-by: Uday Bondhugula <uday@polymagelabs.com>
Closes tensorflow/mlir#317
COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/mlir/pull/317 from bondhugula:quickfix 7fcd945b318c973b2488b702874c87526855c8ef
PiperOrigin-RevId: 285574527
Diffstat (limited to 'mlir/include')
| -rw-r--r-- | mlir/include/mlir/Dialect/StandardOps/Ops.td | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/mlir/include/mlir/Dialect/StandardOps/Ops.td b/mlir/include/mlir/Dialect/StandardOps/Ops.td index 553a612f5a6..2cba150560c 100644 --- a/mlir/include/mlir/Dialect/StandardOps/Ops.td +++ b/mlir/include/mlir/Dialect/StandardOps/Ops.td @@ -1086,7 +1086,16 @@ def SplatOp : Std_Op<"splat", [NoSideEffect]> { %1 = splat %0 : vector<8xi32> %2 = splat %0 : tensor<4x8xi32> - // TODO: handle broadcast to dynamically shaped tensors. + TODO: Extend this operation to broadcast to dynamically shaped tensors in + the same way dynamically shaped memrefs are handled. + + // Broadcasts %s to a 2-d dynamically shaped tensor, with %m, %n binding + // to the sizes of the two dynamic dimensions. + + %m = "foo"() : () -> (index) + %n = "bar"() : () -> (index) + %t = splat %s [%m, %n] : tensor<?x?xi32> + }]; let arguments = (ins AnyTypeOf<[AnyInteger, AnyFloat], |
