Searched refs:base_allocator (Results 1 – 2 of 2) sorted by relevance
780 MemoryMapper(const ThisT& base_allocator, uptr class_id) in MemoryMapper() argument781 : allocator(base_allocator), in MemoryMapper()782 region_base(base_allocator.GetRegionBeginBySizeClass(class_id)), in MemoryMapper()
766 TfGpuId tf_gpu_id, Allocator* base_allocator, in Reinitialize() argument768 stream_device_.Reinitialize(context, gpu_stream, tf_gpu_id, base_allocator, in Reinitialize()