- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 37 for device_name (0.26 sec)
-
tensorflow/c/eager/c_api_cluster_test.cc
EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); const char remote_device_name[] = "/job:localhost/replica:0/task:1/device:CPU:0"; const char local_device_name[] = "/job:localhost/replica:0/task:0/device:CPU:0"; CheckRemoteMatMulExecutesOK(ctx, remote_device_name, local_device_name); TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_tensor_handle.cc
} 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, ", dtype=", DataType_Name(DataType()), ", device=\"", device_name, "\")"); } Status ImmediateExecutionTensorHandle::SummarizeValue(
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 2.1K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
const char* device_name = "/job:localhost/replica:0/task:0/device:CUSTOM:0"; const char* first_device_name = "/job:localhost/replica:0/task:0/device:CPU:0"; const char* second_device_name = "/job:localhost/replica:0/task:0/device:CPU:1"; std::array<const char*, 2> underlying_devices{first_device_name, second_device_name};
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device.h
namespace tensorflow { namespace parallel_device { // Allocate a parallel device named `device_name` which forwards operations to // `underlying_devices`, maintaining "parallel tensors" with components placed // on each underlying device. // // For example if `device_name` is // "/job:localhost/replica:0/task:0/device:CUSTOM:0" // and `underlying_devices` is
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jun 04 21:49:16 GMT 2020 - 2.9K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
*(dev->arrived_flag) = true; return MakeLoggedTensorHandle(context, dev->device_name, std::move(dst), status); } TFE_TensorHandle* CopyTensorFromLoggingDevice(TFE_Context* context, TFE_TensorHandle* tensor, const char* target_device_name, TF_Status* status,
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.h
// `device_name`. TFE_TensorHandle* CreateVarHandle(TFE_Context* ctx, const tensorflow::string& device_name, const tensorflow::string& variable_name); // Create a variable with value `value` and name `variable_name` on a device // with name `device_name`. TFE_TensorHandle* CreateVariable(TFE_Context* ctx, float value,
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Jul 17 23:43:59 GMT 2023 - 7.7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
std::unique_ptr<ParallelDevice> parallel_device) : device_name_(name), parallel_device_(std::move(parallel_device)) {} const std::string& name() const { return device_name_; } const ParallelDevice& device() const { return *parallel_device_; } private: std::string device_name_; std::unique_ptr<ParallelDevice> parallel_device_; };
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test.cc
const char remote_device_name[] = "/job:localhost/replica:0/task:1/device:CPU:0"; auto* h0_task1 = TFE_TensorHandleCopyToDevice(h0_task0, ctx, remote_device_name, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); auto* h1_task1 = TFE_TensorHandleCopyToDevice(h1_task0, ctx, remote_device_name, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Aug 12 00:14:22 GMT 2020 - 5.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
} } // namespace extern "C" { bool TFE_IsCustomDevice(TFE_Context* ctx, const char* device_name) { return tensorflow::unwrap(ctx)->IsCustomDevice(device_name); } void TFE_RegisterCustomDevice(TFE_Context* ctx, TFE_CustomDevice device, const char* device_name, void* device_info, TF_Status* status) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
void Release() override { delete this; } Status Reset(const char* op, const char* raw_device_name) override { if (op_) { return errors::FailedPrecondition("Reset called on already built op."); } if (raw_device_name) { device_name_ = raw_device_name; } op_type_ = op; return absl::OkStatus(); } Status SetOpName(const char* const op_name) override {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1)