Skip to content

Commit 6e6d2a5

Browse files
committed
Rollback APIs for tf2-alpha
1 parent 0095fe1 commit 6e6d2a5

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tensorflow_addons/image/filters.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def func2():
9191
for a in range(ch):
9292
img = image[:, :, a:a + 1]
9393
img = tf.reshape(img, [1, row, col, 1])
94-
slic = tf.image.extract_patches(
94+
slic = tf.image.extract_image_patches(
9595
img, [1, filter_shapex, filter_shapey, 1], [1, 1, 1, 1],
9696
[1, 1, 1, 1],
9797
padding='SAME')

tensorflow_addons/seq2seq/beam_search_decoder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -880,7 +880,7 @@ def _beam_search_step(time, logits, next_cell_state, beam_state, batch_size,
880880
# name="next_beam_word_ids")
881881
# would be a lot cleaner but for reasons unclear, that hides the results of
882882
# the op which prevents capturing it with tfdbg debug ops.
883-
raw_next_word_ids = tf.math.floormod(
883+
raw_next_word_ids = tf.mod(
884884
word_indices, vocab_size, name="next_beam_word_ids")
885885
next_word_ids = tf.cast(raw_next_word_ids, tf.int32)
886886
next_beam_ids = tf.cast(

0 commit comments

Comments
 (0)