Searched refs:replica_args (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 370 for replica, replica_args in enumerate(arguments)
371 for arg in replica_args]
376 for replica_args in arguments:
377 arg_buffers.append(flat_arg_buffers[:len(replica_args)])
378 flat_arg_buffers = flat_arg_buffers[len(replica_args):]
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | cluster_coordinator.py | 348 replica_args = _select_worker_slice(0, self._args)
363 *nest.map_structure(_maybe_as_type_spec, replica_args),
395 replica_args = _select_worker_slice(worker.worker_index, self._args)
399 _maybe_rebuild_remote_values(worker, replica_args) or
411 *nest.map_structure(_maybe_get_remote_value, replica_args),
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 1447 def replicated_fn(replica_id, replica_args, replica_kwargs): argument
1450 result[0] = fn(*replica_args, **replica_kwargs)
|