summaryrefslogtreecommitdiff
path: root/res/TensorFlowPythonExamples/examples/space_to_batch/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'res/TensorFlowPythonExamples/examples/space_to_batch/__init__.py')
-rw-r--r--res/TensorFlowPythonExamples/examples/space_to_batch/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/res/TensorFlowPythonExamples/examples/space_to_batch/__init__.py b/res/TensorFlowPythonExamples/examples/space_to_batch/__init__.py
index e088012e9..b0e3d85ab 100644
--- a/res/TensorFlowPythonExamples/examples/space_to_batch/__init__.py
+++ b/res/TensorFlowPythonExamples/examples/space_to_batch/__init__.py
@@ -1,5 +1,7 @@
import tensorflow as tf
+tf.compat.v1.disable_eager_execution()
+
in_ = tf.compat.v1.placeholder(tf.float32, shape=[1, 2, 2, 1], name="Hole")
pd_ = tf.constant([[0, 0], [0, 0]], name="Hole")
op_ = tf.space_to_batch(in_, pd_, 2)