Searched defs:sub_allocator (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator_test.cc | 50 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local
146 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local
189 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local
232 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local
242 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in TEST() local
|
| D | gpu_bfc_allocator.cc | 76 GPUBFCAllocator::GPUBFCAllocator(SubAllocator* sub_allocator, in GPUBFCAllocator()
82 GPUBFCAllocator::GPUBFCAllocator(SubAllocator* sub_allocator, in GPUBFCAllocator()
|
| D | gpu_process_state.cc | 182 auto* sub_allocator = in GetGPUAllocator() local
337 SubAllocator* sub_allocator = new DeviceHostAllocator( in GetGpuHostAllocator() local
|
| D | gpu_process_state.h | 153 SubAllocator* sub_allocator; // owned by allocator member
|
| D | pool_allocator_test.cc | 149 DeviceHostAllocator* sub_allocator = new DeviceHostAllocator( in TEST() local
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_bfc_allocator.cc | 78 DeviceMemAllocator* sub_allocator, size_t total_memory, const string& name) in PluggableDeviceBFCAllocator()
83 DeviceMemAllocator* sub_allocator, size_t total_memory, in PluggableDeviceBFCAllocator()
|
| D | pluggable_device_process_state.cc | 113 DeviceMemAllocator* sub_allocator = new DeviceMemAllocator( in GetPluggableDeviceAllocator() local
185 SubAllocator* sub_allocator = new DeviceHostAllocator( in GetPluggableDeviceHostAllocator() local
|
| D | pluggable_device_simple_allocator.cc | 22 DeviceMemAllocator* sub_allocator) in PluggableDeviceSimpleAllocator()
|
| D | pluggable_device_process_state.h | 109 SubAllocator* sub_allocator; // owned by allocator member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.cc | 86 SubAllocator* sub_allocator = in GetCPUAllocator() local
|
| D | mkl_cpu_allocator.h | 52 MklSmallSizeAllocator(SubAllocator* sub_allocator, size_t total_memory, in MklSmallSizeAllocator()
|
| D | bfc_allocator.cc | 41 BFCAllocator::BFCAllocator(SubAllocator* sub_allocator, size_t total_memory, in BFCAllocator()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_device.cc | 248 auto sub_allocator = absl::make_unique<tensorflow::DeviceMemAllocator>( in CreateBFCAllocator() local
328 tensorflow::SubAllocator* sub_allocator = new tensorflow::DeviceHostAllocator( in GetGpuHostAllocator() local
|