Searched defs:sub_allocator (Results 1 – 12 of 12) sorted by relevance
90 DeviceMemAllocator* sub_allocator, size_t total_memory, const string& name, in PluggableDeviceBFCAllocator()96 DeviceMemAllocator* sub_allocator, size_t total_memory, in PluggableDeviceBFCAllocator()
113 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in GetPluggableDeviceAllocator() local186 SubAllocator* sub_allocator = new DeviceHostAllocator( in GetPluggableDeviceHostAllocator() local
22 DeviceMemAllocator* sub_allocator) in PluggableDeviceSimpleAllocator()
109 SubAllocator* sub_allocator; // owned by allocator member
182 std::unique_ptr<SubAllocator> sub_allocator = in GetGPUAllocator() local344 SubAllocator* sub_allocator = new DeviceHostAllocator( in GetGpuHostAllocator() local
80 GPUBFCAllocator::GPUBFCAllocator(std::unique_ptr<SubAllocator> sub_allocator, in GPUBFCAllocator()
153 SubAllocator* sub_allocator; // owned by allocator member
149 DeviceHostAllocator* sub_allocator = new DeviceHostAllocator( in TEST() local
224 auto sub_allocator = std::make_unique<tensorflow::DeviceMemAllocator>( in CreateBFCAllocator() local308 std::unique_ptr<tensorflow::SubAllocator> sub_allocator( in GetGpuHostAllocator() local
86 SubAllocator* sub_allocator = in GetCPUAllocator() local
53 MklSmallSizeAllocator(SubAllocator* sub_allocator, size_t total_memory, in MklSmallSizeAllocator()
43 BFCAllocator::BFCAllocator(std::unique_ptr<SubAllocator> sub_allocator, in BFCAllocator()