- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for device_threads_ (0.21 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
: underlying_devices_(devices), default_cancellation_manager_(absl::make_unique<CancellationManager>()) { device_threads_.reserve(devices.size()); for (int device_index = 0; device_index < devices.size(); ++device_index) { device_threads_.emplace_back(new DeviceThread( devices[device_index].c_str(), is_async, in_flight_nodes_limit)); } }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
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.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0)