Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "98ac391ed9081ba28b2e8085d1dc05ba2d75fd67"
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 @@
"source": [
"if tf.test.is_gpu_available() > 0:\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,
"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