experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add EnsureShape to XLALiteWhitelist to pass the CI test

Browse Source 
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2020-04-14 18:13:11 +00:00
parent 88692052ad
commit 1e46a8d079
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/jit/mark_for_compilation_pass.cc
View File

@ -1891,6 +1891,7 @@ absl::flat_hash_set<string> GetKnownXLAWhitelistOp() {
"DynamicStitch", "DynamicStitch",
"Einsum", "Einsum",
"EmptyTensorList", "EmptyTensorList",
"EnsureShape",
"ExtractImagePatches", "ExtractImagePatches",
"Igamma", "Igamma",
"IgammaGradA", "IgammaGradA",