Searched refs:loss_scale_value (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale.py | 241 def __init__(self, loss_scale_value): argument
255 if not isinstance(loss_scale_value, six.integer_types + (float,)):
257 if loss_scale_value < 1:
265 self._loss_scale_value = float(loss_scale_value)
|
| D | loss_scale_test.py | 65 loss_scale_value = 1000
66 loss_scale = loss_scale_module.FixedLossScale(loss_scale_value)
74 self.assertEqual(loss_scale_value, self.evaluate(loss_scale()))
81 self.assertEqual(loss_scale_value, self.evaluate(loss_scale()))
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.mixed_precision.-fixed-loss-scale.pbtxt | 9 argspec: "args=[\'self\', \'loss_scale_value\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.mixed_precision.experimental.-fixed-loss-scale.pbtxt | 9 argspec: "args=[\'self\', \'loss_scale_value\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.train.experimental.-fixed-loss-scale.pbtxt | 9 argspec: "args=[\'self\', \'loss_scale_value\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.mixed_precision.experimental.-fixed-loss-scale.pbtxt | 9 argspec: "args=[\'self\', \'loss_scale_value\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.train.experimental.-fixed-loss-scale.pbtxt | 9 argspec: "args=[\'self\', \'loss_scale_value\'], varargs=None, keywords=None, defaults=None"
|