Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for DeviceName (0.07 sec)

  1. tensorflow/c/eager/abstract_operation.h

      // returned by DeviceName should be "/device:GPU:*" until a particular GPU is
      // chosen for the operation by the device placement logic in the
      // executor. After that, the value returned by DeviceName will be a full
      // device name such as "/job:localhost/replica:0/task:0/device:GPU:1".
      virtual const string& DeviceName() const = 0;
    
      // Sets the operation device name.
      //
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/immediate_execution_tensor_handle.cc

        // The default NumPy-style output can be distractingly long in error
        // messages.
        value_string = absl::StrCat(value_string.substr(0, 100), " [...]");
      }
      absl::Status s;
      const char* device_name = DeviceName(&s);
      if (!s.ok()) {
        device_name = "<error fetching device name>";
      }
      return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:24:07 UTC 2024
    - 2.1K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_cluster_test.cc

      TFE_TensorHandle* var_handle0 = TestVariable(ctx, 1.0, dev0_name);
      EXPECT_NE(var_handle0, nullptr);
    
      absl::Status status2;
      EXPECT_EQ(tensorflow::unwrap(var_handle0)->DeviceName(&status2), dev0_name);
    
      // Rename local device
      // This server def has the task index set to 0.
      string serialized = server_def.SerializeAsString();
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  4. 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)
  5. 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: 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/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_; }
    
      absl::Status SetDeviceName(const char* name) override {
        // TODO(srbs): Implement this.
        device_name_ = name;
        return absl::OkStatus();
      }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
Back to top