Searched refs:CreateSubAllocator (Results 1 – 6 of 6) sorted by relevance
124 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator() function in tensorflow::__anon59fd7d240111::MklCPUAllocatorFactory
44 virtual SubAllocator* CreateSubAllocator(int numa_node) = 0;
149 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator() function in tensorflow::__anon29597ede0111::CPUAllocatorFactory
121 best_entry->factory->CreateSubAllocator(numa_node)); in GetSubAllocator()
87 SubAllocator* CreateSubAllocator(size_t virtual_address_space_size = 1ull in CreateSubAllocator() function337 GPUBFCAllocator a(CreateSubAllocator(1ul << 36), 1uLL << 33, "GPU_0_bfc"); in BM_Allocation()358 GPUBFCAllocator a(CreateSubAllocator(1ul << 36), 1uLL << 33, "GPU_0_bfc"); in BM_AllocationThreaded()399 GPUBFCAllocator a(CreateSubAllocator(1ull << 32), 1 << 30, "GPU_0_bfc"); in BM_AllocationDelayed()
83 static SubAllocator* CreateSubAllocator( in CreateSubAllocator() function167 CreateSubAllocator(options, platform_gpu_id, gpu_visitors_[bus_id], in GetGPUAllocator()