Searched refs:device_assignment_proto (Results 1 – 2 of 2) sorted by relevance
415 xla::DeviceAssignmentProto device_assignment_proto; in GetGeneralTPUExecutionDeviceAssignment() local416 TF_RETURN_IF_ERROR(device_assignment.Serialize(&device_assignment_proto)); in GetGeneralTPUExecutionDeviceAssignment()419 std::move(devices_and_hosts), std::move(device_assignment_proto)); in GetGeneralTPUExecutionDeviceAssignment()
209 DeviceAssignmentProto device_assignment_proto; in ConvertCppCompileOptionsToCCompileOptions() local212 &device_assignment_proto)); in ConvertCppCompileOptionsToCCompileOptions()213 *device_assignment_str = device_assignment_proto.SerializeAsString(); in ConvertCppCompileOptionsToCCompileOptions()