Internal change

PiperOrigin-RevId: 289203075 Change-Id: Ie83a6140dee4403605b4c37e51c159934601bc88
google · Jan 12, 2020 · 71fb291 · 71fb291
1 parent e13cbc4
commit 71fb291
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/qkeras/estimate.py b/qkeras/estimate.py
@@ -34,7 +34,7 @@
 from keras.layers import InputLayer
 from keras.models import Model
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 from .qlayers import QActivation

diff --git a/qkeras/qlayers.py b/qkeras/qlayers.py
@@ -59,7 +59,7 @@
 
 import numpy as np
 import six
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from .safe_eval import safe_eval