Home
last modified time | relevance | path

Searched refs:num_replicas_in_sync (Results 1 – 25 of 41) sorted by relevance

12

/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dloss_scale_optimizer_test.py68 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)
Dkeras_test.py230 y /= strategy.num_replicas_in_sync
349 expected_gradient = loss_scale / strategy.num_replicas_in_sync / 2
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-input-context.pbtxt14 name: "num_replicas_in_sync"
19 …args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'], varargs…
Dtensorflow.distribute.-replica-context.pbtxt10 name: "num_replicas_in_sync"
Dtensorflow.distribute.-strategy.pbtxt10 name: "num_replicas_in_sync"
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt11 name: "num_replicas_in_sync"
Dtensorflow.distribute.-mirrored-strategy.pbtxt11 name: "num_replicas_in_sync"
Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt11 name: "num_replicas_in_sync"
Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt11 name: "num_replicas_in_sync"
Dtensorflow.distribute.-one-device-strategy.pbtxt11 name: "num_replicas_in_sync"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-input-context.pbtxt14 name: "num_replicas_in_sync"
19 …args=[\'self\', \'num_input_pipelines\', \'input_pipeline_id\', \'num_replicas_in_sync\'], varargs…
Dtensorflow.distribute.-replica-context.pbtxt10 name: "num_replicas_in_sync"
Dtensorflow.distribute.-strategy.pbtxt10 name: "num_replicas_in_sync"
Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt11 name: "num_replicas_in_sync"
Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt11 name: "num_replicas_in_sync"
Dtensorflow.distribute.-one-device-strategy.pbtxt11 name: "num_replicas_in_sync"
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt11 name: "num_replicas_in_sync"
Dtensorflow.distribute.-mirrored-strategy.pbtxt11 name: "num_replicas_in_sync"
/external/tensorflow/tensorflow/python/keras/engine/
Ddistributed_training_utils.py456 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/
Ddistribute_lib_test.py153 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)
Ddistribute_lib.py241 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/
Destimator_integration_test.py79 batch_size=batch_size // distribution.num_replicas_in_sync,
84 batch_size=batch_size // distribution.num_replicas_in_sync,
Dminimize_loss_test.py213 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)
Dparameter_server_strategy_test.py469 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/
Dlosses_utils.py180 distribution_strategy_context.get_strategy().num_replicas_in_sync)
240 distribution_strategy_context.get_strategy().num_replicas_in_sync)

12