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