/external/tensorflow/tensorflow/stream_executor/ |
D | temporary_memory_manager.cc | 32 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 …]
|
D | BUILD | 78 "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 …]
|
D | temporary_device_memory.h | 65 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()
|
D | temporary_device_memory.cc | 34 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()
|
D | temporary_memory_manager.h | 77 void MarkFinalized(const DeviceMemoryBase& device_memory, uint64 generation, 91 bool IsFinalized(const DeviceMemoryBase& device_memory, 99 bool HasAllocated(const DeviceMemoryBase& device_memory,
|
D | scratch_allocator.cc | 37 return temporary_->device_memory(); in AllocateBytes()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | allocation_tracker.cc | 219 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()
|
D | allocation_tracker.h | 80 OwningDeviceMemory device_memory; member 103 void AddAllocationOrIncrementRefCount(se::DeviceMemoryBase device_memory, 109 Status DecrementRefCount(se::DeviceMemoryBase device_memory,
|
D | generic_transfer_manager.cc | 123 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()
|
D | transfer_manager.cc | 247 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/ |
D | infeed_manager.h | 56 se::DeviceMemoryBase* device_memory() { return device_memory_.ptr(); } in device_memory() function
|
D | gpu_executable.cc | 298 const ShapeIndex& index, se::DeviceMemoryBase* device_memory) { in ExecuteOnStream() argument 327 *device_memory = src_base; in ExecuteOnStream()
|
D | infeed_thunk.cc | 72 *(buffer->device_memory()), buffer->length()); in ExecuteOnStream()
|
D | gpu_transfer_manager.cc | 113 stream->ThenMemcpy(buffer.device_memory(), source, size); in TransferBufferToInfeedInternal()
|
D | gemm_thunk.cc | 428 scratch_data = scratch_mem->device_memory(); in ExecuteOnStream()
|
D | BUILD | 409 "//tensorflow/stream_executor:device_memory",
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_transfer_manager.cc | 55 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()
|
D | cpu_executable.cc | 223 [&](const ShapeIndex& index, se::DeviceMemoryBase* device_memory) { in CreateResultShapedBuffer() argument 250 *device_memory = buffer.Forget(); in CreateResultShapedBuffer()
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | local_computation_builder.cc | 194 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/ |
D | cudnn_rnn_ops.cc | 313 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/ |
D | rocm_dnn.cc | 2806 return (*transform_scratch)->device_memory(); in MaybeTransformLayout()
|
/external/vulkan-validation-layers/tests/ |
D | layer_validation_tests.cpp | 878 VkDeviceMemory device_memory; in ~VkBufferTest() member 882 bad_index.device_memory = VulkanMemory; in ~VkBufferTest() 885 vkFreeMemory(VulkanDevice, bad_index.device_memory, nullptr); in ~VkBufferTest()
|