Skip to content

Commit 1b28131

Browse files
Merge pull request #1359 from 8bitmp3:patch-1
PiperOrigin-RevId: 387911509
2 parents 23a4c90 + 3e482cf commit 1b28131

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow_probability/examples/jupyter_notebooks/Probabilistic_Layers_VAE.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -203,12 +203,12 @@
203203
"train_dataset = (datasets['train']\n",
204204
" .map(_preprocess)\n",
205205
" .batch(256)\n",
206-
" .prefetch(tf.data.experimental.AUTOTUNE)\n",
206+
" .prefetch(tf.data.AUTOTUNE)\n",
207207
" .shuffle(int(10e3)))\n",
208208
"eval_dataset = (datasets['test']\n",
209209
" .map(_preprocess)\n",
210210
" .batch(256)\n",
211-
" .prefetch(tf.data.experimental.AUTOTUNE))"
211+
" .prefetch(tf.data.AUTOTUNE))"
212212
]
213213
},
214214
{

0 commit comments

Comments
 (0)