Searched refs:CreateSubAllocator (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator_registry.h | 44 virtual SubAllocator* CreateSubAllocator(int numa_node) = 0;
|
| D | allocator_registry.cc | 121 best_entry->factory->CreateSubAllocator(numa_node)); in GetSubAllocator()
|
| D | cpu_allocator_impl.cc | 186 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator() function in tensorflow::__anonb46ab94d0111::CPUAllocatorFactory
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_bfc_allocator_test.cc | 86 std::unique_ptr<SubAllocator> CreateSubAllocator( in CreateSubAllocator() function
336 GPUBFCAllocator a(CreateSubAllocator(1ul << 36), 1uLL << 33, "GPU_0_bfc", {}); in BM_Allocation()
357 GPUBFCAllocator a(CreateSubAllocator(1ul << 36), 1uLL << 33, "GPU_0_bfc", in BM_AllocationThreaded()
399 GPUBFCAllocator a(CreateSubAllocator(1ull << 32), 1 << 30, "GPU_0_bfc", {}); in BM_AllocationDelayed()
|
| D | gpu_process_state.cc | 98 static std::unique_ptr<SubAllocator> CreateSubAllocator( in CreateSubAllocator() function
183 CreateSubAllocator(options, platform_device_id, gpu_visitors_[bus_id], in GetGPUAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | threadpool_device.cc | 262 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator() function in tensorflow::__anon26f457130311::MklCPUAllocatorFactory
|