Commit 5a9c0fdd authored by Eugene Brevdo's avatar Eugene Brevdo Committed by TensorFlower Gardener
Browse files

[tf contrib layers] Add options to layer_norm to configure layer axes. 

This is a fully backwards compatible change that allows the user to decide:
* Which axes to normalize
* The shape of the scale/center variables (these are broadcasted).

PiperOrigin-RevId: 157019098
parent e8dd3379
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