- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for TFE_TensorHandleCopyToDevice (0.15 sec)
-
tensorflow/c/eager/c_api_cluster_test.cc
TFE_TensorHandle* retvals[1]; int num_retvals = 1; TFE_Execute(matmul, &retvals[0], &num_retvals, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); auto* retval_task0 = TFE_TensorHandleCopyToDevice(retvals[0], ctx, local_device_name, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); CheckTFE_TensorHandleHasFloats(retval_task0, {7, 10, 15, 22}); TFE_DeleteTensorHandle(retval_task0);
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/c_api_experimental_test.cc
for (int d = 0; d < num_devices; ++d) { const char* name = TF_DeviceListName(devices, d, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_TensorHandle* copy = TFE_TensorHandleCopyToDevice(m, ctx, name, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); void* data = TFE_TensorHandleDevicePointer(copy, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 03 03:14:26 UTC 2023 - 31.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
// Copy to device TFE_TensorHandle* hdevice = TFE_TensorHandleCopyToDevice(hcpu, ctx, name.c_str(), status.get()); if (TF_GetCode(status.get()) != TF_OK) { ADD_FAILURE() << tag << " -- " << TF_Message(status.get()); continue; } // Copy from device to the same device. TFE_TensorHandle* hdevice2 = TFE_TensorHandleCopyToDevice(hdevice, ctx, name.c_str(), status.get());
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_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);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 23.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); // Copying on to a parallel device must be explicit. TensorHandlePtr failed_copy_on_result(TFE_TensorHandleCopyToDevice( 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()};
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/c_api.h
// If async execution is enabled, the copy may be enqueued and the call will // return "non-ready" handle. Else, this function returns after the copy has // been done. TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopyToDevice( TFE_TensorHandle* h, TFE_Context* ctx, const char* device_name, TF_Status* status); // Debugging/Profiling information for TFE_TensorHandle //
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
std::vector<TensorHandlePtr> components; components.reserve(underlying_devices_.size()); for (const std::string& underlying_device_name : underlying_devices_) { TFE_TensorHandle* t = TFE_TensorHandleCopyToDevice( tensor, context, underlying_device_name.c_str(), status); if (TF_GetCode(status) != TF_OK) return nullptr; components.emplace_back(t); }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
unwrapped_op, reinterpret_cast<tensorflow::ImmediateExecutionTensorHandle**>( retvals), num_retvals); } TFE_TensorHandle* TFE_TensorHandleCopyToDevice(TFE_TensorHandle* h, TFE_Context* ctx, const char* device_name,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)