[MLIR][Shape] Concretize broadcast result type if possible
authorFrederik Gossen <frgossen@google.com>
Wed, 28 Apr 2021 09:56:48 +0000 (11:56 +0200)
committerFrederik Gossen <frgossen@google.com>
Wed, 28 Apr 2021 09:58:32 +0000 (11:58 +0200)
commitdca536103592cf1e92aa8316ed23f33d75da25bc
treef25a952c02a3dc354a7392beb6730ca7b36fb75f
parent2d37f2115d173a2c9117c6dbb4720dc7aefb9f42
[MLIR][Shape] Concretize broadcast result type if possible

As a canonicalization, infer the resulting shape rank if possible.

Differential Revision: https://reviews.llvm.org/D101377
mlir/include/mlir/Dialect/Shape/IR/Shape.h
mlir/lib/Dialect/Shape/IR/Shape.cpp
mlir/test/Dialect/Shape/canonicalize.mlir