Home
last modified time | relevance | path

Searched refs:replica_id_in_sync_group (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.experimental.-value-context.pbtxt10 name: "replica_id_in_sync_group"
15 …argspec: "args=[\'self\', \'replica_id_in_sync_group\', \'num_replicas_in_sync\'], varargs=None, k…
Dtensorflow.distribute.-replica-context.pbtxt15 name: "replica_id_in_sync_group"
24 …argspec: "args=[\'self\', \'strategy\', \'replica_id_in_sync_group\'], varargs=None, keywords=None…
/external/tensorflow/tensorflow/python/distribute/
Dstrategy_gather_test.py130 1, shape=(ctx.replica_id_in_sync_group + 1, 1))
152 1, shape=(1, ctx.replica_id_in_sync_group + 1))
180 1, shape=(1, ctx.replica_id_in_sync_group + 1))
232 return array_ops.ones(shape=(range(1, ctx.replica_id_in_sync_group + 2)))
372 1, shape=(ctx.replica_id_in_sync_group + 1, 1))
400 1, shape=(1, ctx.replica_id_in_sync_group + 1))
429 1, shape=(1, ctx.replica_id_in_sync_group + 1))
497 1, shape=(1, ctx.replica_id_in_sync_group + 1))
554 return array_ops.ones(shape=(range(1, ctx.replica_id_in_sync_group + 2)))
615 return array_ops.constant([x[ctx.replica_id_in_sync_group]])
[all …]
Ddistribute_lib.py563 replica_id_in_sync_group=0, argument
572 self._replica_id_in_sync_group = replica_id_in_sync_group
581 def replica_id_in_sync_group(self): member in ValueContext
588 .format(self.replica_id_in_sync_group, self.num_replicas_in_sync))
2907 def __init__(self, strategy, replica_id_in_sync_group): argument
2919 if not (replica_id_in_sync_group is None or
2920 tensor_util.is_tf_type(replica_id_in_sync_group) or
2921 isinstance(replica_id_in_sync_group, int)):
2924 self._replica_id_in_sync_group = replica_id_in_sync_group
2929 replica_id_in_sync_group)
[all …]
Done_device_strategy.py474 def _get_local_replica_id(self, replica_id_in_sync_group): argument
475 return replica_id_in_sync_group
483 self, strategy, replica_id_in_sync_group=0)
Dsummary_op_util.py45 replica_id = replica_context.replica_id_in_sync_group
Dtpu_strategy.py1449 with _TPUReplicaContext(strategy, replica_id_in_sync_group=replica_id):
1523 def _get_local_replica_id(self, replica_id_in_sync_group): argument
1524 return replica_id_in_sync_group
1532 def __init__(self, strategy, replica_id_in_sync_group=0): argument
1534 self, strategy, replica_id_in_sync_group=replica_id_in_sync_group)
1540 replica_id = tensor_util.constant_value(self.replica_id_in_sync_group)
1574 array_ops.expand_dims_v2(self.replica_id_in_sync_group, 0)
Dvars_test.py337 replica_id = ctx.replica_id_in_sync_group
465 replica_id = ctx.replica_id_in_sync_group
489 replica_id = ctx.replica_id_in_sync_group
510 replica_id = ctx.replica_id_in_sync_group
531 replica_id = ctx.replica_id_in_sync_group
923 replica_id = ctx.replica_id_in_sync_group
959 replica_id = ctx.replica_id_in_sync_group
992 replica_id = ctx.replica_id_in_sync_group
1050 replica_id = ctx.replica_id_in_sync_group
1094 math_ops.equal(replica_ctx.replica_id_in_sync_group,
Ddistribution_strategy_context.py339 _get_default_strategy(), replica_id_in_sync_group=0)
Dmirrored_run.py249 self.replica_id_in_sync_group = (
317 self.replica_id_in_sync_group), \
Dmirrored_strategy_test.py400 return ds_context.get_replica_context().replica_id_in_sync_group
414 return ds_context.get_replica_context().replica_id_in_sync_group
741 ds_context.get_replica_context().replica_id_in_sync_group,
822 ds_context.get_replica_context().replica_id_in_sync_group,
885 ds_context.get_replica_context().replica_id_in_sync_group,
1424 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
1431 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
Ddistribute_lib_test.py76 self._container_strategy(), replica_id_in_sync_group=0):
122 def _get_local_replica_id(self, replica_id_in_sync_group): argument
123 return replica_id_in_sync_group
Dparameter_server_strategy.py701 def _get_local_replica_id(self, replica_id_in_sync_group): argument
702 return replica_id_in_sync_group
Dstrategy_test_lib.py258 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
292 ds_context.get_replica_context().replica_id_in_sync_group)
614 lambda: ds_context.get_replica_context().replica_id_in_sync_group + 1,
Dparameter_server_strategy_v2.py722 replica_id_in_sync_group=constant_op.constant(0, dtypes.int32)):
Dcollective_all_reduce_strategy.py933 def _get_local_replica_id(self, replica_id_in_sync_group): argument
934 return (replica_id_in_sync_group -
Dmirrored_strategy.py857 def _get_local_replica_id(self, replica_id_in_sync_group): argument
858 return replica_id_in_sync_group
Dstrategy_common_test.py58 return ds_context.get_replica_context().replica_id_in_sync_group
Dvalues_test.py193 per_replica.append(val * ctx.replica_id_in_sync_group)
239 return multiple_values[ctx.replica_id_in_sync_group]
260 return multiple_values[ctx.replica_id_in_sync_group]
314 worker_device_id = ctx.replica_id_in_sync_group % len(worker_devices)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-replica-context.pbtxt15 name: "replica_id_in_sync_group"
24 …argspec: "args=[\'self\', \'strategy\', \'replica_id_in_sync_group\'], varargs=None, keywords=None…
/external/tensorflow/tensorflow/python/distribute/integration_test/
Dsaved_model_test.py298 lambda ctx: tf.identity([3., 7.][ctx.replica_id_in_sync_group]))
351 lambda ctx: tf.identity([3., 7.][ctx.replica_id_in_sync_group]))
428 lambda ctx: tf.identity([1., 2.][ctx.replica_id_in_sync_group]))
/external/tensorflow/tensorflow/python/keras/distribute/
Dkeras_optimizer_v2_test.py138 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_test.py117 .replica_id_in_sync_group)
/external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/
DTensorFlow_NumPy_Keras_and_Distribution_Strategy.ipynb261 " replica_id = tf.distribute.get_replica_context().replica_id_in_sync_group\n",
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_interop_test.py211 val = np.asarray(ctx.replica_id_in_sync_group)

12