Home
last modified time | relevance | path

Searched defs:DefaultCPUAllocator (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DCPUAllocator.cpp18 struct C10_API DefaultCPUAllocator final : at::Allocator { struct
20 at::DataPtr allocate(size_t nbytes) override { in allocate()
32 static void ReportAndDelete(void* ptr) { in ReportAndDelete()
40 at::DeleterFnPtr raw_deleter() const override { in raw_deleter()
44 void copy_data(void* dest, const void* src, std::size_t count) const final { in copy_data()
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A DreportMemoryUsage_test.cpp7 TEST(DefaultCPUAllocator, check_reporter) { in TEST() argument