STT-tensorflow/tensorflow/python/keras/layers
Rohan Jain d60d7d3c7e Going back to forcing embedding layer variables on the CPU even within a tf.function as this is breaking some user code.
PiperOrigin-RevId: 321607029
Change-Id: Id159867f51b26e6604a1186d9ce526658ddd1e19
2020-07-22 11:45:58 -07:00
..
legacy_rnn Update RNNCell._rnn_get_variable to use Variable._trainable in TF2 mode. 2020-06-05 23:02:56 -07:00
ops Reduce Dense.__call__ overhead by ~15% 2020-06-01 20:22:44 -07:00
preprocessing Fix a critical breakage in `training` argument default value in inference for 2020-06-25 15:26:51 -07:00
BUILD Move the module test wrt to Keras private API to keras test. 2020-06-16 09:23:32 -07:00
__init__.py Export StringLookup and IntegerLookup layers. 2020-05-29 12:37:56 -07:00
advanced_activations.py
advanced_activations_test.py
convolutional.py Reduce Conv2D.__call__ eager overhead by ~20%, Conv3D.__call__ by ~35%, and 2020-06-12 14:54:58 -07:00
convolutional_recurrent.py Make `return_state` as explicit kwarg in the Conv2DLSTM layer. 2020-06-18 15:55:19 -07:00
convolutional_recurrent_test.py
convolutional_test.py Reduce Conv2D.__call__ overhead by 45%, Conv3D.__call__ overhead by 30%, and 2020-06-11 10:06:48 -07:00
convolutional_transpose_test.py
core.py Fix Keras documentation 2020-06-17 02:12:14 -07:00
core_test.py Fix Lambda layer deserialization polluting core module. 2020-06-05 12:50:10 -07:00
cudnn_recurrent.py
cudnn_recurrent_test.py
dense_attention.py
dense_attention_test.py
einsum_dense.py
einsum_dense_test.py
embeddings.py Going back to forcing embedding layer variables on the CPU even within a tf.function as this is breaking some user code. 2020-07-22 11:45:58 -07:00
embeddings_test.py Support ShardedVariable in `tf.keras.layers.Embedding`. 2020-05-21 11:06:33 -07:00
gru_test.py Fix RNN cell wrt deep copy. 2020-06-01 21:56:22 -07:00
gru_v2_test.py
kernelized.py Make RandomFourierFeatures state saveable. 2020-05-25 17:08:03 -07:00
kernelized_test.py Make RandomFourierFeatures state saveable. 2020-05-25 17:08:03 -07:00
layers_test.py Move the module test wrt to Keras private API to keras test. 2020-06-16 09:23:32 -07:00
local.py
local_test.py
lstm_test.py Fix RNN cell wrt deep copy. 2020-06-01 21:56:22 -07:00
lstm_v2_test.py
merge.py This change adds WIP support for using `KerasTensor` objects in Keras's functional API instead of symbolic graph tf.Tensors. It is controlled by an internal behavior flag that is disabled by default and is not yet exposed in TF's APIs. 2020-06-05 15:47:28 -07:00
merge_test.py
noise.py
noise_test.py
normalization.py Do non-fused mixed precision BatchNormalization in float32. 2020-06-09 18:06:06 -07:00
normalization_test.py Simplify Layer.add_udpate in v2 and update version_selector to use v1 inside a 2020-06-17 11:01:44 -07:00
normalization_v2.py
pooling.py
pooling_test.py
recurrent.py Avoid deprecation warning related to the use of `collections.Sequence`. It will stop working in Python 3.8. 2020-06-17 14:35:15 -07:00
recurrent_test.py
recurrent_v2.py Updated documentation for GRU & LSTM 2020-06-09 10:43:47 +08:00
recurrent_v2_test.py Don't use the cuDNN GPU kernel for LSTMs when inputs are RaggedTensors. 2020-05-29 12:50:57 -07:00
rnn_cell_wrapper_v2.py Updates Keras layer `__call__` to always set `training`, with the following priority order: 2020-06-22 13:13:46 -07:00
rnn_cell_wrapper_v2_test.py
separable_convolutional_test.py
serialization.py Export StringLookup and IntegerLookup layers. 2020-05-29 12:37:56 -07:00
serialization_test.py Make regularizers API more consistent. 2020-05-13 14:27:05 -07:00
simplernn_test.py Fix RNN cell wrt deep copy. 2020-06-01 21:56:22 -07:00
subclassed_layers_test.py
tensorflow_op_layer_test.py Remove keras_tensor skipping from keras tensorflow_op_layer_test 2020-06-12 10:29:30 -07:00
wrappers.py Fix Keras documentation 2020-06-17 02:12:14 -07:00
wrappers_test.py Simplify Layer.add_udpate in v2 and update version_selector to use v1 inside a 2020-06-17 11:01:44 -07:00