Home
last modified time | relevance | path

Searched refs:device_memory (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dtemporary_memory_manager.cc32 DeviceMemoryBase device_memory = it->first; in ForceDeallocateAll() local
33 stream_->parent()->Deallocate(&device_memory); in ForceDeallocateAll()
38 const DeviceMemoryBase& device_memory, uint64 generation, bool must_exist) { in MarkFinalized() argument
40 auto it = records_.find(device_memory); in MarkFinalized()
56 DeviceMemoryBase device_memory = it->first; in DeallocateFinalizedTemporaries() local
57 stream_->parent()->Deallocate(&device_memory); in DeallocateFinalizedTemporaries()
67 bool TemporaryMemoryManager::IsFinalized(const DeviceMemoryBase& device_memory, in IsFinalized() argument
70 auto it = records_.find(device_memory); in IsFinalized()
83 bool TemporaryMemoryManager::HasAllocated(const DeviceMemoryBase& device_memory, in HasAllocated() argument
86 auto it = records_.find(device_memory); in HasAllocated()
[all …]
DBUILD78 "device_memory.h",
133 ":device_memory",
181 "device_memory.h",
217 ":device_memory",
295 ":device_memory",
307 ":device_memory",
339 name = "device_memory",
340 hdrs = ["device_memory.h"],
348 ":device_memory",
365 "device_memory.h",
[all …]
Dtemporary_device_memory.h65 const DeviceMemoryBase& device_memory() const;
93 TemporaryDeviceMemoryBase(Stream* parent, DeviceMemoryBase device_memory,
119 const DeviceMemory<T>& device_memory() const { in device_memory() function
122 TemporaryDeviceMemoryBase::device_memory()); in device_memory()
Dtemporary_device_memory.cc34 const DeviceMemoryBase& TemporaryDeviceMemoryBase::device_memory() const { in device_memory() function in stream_executor::TemporaryDeviceMemoryBase
58 Stream* parent, DeviceMemoryBase device_memory, in TemporaryDeviceMemoryBase() argument
60 : device_memory_(device_memory), in TemporaryDeviceMemoryBase()
Dtemporary_memory_manager.h77 void MarkFinalized(const DeviceMemoryBase& device_memory, uint64 generation,
91 bool IsFinalized(const DeviceMemoryBase& device_memory,
99 bool HasAllocated(const DeviceMemoryBase& device_memory,
Dscratch_allocator.cc37 return temporary_->device_memory(); in AllocateBytes()
/external/tensorflow/tensorflow/compiler/xla/service/
Dallocation_tracker.cc219 se::DeviceMemoryBase device_memory, int device_ordinal) { in AddAllocationOrIncrementRefCount() argument
221 auto it = allocation_map.find(device_memory.opaque()); in AddAllocationOrIncrementRefCount()
223 allocation_map[device_memory.opaque()] = { in AddAllocationOrIncrementRefCount()
224 OwningDeviceMemory(device_memory, device_ordinal, in AddAllocationOrIncrementRefCount()
232 Status AllocationTracker::DecrementRefCount(se::DeviceMemoryBase device_memory, in DecrementRefCount() argument
235 auto it = allocation_map.find(device_memory.opaque()); in DecrementRefCount()
240 allocation.device_memory.Free(); in DecrementRefCount()
Dallocation_tracker.h80 OwningDeviceMemory device_memory; member
103 void AddAllocationOrIncrementRefCount(se::DeviceMemoryBase device_memory,
109 Status DecrementRefCount(se::DeviceMemoryBase device_memory,
Dgeneric_transfer_manager.cc123 se::DeviceMemoryBase device_memory = device_buffer.buffer(index); in TransferLiteralToDeviceAsync() local
126 device_memory.size()); in TransferLiteralToDeviceAsync()
137 &device_memory); in TransferLiteralToDeviceAsync()
146 &device_memory)); in TransferLiteralToDeviceAsync()
Dtransfer_manager.cc247 se::DeviceMemoryBase device_memory = device_buffer.buffer(index); in WriteTupleIndexTablesAsync() local
249 device_memory.size()); in WriteTupleIndexTablesAsync()
260 &device_memory); in WriteTupleIndexTablesAsync()
270 se::DeviceMemoryBase device_memory = device_buffer.buffer({}); in WriteRootTupleIndexTable() local
272 device_memory.size()); in WriteRootTupleIndexTable()
280 stream, elements, device_buffer.on_device_shape(), &device_memory); in WriteRootTupleIndexTable()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinfeed_manager.h56 se::DeviceMemoryBase* device_memory() { return device_memory_.ptr(); } in device_memory() function
Dgpu_executable.cc298 const ShapeIndex& index, se::DeviceMemoryBase* device_memory) { in ExecuteOnStream() argument
327 *device_memory = src_base; in ExecuteOnStream()
Dinfeed_thunk.cc72 *(buffer->device_memory()), buffer->length()); in ExecuteOnStream()
Dgpu_transfer_manager.cc113 stream->ThenMemcpy(buffer.device_memory(), source, size); in TransferBufferToInfeedInternal()
Dgemm_thunk.cc428 scratch_data = scratch_mem->device_memory(); in ExecuteOnStream()
DBUILD409 "//tensorflow/stream_executor:device_memory",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.cc55 se::DeviceMemoryBase* device_memory() { return &device_memory_; } in device_memory() function in xla::__anon79616a0d0111::CpuInfeedBuffer
168 /*host_src=*/source, /*size=*/size, queued_buffer->device_memory()); in TransferBufferToInfeedInternal()
Dcpu_executable.cc223 [&](const ShapeIndex& index, se::DeviceMemoryBase* device_memory) { in CreateResultShapedBuffer() argument
250 *device_memory = buffer.Forget(); in CreateResultShapedBuffer()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc194 se::DeviceMemoryBase* device_memory = in DestructureTuple() local
196 shaped_buffer.set_buffer(*device_memory, index); in DestructureTuple()
197 *device_memory = se::DeviceMemoryBase(); in DestructureTuple()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc313 DeviceMemoryBase SliceDeviceMemory(const DeviceMemoryBase& device_memory, in SliceDeviceMemory() argument
315 const void* base_ptr = device_memory.opaque(); in SliceDeviceMemory()
318 CHECK(offset + size <= device_memory.size()) in SliceDeviceMemory()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2806 return (*transform_scratch)->device_memory(); in MaybeTransformLayout()
/external/vulkan-validation-layers/tests/
Dlayer_validation_tests.cpp878 VkDeviceMemory device_memory; in ~VkBufferTest() member
882 bad_index.device_memory = VulkanMemory; in ~VkBufferTest()
885 vkFreeMemory(VulkanDevice, bad_index.device_memory, nullptr); in ~VkBufferTest()