Searched refs:device_assignment_lib (Results 1 – 4 of 4) sorted by relevance
56 from tensorflow.python.tpu import device_assignment as device_assignment_lib unknown201 device_assignment = device_assignment_lib.DeviceAssignment.build(208 device_assignment2 = device_assignment_lib.DeviceAssignment.build(319 device_assignment = device_assignment_lib.DeviceAssignment.build(327 device_assignment2 = device_assignment_lib.DeviceAssignment(354 device_assignment = device_assignment_lib.DeviceAssignment.build(362 device_assignment2 = device_assignment_lib.DeviceAssignment(1104 device_assignment = device_assignment_lib.DeviceAssignment(1173 device_assignment = device_assignment_lib.DeviceAssignment(1209 device_assignment = device_assignment_lib.DeviceAssignment([all …]
37 from tensorflow.python.tpu import device_assignment as device_assignment_lib unknown106 device_assignment = device_assignment_lib.DeviceAssignment(108 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]:885 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1178 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1717 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1874 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1957 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,2018 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,2175 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,