Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TFE_OpGetDevice (0.07 sec)

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

    void ParallelDeviceExecute(const TFE_Op* original_op, int* num_outputs,
                               TFE_TensorHandle** outputs, TF_Status* status,
                               void* device_info) {
      const char* requested_placement = TFE_OpGetDevice(original_op, status);
      if (*requested_placement == '\0') {
        TF_SetStatus(
            status, TF_INTERNAL,
            "Ops must be placed on the parallel device explicitly, or their inputs "
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

    }
    
    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) {
      return tensorflow::unwrap(op)->DeviceName().c_str();
    }
    
    void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_experimental.h

      // Method to execute an operation.
      //
      // Arguments provide enough information to reconstruct the original `TFE_Op`,
      // or construct a transformed version, by inspecting the passed `op`.
      //
      // TFE_OpGetDevice(op) records the original placement of the operation. It may
      // be an empty string if no device was explicitly requested, but will
      // otherwise be the name of this custom device. Ops are placed onto a custom
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 39.5K bytes
    - Viewed (0)
Back to top