diff options
| author |  Akshay Modi <nareshmodi@google.com> | 2018-06-04 12:05:14 -0700 |
|---|---|---|
| committer |  Gunhan Gulsoy <gunan@google.com> | 2018-06-04 20:39:49 -0700 |
| commit | 6eb43fc26785c4835747a79b3d6a3e094ef1c60f (patch) | |
| tree | 3ec66110fca9b851913ba4dde552ae7327eb2106 /tensorflow/tools/ci_build/builds | |
| parent | a3c642c945b4a27e5d826eb9c9cbc07132cb2bba (diff) | |
Fix test user ops
PiperOrigin-RevId: 199171316
Diffstat (limited to 'tensorflow/tools/ci_build/builds')
| -rwxr-xr-x | tensorflow/tools/ci_build/builds/test_user_ops.sh | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/tools/ci_build/builds/test_user_ops.sh b/tensorflow/tools/ci_build/builds/test_user_ops.sh index c342367bac..25ecee4725 100755 --- a/tensorflow/tools/ci_build/builds/test_user_ops.sh +++ b/tensorflow/tools/ci_build/builds/test_user_ops.sh @@ -239,8 +239,9 @@ function run_op() { fi } -run_op $("${PYTHON_BIN_PATH}" -c "import tensorflow as tf; print(tf.Session('').run(tf.load_op_library('./${USER_OP_SO}').${USER_OP}(${OP_INPUT})))") -run_op $("${PYTHON_BIN_PATH}" -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.load_op_library('./${USER_OP_SO}').${USER_OP}(${OP_INPUT}))") " in eager mode" +run_op "$("${PYTHON_BIN_PATH}" -c "import tensorflow as tf; print(tf.Session('').run(tf.load_op_library('./${USER_OP_SO}').${USER_OP}(${OP_INPUT})))")" +run_op "$("${PYTHON_BIN_PATH}" -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.load_op_library('./${USER_OP_SO}').${USER_OP}(${OP_INPUT}).numpy())")" " in eager mode" + popd |