Home
last modified time | relevance | path

Searched refs:FindDeviceByName (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_set.h63 Device* FindDeviceByName(const string& fullname) const;
Ddevice_set.cc56 Device* DeviceSet::FindDeviceByName(const string& name) const { in FindDeviceByName() function in tensorflow::DeviceSet
Dpartitioning_utils.cc45 const Device* d = device_set.FindDeviceByName(name); in PartitionFunctionGraph()
Dplacer.cc304 devices_->FindDeviceByName(candidate_device_name); in CanAssignToDevice()
Dprocess_function_library_runtime.cc936 dev_set->FindDeviceByName(target)->device_type(); in InstantiateMultiDevice()
1040 target_device = device_set()->FindDeviceByName(target); in GetOutputDevices()
1044 host = device_set()->FindDeviceByName(remote_host); in GetOutputDevices()
Dplacer_test.cc382 .FindDeviceByName( \
389 .FindDeviceByName( \
394 .FindDeviceByName( \
Dgraph_execution_state.cc370 Device* device = device_set.FindDeviceByName( in LookupDevice()
Dcolocation_graph.cc1400 device_set_.FindDeviceByName(assigned_device_name); in InitializeMemberWithAssignedDevice()
Dprocess_function_library_runtime_test.cc1210 cloned_proc_flr->device_set()->FindDeviceByName(composite_device->name()), in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_session.cc1681 Device* d = devices_->FindDeviceByName(name); in BuildAndRegisterPartitions()