- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for remote (0.22 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
A. Unique TensorFlower <******@****.***> 1682633218 -0700
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/c_api_cluster_test.cc
// Check that copying it to the old remote device (named localhost) fails. TFE_TensorHandleCopyToDevice(h0_task0_new, ctx, remote_device_name, status); EXPECT_NE(TF_OK, TF_GetCode(status)) << TF_Message(status); // Copying and executing on the new remote device works. const char new_remote_device_name[] = "/job:worker/replica:0/task:1/device:CPU:0";
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
tensorflow::ContextFromInterface(tensorflow::unwrap(ctx)); auto collective_executor_handle = context->GetCollectiveExecutorHandle(); tensorflow::Notification done; collective_executor_handle->get()->remote_access()->CheckPeerHealth( task, timeout_in_ms, [&done, status](const Status& s) { status->status = s; done.Notify(); }); done.WaitForNotification(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/c_api.cc
namespace { #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) // Creates a placeholder representing an input to the cond or body graph. // TODO(skyewm): remove these from final graph bool CreateInput(const TF_Output& parent_input, TF_Graph* g, const char* name, TF_Output* input, TF_Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); // Pack 3 variable handles into one TFE_TensorHandle. // When remote is false, function device is placed on task0. Handle types are // REMOTE, REMOTE, LOCAL on task0. When remote is true, function device is // placed on task1, Handle types are LOCAL, REMOTE, LOCAL on task1. int num_replicas = 3; std::vector<TFE_TensorHandle*> handles = {h0, h1, h2};
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/gradients/math_grad_test.cc
immediate_execution_ctx_.get(), {x.get()}, UseFunction())); } TEST_P(CppGradients, TestMatMulGrad) { // TODO(vnvo2409): Figure out why `gradient_checker` does not work very // well with `MatMul` and remove `TestMatMul*` in // `mnist_gradients_test` when done. GTEST_SKIP(); float A_vals[] = {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f}; int64_t A_dims[] = {3, 3}; AbstractTensorHandlePtr A; {
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Apr 13 17:32:14 GMT 2023 - 16.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
// Remote device on `worker1`. const char remote_device[] = "/job:localhost/replica:0/task:1/device:CPU:0"; // `ctx_0`, `ctx_1`, `ctx_2` 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 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
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/parallel_device/parallel_device.cc
const TFE_OpAttrs* attributes, int expected_max_outputs, TF_Status* status) { absl::optional<std::vector<MaybeParallelTensorOwned>> result; // TODO(allenl): We should remove "TPU" from these op names at the very least, // or consider other ways of packing/unpacking parallel tensors. if (operation_name == std::string("TPUReplicatedInput")) {
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/unified_api_test.cc
"Tensor expected to have scalar shape found rank: ", shape.dims()); } return absl::OkStatus(); } TEST_P(UnifiedAPI, TestTensorShapeScalar) { if (UseFunction() && UseMlir()) { // TODO(b/173074167): Remove this. GTEST_SKIP() << "MlirTensor::Shape is not implemented yet."; } AbstractContextPtr ctx; { AbstractContext* ctx_raw = nullptr; Status s =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 6.7K bytes - Viewed (0)