Searched refs:DeviceSpecV2 (Results 1 – 8 of 8) sorted by relevance
52 class DeviceSpecV2(object): class402 class DeviceSpecV1(DeviceSpecV2):403 __doc__ = DeviceSpecV2.__doc__404 __slots__ = DeviceSpecV2.__slots__406 @DeviceSpecV2.job.setter411 @DeviceSpecV2.replica.setter416 @DeviceSpecV2.task.setter421 @DeviceSpecV2.device_type.setter426 @DeviceSpecV2.device_index.setter462 to_string.__doc__ = DeviceSpecV2.to_string.__doc__[all …]
27 DeviceSpec = device_spec.DeviceSpecV247 return isinstance(obj, device_spec.DeviceSpecV2)118 if isinstance(spec, device_spec.DeviceSpecV2):
29 ("v2", device_spec.DeviceSpecV2))201 d = device_spec.DeviceSpecV2()202 d.merge_from(device_spec.DeviceSpecV2.from_string("/task:1/cpu:0"))
33 ("v2", device_spec.DeviceSpecV2))
3 is_instance: "<class \'tensorflow.python.framework.device_spec.DeviceSpecV2\'>"
4 is_instance: "<class \'tensorflow.python.framework.device_spec.DeviceSpecV2\'>"
819 job_name = device_spec.DeviceSpecV2.from_string(tpu_devices_flat[0]).job
851 if isinstance(device, device_spec.DeviceSpecV2):871 return device_spec.DeviceSpecV2.from_string(op.device)