Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 388 for Device (0.22 sec)

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

      underlying_devices_vector.reserve(num_underlying_devices);
      for (int device_index = 0; device_index < num_underlying_devices;
           ++device_index) {
        underlying_devices_vector.push_back(underlying_devices[device_index]);
      }
      std::unique_ptr<ParallelDevice> parallel_device(
          new ParallelDevice(underlying_devices_vector));
      *device_info =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

    ParallelDevice::ParallelDevice(const std::vector<std::string>& devices,
                                   bool is_async, int in_flight_nodes_limit)
        : underlying_devices_(devices),
          default_cancellation_manager_(absl::make_unique<CancellationManager>()) {
      device_threads_.reserve(devices.size());
      for (int device_index = 0; device_index < devices.size(); ++device_index) {
        device_threads_.emplace_back(new DeviceThread(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  3. tensorflow/c/eager/parallel_device/parallel_device_lib.h

           ++device_index) {
        auto device_value = absl::make_unique<DataType>();
        *device_value = values[device_index];
        std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)> tensor(
            TF_NewTensor(
                datatype_enum, /*dims=*/nullptr, /*num_dims=*/0,
                device_value.release(), sizeof(DataType),
                [](void* data, size_t, void* arg) {
                  delete reinterpret_cast<DataType*>(data);
                },
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 25 15:21:13 GMT 2023
    - 12.9K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_experimental.h

    //
    // `device_name` must not name an existing physical or custom device. It must
    // follow the format:
    //
    //    /job:<name>/replica:<replica>/task:<task>/device:<type>:<device_num>
    //
    // If the device is successfully registered, `status` is set to TF_OK. Otherwise
    // the device is not usable. In case of a bad status, `device.delete_device` is
    // still called on `device_info` (i.e. the caller does not retain ownership).
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  5. tensorflow/c/eager/immediate_execution_context.h

      virtual bool UsesTFRT() = 0;
    
      // List attributes of available devices
      virtual void ListDevices(std::vector<DeviceAttributes>* devices) = 0;
    
      // Add `devices` into context's device manager. Context's device manager
      // will take ownership and maintain devices' lifetime.
      virtual Status AddDevices(std::vector<std::unique_ptr<Device>> devices) = 0;
    
      // Block until all pending nodes are finished.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 06 08:34:00 GMT 2023
    - 12.3K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.h

    // Controls how to act when we try to run an operation on a given device but
    // some input tensors are not on that device.
    // LINT.IfChange
    // Note: Keep in sync with internal copy of enum in eager/context.h.
    typedef enum TFE_ContextDevicePlacementPolicy {
      // Running operations with input tensors on the wrong device will fail.
      TFE_DEVICE_PLACEMENT_EXPLICIT = 0,
      // Copy the tensor to the right device but log a warning.
      TFE_DEVICE_PLACEMENT_WARN = 1,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
  7. tensorflow/c/eager/c_api_test_util.cc

                          send_device.size());
      TFE_OpSetAttrString(op, "recv_device", recv_device.c_str(),
                          recv_device.size());
      TFE_OpSetAttrInt(op, "send_device_incarnation", send_device_incarnation);
    
      return op;
    }
    
    bool GetDeviceName(TFE_Context* ctx, string* device_name,
                       const char* device_type) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  8. tensorflow/c/eager/c_api.cc

                                  const char* device_name, void* device_info,
                                  TF_Status* status) {
      // Fill in default values for optional functionality.
      if (device.pack == nullptr) {
        device.pack = &DefaultCustomDevicePack;
      }
      auto custom_device = std::make_unique<tensorflow::CustomDeviceAPI>(
          ctx, device, device_info, device_name);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  9. tensorflow/c/eager/c_api_cluster_test.cc

      // Check that copying it to the old remote device (named localhost) fails.
      TFE_TensorHandleCopyToDevice(h0_task0_new, ctx, remote_device_name, status);
      EXPECT_NE(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      // Copying and executing on the new remote device works.
      const char new_remote_device_name[] =
          "/job:worker/replica:0/task:1/device:CPU:0";
      const char new_local_device_name[] =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/dlpack.cc

      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;
      if (parsed_name.has_id) {
        device_id = parsed_name.id;
      }
    
      ctx.device_id = device_id;
      if (device_type == "CPU") {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
Back to top