Searched refs:device_assignment_lib (Results 1 – 4 of 4) sorted by relevance
59 from tensorflow.python.tpu import device_assignment as device_assignment_lib unknown254 device_assignment = device_assignment_lib.DeviceAssignment.build(261 device_assignment2 = device_assignment_lib.DeviceAssignment.build(372 device_assignment = device_assignment_lib.DeviceAssignment.build(380 device_assignment2 = device_assignment_lib.DeviceAssignment(407 device_assignment = device_assignment_lib.DeviceAssignment.build(415 device_assignment2 = device_assignment_lib.DeviceAssignment(1253 device_assignment = device_assignment_lib.DeviceAssignment(1322 device_assignment = device_assignment_lib.DeviceAssignment(1358 device_assignment = device_assignment_lib.DeviceAssignment([all …]
41 from tensorflow.python.tpu import device_assignment as device_assignment_lib unknown111 device_assignment = device_assignment_lib.DeviceAssignment(113 core_assignment=device_assignment_lib.SINGLE_CORE_ASSIGNMENT)
61 from tensorflow.python.tpu import device_assignment as device_assignment_lib # pylint: disable=unu… unknown
55 from tensorflow.python.tpu import device_assignment as device_assignment_lib unknown221 ) -> Optional[device_assignment_lib.DeviceAssignment]:900 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1192 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1733 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1896 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1979 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,2040 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,2197 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,