- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 317 for Device (0.28 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
ParallelDevice::ParallelDevice(const std::vector<std::string>& devices, bool is_async, int in_flight_nodes_limit) : underlying_devices_(devices), default_cancellation_manager_(absl::make_unique<CancellationManager>()) { device_threads_.reserve(devices.size()); for (int device_index = 0; device_index < devices.size(); ++device_index) { device_threads_.emplace_back(new DeviceThread(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental.h
// // `device_name` must not name an existing physical or custom device. It must // follow the format: // // /job:<name>/replica:<replica>/task:<task>/device:<type>:<device_num> // // If the device is successfully registered, `status` is set to TF_OK. Otherwise // the device is not usable. In case of a bad status, `device.delete_device` is // still called on `device_info` (i.e. the caller does not retain ownership).
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_context.h
virtual bool UsesTFRT() = 0; // List attributes of available devices virtual void ListDevices(std::vector<DeviceAttributes>* devices) = 0; // Add `devices` into context's device manager. Context's device manager // will take ownership and maintain devices' lifetime. virtual Status AddDevices(std::vector<std::unique_ptr<Device>> devices) = 0; // Block until all pending nodes are finished.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 06 08:34:00 GMT 2023 - 12.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
send_device.size()); TFE_OpSetAttrString(op, "recv_device", recv_device.c_str(), recv_device.size()); TFE_OpSetAttrInt(op, "send_device_incarnation", send_device_incarnation); return op; } bool GetDeviceName(TFE_Context* ctx, string* device_name, const char* device_type) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/eager/c_api.cc
const char* device_name, void* device_info, TF_Status* status) { // Fill in default values for optional functionality. if (device.pack == nullptr) { device.pack = &DefaultCustomDevicePack; } auto custom_device = std::make_unique<tensorflow::CustomDeviceAPI>( ctx, device, device_info, device_name);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/dlpack.cc
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) { device_id = parsed_name.id; } ctx.device_id = device_id; if (device_type == "CPU") {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
// Remote device on `worker2`. const char remote_device[] = "/job:localhost/replica:0/task:2/device:CPU:0"; // `ctx_0`, `ctx_1` contains `remote_device`. { const std::vector<std::string>& device_names = ListDeviceNames(ctx_0); ASSERT_TRUE(std::find(device_names.begin(), device_names.end(), remote_device) != device_names.end()); } {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/eager/c_api_distributed_test.cc
const string composite_device_name = "/job:localhost/replica:0/task:0/device:COMPOSITE:0"; EXPECT_EQ(TFE_TensorHandleDeviceName(packed_handle, status), composite_device_name); EXPECT_EQ(TFE_TensorHandleBackingDeviceName(packed_handle, status), composite_device_name); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
tensorflow::DeviceType(device_type), std::move(*pjrt_client)); status->status = s; } void TF_ResetPjRtCClient(const char* device_type, TF_Status* status) { status->status = tensorflow::ResetPjRtClient(tensorflow::DeviceType(device_type)); } PJRT_Client* TF_GetPjRtCClient(const char* device_type, TF_Status* status) {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental.cc
// One of the inputs we're trying to pack is on a custom device. We'll let // the first custom device we see handle all of the packing. auto* custom_device_handle = tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( unwrapped_handle); tensorflow::ImmediateExecutionTensorHandle* result; status->status = custom_device_handle->device()->Pack(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3)