- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for pjrt_c_client (0.18 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1}); auto buffer = c_api_client->pjrt_c_client()->client->BufferFromHostBuffer( data.data(), shape.element_type(), shape.dimensions(), /*byte_strides=*/std::nullopt, xla::PjRtClient::HostBufferSemantics::kImmutableOnlyDuringCall, nullptr, c_api_client->pjrt_c_client()->client->addressable_devices()[0]); CHECK_OK(buffer.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) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
if (!pjrt_c_api_client.ok()) { 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()) {
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)