Searched refs:tf_device_id (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_id_manager.cc | 37 Status Insert(const DeviceType& type, TfDeviceId tf_device_id, in Insert() argument
45 {tf_device_id.value(), platform_device_id.value()}); in Insert()
49 "TensorFlow device (", type, ":", tf_device_id.value(), in Insert()
62 bool Find(const DeviceType& type, TfDeviceId tf_device_id, in Find() argument
69 auto id_map_iter = type_id_map_iter->second.find(tf_device_id.value()); in Find()
98 const DeviceType& type, TfDeviceId tf_device_id, in InsertTfPlatformDeviceIdPair() argument
100 return TfToPlatformDeviceIdMap::singleton()->Insert(type, tf_device_id, in InsertTfPlatformDeviceIdPair()
105 const DeviceType& type, TfDeviceId tf_device_id, in TfToPlatformDeviceId() argument
107 if (TfToPlatformDeviceIdMap::singleton()->Find(type, tf_device_id, in TfToPlatformDeviceId()
111 return errors::NotFound("TensorFlow device ", type, ":", tf_device_id.value(), in TfToPlatformDeviceId()
|
| D | device_id_utils.h | 40 TfDeviceId tf_device_id) { in ExecutorForTfDeviceId() argument
43 type, tf_device_id, &platform_device_id)); in ExecutorForTfDeviceId()
51 TfDeviceId tf_device_id) { in CheckValidTfDeviceId() argument
53 TF_CHECK_OK(DeviceIdManager::TfToPlatformDeviceId(type, tf_device_id, in CheckValidTfDeviceId()
58 << " TF " << type << " id: " << tf_device_id << ", platform " << type in CheckValidTfDeviceId()
|
| D | device_id_manager.h | 31 const DeviceType& type, TfDeviceId tf_device_id,
37 TfDeviceId tf_device_id,
|