Searched refs:client_device (Results 1 – 6 of 6) sorted by relevance
51 Device* client_device() const { return client_device_; } in client_device() function
560 const Device* client_device, const Device* default_local_device) { in LocalAddressSpec() argument561 if (client_device != nullptr) { in LocalAddressSpec()562 return DeviceNameUtils::AddressSpace(client_device->parsed_name()); in LocalAddressSpec()589 LocalAddressSpec(device_set->client_device(), default_local_device)), in ColocationGraph()
348 *out_device_attrs = &device_set.client_device()->attributes(); in LookupDevice()353 *out_device_attrs = &device_set.client_device()->attributes(); in LookupDevice()544 &device_set_->client_device()->attributes(); in PruneGraph()
1436 input, device_set_.client_device()->attributes(), FrameAndIter(0, 0)); in CreateExecutors()1441 GetRendezvousKey(output, device_set_.client_device()->attributes(), in CreateExecutors()
1447 const Device* client_device = devices_->client_device(); in ListDevices() local1449 if (dev != client_device) { in ListDevices()1453 *(resp->add_local_device()) = client_device->attributes(); in ListDevices()
462 CHECK(device_set->client_device()) << "No client device found. Missing " in CreateSession()