Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 54 for set_device (0.12 sec)

  1. pkg/volume/volume.go

    type CustomBlockVolumeMapper interface {
    	BlockVolumeMapper
    	// SetUpDevice prepares the volume to the node by the plugin specific way.
    	// For most in-tree plugins, attacher.Attach() and attacher.WaitForAttach()
    	// will do necessary works.
    	// This may be called more than once, so implementations must be idempotent.
    	// SetUpDevice returns stagingPath if device setup was successful
    	SetUpDevice() (stagingPath string, err error)
    
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Tue May 14 06:17:25 UTC 2024
    - 11.8K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/stream_executor/stream_executor.cc

                              SP_DEVICE_MEMORY_BASE_STRUCT_SIZE);
      // All other fields could theoretically be zero/null.
      return absl::OkStatus();
    }
    
    absl::Status ValidateSPDevice(const SP_Device& device) {
      TF_VALIDATE_STRUCT_SIZE(SP_Device, device, SP_DEVICE_STRUCT_SIZE);
      // All other fields could theoretically be zero/null.
      return absl::OkStatus();
    }
    
    absl::Status ValidateSPDeviceFns(const SP_DeviceFns& device_fns) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 27.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc

      // Ensure that users of the launch op's results appear after the launch op
      // in order to preserve the dominance property.
      TF::ReorderOpResultUses(launch_op);
    }
    
    std::string GetDevice(Operation* op) {
      auto device_attr = op->getAttrOfType<StringAttr>("device");
      return device_attr ? device_attr.getValue().str() : "";
    }
    
    bool CanBeIgnoredInCluster(Operation* op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 05 13:30:21 UTC 2023
    - 6.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_device.h

    // registered on 'jit_device'. Returns ownership of a XlaDeviceOpRegistrations
    // object that encapsulates the kernel registrations.
    struct XlaDeviceOpRegistrations {
      std::vector<std::unique_ptr<kernel_factory::OpKernelRegistrar>>
          op_kernel_registrars;
    };
    
    XlaDeviceOpRegistrations* RegisterXlaDeviceKernels(
        const char* device, const char* jit_device,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 13.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/pjrt_tensor_buffer_util_test.cc

    namespace tensorflow {
    namespace {
    
    TEST(PjRtTensorBufferUtilTest, MakeTensorFromPjRtBuffer) {
      DeviceSetup device_setup;
      device_setup.AddDevicesAndSetUp({DEVICE_GPU});
      Device* device = device_setup.GetDevice(DEVICE_GPU);
      std::vector<int64_t> dimensions = {2, 3};
      Tensor dest_cpu_tensor(cpu_allocator(), tensorflow::DT_INT32,
                             tensorflow::TensorShape(dimensions));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Sep 14 18:14:47 UTC 2023
    - 2.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/utils/cluster_util_test.cc

      OwningOpRef<ModuleOp> mlir_module;
      auto status =
          tensorflow::DeserializeMlirModule(string, context, &mlir_module);
      if (!status.ok()) {
        return status;
      }
      return mlir_module;
    }
    
    std::string GetDevice(Operation* op) {
      auto device_attr = op->getAttrOfType<StringAttr>("device");
      return device_attr ? device_attr.getValue().str() : "";
    }
    
    bool CanBeIgnoredInCluster(Operation* op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/xla_device.cc

    XlaDeviceOpRegistrations* RegisterXlaDeviceKernels(
        const char* device, const char* jit_device,
        OpKernel* (*factory)(OpKernelConstruction*),
        StringPiece kernel_class_name) {
      XlaOpRegistry::RegisterCompilationKernels();
      XlaDeviceOpRegistrations* registrations = new XlaDeviceOpRegistrations;
      for (const KernelDef* jit_def : XlaOpRegistry::DeviceKernels(
               jit_device,
               /*include_compilation_only_kernels=*/false)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 21:05:42 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/test_util.h

    // Helps set up devices for unit tests.
    class DeviceSetup {
     public:
      void AddDevicesAndSetUp(
          const std::vector<std::string>& device_names,
          const std::optional<FunctionDef>& fdef = std::nullopt);
      Device* GetDevice(const string& device_name);
      FunctionLibraryRuntime* flr() { return flr_; }
    
     private:
      FunctionLibraryRuntime* flr_;
      std::unique_ptr<DeviceMgr> device_mgr_;
      std::unique_ptr<FunctionLibraryDefinition> lib_def_;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 16 18:03:15 UTC 2023
    - 3.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/test_util.cc

          TF_GRAPH_DEF_VERSION, lib_def_.get(), opts,
          /*default_thread_pool=*/nullptr, /*cluster_flr=*/nullptr);
      flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0");
    }
    
    Device* DeviceSetup::GetDevice(const string& device_name) {
      if (device_mgr_ == nullptr) {
        return nullptr;
      }
    
      string full_device_name = absl::StrCat(
          "/job:localhost/replica:0/task:0/device:", device_name, ":0");
      Device* device;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 11:36:41 UTC 2024
    - 3.7K bytes
    - Viewed (0)
  10. src/cmd/vendor/golang.org/x/sys/unix/ioctl_linux.go

    	err := ioctlPtr(fd, _HIDIOCGRAWUNIQ, unsafe.Pointer(&value[0]))
    	return ByteSliceToString(value[:]), err
    }
    
    // IoctlIfreq performs an ioctl using an Ifreq structure for input and/or
    // output. See the netdevice(7) man page for details.
    func IoctlIfreq(fd int, req uint, value *Ifreq) error {
    	// It is possible we will add more fields to *Ifreq itself later to prevent
    	// misuse, so pass the raw *ifreq directly.
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed Nov 29 21:28:33 UTC 2023
    - 8.1K bytes
    - Viewed (0)
Back to top