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
406 Status Member::AssignDevice(const Node& node, bool allow_soft_placement) { in AssignDevice() function in tensorflow::Member