| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | process_state.cc | 86 SubAllocator* sub_allocator = in GetCPUAllocator() local
102 DCHECK(sub_allocator); in GetCPUAllocator()
107 absl::WrapUnique(sub_allocator), cpu_mem_limit, in GetCPUAllocator()
112 } else if (sub_allocator) { in GetCPUAllocator()
113 DCHECK(sub_allocator); in GetCPUAllocator()
116 sub_allocator, new NoopRounder, "cpu_pool"); in GetCPUAllocator()
121 DCHECK(!sub_allocator); in GetCPUAllocator()
134 if (!sub_allocator) { in GetCPUAllocator()
|
| H A D | mkl_cpu_allocator.h | 53 MklSmallSizeAllocator(SubAllocator* sub_allocator, size_t total_memory, in MklSmallSizeAllocator() argument
55 : sub_allocator_(sub_allocator), name_(name) { in MklSmallSizeAllocator()
|
| H A D | bfc_allocator.cc | 43 BFCAllocator::BFCAllocator(std::unique_ptr<SubAllocator> sub_allocator, in BFCAllocator() argument
47 coalesce_regions_(sub_allocator->SupportsCoalescing()), in BFCAllocator()
48 sub_allocator_(std::move(sub_allocator)), in BFCAllocator()
|
| H A D | bfc_allocator.h | 70 BFCAllocator(std::unique_ptr<SubAllocator> sub_allocator, size_t total_memory,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_process_state.cc | 113 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in GetPluggableDeviceAllocator() local
127 sub_allocator, total_bytes, options, in GetPluggableDeviceAllocator()
131 device_allocator = new PluggableDeviceSimpleAllocator(sub_allocator); in GetPluggableDeviceAllocator()
135 device_allocator, sub_allocator}; in GetPluggableDeviceAllocator()
186 SubAllocator* sub_allocator = new DeviceHostAllocator( in GetPluggableDeviceHostAllocator() local
203 absl::WrapUnique(sub_allocator), pluggable_device_host_mem_limit, in GetPluggableDeviceHostAllocator()
213 sub_allocator}); in GetPluggableDeviceHostAllocator()
|
| H A D | pluggable_device_bfc_allocator.cc | 90 DeviceMemAllocator* sub_allocator, size_t total_memory, const string& name, in PluggableDeviceBFCAllocator() argument
92 : PluggableDeviceBFCAllocator(sub_allocator, total_memory, GPUOptions(), in PluggableDeviceBFCAllocator()
96 DeviceMemAllocator* sub_allocator, size_t total_memory, in PluggableDeviceBFCAllocator() argument
99 : BFCAllocator(absl::WrapUnique(sub_allocator), total_memory, name, [&] { in PluggableDeviceBFCAllocator()
|
| H A D | pluggable_device_bfc_allocator.h | 36 PluggableDeviceBFCAllocator(DeviceMemAllocator* sub_allocator,
39 PluggableDeviceBFCAllocator(DeviceMemAllocator* sub_allocator,
|
| H A D | pluggable_device_simple_allocator.cc | 22 DeviceMemAllocator* sub_allocator) in PluggableDeviceSimpleAllocator() argument
23 : sub_allocator_(sub_allocator) {} in PluggableDeviceSimpleAllocator()
|
| H A D | pluggable_device_simple_allocator.h | 32 explicit PluggableDeviceSimpleAllocator(DeviceMemAllocator* sub_allocator);
|
| H A D | pluggable_device_process_state.h | 109 SubAllocator* sub_allocator; // owned by allocator member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_process_state.cc | 182 std::unique_ptr<SubAllocator> sub_allocator = in GetGPUAllocator() local
185 SubAllocator* sub_allocator_ptr = sub_allocator.get(); in GetGPUAllocator()
188 std::move(sub_allocator), total_bytes, in GetGPUAllocator()
344 SubAllocator* sub_allocator = new DeviceHostAllocator( in GetGpuHostAllocator() local
360 new BFCAllocator(absl::WrapUnique(sub_allocator), gpu_host_mem_limit, in GetGpuHostAllocator()
370 nullptr, sub_allocator, in GetGpuHostAllocator()
|
| H A D | gpu_bfc_allocator.cc | 80 GPUBFCAllocator::GPUBFCAllocator(std::unique_ptr<SubAllocator> sub_allocator, in GPUBFCAllocator() argument
83 : BFCAllocator(std::move(sub_allocator), total_memory, name, [&] { in GPUBFCAllocator()
|
| H A D | pool_allocator_test.cc | 149 DeviceHostAllocator* sub_allocator = new DeviceHostAllocator( in TEST() local
154 sub_allocator, new NoopRounder, "pool"); in TEST()
|
| H A D | gpu_bfc_allocator.h | 55 GPUBFCAllocator(std::unique_ptr<SubAllocator> sub_allocator,
|
| H A D | gpu_process_state.h | 153 SubAllocator* sub_allocator; // owned by allocator member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | gpu_device.cc | 224 auto sub_allocator = std::make_unique<tensorflow::DeviceMemAllocator>( in CreateBFCAllocator() local
256 std::move(sub_allocator), allocator_memory, in CreateBFCAllocator()
308 std::unique_ptr<tensorflow::SubAllocator> sub_allocator( in GetGpuHostAllocator() local
318 std::move(sub_allocator), kGpuHostMemoryLimitBytes, in GetGpuHostAllocator()
|