Searched defs:devices_to_lock (Results 1 – 1 of 1) sorted by relevance
1769 std::vector<int>* devices_to_lock) { in BuildFullMeshDeviceAssignment()1794 std::vector<int>* devices_to_lock, in BuildGeneralDeviceAssignment()1834 std::vector<int>* devices_to_lock, in BuildDeviceAssignment()4428 std::vector<int>* devices_to_lock, in GetDeviceTopology()4625 absl::Span<const int> devices_to_lock, Node** control_after_compilation, in BuildCompilationStatusReturnNodes()4786 std::vector<int> devices_to_lock; in RewriteTPUReplicateNode() local