Skip to content

Commit dc479ee

Browse files
committed
Added missing LayerNorm test case into TestLayersSameGraph
1 parent 74985b2 commit dc479ee

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/nnx/nn/linear_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,7 @@ class TestLayersSameGraph(parameterized.TestCase):
256256
("kernel_init", "out_kernel_init", "bias_init", "out_bias_init"),
257257
),
258258
(nnx.BatchNorm, (3,), ("scale_init", "bias_init")),
259+
(nnx.LayerNorm, (3,), ("scale_init", "bias_init")),
259260
(nnx.RMSNorm, (3,), ("scale_init",)),
260261
(nnx.GroupNorm, (6, 3), ("scale_init", "bias_init")),
261262
(nnx.InstanceNorm, (6,), ("scale_init", "bias_init")),

0 commit comments

Comments
 (0)