Searched defs:AssignDevice (Results 1 – 3 of 3) sorted by relevance
27 OpSharding AssignDevice(int device) { in AssignDevice() function
29 HloSharding HloSharding::AssignDevice(int64 device_id) { in AssignDevice() function in xla::HloSharding
440 Status Member::AssignDevice(const Node& node) { in AssignDevice() function in tensorflow::Member