attention
|
Remove useless property from mha layer (#242)
|
2023-06-01 13:49:22 -07:00 |
convolutional
|
Add mobilenet and efficientnet models.
|
2023-05-22 16:39:24 -07:00 |
core
|
Switch to using take for embedding layers (#229)
|
2023-05-30 16:32:46 -07:00 |
merging
|
Add DenseNet
|
2023-05-22 16:51:41 -07:00 |
pooling
|
Finish adding SavedModel support
|
2023-06-01 18:48:52 -07:00 |
regularization
|
Allow dynamic shapes for JAX.
|
2023-05-20 19:01:01 -07:00 |
reshaping
|
Fixes
|
2023-05-31 13:45:14 -07:00 |
rnn
|
Minor fixes
|
2023-05-28 14:35:22 -07:00 |
input_spec.py
|
Add some numpy ops (#1)
|
2023-04-12 11:31:58 -07:00 |
layer_test.py
|
Add layers.MultiHeadAttention (#169)
|
2023-05-16 18:22:42 -07:00 |
layer.py
|
Minor fixes
|
2023-05-30 09:19:46 -07:00 |