Searched defs:tf_device_assignment (Results 1 – 1 of 1) sorted by relevance
1715 std::vector<std::vector<string>>* tf_device_assignment, in BuildFullMeshDeviceAssignment()1740 std::vector<std::vector<string>>* tf_device_assignment, in BuildGeneralDeviceAssignment()1780 std::vector<std::vector<string>>* tf_device_assignment, in BuildDeviceAssignment()3962 const std::vector<std::vector<string>>& tf_device_assignment, in ReplicateOutsideCompilationNodes()4374 std::vector<std::vector<string>>* tf_device_assignment, in GetDeviceTopology()4680 const std::vector<std::vector<string>>& tf_device_assignment, in UpdateHeadTailOutsideCompilation()4728 std::vector<std::vector<string>> tf_device_assignment; in RewriteTPUReplicateNode() local