- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for PjRtBuffer (0.19 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
if (av_tensor == nullptr || av_tensor->GetBuffer() == nullptr) { return absl::InternalError("Input tensor does not have PjRtBuffer."); } auto* c_api_buffer = dynamic_cast<xla::PjRtCApiBuffer*>(av_tensor->GetBuffer().get()); if (c_api_buffer == nullptr) { return absl::InternalError( "The PjRtBuffer in the tensor is not type PjRtCApiBuffer."); } return c_api_buffer->c_buffer(); }
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/tensor_pjrt_buffer_util_test.cc
EXPECT_THAT( GetPjRtCBufferFromTensor(&tensor), StatusIs(error::INTERNAL, HasSubstr(absl::StrCat( "Input tensor does not have PjRtBuffer")))); } TEST(TensorPjRtBufferUtilTest, GetPjRtCBufferFromTensorIncoorectType) { auto allocator = std::make_unique<AsyncValueAllocator>(); tensorflow::Tensor tensor(allocator.get(), DT_FLOAT, {1});
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)