fix isValidDim for block arg case
authorUday Bondhugula <uday@polymagelabs.com>
Fri, 20 Dec 2019 17:43:34 +0000 (09:43 -0800)
committerA. Unique TensorFlower <gardener@tensorflow.org>
Fri, 20 Dec 2019 17:44:03 +0000 (09:44 -0800)
commite5691c512f7335fe90093bb9989da1760d80701b
treeaa9054cc7b1b09c157729a7cc0c62b5f8ee6c089
parent42d46b4efa9d7b596c74c101a777f4464a8f1fd0
fix isValidDim for block arg case

- a block argument associated with an arbitrary op can't be a valid
  dimensional identifier; it has to be the block argument of either
  a function op or an affine.for.

Signed-off-by: Uday Bondhugula <uday@polymagelabs.com>
Closes tensorflow/mlir#331

COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/mlir/pull/331 from bondhugula:valid_dim 3273b4fcbaa31fb7b6671d93c9e42a6b2a6a4e4c
PiperOrigin-RevId: 286593693
mlir/lib/Dialect/AffineOps/AffineOps.cpp
mlir/test/AffineOps/invalid.mlir
mlir/test/Transforms/slicing-utils.mlir