Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for parallel_device_execute (0.16 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

                                /*in_flight_nodes_limit=*/in_flight_nodes_limit)),
            op_(nullptr),
            thread_(tensorflow::Env::Default()->StartThread(
                tensorflow::ThreadOptions(), "parallel_device_execute",
                std::bind(&DeviceThread::Run, this))) {}
      ~DeviceThread();
    
      // Requests that the worker thread execute the specified operation. Blocks
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device.cc

    //
    // Since this function is used to satisfy the TFE_CustomDevice C API,
    // device_info is passed in using a C-style generic. It must always be a
    // ParallelDevice.
    void ParallelDeviceExecute(const TFE_Op* original_op, int* num_outputs,
                               TFE_TensorHandle** outputs, TF_Status* status,
                               void* device_info) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
Back to top