Searched defs:tf_device_assignment (Results 1 – 1 of 1) sorted by relevance
1505 std::vector<std::vector<string>>* tf_device_assignment) { in BuildFullMeshDeviceAssignment()1528 std::vector<std::vector<string>>* tf_device_assignment, in BuildGeneralDeviceAssignment()1566 std::vector<std::vector<string>>* tf_device_assignment, in BuildDeviceAssignment()3508 const std::vector<std::vector<string>>& tf_device_assignment, in ReplicateOutsideCompilationNodes()3920 std::vector<std::vector<string>>* tf_device_assignment, in GetDeviceTopology()4179 const std::vector<std::vector<string>>& tf_device_assignment, in UpdateHeadTailOutsideCompilation()4227 std::vector<std::vector<string>> tf_device_assignment; in RewriteTPUReplicateNode() local