Fix Test breakage on with context for dropout (#38)

This commit is contained in:
Ramesh Sampath 2023-04-25 16:34:53 -05:00 committed by Francois Chollet
parent 1f80d8c1ed
commit 6034134d95

@ -106,7 +106,6 @@ def truncated_normal(shape, mean=0.0, stddev=1.0, dtype=None, seed=None):
def dropout(inputs, rate, noise_shape=None, seed=None): def dropout(inputs, rate, noise_shape=None, seed=None):
seed = tf_draw_seed(seed) seed = tf_draw_seed(seed)
with tf.init_scope():
return tf.nn.experimental.stateless_dropout( return tf.nn.experimental.stateless_dropout(
inputs, inputs,
rate=rate, rate=rate,