- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TF_GetPjRtCBuffer (0.18 sec)
-
tensorflow/c/experimental/next_pluggable_device/c_api.h
// Gets the `PJRT_Buffer*` stored in the tensor. The status will contain error // if the tensor does not have a `PjRtCApiBuffer`. TF_CAPI_EXPORT extern PJRT_Buffer* TF_GetPjRtCBuffer(TF_Tensor* c_tensor, TF_Status* status); // Creates a `PjRtCApiBuffer` with the `PJRT_Buffer*` passed in and set to the // tensor.
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Dec 20 20:01:06 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
status->status = pjrt_c_api_client.status(); return nullptr; } status->status = absl::OkStatus(); return (*pjrt_c_api_client)->pjrt_c_client(); } PJRT_Buffer* TF_GetPjRtCBuffer(TF_Tensor* c_tensor, TF_Status* status) { tensorflow::Tensor tensor; auto s = tensorflow::TF_TensorToTensor(c_tensor, &tensor); if (!s.ok()) { status->status = s; return nullptr; }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1)