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:
Brian Atkinson 2019-10-23 17:44:50 -07:00 committed by TensorFlower Gardener
parent d752708859
commit e0b4792e7b

View File

@ -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,