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

Enable skipped test as Variable.assign(name=xxx) now works with CentralStorage 

PiperOrigin-RevId: 312701125
Change-Id: I2c43d1da9cf97b359293498085f491908a3ad4ab
This commit is contained in:
Priya Gupta 2020-05-21 10:58:17 -07:00 committed by TensorFlower Gardener
parent 37b60af536
commit 31eeaec3b4
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/keras/distribute/keras_utils_test.py
View File

@ -26,7 +26,6 @@ import numpy as np
from tensorflow.python import keras
from tensorflow.python.data.ops import dataset_ops
from tensorflow.python.distribute import combinations
from tensorflow.python.distribute import parameter_server_strategy
from tensorflow.python.distribute import strategy_combinations
from tensorflow.python.distribute import tpu_strategy
from tensorflow.python.distribute import values
@ -398,9 +397,6 @@ class TestDistributionStrategyWithNormalizationLayer(test.TestCase,
optimizer=strategy_combinations
.gradient_descent_optimizer_keras_v2_fn)))
def test_batchnorm_correctness(self, distribution, fused, optimizer):
if isinstance(distribution.extended,
parameter_server_strategy.ParameterServerStrategyExtended):
self.skipTest('b/152353796')
with self.cached_session():
with distribution.scope():
model = keras.models.Sequential()