diff options
Diffstat (limited to 'res/TensorFlowPythonExamples/examples/max_pool_with_argmax/__init__.py')
| -rwxr-xr-x | res/TensorFlowPythonExamples/examples/max_pool_with_argmax/__init__.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/res/TensorFlowPythonExamples/examples/max_pool_with_argmax/__init__.py b/res/TensorFlowPythonExamples/examples/max_pool_with_argmax/__init__.py index 487858cc5..78daa034c 100755 --- a/res/TensorFlowPythonExamples/examples/max_pool_with_argmax/__init__.py +++ b/res/TensorFlowPythonExamples/examples/max_pool_with_argmax/__init__.py @@ -1,5 +1,7 @@ import tensorflow as tf +tf.compat.v1.disable_eager_execution() + in_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(1, 4, 4, 1), name="Hole") op_ = tf.compat.v1.nn.max_pool_with_argmax( in_, ksize=[1, 2, 2, 1], strides=[1, 1, 1, 1], padding="VALID") |