Searched refs:alloc_visitors (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_host_allocator.h | 30 const std::vector<Visitor>& alloc_visitors, in GpuHostAllocator() argument
32 : SubAllocator(alloc_visitors, free_visitors), in GpuHostAllocator()
|
| D | gpu_bfc_allocator.h | 42 const std::vector<Visitor>& alloc_visitors, in GPUMemAllocator() argument
44 : SubAllocator(alloc_visitors, free_visitors), in GPUMemAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | pool_allocator.h | 151 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator() argument
153 : SubAllocator(alloc_visitors, free_visitors), numa_node_(numa_node) {} in BasicCPUAllocator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 238 SubAllocator::SubAllocator(const std::vector<Visitor>& alloc_visitors, in SubAllocator() argument
240 : alloc_visitors_(alloc_visitors), free_visitors_(free_visitors) {} in SubAllocator()
|
| D | allocator.h | 422 SubAllocator(const std::vector<Visitor>& alloc_visitors,
|