- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for TFE_TensorHandleBackingDeviceName (0.31 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_test.cc
// Verify that the mirrors are placed on the component devices. std::string first_device = TFE_TensorHandleBackingDeviceName( second_components[0].get(), status.get()); ASSERT_EQ(second_underlying_devices[0], first_device); std::string second_device = TFE_TensorHandleBackingDeviceName( second_components[1].get(), status.get()); ASSERT_EQ(second_underlying_devices[1], second_device);
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
// a copy, returns the destination device of the copy. Note that the returned // device name is not always the device holding the tensor handle's memory. If // you want the latter, use TFE_TensorHandleBackingDeviceName. This function // will block till the operation that produces `h` has completed. TF_CAPI_EXPORT extern const char* TFE_TensorHandleDeviceName( TFE_TensorHandle* h, TF_Status* status);
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/c_api_remote_test_util.cc
ASSERT_TRUE(remote_arg->HasLocalMirror(nullptr)); } if (remote_func_outputs) { const string backing_device = TFE_TensorHandleBackingDeviceName(retvals[0], status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); EXPECT_EQ(backing_device, task2_name); } auto* retval_task0 = TFE_TensorHandleCopyToDevice(
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Fri Dec 11 22:56:03 UTC 2020 - 9.1K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_test.cc
auto value_cleaner = tensorflow::gtl::MakeCleanup( [var_value]() { TFE_DeleteTensorHandle(var_value); }); ASSERT_EQ(tensorflow::string(name), tensorflow::string( TFE_TensorHandleBackingDeviceName(var_value, status.get()))); TFE_TensorHandle* var_value_unpacked = UnpackTensorHandle(var_value, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 27 23:39:24 UTC 2020 - 18.4K bytes - Viewed (0) -
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); // Register and run a function which returns the sum of 3 variables.
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/c_api_test.cc
ASSERT_EQ(device_name, nullptr); ASSERT_EQ("Invalid handle", string(TF_Message(status.get()))); TF_SetStatus(status.get(), TF_OK, ""); device_name = TFE_TensorHandleBackingDeviceName(h, status.get()); ASSERT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(status.get())); ASSERT_EQ(device_name, nullptr); ASSERT_EQ("Invalid handle", string(TF_Message(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.cc
if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return nullptr; } return tensorflow::unwrap(h)->DeviceName(&status->status); } const char* TFE_TensorHandleBackingDeviceName(TFE_TensorHandle* h, TF_Status* status) { if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return nullptr;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)