| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| buffer_allocations.h | 48 // `memory_allocator` is what this function uses to allocate device memory.
53 DeviceMemoryAllocator* memory_allocator);
65 DeviceMemoryAllocator* memory_allocator() const { return memory_allocator_; } function in class:xla::gpu::BufferAllocations
87 DeviceMemoryAllocator* memory_allocator,
91 memory_allocator_(memory_allocator),
|
| gpu_executable.cc | 240 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); local
280 assignment_.get(), executor->device_ordinal(), memory_allocator));
283 !memory_allocator->AllowsAsynchronousDeallocation();
291 memory_allocator, device_ordinal);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| backend.h | 91 DeviceMemoryAllocator* memory_allocator() const { function in class:xla::Backend
|
| shaped_buffer.h | 157 DeviceMemoryAllocator* memory_allocator() const { return allocator_; } function in class:xla::ScopedShapedBuffer
|
| /external/libchrome/base/metrics/ |
| persistent_histogram_allocator.h | 217 PersistentMemoryAllocator* memory_allocator() { function in class:base::PersistentHistogramAllocator
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_executable.cc | 78 DeviceMemoryAllocator* memory_allocator, int device_ordinal,
114 TF_ASSIGN_OR_RETURN(owning_buffers[i], memory_allocator->Allocate(
300 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); local
305 CreateBufferTable(memory_allocator, stream->parent()->device_ordinal(),
|
| /external/v8/src/heap/ |
| heap.h | 792 MemoryAllocator* memory_allocator() { return memory_allocator_; } function in class:v8::internal::GarbageCollectionReason::Heap
[all...] |
