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

Enable tf.Module + sharded variable test given the integration is in place.

Browse Source 
PiperOrigin-RevId: 338319945
Change-Id: I6a2a0bd397f035e6fbce9fce4eda5e154ed38604
This commit is contained in:
Chenkai Kuang 2020-10-21 12:45:04 -07:00 committed by TensorFlower Gardener
parent 898c3b6d33
commit 9fe37b34f5
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/python/distribute/sharded_variable_test.py
View File

@ -364,7 +364,6 @@ class ShardedVariableTest(test.TestCase):
self.assertAllEqual(variables, got)
def test_tf_module(self):
self.skipTest('integration with tf.module is not added yet.')
class Model(module.Module):