Searched refs:owned_buffer (Results 1 – 3 of 3) sorted by relevance
143 : buffer(b), weak_type(weak_type), owned_buffer(nullptr) {} in DevicePutResult()147 owned_buffer(std::move(new_buffer)) {} in DevicePutResult()151 std::unique_ptr<xla::PjRtBuffer> owned_buffer; member
1014 if (on_device.owned_buffer) { in ConvertArgsToBuffers()1015 keep_alive.emplace_back(std::move(on_device.owned_buffer)); in ConvertArgsToBuffers()1243 if (results->owned_buffer) { in BuildJaxjitSubmodule()1245 pyclient, std::move(results->owned_buffer), xla::Traceback::Get()); in BuildJaxjitSubmodule()
312 if (absl::optional<se::OwningDeviceMemory> owned_buffer = in CreateResultShapedBuffer() local314 result_buffer = owned_buffer->Release(); in CreateResultShapedBuffer()