提交 b60438ac 编写于 作者: A A. Unique TensorFlower 提交者: TensorFlower Gardener

Add a test case that covers empty input for tf.keras.layers.LayerNormalization.

PiperOrigin-RevId: 381368343
上级 a1446dc8
......@@ -74,6 +74,9 @@ class LayerNormalizationTest(keras_parameterized.TestCase):
keras.layers.LayerNormalization,
kwargs={'axis': (-3, -2, -1)},
input_shape=(2, 8, 8, 3))
testing_utils.layer_test(
keras.layers.LayerNormalization,
input_shape=(1, 0, 10))
@keras_parameterized.run_all_keras_modes
def test_non_fused_layernorm(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册