Searched defs:loss_scale (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/experimental/ |
| H A D | mixed_precision_test.py | 93 def __init__(self, inner_optimizer, loss_scale): argument
111 def __init__(self, inner_optimizer, loss_scale): argument
117 def create_lso(inner_optimizer, loss_scale): argument
|
| H A D | mixed_precision.py | 60 def _wrap_optimizer(opt, loss_scale): argument
84 def enable_mixed_precision_graph_rewrite_v1(opt, loss_scale='dynamic'): argument
|
| H A D | loss_scale_optimizer.py | 60 def __init__(self, opt, loss_scale): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| H A D | policy.py | 335 def __init__(self, name, loss_scale='auto'): argument
374 def loss_scale(self): member in PolicyV1
|
| H A D | loss_scale.py | 25 def serialize(loss_scale): argument
|
| H A D | loss_scale_optimizer.py | 458 def loss_scale(self): member in LossScaleOptimizer
953 def __init__(self, optimizer, loss_scale): argument
|