tf.nn.fused_batch_norm is now tf.compat.v1.nn.fused_batch_norm
PiperOrigin-RevId: 276388087 Change-Id: I48d033d2d08f6f3bc8b391f1db1e877933f5112e
This commit is contained in:
parent
d752708859
commit
e0b4792e7b
@ -44,7 +44,7 @@ def make_fused_batch_norm_tests(options):
|
||||
variance = create_tensor_data(parameters["dtype"], scale_shape)
|
||||
|
||||
x = create_tensor_data(parameters["dtype"], parameters["input_shape"])
|
||||
[x_norm, _, _] = tf.nn.fused_batch_norm(
|
||||
[x_norm, _, _] = tf.compat.v1.nn.fused_batch_norm(
|
||||
x,
|
||||
scale,
|
||||
offset,
|
||||
|
Loading…
Reference in New Issue
Block a user