Internal change.

PiperOrigin-RevId: 242467951

Author: T2T Team

tensor2tensor/layers/common_layers_test.py

@@ -214,10 +214,6 @@ def testLayerNorm(self):
     res = self.evaluate(y)
     self.assertEqual(res.shape, (5, 7, 11))
 
-    # Testing layer collection.
-    layer_collection = kfac.LayerCollection()
-    common_layers.layer_norm(x, layer_collection=layer_collection)
-    self.assertLen(layer_collection.get_blocks(), 1)
 
   @test_utils.run_in_graph_and_eager_modes()
   def testGroupNorm(self):
@@ -317,13 +313,6 @@ def testApplyNormNone(self):
     self.assertEqual(actual.shape, (5, 2, 1, 11))
     self.assertAllClose(actual, x1, atol=1e-03)
 
-  @test_utils.run_in_graph_mode_only()
-  def testApplyNormWithLayerCollection(self):
-    x = np.random.rand(5, 2, 1, 11)
-    layer_collection = kfac.LayerCollection()
-    common_layers.apply_norm(x, "layer", depth=11, epsilon=1e-6,
-                             layer_collection=layer_collection)
-    self.assertLen(layer_collection.get_blocks(), 1)
 
   @test_utils.run_in_graph_mode_only()
   def testDenseWithLayerCollection(self):