Searched defs:replica_id (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_context.py | 142 def device_for_replica(self, replica_id): argument
527 …def _placement_function(_sentinal=None, replica_id=None, host_id=None): # pylint: disable=invalid… argument
674 def device_for_replica(self, replica_id): argument
|
| D | tensor_tracer.py | 1254 def _make_flush_fun(replica_id): argument
1298 def _f(replica_id): argument
|
| D | tpu_estimator.py | 973 def tpu_ordinal_function_impl(replica_id): argument
979 def device_function_impl(replica_id): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.cc | 96 Status Client::TransferToInfeed(const LiteralSlice& literal, int64 replica_id, in TransferToInfeed()
119 const Shape* shape_with_layout, int64 replica_id, in TransferFromOutfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 339 int64 replica_id = 2; field
350 int64 replica_id = 2; field
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops_test.py | 1372 def __init__(self, graph, replica_id): argument
1429 def __init__(self, graph, replica_id, op_to_colocate_with): argument
1494 def __init__(self, graph, replica_id, dependency_op): argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 167 def replicated_fn(replica_id, replica_args, replica_kwargs): argument
|
| D | values.py | 91 def is_device_in_replica(self, device, replica_id): argument
153 def is_device_in_replica(self, device, replica_id): argument
217 def is_device_in_replica(self, device, replica_id): argument
1385 def select_replica(replica_id, structured): argument
|
| D | mirrored_strategy.py | 805 def __init__(self, dist, coord, replica_id, device_map, variable_creator_fn, argument
|