Searched refs:ScratchBufAllocator (Results 1 – 1 of 1) sorted by relevance
45 class ScratchBufAllocator : public se::ScratchAllocator { class47 explicit ScratchBufAllocator(se::DeviceMemoryBase scratch) in ScratchBufAllocator() function in xla::gpu::__anon53f8f4230111::ScratchBufAllocator50 ~ScratchBufAllocator() override = default;500 ScratchBufAllocator scratch_allocator(scratch_buf); in RunGpuConv()