Searched refs:NonFusedAdam (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adam.py | 256 class NonFusedAdam(optimizer_v2.OptimizerV2): class
367 super(NonFusedAdam, self).__init__(name, **kwargs)
387 super(NonFusedAdam, self)._prepare_local(var_device, var_dtype, apply_state)
417 super(NonFusedAdam, self).set_weights(weights)
468 config = super(NonFusedAdam, self).get_config()
|
| D | adam_test.py | 586 opt = adam.NonFusedAdam()
620 optimizer = adam.NonFusedAdam(3.0)
643 repeated_update = adam.NonFusedAdam().apply_gradients(
645 aggregated_update = adam.NonFusedAdam().apply_gradients(
681 opt = adam.NonFusedAdam(learning_rate=learning_rate)
731 opt = adam.NonFusedAdam(amsgrad=True)
778 opt_repeated = adam.NonFusedAdam(amsgrad=True)
779 opt_aggregated = adam.NonFusedAdam(amsgrad=True)
831 opt = adam.NonFusedAdam(
878 opt = adam.NonFusedAdam(
[all …]
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | optimizer_benchmarks_test.py | 55 ("NonFusedAdam", adam.NonFusedAdam(), 10),
|