Searched refs:GPUMemAllocator (Results 1 – 5 of 5) sorted by relevance
54 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()87 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()149 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()188 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()198 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()207 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()219 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()239 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()305 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()309 sub_allocator = new GPUMemAllocator( in TEST()[all …]
38 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()66 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()102 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()135 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()180 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()226 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()236 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in TEST()
35 class GPUMemAllocator : public SubAllocator {40 explicit GPUMemAllocator(se::StreamExecutor* stream_exec, in GPUMemAllocator() function50 ~GPUMemAllocator() override {} in ~GPUMemAllocator()82 TF_DISALLOW_COPY_AND_ASSIGN(GPUMemAllocator);89 GPUBFCAllocator(GPUMemAllocator* sub_allocator, size_t total_memory,91 GPUBFCAllocator(GPUMemAllocator* sub_allocator, size_t total_memory,
58 GPUBFCAllocator::GPUBFCAllocator(GPUMemAllocator* sub_allocator, in GPUBFCAllocator()62 GPUBFCAllocator::GPUBFCAllocator(GPUMemAllocator* sub_allocator, in GPUBFCAllocator()
108 GPUMemAllocator* sub_allocator = new GPUMemAllocator( in GetGPUAllocator()