From e0b4792e7bd36ab4ebc49911c65e1b7dc9bdc915 Mon Sep 17 00:00:00 2001 From: Brian Atkinson Date: Wed, 23 Oct 2019 17:44:50 -0700 Subject: [PATCH] tf.nn.fused_batch_norm is now tf.compat.v1.nn.fused_batch_norm PiperOrigin-RevId: 276388087 Change-Id: I48d033d2d08f6f3bc8b391f1db1e877933f5112e --- tensorflow/lite/testing/op_tests/fused_batch_norm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/lite/testing/op_tests/fused_batch_norm.py b/tensorflow/lite/testing/op_tests/fused_batch_norm.py index a884597731c..10ed48ed013 100644 --- a/tensorflow/lite/testing/op_tests/fused_batch_norm.py +++ b/tensorflow/lite/testing/op_tests/fused_batch_norm.py @@ -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,