Home
last modified time | relevance | path

Searched refs:_DeviceAttributes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dcluster_resolver_test.py59 session._DeviceAttributes(name, "GPU", 1024, 0)
83 session._DeviceAttributes(name, name[26:29], 1024, 0)
107 session._DeviceAttributes(name, name[27:30], 1024, 0)
Dtpu_cluster_resolver_test.py577 session._DeviceAttributes(
602 session._DeviceAttributes(
640 session._DeviceAttributes(
Dtfconfig_cluster_resolver_test.py275 session._DeviceAttributes(name, name[27:30], 1024, 0)
Dcluster_resolver.py56 …devices.append(session._DeviceAttributes(name, device_type, 0, 0)) # pylint: disable=protected-ac…
/external/tensorflow/tensorflow/python/tpu/
Dtpu_system_metadata.py72 …devices.append(session_lib._DeviceAttributes(name, device_type, 0, 0)) # pylint: disable=protecte…
/external/tensorflow/tensorflow/python/client/
Dsession.py576 class _DeviceAttributes(object): class
713 _DeviceAttributes(name, device_type, memory, incarnation))
Dsession_test.py1990 attrs = session._DeviceAttributes(
2000 attrs = session._DeviceAttributes('/job:worker/replica:0/task:3/cpu:1',