Searched defs:DeviceAssignment (Results 1 – 7 of 7) sorted by relevance
43 DeviceAssignment() {} in DeviceAssignment() function44 DeviceAssignment(int replica_count, int computation_count) in DeviceAssignment() function
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,
62 class DeviceAssignment(object): class
66 typedef struct DeviceAssignment { struct69 } DeviceAssignment; typedef
39 class DeviceAssignment; variable
10 message DeviceAssignment { message
294 DeviceAssignment = _xla.DeviceAssignment variable