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

Function AsRemoteDevices

tensorflow/core/distributed_runtime/remote_device.cc:51–67  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

49};
50
51void AsRemoteDevices(
52 Env* env,
53 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
54 LookupLocalDevice lookup_local_device,
55 std::vector<std::unique_ptr<Device>>* remote_devices) {
56 for (const auto& da : device_attributes) {
57 Device* local_device;
58 if (lookup_local_device != nullptr &&
59 lookup_local_device(da.name(), &local_device).ok()) {
60 remote_devices->emplace_back(RenamedDevice::NewRenamedDevice(
61 local_device->name(), local_device, false, false));
62 } else {
63 auto d = new RemoteDevice(env, da);
64 remote_devices->emplace_back(d);
65 }
66 }
67}
68
69void NewRemoteDevices(Env* env, WorkerCacheInterface* worker_cache,
70 const string& worker_name, NewRemoteDevicesDone done) {

Callers 1

CreateSessionMethod · 0.85

Calls 3

nameMethod · 0.65
okMethod · 0.45
emplace_backMethod · 0.45

Tested by

no test coverage detected