Searched defs:sub_allocator (Results 1 – 10 of 10) sorted by relevance
50 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local146 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local189 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local232 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local242 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local
76 GPUBFCAllocator::GPUBFCAllocator(SubAllocator* sub_allocator, in GPUBFCAllocator()80 GPUBFCAllocator::GPUBFCAllocator(SubAllocator* sub_allocator, in GPUBFCAllocator()
166 auto* sub_allocator = in GetGPUAllocator() local299 SubAllocator* sub_allocator = new DeviceHostAllocator( in GetGpuHostAllocator() local
148 SubAllocator* sub_allocator; // owned by allocator member
149 DeviceHostAllocator* sub_allocator = new DeviceHostAllocator( in TEST() local
86 SubAllocator* sub_allocator = in GetCPUAllocator() local
53 FakeSubAllocator* sub_allocator = new FakeSubAllocator; in BM_Allocator() local
55 MklSmallSizeAllocator(SubAllocator* sub_allocator, size_t total_memory, in MklSmallSizeAllocator()
40 BFCAllocator::BFCAllocator(SubAllocator* sub_allocator, size_t total_memory, in BFCAllocator()
108 auto sub_allocator = absl::make_unique<tensorflow::DeviceMemAllocator>( in CreateBFCAllocator() local163 tensorflow::SubAllocator* sub_allocator = new tensorflow::DeviceHostAllocator( in GetGpuHostAllocator() local