Skip to content

GitLab

Commit 40b84bce authored by Mark Daoust's avatar Mark Daoust
Browse files

autograph.utils.set_element_type --> autograph.set_element_type

parent 3ec66da0
Show whitespace changes
Inline Side-by-side
Showing with 18 additions and 5 deletions
samples/core/guide/autograph.ipynb
View file @ 40b84bce
......@@ -488,7 +488,7 @@
"def f(n):\n",
" z = []\n",
" # We ask you to tell us the element dtype of the list\n",
" autograph.utils.set_element_type(z, tf.int32)\n",
" autograph.set_element_type(z, tf.int32)\n",
" \n",
" for i in range(n):\n",
" z.append(i)\n",
......@@ -746,13 +746,13 @@
" # to convert these lists into their graph equivalent,\n",
" # we need to specify the element type of the lists.\n",
" train_losses = []\n",
" autograph.utils.set_element_type(train_losses, tf.float32)\n",
" autograph.set_element_type(train_losses, tf.float32)\n",
" test_losses = []\n",
" autograph.utils.set_element_type(test_losses, tf.float32)\n",
" autograph.set_element_type(test_losses, tf.float32)\n",
" train_accuracies = []\n",
" autograph.utils.set_element_type(train_accuracies, tf.float32)\n",
" autograph.set_element_type(train_accuracies, tf.float32)\n",
" test_accuracies = []\n",
" autograph.utils.set_element_type(test_accuracies, tf.float32)\n",
" autograph.set_element_type(test_accuracies, tf.float32)\n",
" \n",
" # This entire training loop will be run in-graph.\n",
" i = tf.constant(0)\n",
......@@ -835,6 +835,19 @@
],
"execution_count": 0,
"outputs": []
},
{
"metadata": {
"id": "qj0pLwit746K",
"colab_type": "code",
"colab": {}
},
"cell_type": "code",
"source": [
""
],
"execution_count": 0,
"outputs": []
}
]
}
\ No newline at end of file
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