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

Method AddDevice

tensorflow/core/common_runtime/device_set.cc:33–39  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

31DeviceSet::~DeviceSet() {}
32
33void DeviceSet::AddDevice(Device* device) {
34 devices_.push_back(device);
35 for (const string& name :
36 DeviceNameUtils::GetNamesForDeviceMappings(device->parsed_name())) {
37 device_by_name_.insert({name, device});
38 }
39}
40
41void DeviceSet::FindMatchingDevices(const DeviceNameUtils::ParsedName& spec,
42 std::vector<Device*>* devices) const {

Callers 7

DirectSessionMethod · 0.45
PlacerTestMethod · 0.45
TEST_FFunction · 0.45
InitBaseGraphMethod · 0.45
OptimizeGraphMethod · 0.45
SetUpMethod · 0.45

Calls 2

push_backMethod · 0.45
insertMethod · 0.45

Tested by 3

PlacerTestMethod · 0.36
TEST_FFunction · 0.36
SetUpMethod · 0.36