|
|
|
@ -26,12 +26,15 @@ def import_tf():
|
|
|
|
pass
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
# Enable eager execution in early versions to unlock dataset operations
|
|
|
|
# Enable eager execution in early versions to unlock dataset operations
|
|
|
|
|
|
|
|
eager_enabled = False
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
tf.compat.v1.enable_eager_execution()
|
|
|
|
tf.compat.v1.enable_eager_execution()
|
|
|
|
|
|
|
|
eager_enabled = True
|
|
|
|
except AttributeError:
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
pass
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
tf.enable_eager_execution()
|
|
|
|
if not eager_enabled:
|
|
|
|
|
|
|
|
tf.enable_eager_execution()
|
|
|
|
except AttributeError:
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|