- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for CreateCBuffer (0.24 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
#include "tsl/protobuf/error_codes.pb.h" namespace tensorflow { namespace { using ::testing::HasSubstr; using ::testing::NotNull; using ::tsl::testing::StatusIs; PJRT_Buffer* CreateCBuffer() { auto status = pjrt::PjrtApi(DEVICE_CPU); if (!status.ok()) { CHECK_OK(pjrt::SetPjrtApi(DEVICE_CPU, GetPjrtApi())); } auto pjrt_client = xla::GetCApiClient(DEVICE_CPU); CHECK_OK(pjrt_client.status());
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)