Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for device_ids (0.06 sec)

  1. tensorflow/c/eager/dlpack.cc

      DeviceNameUtils::ParsedName parsed_name;
      tensorflow::DeviceNameUtils::ParseFullName(device_name, &parsed_name);
      std::string device_type = parsed_name.type;
      int device_id = 0;
      if (parsed_name.has_id) {
        device_id = parsed_name.id;
      }
    
      ctx.device_id = device_id;
      if (device_type == "CPU") {
        ctx.device_type = DLDeviceType::kDLCPU;
      } else if (device_type == "GPU") {
    #if TENSORFLOW_USE_ROCM
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/immediate_execution_tensor_handle.h

      // 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.
      virtual int DeviceId(absl::Status* status) const = 0;
      // Returns a tensor for the handle. If tensor is remote, it will be copied.
      virtual AbstractTensorInterface* Resolve(absl::Status* status) = 0;
    
      std::string DebugString() const override;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_lib.h

          absl::Span<const DataType> values, TFE_Context* context,
          TF_Status* status) const;
    
      // A parallel tensor with scalar integers numbering component devices.
      std::unique_ptr<ParallelTensor> DeviceIDs(TFE_Context* context,
                                                TF_Status* status) const;
    
      // The number of devices operations run on.
      size_t num_underlying_devices() const { return underlying_devices_.size(); }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

        components.emplace_back(t);
      }
      return ParallelTensor::FromTensorHandles(*this, std::move(components),
                                               status);
    }
    
    std::unique_ptr<ParallelTensor> ParallelDevice::DeviceIDs(
        TFE_Context* context, TF_Status* status) const {
      std::vector<int32_t> ids;
      ids.reserve(num_underlying_devices());
      for (int i = 0; i < num_underlying_devices(); ++i) {
        ids.push_back(i);
      }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_experimental.cc

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