Home
last modified time | relevance | path

Searched refs:VisitAlloc (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_host_allocator.h48 VisitAlloc(ptr, numa_node_, num_bytes); in Alloc()
Dgpu_bfc_allocator.h60 VisitAlloc(ptr, gpu_id_.value(), num_bytes); in Alloc()
/external/tensorflow/tensorflow/core/framework/
Dallocator.cc242 void SubAllocator::VisitAlloc(void* ptr, int index, size_t num_bytes) { in VisitAlloc() function in tensorflow::SubAllocator
Dallocator.h432 void VisitAlloc(void* ptr, int index, size_t num_bytes);
/external/tensorflow/tensorflow/core/common_runtime/
Dpool_allocator.cc268 VisitAlloc(ptr, numa_node_, num_bytes); in Alloc()