Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for DeviceName (0.29 sec)

  1. tensorflow/c/eager/c_api.cc

    #if !defined(IS_MOBILE_PLATFORM)
    #include "tensorflow/core/common_runtime/eager/context_distributed_manager.h"
    #endif  // !IS_MOBILE_PLATFORM
    
    using tensorflow::string;
    
    namespace {
    
    string DeviceName(const tensorflow::Device* d) {
      return (d == nullptr) ? "cpu:0" : d->name();
    }
    
    // Annotate eager runtime construction context to the given `function_def` as
    // an attribute.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc

      void Release() override { delete this; }
    
      Status Reset(const char* op, const char* raw_device_name) override;
    
      const string& Name() const override;
    
      const string& DeviceName() const override;
    
      Status SetDeviceName(const char* name) override;
    
      Status AddInput(AbstractTensorHandle* input) override;
      Status AddInputList(absl::Span<AbstractTensorHandle* const> inputs) override;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  3. pkg/kubelet/cm/dra/manager_test.go

    	s.prepareResourceCalls.Add(1)
    
    	if s.timeout != nil {
    		time.Sleep(*s.timeout)
    	}
    
    	if s.prepareResourcesResponse == nil {
    		deviceName := "claim-" + req.Claims[0].Uid
    		result := s.driverName + "/" + driverClassName + "=" + deviceName
    		return &drapbv1.NodePrepareResourcesResponse{
    			Claims: map[string]*drapbv1.NodePrepareResourceResponse{
    				req.Claims[0].Uid: {
    					CDIDevices: []string{result},
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri May 03 13:30:29 UTC 2024
    - 42K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental_graph.cc

                                              g_->graph.NewName(op_name).c_str()));
        return absl::OkStatus();
      }
      const string& Name() const override { return op_type_; }
      const string& DeviceName() const override { return device_name_; }
    
      Status SetDeviceName(const char* name) override {
        // TODO(srbs): Implement this.
        device_name_ = name;
        return absl::OkStatus();
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 20:00:09 UTC 2024
    - 15.4K bytes
    - Viewed (0)
Back to top