Add EnsureShape to XLALiteWhitelist to pass the CI test

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
2020-04-14 18:13:11 +00:00 · 2020-04-14 18:13:11 +00:00 · 1e46a8d079
commit 1e46a8d079
parent 88692052ad
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/compiler/jit/mark_for_compilation_pass.cc
+++ b/tensorflow/compiler/jit/mark_for_compilation_pass.cc
@ -1891,6 +1891,7 @@ absl::flat_hash_set<string> GetKnownXLAWhitelistOp() {
                                     "DynamicStitch",
                                     "Einsum",
                                     "EmptyTensorList",
+                                     "EnsureShape",
                                     "ExtractImagePatches",
                                     "Igamma",
                                     "IgammaGradA",