- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TFE_TensorHandleDeviceMemorySize (0.36 sec)
-
tensorflow/c/eager/c_api.cc
// completed. This is only valid on local TFE_TensorHandles. Returns the size in // bytes of the memory pointed to by the device pointer returned above. size_t TFE_TensorHandleDeviceMemorySize(TFE_TensorHandle* h, TF_Status* status) { if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return 0; }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
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); size_t size = TFE_TensorHandleDeviceMemorySize(copy, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); int64_t dims[] = {2, 2}; TFE_TensorHandle* copy_aliased = TFE_NewTensorHandleFromDeviceMemory(
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_experimental.h
// completed. This is only valid on local TFE_TensorHandles. Returns the size in // bytes of the memory pointed to by the device pointer returned above. TF_CAPI_EXPORT extern size_t TFE_TensorHandleDeviceMemorySize(TFE_TensorHandle*, TF_Status*); // Creates a new TensorHandle from memory residing in the physical device
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 39.5K bytes - Viewed (0)