| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | loss_scale_optimizer_test.py | 68 loss = lambda: grad_check_fn(var) / strategy.num_replicas_in_sync
82 self.assertEqual(loss_scale % strategy.num_replicas_in_sync, 0)
84 strategy, var, opt, loss_scale / strategy.num_replicas_in_sync)
|
| D | keras_test.py | 230 y /= strategy.num_replicas_in_sync
349 expected_gradient = loss_scale / strategy.num_replicas_in_sync / 2
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-input-context.pbtxt | 14 name: "num_replicas_in_sync"
19 …args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'], varargs…
|
| D | tensorflow.distribute.-replica-context.pbtxt | 10 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.-strategy.pbtxt | 10 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 11 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 11 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 11 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 11 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 11 name: "num_replicas_in_sync"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-input-context.pbtxt | 14 name: "num_replicas_in_sync"
19 …args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'], varargs…
|
| D | tensorflow.distribute.-replica-context.pbtxt | 10 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.-strategy.pbtxt | 10 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 11 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 11 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 11 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 11 name: "num_replicas_in_sync"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 11 name: "num_replicas_in_sync"
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | distributed_training_utils.py | 456 global_batch_size *= distribution_strategy.num_replicas_in_sync
478 global_batch_size *= distribution_strategy.num_replicas_in_sync
491 if global_batch_size % distribution_strategy.num_replicas_in_sync:
495 global_batch_size, distribution_strategy.num_replicas_in_sync))
496 batch_size = global_batch_size // distribution_strategy.num_replicas_in_sync
580 None for _ in range(len(model.outputs) * strategy.num_replicas_in_sync)
872 num_replicas = model._distribution_strategy.num_replicas_in_sync
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib_test.py | 153 num_input_pipelines=2, input_pipeline_id=1, num_replicas_in_sync=6)
154 self.assertEqual(6, input_context.num_replicas_in_sync)
160 num_input_pipelines=2, input_pipeline_id=1, num_replicas_in_sync=6)
|
| D | distribute_lib.py | 241 num_replicas_in_sync=1): argument
252 self._num_replicas_in_sync = num_replicas_in_sync
255 def num_replicas_in_sync(self): member in InputContext
562 def num_replicas_in_sync(self): member in DistributionStrategy
1454 def num_replicas_in_sync(self): member in ReplicaContext
1456 return self._strategy.num_replicas_in_sync
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | estimator_integration_test.py | 79 batch_size=batch_size // distribution.num_replicas_in_sync,
84 batch_size=batch_size // distribution.num_replicas_in_sync,
|
| D | minimize_loss_test.py | 213 num_replicas = distribution.num_replicas_in_sync
356 self.assertNear(weight, 2 + 21.2 * distribution.num_replicas_in_sync,
474 distribution.num_replicas_in_sync)
|
| D | parameter_server_strategy_test.py | 469 self.assertEqual(x_val, 10.0 + 1.0 * num_workers * d.num_replicas_in_sync)
470 self.assertEqual(y_val, 20.0 + 1.0 * num_workers * d.num_replicas_in_sync)
472 return (x_val == 10.0 + 1.0 * num_workers * d.num_replicas_in_sync and
473 y_val == 20.0 + 1.0 * num_workers * d.num_replicas_in_sync and
627 self.assertEqual(2, strategy.num_replicas_in_sync)
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | losses_utils.py | 180 distribution_strategy_context.get_strategy().num_replicas_in_sync)
240 distribution_strategy_context.get_strategy().num_replicas_in_sync)
|