| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.h | 105 const SubAllocator::Visitor& visitor);
110 const SubAllocator::Visitor& visitor);
115 const SubAllocator::Visitor& visitor);
148 SubAllocator* sub_allocator; // owned by allocator
152 std::vector<std::vector<SubAllocator::Visitor>> gpu_visitors_
156 std::vector<std::vector<SubAllocator::Visitor>> gpu_host_alloc_visitors_
158 std::vector<std::vector<SubAllocator::Visitor>> gpu_host_free_visitors_
|
| D | gpu_process_state.cc | 83 static SubAllocator* CreateSubAllocator( in CreateSubAllocator()
85 const std::vector<SubAllocator::Visitor>& alloc_visitors, in CreateSubAllocator()
299 SubAllocator* sub_allocator = new DeviceHostAllocator( in GetGpuHostAllocator()
346 const SubAllocator::Visitor& visitor) { in AddGPUAllocVisitor()
355 gpu_visitors_.push_back(std::vector<SubAllocator::Visitor>()); in AddGPUAllocVisitor()
362 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostAllocVisitor()
370 gpu_host_alloc_visitors_.push_back(std::vector<SubAllocator::Visitor>()); in AddGpuHostAllocVisitor()
377 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostFreeVisitor()
385 gpu_host_free_visitors_.push_back(std::vector<SubAllocator::Visitor>()); in AddGpuHostFreeVisitor()
|
| D | gpu_bfc_allocator.h | 37 GPUBFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
39 GPUBFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
|
| D | gpu_bfc_allocator.cc | 76 GPUBFCAllocator::GPUBFCAllocator(SubAllocator* sub_allocator, in GPUBFCAllocator()
80 GPUBFCAllocator::GPUBFCAllocator(SubAllocator* sub_allocator, in GPUBFCAllocator()
|
| D | gpu_bfc_allocator_test.cc | 60 : public ::testing::TestWithParam<SubAllocator* (*)(size_t)> {};
63 SubAllocator* CreateVirtualMemorySubAllocator( in CreateVirtualMemorySubAllocator()
78 SubAllocator* CreateGPUMemAllocator(size_t) { in CreateGPUMemAllocator()
87 SubAllocator* CreateSubAllocator(size_t virtual_address_space_size = 1ull in CreateSubAllocator()
431 : public ::testing::TestWithParam<SubAllocator* (*)(size_t)> {
|
| D | gpu_virtual_mem_allocator.h | 41 class GpuVirtualMemAllocator : public SubAllocator {
|
| D | pool_allocator_test.cc | 134 SubAllocator::Visitor alloc_visitor = in TEST()
141 SubAllocator::Visitor free_visitor = in TEST()
|
| D | gpu_virtual_mem_allocator.cc | 116 : SubAllocator(alloc_visitors, free_visitors), in GpuVirtualMemAllocator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 95 SubAllocator::SubAllocator(const std::vector<Visitor>& alloc_visitors, in SubAllocator() function in tensorflow::SubAllocator
99 void SubAllocator::VisitAlloc(void* ptr, int index, size_t num_bytes) { in VisitAlloc()
105 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;
79 SubAllocator* GetSubAllocator(int numa_node);
102 std::vector<std::unique_ptr<SubAllocator>> sub_allocators;
|
| D | cpu_allocator_impl.cc | 149 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator()
154 class CPUSubAllocator : public SubAllocator {
157 : SubAllocator({}, {}), cpu_allocator_(cpu_allocator) {} in CPUSubAllocator()
|
| D | allocator.h | 431 class SubAllocator {
438 SubAllocator(const std::vector<Visitor>& alloc_visitors,
441 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);
102 std::vector<SubAllocator::Visitor> cpu_alloc_visitors_ TF_GUARDED_BY(mu_);
103 std::vector<SubAllocator::Visitor> cpu_free_visitors_ TF_GUARDED_BY(mu_);
|
| D | pool_allocator.h | 57 SubAllocator* allocator, RoundUpInterface* size_rounder,
124 std::unique_ptr<SubAllocator> allocator_;
150 class BasicCPUAllocator : public SubAllocator {
154 : SubAllocator(alloc_visitors, free_visitors), numa_node_(numa_node) {} in BasicCPUAllocator()
|
| D | process_state.cc | 86 SubAllocator* sub_allocator = in GetCPUAllocator()
137 void ProcessState::AddCPUAllocVisitor(SubAllocator::Visitor visitor) { in AddCPUAllocVisitor()
146 void ProcessState::AddCPUFreeVisitor(SubAllocator::Visitor visitor) { in AddCPUFreeVisitor()
|
| D | bfc_allocator_test.cc | 27 class FakeSubAllocator : public SubAllocator {
29 FakeSubAllocator() : SubAllocator({}, {}), alloc_counter_(0) {} in FakeSubAllocator()
|
| 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.
315 SubAllocator* sub_allocator_; // not owned by this class
|
| D | threadpool_device.cc | 124 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator()
|
| D | bfc_allocator.h | 52 BFCAllocator(SubAllocator* sub_allocator, size_t total_memory,
584 std::unique_ptr<SubAllocator> sub_allocator_;
|
| D | pool_allocator.cc | 37 SubAllocator* allocator, in PoolAllocator()
|
| D | bfc_allocator.cc | 40 BFCAllocator::BFCAllocator(SubAllocator* sub_allocator, size_t total_memory, in BFCAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_host_allocator.h | 26 class DeviceHostAllocator : public SubAllocator {
32 : SubAllocator(alloc_visitors, free_visitors), in DeviceHostAllocator()
|
| D | device_mem_allocator.h | 26 class DeviceMemAllocator : public SubAllocator {
36 : SubAllocator(alloc_visitors, free_visitors), in DeviceMemAllocator()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_device.cc | 111 /*alloc_visitors=*/std::vector<tensorflow::SubAllocator::Visitor>(), in CreateBFCAllocator()
112 /*free_visitors=*/std::vector<tensorflow::SubAllocator::Visitor>()); in CreateBFCAllocator()
163 tensorflow::SubAllocator* sub_allocator = new tensorflow::DeviceHostAllocator( in GetGpuHostAllocator()
|