Searched defs:staging_buffer (Results 1 – 2 of 2) sorted by relevance
340 void* staging_buffer = nullptr; in CopyCPUTensorToGPU() local
752 void* staging_buffer = host_memory_allocator()->AllocateRaw( in BufferFromHostBuffer() local788 std::shared_ptr<void> staging_buffer; in BufferFromHostBuffer() local