keras/keras_core/layers
Neel Kovelamudi a333b9bd1b Adds UnitNormalization and LayerNormalization (#92)
* Adds unit normalization and tests

* Adds layer normalization and initial tests

* Fixes formatting in docstrings

* Fix type issues for JAX

* Fix nits
2023-05-05 22:52:39 +00:00
..
activations Add Normalization layer. 2023-05-05 14:27:30 -07:00
attention Add layer init files 2023-05-03 11:12:40 -07:00
convolutional Add layer init files 2023-05-03 11:12:40 -07:00
core Add EisumDense layer. 2023-05-04 18:25:08 -07:00
merging Style nits 2023-05-04 09:51:20 -07:00
normalization Adds UnitNormalization and LayerNormalization (#92) 2023-05-05 22:52:39 +00:00
pooling Fix typos 2023-05-02 22:44:46 -07:00
preprocessing Add Rescaling layer. 2023-05-05 14:46:20 -07:00
regularization Standardize annotation to skip tests that require dynamic shapes. (#65) 2023-05-01 11:14:28 -07:00
reshaping Add keras.layers.Reshape. (#58) 2023-05-01 10:21:40 -07:00
__init__.py Adds UnitNormalization and LayerNormalization (#92) 2023-05-05 22:52:39 +00:00
input_spec.py Add some numpy ops (#1) 2023-04-12 11:31:58 -07:00
layer_test.py Add loss collection in stateless call as well stateless call tests 2023-05-04 14:52:00 -07:00
layer.py Add EisumDense layer. 2023-05-04 18:25:08 -07:00
layers_utils.py Initial draft of the codebase. 2023-04-09 12:21:45 -07:00