Searched refs:VisitAlloc (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_host_allocator.h | 48 VisitAlloc(ptr, numa_node_, num_bytes); in Alloc()
|
| D | gpu_bfc_allocator.h | 60 VisitAlloc(ptr, gpu_id_.value(), num_bytes); in Alloc()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 242 void SubAllocator::VisitAlloc(void* ptr, int index, size_t num_bytes) { in VisitAlloc() function in tensorflow::SubAllocator
|
| D | allocator.h | 432 void VisitAlloc(void* ptr, int index, size_t num_bytes);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | pool_allocator.cc | 268 VisitAlloc(ptr, numa_node_, num_bytes); in Alloc()
|