Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 11 - 12 of 12 for TFE_OpSetDevice (0.1 seconds)

  1. tensorflow/c/eager/c_api_experimental.h

    // creating a new op every time. If `raw_device_name` is `NULL` or empty, it
    // does not set the device name. If it's not `NULL`, then it attempts to parse
    // and set the device name. It's effectively `TFE_OpSetDevice`, but it is faster
    // than separately calling it because if the existing op has the same
    // `raw_device_name`, it skips parsing and just leave as it is.
    TF_CAPI_EXPORT extern void TFE_OpReset(TFE_Op* op_to_reset,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Click Count (1)
  2. tensorflow/c/eager/c_api.cc

      return tensorflow::unwrap(op)->Name().c_str();
    }
    
    TFE_Context* TFE_OpGetContext(const TFE_Op* op, TF_Status* status) {
      return tensorflow::wrap(tensorflow::unwrap(op)->GetContext());
    }
    
    void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) {
      status->status = tensorflow::unwrap(op)->SetDeviceName(device_name);
    }
    
    const char* TFE_OpGetDevice(const TFE_Op* op, TF_Status* status) {
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Fri Nov 07 05:55:21 GMT 2025
    - 43.9K bytes
    - Click Count (0)
Back to Top