Searched defs:tf_device_assignment (Results 1 – 1 of 1) sorted by relevance
1768 std::vector<std::vector<string>>* tf_device_assignment, in BuildFullMeshDeviceAssignment()1793 std::vector<std::vector<string>>* tf_device_assignment, in BuildGeneralDeviceAssignment()1833 std::vector<std::vector<string>>* tf_device_assignment, in BuildDeviceAssignment()4015 const std::vector<std::vector<string>>& tf_device_assignment, in ReplicateOutsideCompilationNodes()4427 std::vector<std::vector<string>>* tf_device_assignment, in GetDeviceTopology()4737 const std::vector<std::vector<string>>& tf_device_assignment, in UpdateHeadTailOutsideCompilation()4785 std::vector<std::vector<string>> tf_device_assignment; in RewriteTPUReplicateNode() local