Merge pull request #3531 from chiamp:normalize

PiperOrigin-RevId: 588113491
google · Dec 5, 2023 · 512a6d8 · 512a6d8
2 parents 230b0d7 + 28d41d9
commit 512a6d8
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/flax/experimental/nnx/__init__.py b/flax/experimental/nnx/__init__.py
@@ -49,7 +49,6 @@
 from .nnx.nn.activations import log_sigmoid as log_sigmoid
 from .nnx.nn.activations import log_softmax as log_softmax
 from .nnx.nn.activations import logsumexp as logsumexp
-from .nnx.nn.activations import normalize as normalize
 from .nnx.nn.activations import one_hot as one_hot
 from .nnx.nn.activations import relu as relu
 from .nnx.nn.activations import relu6 as relu6

diff --git a/flax/experimental/nnx/nnx/nn/activations.py b/flax/experimental/nnx/nnx/nn/activations.py
@@ -25,7 +25,6 @@
   log_sigmoid,
   log_softmax,
   logsumexp,
-  normalize,
   one_hot,
   relu,
   relu6,
@@ -53,7 +52,6 @@
   'log_sigmoid',
   'log_softmax',
   'logsumexp',
-  'normalize',
   'one_hot',
   'relu',
   'relu6',