Skip to content

GitLab

Commit 554bb86f authored by Akshay Agrawal's avatar Akshay Agrawal
Browse files

Fix eager gpu cell

parent d640ab9c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
samples/outreach/demos/eager_execution.ipynb
View file @ 554bb86f
...@@ -284,7 +284,8 @@ ...@@ -284,7 +284,8 @@
"source": [ "source": [
"if tf.test.is_gpu_available() > 0:\n", "if tf.test.is_gpu_available() > 0:\n",
" with tf.device(tf.test.gpu_device_name()):\n", " with tf.device(tf.test.gpu_device_name()):\n",
" print(tf.matmul(A, A))" " B = tf.constant([[2.0, 0.0], [0.0, 3.0]])\n",
" print(tf.matmul(B, B))"
], ],
"execution_count": 0, "execution_count": 0,
"outputs": [] "outputs": []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment