From 71fb291ac9c4c2ad0bf2575279b57ee889f847c7 Mon Sep 17 00:00:00 2001 From: Hao Zhuang Date: Fri, 10 Jan 2020 18:54:37 -0800 Subject: [PATCH] Internal change PiperOrigin-RevId: 289203075 Change-Id: Ie83a6140dee4403605b4c37e51c159934601bc88 --- qkeras/estimate.py | 2 +- qkeras/qlayers.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/qkeras/estimate.py b/qkeras/estimate.py index a507f70f..5028791d 100644 --- 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 index e5cf8e90..81c01331 100644 --- 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