Searched defs:AssignDevice (Results 1 – 3 of 3) sorted by relevance
33 OpSharding AssignDevice(int device) { in AssignDevice() function
35 HloSharding HloSharding::AssignDevice(int64_t device_id, in AssignDevice() function in xla::HloSharding
467 Status Member::AssignDevice(const Node& node) { in AssignDevice() function in tensorflow::Member