- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for GetPjRtCApiClient (0.25 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
absl::Status SetPjRtCBufferToTensor(PJRT_Buffer* c_buffer, xla::PjRtCApiClient* c_api_client, Tensor* tensor); absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type); absl::Status ResetPjRtClient(const DeviceType& device_type); } // namespace tensorflow
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 09 19:12:59 GMT 2023 - 1.5K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
TF_EXPECT_OK(SetPjRtCBufferToTensor( c_buffer, down_cast<xla::PjRtCApiClient*>(pjrt_client.get()), &tensor)); } TEST(TensorPjRtBufferUtilTest, GetPjRtCApiClientNotFound) { EXPECT_THAT( GetPjRtCApiClient(tensorflow::DeviceType(DEVICE_CPU)), StatusIs(error::NOT_FOUND, HasSubstr(absl::StrCat("PjRt client not found for device type ", DEVICE_CPU)))); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
std::move(buffer))); } else { av_tensor->SetBuffer(std::move(buffer)); } return absl::OkStatus(); } absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type) { TF_ASSIGN_OR_RETURN(tsl::StatusOr<xla::PjRtClient*> pjrt_client, tensorflow::GetPjRtClient(device_type));
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 3.7K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
} PJRT_Client* TF_GetPjRtCClient(const char* device_type, TF_Status* status) { absl::StatusOr<xla::PjRtCApiClient*> pjrt_c_api_client = tensorflow::GetPjRtCApiClient(tensorflow::DeviceType(device_type)); if (!pjrt_c_api_client.ok()) { status->status = pjrt_c_api_client.status(); return nullptr; } status->status = absl::OkStatus();
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)