Searched refs:tf_device_assignment (Results 1 – 3 of 3) sorted by relevance
270 std::vector<std::vector<string>>* tf_device_assignment,478 const std::vector<std::vector<string>>& tf_device_assignment,523 std::vector<std::vector<string>>* tf_device_assignment,578 const std::vector<std::vector<string>>& tf_device_assignment,
1505 std::vector<std::vector<string>>* tf_device_assignment) { in BuildFullMeshDeviceAssignment() argument1515 (*tf_device_assignment)[i] = {tpu_devices[task][device]->name()}; in BuildFullMeshDeviceAssignment()1528 std::vector<std::vector<string>>* tf_device_assignment, in BuildGeneralDeviceAssignment() argument1554 (*tf_device_assignment)[replica].push_back( in BuildGeneralDeviceAssignment()1566 std::vector<std::vector<string>>* tf_device_assignment, in BuildDeviceAssignment() argument1592 tf_device_assignment->clear(); in BuildDeviceAssignment()1593 tf_device_assignment->resize(num_replicas); in BuildDeviceAssignment()1632 (*tf_device_assignment)[0] = {tpu_devices[0][0]->name()}; in BuildDeviceAssignment()1641 tf_device_assignment); in BuildDeviceAssignment()1658 topology, tf_device_assignment, xla_device_assignment); in BuildDeviceAssignment()[all …]
924 "transforms/tf_device_assignment.cc",