| commit | 11399f69b6b92363bbe6ae856510f16ca23d5f03 | [log] [tgz] |
|---|---|---|
| author | Haoliang Zhang <haoliang@google.com> | Wed Jan 29 14:35:24 2020 -0800 |
| committer | TensorFlower Gardener <gardener@tensorflow.org> | Wed Jan 29 14:40:51 2020 -0800 |
| tree | efaa516abd9640239e901338aab7984e6c1b9815 | |
| parent | 58b1c0f401436f3311c551977c18abc897938a23 [diff] |
Don't match pattern against 1-d input in dilated conv pass. PiperOrigin-RevId: 292220461 Change-Id: I8297e51fac26c61fa8f570bd6719097e4f983cb2
diff --git a/tensorflow/compiler/mlir/lite/transforms/dilated_conv.h b/tensorflow/compiler/mlir/lite/transforms/dilated_conv.h index 5d9f8d0..c3d3df1 100644 --- a/tensorflow/compiler/mlir/lite/transforms/dilated_conv.h +++ b/tensorflow/compiler/mlir/lite/transforms/dilated_conv.h
@@ -217,6 +217,9 @@ if (stb_bs_attr.getValue({i}) != bts_bs_attr.getValue({i})) return {}; } + // TODO(haoliang): support 1-D dilated conv. + if (stb_bs_attr.getNumElements() < 2) return {}; + int dilation_h_factor = stb_bs_attr.getValue({0}).cast<IntegerAttr>().getInt(); int dilation_w_factor =