Home
last modified time | relevance | path

Searched defs:CreateSubAllocator (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcpu_allocator_impl.cc186 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator() function in tensorflow::__anon2523aecd0111::CPUAllocatorFactory
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dthreadpool_device.cc262 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator() function in tensorflow::__anon8e2c0c930311::MklCPUAllocatorFactory
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_process_state.cc98 static std::unique_ptr<SubAllocator> CreateSubAllocator( in CreateSubAllocator() function
H A Dgpu_bfc_allocator_test.cc86 std::unique_ptr<SubAllocator> CreateSubAllocator( in CreateSubAllocator() function