Searched defs:DeviceAssignment (Results 1 – 7 of 7) sorted by relevance
41 DeviceAssignment() {} in DeviceAssignment() function42 DeviceAssignment(int replica_count, int computation_count) in DeviceAssignment() function
919 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1211 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1788 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,1952 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,2036 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,2098 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,2255 device_assignment: Optional[device_assignment_lib.DeviceAssignment] = None,
57 class DeviceAssignment(object): class
66 typedef struct DeviceAssignment { struct69 } DeviceAssignment; argument
39 class DeviceAssignment; variable
10 message DeviceAssignment { message
267 DeviceAssignment = _xla.DeviceAssignment variable