diff options
Diffstat (limited to 'res/TensorFlowPythonExamples/examples/segment_sum/__init__.py')
-rwxr-xr-x | res/TensorFlowPythonExamples/examples/segment_sum/__init__.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/res/TensorFlowPythonExamples/examples/segment_sum/__init__.py b/res/TensorFlowPythonExamples/examples/segment_sum/__init__.py index 24d15bb8b..c15746a66 100755 --- a/res/TensorFlowPythonExamples/examples/segment_sum/__init__.py +++ b/res/TensorFlowPythonExamples/examples/segment_sum/__init__.py @@ -1,5 +1,7 @@ import tensorflow as tf +tf.compat.v1.disable_eager_execution() + lhs_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(4, 4, 4, 4), name="Hole") rhs_ = tf.compat.v1.placeholder(dtype=tf.int32, shape=(4, ), name="Hole") op_ = tf.compat.v1.math.segment_sum(lhs_, rhs_) |