Add axis to layernorm onnx op

a80f5b19 · turneram · 5818ebde · a80f5b19 · a80f5b19
Commit a80f5b19 authored Jun 01, 2022 by turneram
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 4 deletions

test/onnx/gen_onnx.py test/onnx/gen_onnx.py +2 -1

test/onnx/layernorm_op_test.onnx test/onnx/layernorm_op_test.onnx +4 -3

No files found.
--- a/test/onnx/gen_onnx.py
+++ b/test/onnx/gen_onnx.py
@@ -2654,7 +2654,8 @@ def layernorm_op_test():
    node = onnx.helper.make_node('LayerNormalization',
                                 inputs=['x', 'w', 'b'],
                                 outputs=["output"],
-                                 epsilon=1e-5)
+                                 epsilon=1e-5,
+                                 axis=-1)

    return ([node], [x, w, b], [output])


--- a/test/onnx/layernorm_op_test.onnx
+++ b/test/onnx/layernorm_op_test.onnx
-layernorm_op_test:
-8
+layernorm_op_test:
+N
 x
 w
-boutput"LayerNormalization*
+boutput"LayerNormalization*
+axis*
 epsilon'7layernorm_op_testZ
 x