MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / set_client_device

Method set_client_device

tensorflow/core/common_runtime/device_set.h:42–45  ·  view source on GitHub ↗

Set the device designated as the "client". This device must also be registered via AddDevice().

Source from the content-addressed store, hash-verified

40 // Set the device designated as the "client". This device
41 // must also be registered via AddDevice().
42 void set_client_device(Device* device) {
43 DCHECK(client_device_ == nullptr);
44 client_device_ = device;
45 }
46
47 // Returns a pointer to the device designated as the "client".
48 Device* client_device() const { return client_device_; }

Callers 2

DirectSessionMethod · 0.80
CreateSessionMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected