Searched refs:_DeviceAttributes (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | cluster_resolver_test.py | 59 session._DeviceAttributes(name, "GPU", 1024, 0)
83 session._DeviceAttributes(name, name[26:29], 1024, 0)
107 session._DeviceAttributes(name, name[27:30], 1024, 0)
|
| D | tpu_cluster_resolver_test.py | 577 session._DeviceAttributes(
602 session._DeviceAttributes(
640 session._DeviceAttributes(
|
| D | tfconfig_cluster_resolver_test.py | 275 session._DeviceAttributes(name, name[27:30], 1024, 0)
|
| D | cluster_resolver.py | 56 …devices.append(session._DeviceAttributes(name, device_type, 0, 0)) # pylint: disable=protected-ac…
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_system_metadata.py | 72 …devices.append(session_lib._DeviceAttributes(name, device_type, 0, 0)) # pylint: disable=protecte…
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session.py | 576 class _DeviceAttributes(object): class
713 _DeviceAttributes(name, device_type, memory, incarnation))
|
| D | session_test.py | 1990 attrs = session._DeviceAttributes(
2000 attrs = session._DeviceAttributes('/job:worker/replica:0/task:3/cpu:1',
|