Searched refs:FixedLossScale (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scaling_gradient_tape_test.py | 83 loss_scale=[loss_scale_module.FixedLossScale,
103 loss_scale=[loss_scale_module.FixedLossScale,
123 loss_scale=[loss_scale_module.FixedLossScale,
155 loss_scale=[loss_scale_module.FixedLossScale,
176 loss_scale=[loss_scale_module.FixedLossScale,
212 with lsgt.LossScaleGradientTape(loss_scale_module.FixedLossScale(32),
221 loss_scale=[loss_scale_module.FixedLossScale,
247 loss_scale=[loss_scale_module.FixedLossScale,
259 loss_scale=[loss_scale_module.FixedLossScale,
272 loss_scale=[loss_scale_module.FixedLossScale,
[all …]
|
| D | loss_scale_test.py | 66 loss_scale = loss_scale_module.FixedLossScale(loss_scale_value)
87 loss_scale = loss_scale_module.FixedLossScale.from_config(config)
92 scalar = loss_scale_module.FixedLossScale(123)
97 loss_scale = loss_scale_module.FixedLossScale(123)
|
| D | loss_scale.py | 222 class FixedLossScale(LossScale): class
254 super(FixedLossScale, self).__init__()
470 return FixedLossScale(identifier)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale.py | 36 'FixedLossScale': loss_scale_module.FixedLossScale,
54 return loss_scale_module.FixedLossScale(identifier)
|
| D | loss_scale_benchmark.py | 71 loss_scale = loss_scale_module.FixedLossScale(2.)
|
| D | loss_scale_optimizer.py | 775 if isinstance(loss_scale, loss_scale_module.FixedLossScale):
1030 elif isinstance(loss_scale, loss_scale_module.FixedLossScale):
1102 config['loss_scale'] = loss_scale_module.FixedLossScale(
|
| D | loss_scale_optimizer_test.py | 542 loss_scale = tf_loss_scale_module.FixedLossScale(2.)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.mixed_precision.-fixed-loss-scale.pbtxt | 1 path: "tensorflow.mixed_precision.FixedLossScale"
3 is_instance: "<class \'tensorflow.python.training.experimental.loss_scale.FixedLossScale\'>"
|
| D | tensorflow.mixed_precision.experimental.-fixed-loss-scale.pbtxt | 1 path: "tensorflow.mixed_precision.experimental.FixedLossScale"
3 is_instance: "<class \'tensorflow.python.training.experimental.loss_scale.FixedLossScale\'>"
|
| D | tensorflow.train.experimental.-fixed-loss-scale.pbtxt | 1 path: "tensorflow.train.experimental.FixedLossScale"
3 is_instance: "<class \'tensorflow.python.training.experimental.loss_scale.FixedLossScale\'>"
|
| D | tensorflow.mixed_precision.experimental.pbtxt | 8 name: "FixedLossScale"
|
| D | tensorflow.train.experimental.pbtxt | 8 name: "FixedLossScale"
|
| D | tensorflow.mixed_precision.pbtxt | 8 name: "FixedLossScale"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.mixed_precision.experimental.-fixed-loss-scale.pbtxt | 1 path: "tensorflow.mixed_precision.experimental.FixedLossScale"
3 is_instance: "<class \'tensorflow.python.training.experimental.loss_scale.FixedLossScale\'>"
|
| D | tensorflow.train.experimental.-fixed-loss-scale.pbtxt | 1 path: "tensorflow.train.experimental.FixedLossScale"
3 is_instance: "<class \'tensorflow.python.training.experimental.loss_scale.FixedLossScale\'>"
|
| D | tensorflow.mixed_precision.experimental.pbtxt | 8 name: "FixedLossScale"
|
| D | tensorflow.train.experimental.pbtxt | 8 name: "FixedLossScale"
|