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