Searched refs:alloc_visitors (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_host_allocator.h | 30 const std::vector<Visitor>& alloc_visitors, in DeviceHostAllocator() argument
32 : SubAllocator(alloc_visitors, free_visitors), in DeviceHostAllocator()
|
| D | device_mem_allocator.h | 34 const std::vector<Visitor>& alloc_visitors, in DeviceMemAllocator() argument
36 : SubAllocator(alloc_visitors, free_visitors), in DeviceMemAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_virtual_mem_allocator.cc | 62 GpuVirtualMemAllocator::Create(const std::vector<Visitor>& alloc_visitors, in Create() argument
107 alloc_visitors, free_visitors, gpu_context, gpu_id, in Create()
112 const std::vector<Visitor>& alloc_visitors, in GpuVirtualMemAllocator() argument
116 : SubAllocator(alloc_visitors, free_visitors), in GpuVirtualMemAllocator()
|
| D | gpu_virtual_mem_allocator.h | 45 Create(const std::vector<Visitor>& alloc_visitors,
75 const std::vector<Visitor>& alloc_visitors,
|
| D | gpu_process_state.cc | 85 const std::vector<SubAllocator::Visitor>& alloc_visitors, in CreateSubAllocator() argument
99 /*use_unified_memory=*/true, alloc_visitors, in CreateSubAllocator()
121 alloc_visitors, {}, *gpu_context, platform_gpu_id, in CreateSubAllocator()
132 alloc_visitors, {}); in CreateSubAllocator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 95 SubAllocator::SubAllocator(const std::vector<Visitor>& alloc_visitors, in SubAllocator() argument
97 : alloc_visitors_(alloc_visitors), free_visitors_(free_visitors) {} in SubAllocator()
|
| D | allocator.h | 438 SubAllocator(const std::vector<Visitor>& alloc_visitors,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | pool_allocator.h | 152 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator() argument
154 : SubAllocator(alloc_visitors, free_visitors), numa_node_(numa_node) {} in BasicCPUAllocator()
|