From 1e46a8d079ef63bdebf54f77cae1888f871eb673 Mon Sep 17 00:00:00 2001
From: Yong Tang <yong.tang.github@outlook.com>
Date: Tue, 14 Apr 2020 18:13:11 +0000
Subject: [PATCH] Add EnsureShape to XLALiteWhitelist to pass the CI test

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
---
 tensorflow/compiler/jit/mark_for_compilation_pass.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/compiler/jit/mark_for_compilation_pass.cc b/tensorflow/compiler/jit/mark_for_compilation_pass.cc
index 77496fe7960..cb74d135418 100644
--- 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",