| /external/tensorflow/tensorflow/python/client/ |
| D | session_list_devices_test.py | 41 self.assertTrue(all(d.incarnation != 0 for d in devices))
61 self.assertTrue(all(d.incarnation != 0 for d in devices))
81 self.assertTrue(all(d.incarnation != 0 for d in devices))
|
| D | session.py | 606 def __init__(self, name, device_type, memory_limit_bytes, incarnation): argument
610 self._incarnation = incarnation
625 def incarnation(self): member in _DeviceAttributes
633 self.incarnation,
748 incarnation = tf_session.TF_DeviceListIncarnation(raw_device_list, i)
750 _DeviceAttributes(name, device_type, memory, incarnation))
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 52 req_.set_src_incarnation(server_attributes.incarnation()); in RecvBufCall()
225 remote_incarnations.insert(da.incarnation()); in CheckPeerHealth()
228 if (!remote_incarnations.contains(attr.incarnation())) { in CheckPeerHealth()
230 attr.name(), " with incarnation ", attr.incarnation(), in CheckPeerHealth()
|
| D | device_resolver_distributed.cc | 66 if (!success && it->second.incarnation() != attr.incarnation()) { in UpdateDeviceAttributes()
|
| D | session_mgr.cc | 230 device->attributes().incarnation() != da.incarnation()) { in UpdateSession()
|
| D | worker.cc | 443 if ((*src_dev)->attributes().incarnation() != parsed.src_incarnation) { in PrepareRecvTensor()
446 parsed.src_incarnation, " vs. ", (*src_dev)->attributes().incarnation(), in PrepareRecvTensor()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_attributes.proto | 48 // initialized. "incarnation" should never be 0.
49 fixed64 incarnation = 6; field
|
| D | device.cc | 49 } while (da.incarnation() == 0); // This proto field must not be zero in BuildDeviceAttributes()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_test.cc | 85 int64 incarnation = random::New64(); in CreateDevice() local
86 while (incarnation == 0) { in CreateDevice()
87 incarnation = random::New64(); in CreateDevice()
89 attr.set_incarnation(incarnation); in CreateDevice()
271 th->resource_device()->attributes().incarnation())); in TEST()
277 th->resource_device()->attributes().incarnation())); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous_test.cc | 33 const uint64 incarnation) { in NewDevice() argument
44 attrs.set_incarnation(incarnation); in NewDevice()
49 const uint64 incarnation) { in InitializeDevice() argument
51 devices.push_back(NewDevice(device, type, incarnation)); in InitializeDevice()
|
| D | dynamic_device_mgr.cc | 146 device_incarnation_set_.insert(d->attributes().incarnation()); in AddDevices()
180 device_incarnation_set_.erase(d->attributes().incarnation()); in RemoveDevices()
|
| D | buf_rendezvous.cc | 158 device->attributes().incarnation() != device_incarnation) { in ConsumeBuf()
161 " vs. ", device->attributes().incarnation(), in ConsumeBuf()
|
| D | buf_rendezvous.h | 107 const uint64 incarnation, const ConsumerCallback& done,
|
| D | dynamic_device_mgr_test.cc | 58 } while (attr.incarnation() == 0); in CreateDevice()
84 const int64 d1_incarnation = d1->attributes().incarnation(); in TEST()
|
| D | device_mgr.cc | 48 device_incarnation_set_.insert(d->attributes().incarnation()); in StaticDeviceMgr()
|
| D | collective_rma_local.cc | 93 from_device->attributes().incarnation(), in RecvFromPeer()
|
| D | partitioning_utils.cc | 49 return d->attributes().incarnation(); in PartitionFunctionGraph()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MultiDeviceIteratorInit.pbtxt | 24 An int64 indicating which incarnation of the MultiDeviceIterator
|
| D | api_def_MultiDeviceIteratorGetNextFromShard.pbtxt | 18 Which incarnation of the MultiDeviceIterator is running.
|
| D | api_def_Send.pbtxt | 25 The current incarnation of send_device.
|
| D | api_def_Recv.pbtxt | 25 The current incarnation of send_device.
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | subgraph.cc | 248 static_cast<int64>(device_info().incarnation())) in AddNode()
286 static_cast<int64>(device_info().incarnation())) in AddNode()
|
| D | graph_partition.cc | 958 int64 incarnation = PartitionOptions::kIllegalIncarnation; in SetIncarnation() local
959 if (!TryGetNodeAttr(*ndef, "send_device_incarnation", &incarnation) || in SetIncarnation()
960 (incarnation == PartitionOptions::kIllegalIncarnation)) { in SetIncarnation()
961 incarnation = opts.get_incarnation(send_device); in SetIncarnation()
962 SetAttrValue(incarnation, in SetIncarnation()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor_test.cc | 132 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key() argument
136 Rendezvous::ParseKey(Rendezvous::CreateKey(sender, incarnation, receiver, in Key()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 134 static_cast<int64>(send_device_->attributes().incarnation())); in StartSend()
287 static_cast<int64>(send_device_->attributes().incarnation())); in StartRecv()
|