Searched defs: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::ScratchBufAllocator