keras/keras_core/layers/__init__.py
Neel Kovelamudi 2d40cb20b9 Adds CategoryEncoding layer, bincount op, and tests (#161)
* Adds unit normalization and tests

* Adds layer normalization and initial tests

* Fixes formatting in docstrings

* Fix type issues for JAX

* Fix nits

* Initial stash for group_normalization and spectral_normalization

* Adds spectral normalization and tests

* Adds group normalization and tests

* Formatting fixes

* Fix small nit in docstring

* Fix docstring and tests

* Adds RandomContrast and associated tests

* Remove arithmetic comment

* Adds RandomBrightness and tests

* Fix docstring and format

* Fix nits and add backend generator

* Inlines random_contrast helper

* Add bincount op

* Add CategoryEncoding layer and tests

* Fix formatting

* Fix JAX issues

* Fix JAX bincount

* Formatting and small fix

* Fix nits and docstrings

* Add args to bincount op test
2023-05-14 00:07:43 +00:00

109 lines
6.0 KiB
Python

from keras_core.layers.activations.activation import Activation
from keras_core.layers.attention.additive_attention import AdditiveAttention
from keras_core.layers.attention.attention import Attention
from keras_core.layers.convolutional.conv1d import Conv1D
from keras_core.layers.convolutional.conv1d_transpose import Conv1DTranspose
from keras_core.layers.convolutional.conv2d import Conv2D
from keras_core.layers.convolutional.conv2d_transpose import Conv2DTranspose
from keras_core.layers.convolutional.conv3d import Conv3D
from keras_core.layers.convolutional.conv3d_transpose import Conv3DTranspose
from keras_core.layers.convolutional.depthwise_conv1d import DepthwiseConv1D
from keras_core.layers.convolutional.depthwise_conv2d import DepthwiseConv2D
from keras_core.layers.convolutional.separable_conv1d import SeparableConv1D
from keras_core.layers.convolutional.separable_conv2d import SeparableConv2D
from keras_core.layers.core.dense import Dense
from keras_core.layers.core.einsum_dense import EinsumDense
from keras_core.layers.core.embedding import Embedding
from keras_core.layers.core.identity import Identity
from keras_core.layers.core.input_layer import Input
from keras_core.layers.core.input_layer import InputLayer
from keras_core.layers.core.masking import Masking
from keras_core.layers.core.wrapper import Wrapper
from keras_core.layers.layer import Layer
from keras_core.layers.merging.add import Add
from keras_core.layers.merging.add import add
from keras_core.layers.merging.average import Average
from keras_core.layers.merging.average import average
from keras_core.layers.merging.concatenate import Concatenate
from keras_core.layers.merging.concatenate import concatenate
from keras_core.layers.merging.dot import Dot
from keras_core.layers.merging.dot import dot
from keras_core.layers.merging.maximum import Maximum
from keras_core.layers.merging.maximum import maximum
from keras_core.layers.merging.minimum import Minimum
from keras_core.layers.merging.minimum import minimum
from keras_core.layers.merging.multiply import Multiply
from keras_core.layers.merging.multiply import multiply
from keras_core.layers.merging.subtract import Subtract
from keras_core.layers.merging.subtract import subtract
from keras_core.layers.normalization.batch_normalization import (
BatchNormalization,
)
from keras_core.layers.normalization.group_normalization import (
GroupNormalization,
)
from keras_core.layers.normalization.layer_normalization import (
LayerNormalization,
)
from keras_core.layers.normalization.spectral_normalization import (
SpectralNormalization,
)
from keras_core.layers.normalization.unit_normalization import UnitNormalization
from keras_core.layers.pooling.average_pooling1d import AveragePooling1D
from keras_core.layers.pooling.average_pooling2d import AveragePooling2D
from keras_core.layers.pooling.average_pooling3d import AveragePooling3D
from keras_core.layers.pooling.global_average_pooling1d import (
GlobalAveragePooling1D,
)
from keras_core.layers.pooling.global_average_pooling2d import (
GlobalAveragePooling2D,
)
from keras_core.layers.pooling.global_average_pooling3d import (
GlobalAveragePooling3D,
)
from keras_core.layers.pooling.global_max_pooling1d import GlobalMaxPooling1D
from keras_core.layers.pooling.global_max_pooling2d import GlobalMaxPooling2D
from keras_core.layers.pooling.global_max_pooling3d import GlobalMaxPooling3D
from keras_core.layers.pooling.max_pooling1d import MaxPooling1D
from keras_core.layers.pooling.max_pooling2d import MaxPooling2D
from keras_core.layers.pooling.max_pooling3d import MaxPooling3D
from keras_core.layers.preprocessing.category_encoding import CategoryEncoding
from keras_core.layers.preprocessing.center_crop import CenterCrop
from keras_core.layers.preprocessing.discretization import Discretization
from keras_core.layers.preprocessing.hashing import Hashing
from keras_core.layers.preprocessing.integer_lookup import IntegerLookup
from keras_core.layers.preprocessing.normalization import Normalization
from keras_core.layers.preprocessing.random_brightness import RandomBrightness
from keras_core.layers.preprocessing.random_contrast import RandomContrast
from keras_core.layers.preprocessing.random_crop import RandomCrop
from keras_core.layers.preprocessing.random_flip import RandomFlip
from keras_core.layers.preprocessing.random_rotation import RandomRotation
from keras_core.layers.preprocessing.random_translation import RandomTranslation
from keras_core.layers.preprocessing.random_zoom import RandomZoom
from keras_core.layers.preprocessing.rescaling import Rescaling
from keras_core.layers.preprocessing.resizing import Resizing
from keras_core.layers.preprocessing.string_lookup import StringLookup
from keras_core.layers.preprocessing.text_vectorization import TextVectorization
from keras_core.layers.regularization.activity_regularization import (
ActivityRegularization,
)
from keras_core.layers.regularization.dropout import Dropout
from keras_core.layers.regularization.gaussian_dropout import GaussianDropout
from keras_core.layers.regularization.gaussian_noise import GaussianNoise
from keras_core.layers.regularization.spatial_dropout import SpatialDropout1D
from keras_core.layers.regularization.spatial_dropout import SpatialDropout2D
from keras_core.layers.regularization.spatial_dropout import SpatialDropout3D
from keras_core.layers.reshaping.cropping1d import Cropping1D
from keras_core.layers.reshaping.cropping2d import Cropping2D
from keras_core.layers.reshaping.cropping3d import Cropping3D
from keras_core.layers.reshaping.flatten import Flatten
from keras_core.layers.reshaping.permute import Permute
from keras_core.layers.reshaping.repeat_vector import RepeatVector
from keras_core.layers.reshaping.reshape import Reshape
from keras_core.layers.reshaping.up_sampling1d import UpSampling1D
from keras_core.layers.rnn.gru import GRU
from keras_core.layers.rnn.lstm import LSTM
from keras_core.layers.rnn.rnn import RNN
from keras_core.layers.rnn.simple_rnn import SimpleRNN
from keras_core.layers.rnn.stacked_rnn_cells import StackedRNNCells