Searched defs:owned_buffer (Results 1 – 3 of 3) sorted by relevance
47 std::unique_ptr<PjRtBuffer> owned_buffer; member
499 for (auto& owned_buffer : sharded_arg.owned_buffers) { in Call() local
305 if (std::optional<se::OwningDeviceMemory> owned_buffer = in CreateResultShapedBuffer() local