| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.h | 98 const SubAllocator::Visitor& visitor);
103 const SubAllocator::Visitor& visitor);
108 const SubAllocator::Visitor& visitor);
140 SubAllocator* sub_allocator; // owned by allocator
144 std::vector<std::vector<SubAllocator::Visitor>> gpu_visitors_ GUARDED_BY(mu_);
147 std::vector<std::vector<SubAllocator::Visitor>> gpu_host_alloc_visitors_
149 std::vector<std::vector<SubAllocator::Visitor>> gpu_host_free_visitors_
|
| D | gpu_process_state.cc | 231 SubAllocator* sub_allocator = in GetGpuHostAllocator()
277 const SubAllocator::Visitor& visitor) { in AddGPUAllocVisitor()
285 gpu_visitors_.push_back(std::vector<SubAllocator::Visitor>()); in AddGPUAllocVisitor()
292 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostAllocVisitor()
299 gpu_host_alloc_visitors_.push_back(std::vector<SubAllocator::Visitor>()); in AddGpuHostAllocVisitor()
306 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostFreeVisitor()
313 gpu_host_free_visitors_.push_back(std::vector<SubAllocator::Visitor>()); in AddGpuHostFreeVisitor()
|
| D | gpu_host_allocator.h | 26 class GpuHostAllocator : public SubAllocator {
32 : SubAllocator(alloc_visitors, free_visitors), in GpuHostAllocator()
|
| D | gpu_bfc_allocator.h | 35 class GPUMemAllocator : public SubAllocator {
44 : SubAllocator(alloc_visitors, free_visitors), in GPUMemAllocator()
|
| D | pool_allocator_test.cc | 131 SubAllocator::Visitor alloc_visitor = in TEST()
138 SubAllocator::Visitor free_visitor = in TEST()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 185 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator()
190 class CPUSubAllocator : public SubAllocator {
193 : SubAllocator({}, {}), cpu_allocator_(cpu_allocator) {} in CPUSubAllocator()
238 SubAllocator::SubAllocator(const std::vector<Visitor>& alloc_visitors, in SubAllocator() function in tensorflow::SubAllocator
242 void SubAllocator::VisitAlloc(void* ptr, int index, size_t num_bytes) { in VisitAlloc()
248 void SubAllocator::VisitFree(void* ptr, int index, size_t num_bytes) { in VisitFree()
|
| D | allocator_registry.h | 44 virtual SubAllocator* CreateSubAllocator(int numa_node) = 0;
78 SubAllocator* GetSubAllocator(int numa_node);
101 std::vector<std::unique_ptr<SubAllocator>> sub_allocators;
|
| D | allocator.h | 415 class SubAllocator {
422 SubAllocator(const std::vector<Visitor>& alloc_visitors,
425 virtual ~SubAllocator() {} in ~SubAllocator()
|
| D | allocator_registry.cc | 90 SubAllocator* AllocatorFactoryRegistry::GetSubAllocator(int numa_node) { in GetSubAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.h | 72 void AddCPUAllocVisitor(SubAllocator::Visitor v);
76 void AddCPUFreeVisitor(SubAllocator::Visitor v);
101 std::vector<SubAllocator::Visitor> cpu_alloc_visitors_ GUARDED_BY(mu_);
102 std::vector<SubAllocator::Visitor> cpu_free_visitors_ GUARDED_BY(mu_);
|
| D | pool_allocator.h | 56 SubAllocator* allocator, RoundUpInterface* size_rounder,
123 std::unique_ptr<SubAllocator> allocator_;
149 class BasicCPUAllocator : public SubAllocator {
153 : SubAllocator(alloc_visitors, free_visitors), numa_node_(numa_node) {} in BasicCPUAllocator()
|
| D | process_state.cc | 78 SubAllocator* sub_allocator = in GetCPUAllocator()
125 void ProcessState::AddCPUAllocVisitor(SubAllocator::Visitor visitor) { in AddCPUAllocVisitor()
134 void ProcessState::AddCPUFreeVisitor(SubAllocator::Visitor visitor) { in AddCPUFreeVisitor()
|
| D | mkl_cpu_allocator.h | 55 MklSmallSizeAllocator(SubAllocator* sub_allocator, size_t total_memory, in MklSmallSizeAllocator()
117 SubAllocator* sub_allocator_; // Not owned by this class.
313 SubAllocator* sub_allocator_; // not owned by this class
|
| D | threadpool_device.cc | 114 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator()
|
| D | bfc_allocator.h | 49 BFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
438 std::unique_ptr<SubAllocator> sub_allocator_;
|
| D | pool_allocator.cc | 37 SubAllocator* allocator, in PoolAllocator()
|
| D | bfc_allocator.cc | 33 BFCAllocator::BFCAllocator(SubAllocator* sub_allocator, size_t total_memory, in BFCAllocator()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_mgr.cc | 266 SubAllocator::Visitor alloc_visitor = [](void* ptr, int numa_node, in RegMemVisitors()
271 SubAllocator::Visitor free_visitor = [](void* ptr, int numa_node, in RegMemVisitors()
293 SubAllocator::Visitor cuda_alloc_visitor = [](void* ptr, int gpu_id, in RegMemVisitors()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.cc | 236 SubAllocator::Visitor alloc_visitor = [this](void* ptr, int numa_node, in Init()
242 SubAllocator::Visitor free_visitor = [this](void* ptr, int numa_node, in Init()
259 SubAllocator::Visitor cuda_alloc_visitor = [this](void* ptr, int gpu_id, in Init()
|