/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)
|