From 6e6d2a533994c473477be258231eded0240ddc21 Mon Sep 17 00:00:00 2001 From: Sean Morgan Date: Tue, 30 Apr 2019 12:17:03 -0400 Subject: [PATCH] Rollback APIs for tf2-alpha --- tensorflow_addons/image/filters.py | 2 +- tensorflow_addons/seq2seq/beam_search_decoder.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow_addons/image/filters.py b/tensorflow_addons/image/filters.py index 7108119a1f..2017dd1f35 100644 --- 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 index 338d8d5456..1c7f81745e 100644 --- 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(