- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 25 for device_name (0.12 sec)
-
tensorflow/c/eager/c_api_test_util.cc
TFE_OpSetAttrShape(op, "shape", {}, 0, status); TFE_OpSetAttrString(op, "container", "localhost", 0); TFE_OpSetAttrString(op, "shared_name", "", 0); if (!device_name.empty()) { TFE_OpSetDevice(op, device_name.c_str(), status); } if (TF_GetCode(status) != TF_OK) return nullptr; TFE_TensorHandle* var_handle = nullptr; int num_retvals = 1; TFE_Execute(op, &var_handle, &num_retvals, status);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
cpu_value.get(), context.get(), device_name, status.get())); EXPECT_EQ(TF_GetCode(status.get()), TF_UNIMPLEMENTED); std::array<TFE_TensorHandle*, 2> components{cpu_value.get(), cpu_value.get()}; TensorHandlePtr device_value = CreatePerDeviceValues( context.get(), components, device_name, status.get()); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Tue Aug 06 23:56:17 UTC 2024 - 29.4K bytes - Viewed (0) -
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
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Jun 04 21:49:16 UTC 2020 - 2.9K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_tensor_handle.cc
} 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, ", dtype=", DataType_Name(DataType()), ", device=\"", device_name, "\")"); } absl::Status ImmediateExecutionTensorHandle::SummarizeValue(
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_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,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Jul 17 23:43:59 UTC 2023 - 7.7K 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) {
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/custom_device_testutil.cc
#include "tensorflow/c/tf_status.h" #include "tensorflow/core/lib/gtl/cleanup.h" #include "tensorflow/core/platform/test.h" namespace { struct LoggingDevice { tensorflow::string device_name; tensorflow::string underlying_device; // Set to true whenever a TensorHandle is copied onto the device bool* arrived_flag; // Set to true whenever an operation is executed bool* executed_flag;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Mar 03 20:47:31 UTC 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
const char* second_device = "/job:worker/replica:0/task:2/device:CPU:0"; const char* device_name = "/job:localhost/replica:0/task:0/device:CUSTOM:0"; std::array<const char*, 2> underlying_devices{first_device, second_device}; RegisterParallelDevice(context.get(), device_name, underlying_devices, status.get());
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Jul 10 07:18:05 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
DLDevice GetDlContext(TFE_TensorHandle* h, TF_Status* status) { DLDevice ctx; const char* device_name = tensorflow::unwrap(h)->BackingDeviceName(&status->status); DeviceNameUtils::ParsedName parsed_name; tensorflow::DeviceNameUtils::ParseFullName(device_name, &parsed_name); std::string device_type = parsed_name.type; int device_id = 0; if (parsed_name.has_id) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 12.9K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_context.h
//===--------------------------------------------------------------------===// virtual CustomDeviceOpHandler& GetCustomDeviceOpHandler() = 0; // Returns whether `device_name` is registered as a custom device. virtual bool IsCustomDevice(const string& device_name) = 0; // Register a custom device. It will return error is the device name is // already registered.
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 12.3K bytes - Viewed (0)