- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for DeviceName (0.06 sec)
-
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) -
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) -
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) -
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) -
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) -
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)