- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for device_threads_ (0.13 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
StatusPtr first_bad_status(nullptr); for (int device_index = 0; device_index < underlying_devices_.size(); ++device_index) { DeviceThread* device_thread = device_threads_[device_index].get(); per_device_output_tensors.push_back(device_thread->Join(status)); // We will run every Join even if there are bad statuses in case the user // wants to recover and continue running ops on the parallel device (which
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
// TODO(allenl): Keep a map from outer thread to list of inner threads rather // than a single list of threads so aliased nested parallel devices don't // re-use a thread. std::vector<std::unique_ptr<DeviceThread>> device_threads_; // A cancellation manager to use if the caller does not provide one. When ops // are executed asynchronously this must outlive the queued op, so it can't be // function-local to Execute.
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 13.1K bytes - Viewed (0)