- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 183 for device_ (0.16 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
#include "tensorflow/c/eager/parallel_device/parallel_device_lib.h" #include "tensorflow/c/c_api.h" #include "tensorflow/c/c_api_experimental.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_experimental.h" #include "tensorflow/c/eager/parallel_device/parallel_device_testlib.h" #include "tensorflow/c/eager/tfe_context_internal.h"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
const char* first_device = "/job:worker/replica:0/task:1/device:CPU:0"; 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());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 22:09:57 GMT 2023 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
absl::optional<int64_t> step_id) const { for (int device_index = 0; device_index < underlying_devices_.size(); ++device_index) { DeviceThread* device_thread = device_threads_[device_index].get(); std::vector<TFE_TensorHandle*> device_inputs; device_inputs.reserve(inputs.size()); for (int input_index = 0; input_index < inputs.size(); ++input_index) {
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/parallel_device/parallel_device_test.cc
TFE_ContextListDevices(context.get(), status.get()), TF_DeleteDeviceList); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); bool has_tpu = false; for (int device_index = 0; device_index < TF_DeviceListCount(devices.get()); ++device_index) { std::string device_type = TF_DeviceListType(devices.get(), device_index, status.get());
C++ - Registered: Tue Apr 30 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_lib.h
++device_index) { auto device_value = absl::make_unique<DataType>(); *device_value = values[device_index]; std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)> tensor( TF_NewTensor( datatype_enum, /*dims=*/nullptr, /*num_dims=*/0, device_value.release(), sizeof(DataType), [](void* data, size_t, void* arg) { delete reinterpret_cast<DataType*>(data); },
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
// Create and modify a variable placed on a parallel device which composes // `first_device` and `second_device`. void BasicTestsForTwoDevices(TFE_Context* context, const char* first_device, const char* second_device) { // Register the custom device std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
underlying_devices_vector.reserve(num_underlying_devices); for (int device_index = 0; device_index < num_underlying_devices; ++device_index) { underlying_devices_vector.push_back(underlying_devices[device_index]); } std::unique_ptr<ParallelDevice> parallel_device( new ParallelDevice(underlying_devices_vector)); *device_info =
C++ - Registered: Tue Apr 30 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_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_test.cc
TensorHandleSilentCopy(false, TFE_DEVICE_PLACEMENT_SILENT, TFE_DEVICE_PLACEMENT_SILENT, false); } TEST(CAPI, TensorHandleSilentCopyAsync) { TensorHandleSilentCopy(true, TFE_DEVICE_PLACEMENT_SILENT, TFE_DEVICE_PLACEMENT_SILENT, false); } TEST(CAPI, TensorHandleSilentCopyLocalPolicy) { TensorHandleSilentCopy(false, TFE_DEVICE_PLACEMENT_EXPLICIT,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K 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)