Home
last modified time | relevance | path

Searched refs:free_visitors (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_host_allocator.h31 const std::vector<Visitor>& free_visitors) in DeviceHostAllocator() argument
32 : SubAllocator(alloc_visitors, free_visitors), in DeviceHostAllocator()
Ddevice_mem_allocator.h35 const std::vector<Visitor>& free_visitors) in DeviceMemAllocator() argument
36 : SubAllocator(alloc_visitors, free_visitors), in DeviceMemAllocator()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_virtual_mem_allocator.cc63 const std::vector<Visitor>& free_visitors, in Create() argument
107 alloc_visitors, free_visitors, gpu_context, gpu_id, in Create()
113 const std::vector<Visitor>& free_visitors, GpuContext& gpu_context, in GpuVirtualMemAllocator() argument
116 : SubAllocator(alloc_visitors, free_visitors), in GpuVirtualMemAllocator()
Dgpu_virtual_mem_allocator.h46 const std::vector<Visitor>& free_visitors,
76 const std::vector<Visitor>& free_visitors,
/external/tensorflow/tensorflow/core/framework/
Dallocator.cc96 const std::vector<Visitor>& free_visitors) in SubAllocator() argument
97 : alloc_visitors_(alloc_visitors), free_visitors_(free_visitors) {} in SubAllocator()
Dallocator.h439 const std::vector<Visitor>& free_visitors);
/external/tensorflow/tensorflow/core/common_runtime/
Dpool_allocator.h153 const std::vector<Visitor>& free_visitors) in BasicCPUAllocator() argument
154 : SubAllocator(alloc_visitors, free_visitors), numa_node_(numa_node) {} in BasicCPUAllocator()