From de9d6e224e58c9759d7b301e499e89194c350751 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Wed, 7 Aug 2019 14:36:24 +0200
Subject: [PATCH] tf.pyfunc -> tensorflow.compat.v1.py_func
---
src/pystencils_autodiff/backends/_tensorflow.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/pystencils_autodiff/backends/_tensorflow.py b/src/pystencils_autodiff/backends/_tensorflow.py
index 4ecae0f..e0e6f0b 100644
--- a/src/pystencils_autodiff/backends/_tensorflow.py
+++ b/src/pystencils_autodiff/backends/_tensorflow.py
@@ -1,7 +1,7 @@
import tensorflow as tf
+from tensorflow.compat.v1 import get_default_graph, py_func
import pystencils_autodiff
-from tensorflow.compat.v1 import get_default_graph
_num_generated_ops = 0
@@ -33,7 +33,7 @@ def _py_func(func, inp, Tout, stateful=False, name=None, grad=None):
# Add gradient override map
with g.gradient_override_map({"PyFunc": rnd_name, "PyFuncStateless": rnd_name}):
- return tf.py_func(func, inp, Tout, stateful=stateful, name=name)
+ return py_func(func, inp, Tout, stateful=stateful, name=name)
def tensorflowop_from_autodiffop(autodiffop: pystencils_autodiff.AutoDiffOp,
--
GitLab