| /external/tensorflow/tensorflow/compiler/xla/service/ |
| allocation_tracker.h | 80 OwningDeviceMemory device_memory; member in struct:xla::AllocationTracker::Allocation
103 void AddAllocationOrIncrementRefCount(se::DeviceMemoryBase device_memory,
109 Status DecrementRefCount(se::DeviceMemoryBase device_memory,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| temporary_device_memory.cc | 34 const DeviceMemoryBase& TemporaryDeviceMemoryBase::device_memory() const { function in class:stream_executor::TemporaryDeviceMemoryBase
58 Stream* parent, DeviceMemoryBase device_memory,
60 : device_memory_(device_memory),
|
| temporary_device_memory.h | 23 // // temporary_memory->device_memory() and similar.
33 // after finalization. If temporary_memory->device_memory() is used after the
44 #include "tensorflow/stream_executor/device_memory.h"
65 const DeviceMemoryBase& device_memory() const;
93 TemporaryDeviceMemoryBase(Stream* parent, DeviceMemoryBase device_memory,
118 // Type-safe wrapper around TemporaryDeviceMemoryBase::device_memory.
119 const DeviceMemory<T>& device_memory() const { function in class:stream_executor::TemporaryDeviceMemory
122 TemporaryDeviceMemoryBase::device_memory());
|
| temporary_memory_manager.cc | 32 DeviceMemoryBase device_memory = it->first; local
33 stream_->parent()->Deallocate(&device_memory);
38 const DeviceMemoryBase& device_memory, uint64 generation, bool must_exist) {
40 auto it = records_.find(device_memory);
56 DeviceMemoryBase device_memory = it->first; local
57 stream_->parent()->Deallocate(&device_memory);
67 bool TemporaryMemoryManager::IsFinalized(const DeviceMemoryBase& device_memory,
70 auto it = records_.find(device_memory);
83 bool TemporaryMemoryManager::HasAllocated(const DeviceMemoryBase& device_memory,
86 auto it = records_.find(device_memory);
97 DeviceMemoryBase device_memory = local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| infeed_manager.h | 56 se::DeviceMemoryBase* device_memory() { return device_memory_.ptr(); } function in class:xla::gpu::InfeedBuffer
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_transfer_manager.cc | 55 se::DeviceMemoryBase* device_memory() { return &device_memory_; } function in class:xla::__anon44340::CpuInfeedBuffer
168 /*host_src=*/source, /*size=*/size, queued_buffer->device_memory());
|
