Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 649 for Device (0.06 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.h

     private:
      ParallelTensor(const ParallelDevice& device,
                     std::vector<TensorHandlePtr> tensors,
                     absl::Span<const int64_t> shape, const TF_DataType dtype)
          : device_(device),
            tensors_(std::move(tensors)),
            shape_(std::vector<int64_t>(shape.begin(), shape.end())),
            dtype_(dtype) {}
      ParallelTensor(const ParallelDevice& device,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_experimental.h

                                void* device_info) = nullptr;
    
      // Pins the op to `device` based on inputs to `op`. Returns true
      // signifying to pin to the current custom device. Returns false
      // to pin to the physical device.
      //
      // This function is guaranteed to be called only when all of the custom-device
      // inputs are on this device.
      bool (*shall_pin_to_this_device)(const TFE_Op* op, TF_Status* s) = nullptr;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 39.5K bytes
    - Viewed (0)
  3. 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 absl::Status AddDevices(
          std::vector<std::unique_ptr<Device>> devices) = 0;
    
      // Block until all pending nodes are finished.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device.cc

                                int num_underlying_devices,
                                TFE_CustomDevice* device, void** device_info) {
      device->copy_tensor_to_device = &CopyToParallelDevice;
      device->copy_tensor_from_device = &CopyTensorFromParallelDevice;
      device->delete_device = &DeleteParallelDevice;
      device->execute = &ParallelDeviceExecute;
      std::vector<std::string> underlying_devices_vector;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api.cc

            "TFE_TensorHandleDevicePointer may not be called on a ",
            handle->TypeString(), " tensor handle.");
        return nullptr;
      }
      tensorflow::Device* device(handle->device());
      if (device != nullptr) {
        status->status = device->Sync();
        if (!status->status.ok()) {
          return nullptr;
        }
      }
      const tensorflow::Tensor* tensor;
      status->status = handle->Tensor(&tensor);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  6. 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(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  7. tensorflow/c/eager/abstract_operation.h

      virtual const string& Name() const = 0;
    
      // Returns the operation's device name.
      //
      // The value returned may be different from the one set by SetDeviceName, but
      // it will be compatible with it: the name will be updated by device placement
      // logic to refer to the specific device chosen.
      //
      // Example: If one calls `op->SetDeviceName("/device:GPU")`, the value
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_cluster_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      // The device of var_handle0 is local device which is the same before and
      // after cluster update. Remove resource with valid device should succeed.
      TFE_Op* op = TFE_NewOp(ctx, "DestroyResourceOp", status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpAddInput(op, var_handle0, status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  9. internal/mountinfo/mountinfo_linux_test.go

    	}
    	// Failure case where we detected successfully cross device mounts.
    	{
    		absPaths := []string{"/path/to/1"}
    		if err = checkCrossDevice(absPaths, mountsPath); err == nil {
    			t.Fatal("Expected to fail, but found success")
    		}
    
    		mp := []mountInfo{
    			{"/dev/2", "/path/to/1/2", "type2", []string{"flags"}, "2", "2"},
    		}
    Registered: Sun Nov 03 19:28:11 UTC 2024
    - Last Modified: Thu Jan 18 07:03:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/immediate_execution_tensor_handle.h

      // devices.
      virtual absl::Status Dim(int dim_index, int64_t* dim) const = 0;
    
      // 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.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.3K bytes
    - Viewed (0)
Back to top