Rollback APIs for tf2-alpha

tensorflow · Apr 30, 2019 · 6e6d2a5 · 6e6d2a5
1 parent 0095fe1
commit 6e6d2a5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorflow_addons/image/filters.py b/tensorflow_addons/image/filters.py
@@ -91,7 +91,7 @@ def func2():
         for a in range(ch):
             img = image[:, :, a:a + 1]
             img = tf.reshape(img, [1, row, col, 1])
-            slic = tf.image.extract_patches(
+            slic = tf.image.extract_image_patches(
                 img, [1, filter_shapex, filter_shapey, 1], [1, 1, 1, 1],
                 [1, 1, 1, 1],
                 padding='SAME')

diff --git a/tensorflow_addons/seq2seq/beam_search_decoder.py b/tensorflow_addons/seq2seq/beam_search_decoder.py
@@ -880,7 +880,7 @@ def _beam_search_step(time, logits, next_cell_state, beam_state, batch_size,
     #       name="next_beam_word_ids")
     # would be a lot cleaner but for reasons unclear, that hides the results of
     # the op which prevents capturing it with tfdbg debug ops.
-    raw_next_word_ids = tf.math.floormod(
+    raw_next_word_ids = tf.mod(
         word_indices, vocab_size, name="next_beam_word_ids")
     next_word_ids = tf.cast(raw_next_word_ids, tf.int32)
     next_beam_ids = tf.cast(