Searched defs:CachingHostAllocatorInterface (Results 1 – 1 of 1) sorted by relevance
354 struct CachingHostAllocatorInterface : public at::Allocator { struct355 CachingHostAllocatorInterface() : impl_(std::make_unique<T>()) {} in CachingHostAllocatorInterface() function357 at::DataPtr allocate(size_t size) override { in allocate()361 void free(void* ctx) { in free()366 bool record_event(void* ptr, void* ctx, S stream) { in record_event()370 void empty_cache() { in empty_cache()374 void copy_data(void* dest, const void* src, std::size_t count) in copy_data()379 std::unique_ptr<T> impl_;