Searched refs:pjrt_buffer (Results 1 – 2 of 2) sorted by relevance
358 TF_ASSIGN_OR_RETURN(auto pjrt_buffer, in DLPackManagedTensorToBuffer()367 return std::make_unique<PyBuffer>(std::move(client), std::move(pjrt_buffer), in DLPackManagedTensorToBuffer()
1557 auto pjrt_buffer = absl::make_unique<PjRtStreamExecutorBuffer>( in OutputBufferHelper() local1559 RecordUsage(pjrt_buffer->GetBufferWithUsageHold(), local_device, local_device, in OutputBufferHelper()1562 return std::unique_ptr<PjRtBuffer>(std::move(pjrt_buffer)); in OutputBufferHelper()