keras/keras_core
Haifeng Jin 77b4fcc3dc Add steps_per_execution to TensorFlow backend (#153)
* Add `steps_per_execution` to TensorFlow backend

* add test

* fix the test

---------

Co-authored-by: Haifeng Jin <haifeng-jin@users.noreply.github.com>
2023-05-12 14:57:32 -07:00
..
activations Fix discrepancy with softmax when axis=None (#143) 2023-05-11 10:26:10 -07:00
backend Add steps_per_execution to TensorFlow backend (#153) 2023-05-12 14:57:32 -07:00
callbacks Merge branches 'main' and 'main' of github.com:keras-team/keras-core 2023-05-10 21:16:27 -07:00
constraints Add bulk of serialization logic. 2023-04-21 10:00:32 -07:00
datasets Remove preprocessing file 2023-05-04 09:41:30 -07:00
initializers Increase initializer test coverage 2023-05-09 16:41:25 -07:00
layers Docstring fix (#158) 2023-05-12 14:08:39 -07:00
losses Populate metrics and losses for deserialization 2023-05-09 16:26:12 -07:00
metrics PyTorch numpy API 2023-05-12 17:21:49 +00:00
mixed_precision Refactor backend. 2023-05-03 15:33:40 -07:00
models Add support for h5 weights loading. 2023-05-08 11:34:55 -07:00
operations Add put operation (#126) 2023-05-12 17:12:23 -04:00
optimizers Implement the LearningRateScheduler callback (#113) 2023-05-09 10:54:31 -06:00
regularizers Add normalize() to numerical utils. 2023-04-29 15:08:38 -07:00
saving Merge branch 'main' of github.com:keras-team/keras-core 2023-05-08 12:16:12 -07:00
testing Implement the LearningRateScheduler callback (#113) 2023-05-09 10:54:31 -06:00
trainers Add steps_per_execution to TensorFlow backend (#153) 2023-05-12 14:57:32 -07:00
utils Increase initializer test coverage 2023-05-09 16:41:25 -07:00
__init__.py Add bulk of serialization logic. 2023-04-21 10:00:32 -07:00
api_export.py Added initial Adam tests 2023-04-24 22:07:56 +00:00