tf.data.AUTOTUNE <- tf.data.experimental.AUTOTUNE

Author: 8bitmp3

tensorflow_probability/examples/jupyter_notebooks/Probabilistic_Layers_VAE.ipynb

@@ -203,12 +203,12 @@
         "train_dataset = (datasets['train']\n",
         "                 .map(_preprocess)\n",
         "                 .batch(256)\n",
-        "                 .prefetch(tf.data.experimental.AUTOTUNE)\n",
+        "                 .prefetch(tf.data.AUTOTUNE)\n",
         "                 .shuffle(int(10e3)))\n",
         "eval_dataset = (datasets['test']\n",
         "                .map(_preprocess)\n",
         "                .batch(256)\n",
-        "                .prefetch(tf.data.experimental.AUTOTUNE))"
+        "                .prefetch(tf.data.AUTOTUNE))"
       ]
     },
     {