- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TFE_TensorHandleDeviceName (0.49 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.cc
for (int i = 0; i < num_inputs; ++i) { TFE_TensorHandle* input = TFE_OpGetFlatInput(original_op, i, status); if (TF_GetCode(status) != TF_OK) return; const char* tensor_handle_device = TFE_TensorHandleDeviceName(input, status); if (TF_GetCode(status) != TF_OK) return; if (named_device->name() == tensor_handle_device) { // We assume that any tensors already placed on this device are // ParallelTensors.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 29 22:05:31 UTC 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
EXPECT_EQ(TFE_TensorHandleNumElements(packed_handle, status), 1); 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);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api.h
// 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); // Returns the name of the device in whose memory `h` resides. // // This function will block till the operation that produces `h` has completed.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
ASSERT_EQ(t, nullptr); ASSERT_EQ("Invalid handle", string(TF_Message(status.get()))); TF_SetStatus(status.get(), TF_OK, ""); const char* device_name = TFE_TensorHandleDeviceName(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: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0)