Searched refs:VisitAlloc (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_host_allocator.h | 50 VisitAlloc(ptr, numa_node_, num_bytes); in Alloc()
|
| D | device_mem_allocator.h | 54 VisitAlloc(ptr, device_id_.value(), num_bytes); in Alloc()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 99 void SubAllocator::VisitAlloc(void* ptr, int index, size_t num_bytes) { in VisitAlloc() function in tensorflow::SubAllocator
|
| D | allocator.h | 456 void VisitAlloc(void* ptr, int index, size_t num_bytes);
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_virtual_mem_allocator.cc | 168 VisitAlloc(reinterpret_cast<void*>(next_va), gpu_id_.value(), padded_bytes); in Alloc()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | pool_allocator.cc | 271 VisitAlloc(ptr, numa_node_, num_bytes); in Alloc()
|