Home
last modified time | relevance | path

Searched refs:device_proto (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc258 DeviceProto* device_proto = local_topology.add_devices(); in BuildDistributedDevices() local
259 device_proto->set_local_device_ordinal(local_device->device_ordinal()); in BuildDistributedDevices()
260 device_proto->set_name(desc->name()); in BuildDistributedDevices()
261 device_proto->set_vendor(desc->device_vendor()); in BuildDistributedDevices()
271 for (const DeviceProto& device_proto : node.devices()) { in BuildDistributedDevices() local
272 GlobalDeviceId global_device_id(device_proto.global_device_id()); in BuildDistributedDevices()
276 TF_RET_CHECK(device_proto.local_device_ordinal() >= 0 && in BuildDistributedDevices()
277 device_proto.local_device_ordinal() < in BuildDistributedDevices()
279 TF_RET_CHECK(local_device_states[device_proto.local_device_ordinal()] != in BuildDistributedDevices()
282 std::move(local_device_states[device_proto.local_device_ordinal()]); in BuildDistributedDevices()
[all …]
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_events_reader.py1031 device_proto = debug_event.debugged_device
1032 self._device_by_id[device_proto.device_id] = DebuggedDevice(
1033 device_proto.device_name, device_proto.device_id)