Skip to content
Snippets Groups Projects
Commit a06cd681 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

use tf.compat.v1 session

parent 9d6add19
No related branches found
No related tags found
No related merge requests found
Pipeline #18169 failed
Stage: test
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
tests/test_tfmad.py
+ 1
− 1
View file @ a06cd681
...@@ -275,7 +275,7 @@ def test_tfmad_gradient_check_tensorflow_native(with_offsets, with_cuda, gradien ...@@ -275,7 +275,7 @@ def test_tfmad_gradient_check_tensorflow_native(with_offsets, with_cuda, gradien
out_tensor = auto_diff.create_tensorflow_op(use_cuda=with_cuda, backend='tensorflow_native')(a=a_tensor, b=b_tensor) out_tensor = auto_diff.create_tensorflow_op(use_cuda=with_cuda, backend='tensorflow_native')(a=a_tensor, b=b_tensor)
with tf.Session() as sess: with tf.compat.v1.Session() as sess:
sess.run(tf.global_variables_initializer()) sess.run(tf.global_variables_initializer())
sess.run(out_tensor) sess.run(out_tensor)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment