Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for device_threads_ (0.21 sec)

  1. 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)
  2. 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)
Back to top