PiperOrigin-RevId: 448282936
Showing 2 of 2 files from the diff.
flax/linen/__init__.py
changed.
flax/linen/activation.py
changed.
@@ -23,10 +23,19 @@
Loading
23 | 23 | elu as elu, |
|
24 | 24 | gelu as gelu, |
|
25 | 25 | glu as glu, |
|
26 | + | hard_sigmoid as hard_sigmoid, |
|
27 | + | hard_silu as hard_silu, |
|
28 | + | hard_swish as hard_swish, |
|
29 | + | hard_tanh as hard_tanh, |
|
26 | 30 | leaky_relu as leaky_relu, |
|
27 | 31 | log_sigmoid as log_sigmoid, |
|
28 | 32 | log_softmax as log_softmax, |
|
33 | + | logsumexp as logsumexp, |
|
34 | + | normalize as normalize, |
|
35 | + | one_hot as one_hot, |
|
29 | 36 | relu as relu, |
|
37 | + | relu6 as relu6, |
|
38 | + | selu as selu, |
|
30 | 39 | sigmoid as sigmoid, |
|
31 | 40 | silu as silu, |
|
32 | 41 | soft_sign as soft_sign, |
@@ -27,12 +27,15 @@
Loading
27 | 27 | from jax.nn import gelu |
|
28 | 28 | from jax.nn import glu |
|
29 | 29 | from jax.nn import hard_sigmoid |
|
30 | + | from jax.nn import hard_silu |
|
30 | 31 | from jax.nn import hard_swish |
|
31 | 32 | from jax.nn import hard_tanh |
|
32 | 33 | from jax.nn import leaky_relu |
|
33 | 34 | from jax.nn import log_sigmoid |
|
34 | 35 | from jax.nn import log_softmax |
|
36 | + | from jax.nn import logsumexp |
|
35 | 37 | from jax.nn import normalize |
|
38 | + | from jax.nn import one_hot |
|
36 | 39 | from jax.nn import relu |
|
37 | 40 | from jax.nn import relu6 |
|
38 | 41 | from jax.nn import selu |
Files | Coverage |
---|---|
flax | 74.97% |
Project Totals (59 files) | 74.97% |
2315037664
Sunburst
The inner-most circle is the entire project, moving away from the center are folders then, finally, a single file.
The size and color of each slice is representing the number of statements and the coverage, respectively.
Icicle
The top section represents the entire project. Proceeding with folders and finally individual files.
The size and color of each slice is representing the number of statements and the coverage, respectively.