Home
last modified time | relevance | path

Searched defs:alloc_visitors (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_virtual_mem_allocator.cc63 const std::vector<Visitor>& alloc_visitors, in Create()
112 const std::vector<Visitor>& alloc_visitors, in GpuVirtualMemAllocator()
H A Dgpu_process_state.cc100 const std::vector<SubAllocator::Visitor>& alloc_visitors, in CreateSubAllocator()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/
H A Ddevice_host_allocator.h30 const std::vector<Visitor>& alloc_visitors, in DeviceHostAllocator()
H A Ddevice_mem_allocator.h34 const std::vector<Visitor>& alloc_visitors, in DeviceMemAllocator()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dallocator.cc93 SubAllocator::SubAllocator(const std::vector<Visitor>& alloc_visitors, in SubAllocator()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dpool_allocator.h156 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator()