| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scaling_gradient_tape_test.py | 88 def test_basic_tapes(self, loss_scale, strategy_fn, use_tf_function): argument
108 def test_output_gradients(self, loss_scale, strategy_fn, use_tf_function): argument
128 def test_multiple_source_types(self, loss_scale, strategy_fn, argument
160 def test_loss_scale_of_one(self, loss_scale, strategy_fn, argument
182 def test_nested_tapes(self, loss_scale, strategy_fn, use_tf_function, argument
226 def test_persistent_tapes(self, loss_scale, strategy_fn, use_tf_function): argument
250 def test_nested_sources(self, loss_scale): argument
262 def test_nested_targets(self, loss_scale): argument
276 def test_different_dtypes(self, loss_scale, strategy_fn): argument
308 def test_none_gradients(self, loss_scale, strategy_fn, use_tf_function): argument
[all …]
|
| D | mixed_precision.py | 42 def _wrap_optimizer(opt, loss_scale, use_v1_behavior): argument
72 def enable_mixed_precision_graph_rewrite(opt, loss_scale='dynamic'): argument
220 def enable_mixed_precision_graph_rewrite_v1(opt, loss_scale='dynamic'): argument
332 def _enable_mixed_precision_graph_rewrite_base(opt, loss_scale, argument
|
| D | loss_scaling_gradient_tape.py | 93 loss_scale, argument
205 distribution, loss_scale_gradient_tapes, loss_scale, target, sources, argument
|
| D | loss_scale_optimizer.py | 64 def __init__(self, opt, loss_scale): argument
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale.py | 30 def serialize(loss_scale): argument
|
| D | policy.py | 343 def __init__(self, name, loss_scale='auto'): argument
381 def loss_scale(self): member in PolicyV1
|
| D | loss_scale_optimizer.py | 567 def loss_scale(self): member in LossScaleOptimizer
1014 def __init__(self, optimizer, loss_scale): argument
|