Searched defs:CreateSubAllocator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ | ||
| D | threadpool_device.cc | 124 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator() function in tensorflow::__anon59fd7d240111::MklCPUAllocatorFactory |
| /external/tensorflow/tensorflow/core/framework/ | ||
| D | cpu_allocator_impl.cc | 149 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator() function in tensorflow::__anon29597ede0111::CPUAllocatorFactory |
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ | ||
| D | gpu_process_state.cc | 83 static SubAllocator* CreateSubAllocator( in CreateSubAllocator() function |
| D | gpu_bfc_allocator_test.cc | 87 SubAllocator* CreateSubAllocator(size_t virtual_address_space_size = 1ull in CreateSubAllocator() function |