Searched defs:AssignDevice (Results 1 – 2 of 2) sorted by relevance
27 OpSharding AssignDevice(int device) { in AssignDevice() function
25 HloSharding HloSharding::AssignDevice(int64 device_id) { in AssignDevice() function in xla::HloSharding