Commit 37528511 authored by Corey Wharton's avatar Corey Wharton Committed by Rasmus Munk Larsen
Browse files

Add layer_norm op to contrib.layers. (#3671) 

* Add layer_norm op to contrib.layers.

* Fix unit tests for layer_norm op.

* Remove the epsilon parameter and change defaults for the layer_norm op.

* Add unit test on 4-D input tensor for the layer_norm op.

* Address code review comments for #3671.
parent 5c071987
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment