experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

tf.nn.fused_batch_norm is now tf.compat.v1.nn.fused_batch_norm

Browse Source 
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
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/testing/op_tests/fused_batch_norm.py
View File

@ -44,7 +44,7 @@ def make_fused_batch_norm_tests(options):
variance = create_tensor_data(parameters["dtype"], scale_shape) variance = create_tensor_data(parameters["dtype"], scale_shape)
x = create_tensor_data(parameters["dtype"], parameters["input_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, x,
scale, scale,
offset, offset,