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
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,
62 class DeviceAssignment(object): class
66 typedef struct DeviceAssignment { struct69 } DeviceAssignment; typedef
39 class DeviceAssignment; variable
10 message DeviceAssignment { message
322 DeviceAssignment = _xla.DeviceAssignment variable