| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | 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) {}
87 BufferAllocations(BufferAllocation::Index buffer_count, int device_ordinal, DeviceMemoryAllocator* memory_allocator) argument
|
| H A D | buffer_allocations.cc | 43 DeviceMemoryAllocator* memory_allocator) {
46 new BufferAllocations(num_buffers, device_ordinal, memory_allocator));
72 TF_ASSIGN_OR_RETURN(buffer_address, memory_allocator->Allocate(
84 "Address returned by memory_allocator->Allocate must be a "
41 Build( const BufferAssignment& buffer_assignment, int device_ordinal, DeviceMemoryAllocator* memory_allocator) argument
|
| H A D | fft_thunk.cc | 33 int device_ordinal, DeviceMemoryAllocator* memory_allocator)
34 : device_ordinal_(device_ordinal), memory_allocator_(memory_allocator) {}
134 buffer_allocations.memory_allocator());
32 FftScratchAllocator( int device_ordinal, DeviceMemoryAllocator* memory_allocator) argument
|
| H A D | gpu_executable.cc | 254 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); local
281 memory_allocator));
284 !memory_allocator->AllowsAsynchronousDeallocation();
|
| H A D | cudnn_convolution_algorithm_picker.cc | 37 ScratchAllocator(int device_ordinal, DeviceMemoryAllocator* memory_allocator) argument
38 : device_ordinal_(device_ordinal), memory_allocator_(memory_allocator) {}
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | backend.h | 84 DeviceMemoryAllocator* memory_allocator() const { function in class:xla::Backend
|
| H A D | shaped_buffer.h | 137 DeviceMemoryAllocator* memory_allocator() const { return allocator_; } function in class:xla::ScopedShapedBuffer
|
| /external/libchrome/base/metrics/ |
| H A D | persistent_histogram_allocator.h | 217 PersistentMemoryAllocator* memory_allocator() { function in class:base::PersistentHistogramAllocator
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | 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,
76 AllocateBuffers( DeviceMemoryAllocator* memory_allocator, int device_ordinal, std::vector<perftools::gputools::DeviceMemoryBase>* buffers) argument
|
| H A D | 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(
326 AllocateBuffers( DeviceMemoryAllocator* memory_allocator, int device_ordinal, std::vector<perftools::gputools::DeviceMemoryBase>* buffers) argument
|
| /external/v8/src/heap/ |
| H A D | heap.h | 1039 MemoryAllocator* memory_allocator() { return memory_allocator_; } function in class:v8::internal::GarbageCollectionReason::Heap
|