- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for src_buffer (0.11 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
TF_ASSERT_OK_AND_ASSIGN(auto c_buffer, GetPjRtCBufferFromTensor(&tensor)); EXPECT_THAT(c_buffer, NotNull()); } TEST(TensorPjRtBufferUtilTest, SetPjRtCBufferToTensorNotAsyncValueTensor) { tensorflow::Tensor tensor(DT_FLOAT, {1}); TF_ASSERT_OK_AND_ASSIGN(auto pjrt_client, xla::GetCApiClient(DEVICE_CPU)); PJRT_Buffer* c_buffer = CreateCBuffer(); TF_EXPECT_OK(SetPjRtCBufferToTensor(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 01 16:29:40 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
} return c_api_buffer->c_buffer(); } absl::Status SetPjRtCBufferToTensor(PJRT_Buffer* c_buffer, xla::PjRtCApiClient* c_api_client, Tensor* tensor) { auto buffer = std::make_unique<xla::PjRtCApiBuffer>(c_api_client, c_buffer); tensorflow::AsyncValueTensor* av_tensor =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
#include "tensorflow/core/framework/tensor.h" namespace tensorflow { absl::StatusOr<PJRT_Buffer*> GetPjRtCBufferFromTensor(const Tensor* tensor); absl::Status SetPjRtCBufferToTensor(PJRT_Buffer* c_buffer, xla::PjRtCApiClient* c_api_client, Tensor* tensor); absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 1.5K bytes - Viewed (0)