Searched refs:staging_buffer (Results 1 – 1 of 1) sorted by relevance
696 void* staging_buffer = host_memory_allocator()->AllocateRaw( in BufferFromHostBuffer() local698 buffer = se::DeviceMemoryBase(staging_buffer, size); in BufferFromHostBuffer()699 std::memcpy(staging_buffer, data, size); in BufferFromHostBuffer()703 on_delete_callback = [staging_buffer, host_memory_allocator = in BufferFromHostBuffer()705 host_memory_allocator->DeallocateRaw(staging_buffer); in BufferFromHostBuffer()732 std::shared_ptr<void> staging_buffer; in BufferFromHostBuffer() local737 staging_buffer = std::shared_ptr<void>( in BufferFromHostBuffer()748 std::memcpy(staging_buffer.get(), data, size); in BufferFromHostBuffer()767 staging_buffer{std::move(staging_buffer)}, in BufferFromHostBuffer()782 if (staging_buffer) { in BufferFromHostBuffer()[all …]