/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
buffer_allocations.h | 47 // `memory_allocator` is what this function uses to allocate device memory. 52 DeviceMemoryAllocator* memory_allocator); 62 DeviceMemoryAllocator* memory_allocator() const { return memory_allocator_; } function in class:xla::gpu::BufferAllocations 88 DeviceMemoryAllocator* memory_allocator) 91 memory_allocator_(memory_allocator) {}
|
gpu_executable.cc | 254 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); local 281 memory_allocator)); 284 !memory_allocator->AllowsAsynchronousDeallocation();
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
shaped_buffer.h | 137 DeviceMemoryAllocator* memory_allocator() const { return allocator_; } function in class:xla::ScopedShapedBuffer
|
backend.h | 84 DeviceMemoryAllocator* memory_allocator() const { function in class:xla::Backend
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
cpu_executable.cc | 77 DeviceMemoryAllocator* memory_allocator, int device_ordinal, 103 TF_ASSIGN_OR_RETURN((*buffers)[i], memory_allocator->Allocate( 296 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); local 300 memory_allocator, stream->parent()->device_ordinal(), &buffers)); 310 TF_RETURN_IF_ERROR(DeallocateTempBuffers(memory_allocator, stream, buffers, 328 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); local 332 memory_allocator, stream->parent()->device_ordinal(), &buffers)); 342 buffers_in_result, memory_allocator, stream]() { 348 TF_CHECK_OK(DeallocateTempBuffers(memory_allocator, stream, buffers,
|
parallel_cpu_executable.cc | 327 DeviceMemoryAllocator* memory_allocator, int device_ordinal, 353 TF_ASSIGN_OR_RETURN((*buffers)[i], memory_allocator->Allocate( 461 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); local 469 memory_allocator, stream->parent()->device_ordinal(), &buffers)); 508 TF_RETURN_IF_ERROR(memory_allocator->Deallocate(
|
/external/libchrome/base/metrics/ |
persistent_histogram_allocator.h | 217 PersistentMemoryAllocator* memory_allocator() { function in class:base::PersistentHistogramAllocator
|
/external/v8/src/heap/ |
heap.h | 1039 MemoryAllocator* memory_allocator() { return memory_allocator_; } function in class:v8::internal::GarbageCollectionReason::Heap [all...] |