Searched defs:CreateSubAllocator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ | ||
| D | cpu_allocator_impl.cc | 186 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator() function in tensorflow::__anon99240fbc0111::CPUAllocatorFactory |
| /external/tensorflow/tensorflow/core/common_runtime/ | ||
| D | threadpool_device.cc | 262 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator() function in tensorflow::__anondc414d020311::MklCPUAllocatorFactory |
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ | ||
| D | gpu_process_state.cc | 98 static std::unique_ptr<SubAllocator> CreateSubAllocator( in CreateSubAllocator() function |
| D | gpu_bfc_allocator_test.cc | 86 std::unique_ptr<SubAllocator> CreateSubAllocator( in CreateSubAllocator() function |