Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for BackingDeviceName (0.22 sec)

  1. tensorflow/c/eager/immediate_execution_tensor_handle.h

      // Returns the device which created the handle.
      virtual const char* DeviceName(absl::Status* status) const = 0;
      // Returns the device where the tensor was placed.
      virtual const char* BackingDeviceName(absl::Status* status) const = 0;
      // Returns the device type which created the handle.
      virtual const char* DeviceType(absl::Status* status) const = 0;
      // Returns the device ID which created the handle.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/dlpack.cc

    }
    
    // Gets DLPack's DLDevice from eager tensor handle.
    DLDevice GetDlContext(TFE_TensorHandle* h, TF_Status* status) {
      DLDevice ctx;
      const char* device_name =
          tensorflow::unwrap(h)->BackingDeviceName(&status->status);
      DeviceNameUtils::ParsedName parsed_name;
      tensorflow::DeviceNameUtils::ParseFullName(device_name, &parsed_name);
      std::string device_type = parsed_name.type;
      int device_id = 0;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.9K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.cc

      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
      }
      return tensorflow::unwrap(h)->BackingDeviceName(&status->status);
    }
    
    TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopySharingTensor(
        TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top