Searched defs:devices_to_lock (Results 1 – 1 of 1) sorted by relevance
1716 std::vector<int>* devices_to_lock) { in BuildFullMeshDeviceAssignment()1741 std::vector<int>* devices_to_lock, in BuildGeneralDeviceAssignment()1781 std::vector<int>* devices_to_lock, in BuildDeviceAssignment()4375 std::vector<int>* devices_to_lock, in GetDeviceTopology()4572 absl::Span<const int> devices_to_lock, Node** control_after_compilation, in BuildCompilationStatusReturnNodes()4729 std::vector<int> devices_to_lock; in RewriteTPUReplicateNode() local