Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/stream_executor/
Dtemporary_memory_manager.cc31 DeviceMemoryBase device_memory = it->first; in ForceDeallocateAll() local
32 stream_->parent()->Deallocate(&device_memory); in ForceDeallocateAll()
37 const DeviceMemoryBase& device_memory, uint64 generation, bool must_exist) { in MarkFinalized() argument
39 auto it = records_.find(device_memory); in MarkFinalized()
55 DeviceMemoryBase device_memory = it->first; in DeallocateFinalizedTemporaries() local
56 stream_->parent()->Deallocate(&device_memory); in DeallocateFinalizedTemporaries()
66 bool TemporaryMemoryManager::IsFinalized(const DeviceMemoryBase& device_memory, in IsFinalized() argument
69 auto it = records_.find(device_memory); in IsFinalized()
82 bool TemporaryMemoryManager::HasAllocated(const DeviceMemoryBase& device_memory, in HasAllocated() argument
85 auto it = records_.find(device_memory); in HasAllocated()
[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()
DBUILD77 "device_memory.h",
133 ":device_memory",
248 ":device_memory",
261 ":device_memory",
295 name = "device_memory",
296 hdrs = ["device_memory.h"],
304 ":device_memory",
321 "device_memory.h",
412 ":device_memory",
432 ":device_memory",
[all …]
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.cc36 return temporary_->device_memory(); in AllocateBytes()
/external/tensorflow/tensorflow/compiler/xla/python/
Dshared_device_buffer_test.cc38 ASSERT_EQ(buffer->device_memory().size(), 1); in TEST()
39 EXPECT_FALSE(buffer->device_memory()[0].is_null()); in TEST()
64 ASSERT_EQ(tuple_buffer->device_memory().size(), 1); in TEST()
67 EXPECT_FALSE(tuple_buffer->device_memory()[0].is_null()); in TEST()
111 ASSERT_EQ(a_buffer->device_memory().size(), 1); in TEST()
112 ASSERT_EQ(b_buffer->device_memory().size(), 1); in TEST()
113 ASSERT_EQ(c_buffer->device_memory().size(), 1); in TEST()
114 ASSERT_EQ(ab_tuple_buffer->device_memory().size(), 1); in TEST()
115 ASSERT_EQ(abc_tuple_buffer->device_memory().size(), 1); in TEST()
117 abc_tuple_buffer->device_memory()[0], c_buffer->device_memory()[0], in TEST()
[all …]
Dshared_device_buffer.cc118 se::OwningDeviceMemory device_memory, in MakeTuple()
124 std::initializer_list<se::DeviceMemoryBase>{device_memory.Release()}, in MakeTuple()
138 se::OwningDeviceMemory device_memory, in MakeArray()
142 device_buffers.push_back(std::move(device_memory)); in MakeArray()
156 for (const se::DeviceMemoryBase& buf : buffer.device_memory()) { in PopulateShapedBufferFromBuffer()
181 absl::Span<se::DeviceMemoryBase const> device_memory, in SharedDeviceBuffer() argument
187 device_memory_(device_memory.begin(), device_memory.end()), in SharedDeviceBuffer()
193 absl::Span<se::OwningDeviceMemory> device_memory, in SharedDeviceBuffer() argument
198 CHECK(!device_memory.empty()); in SharedDeviceBuffer()
199 allocator_ = device_memory.front().allocator(); in SharedDeviceBuffer()
[all …]
Dshared_device_buffer.h123 absl::InlinedVector<se::DeviceMemoryBase, 1>& device_memory() { in device_memory() function
126 const absl::InlinedVector<se::DeviceMemoryBase, 1>& device_memory() const { in device_memory() function
135 absl::Span<se::DeviceMemoryBase const> device_memory,
139 SharedDeviceBuffer(absl::Span<se::OwningDeviceMemory> device_memory,
Ddlpack.cc259 TF_RET_CHECK(pack->buffer->device_memory().size() == 1); in BufferToDLPackManagedTensor()
260 dt.data = pack->buffer->device_memory().front().opaque(); in BufferToDLPackManagedTensor()
DBUILD125 "//tensorflow/stream_executor:device_memory",
269 "//tensorflow/stream_executor:device_memory",
/external/tensorflow/tensorflow/compiler/xla/service/
Dallocation_tracker.cc220 se::DeviceMemoryBase device_memory, int device_ordinal) { in AddAllocationOrIncrementRefCount() argument
222 auto it = allocation_map.find(device_memory.opaque()); in AddAllocationOrIncrementRefCount()
224 allocation_map[device_memory.opaque()] = { in AddAllocationOrIncrementRefCount()
225 se::OwningDeviceMemory(device_memory, device_ordinal, in AddAllocationOrIncrementRefCount()
233 Status AllocationTracker::DecrementRefCount(se::DeviceMemoryBase device_memory, in DecrementRefCount() argument
236 auto it = allocation_map.find(device_memory.opaque()); in DecrementRefCount()
241 TF_RETURN_IF_ERROR(allocation.device_memory.Free()); in DecrementRefCount()
Dallocation_tracker.h80 se::OwningDeviceMemory device_memory; member
103 void AddAllocationOrIncrementRefCount(se::DeviceMemoryBase device_memory,
109 Status DecrementRefCount(se::DeviceMemoryBase device_memory,
Dgeneric_transfer_manager.cc121 se::DeviceMemoryBase device_memory = device_buffer.buffer(index); in TransferLiteralToDeviceAsync() local
124 device_memory.size()); in TransferLiteralToDeviceAsync()
135 &device_memory); in TransferLiteralToDeviceAsync()
144 &device_memory)); in TransferLiteralToDeviceAsync()
Dtransfer_manager.cc249 se::DeviceMemoryBase device_memory = device_buffer.buffer(index); in WriteTupleIndexTablesAsync() local
251 device_memory.size()); in WriteTupleIndexTablesAsync()
262 &device_memory); in WriteTupleIndexTablesAsync()
275 se::DeviceMemoryBase device_memory = device_buffer.buffer({}); in WriteRootTupleIndexTable() local
277 device_memory.size()); in WriteRootTupleIndexTable()
285 stream, elements, device_buffer.on_device_shape(), &device_memory); in WriteRootTupleIndexTable()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinfeed_manager.h57 se::DeviceMemoryBase* device_memory() { return device_memory_.ptr(); } in device_memory() function
Dinfeed_thunk.cc74 *(buffer->device_memory()), buffer->length()); in ExecuteOnStream()
Dgpu_executable.cc383 const ShapeIndex& index, se::DeviceMemoryBase* device_memory) { in ExecuteAsyncOnStream() argument
412 *device_memory = src_base; in ExecuteAsyncOnStream()
Dgpu_transfer_manager.cc113 stream->ThenMemcpy(buffer.device_memory(), source, size); in TransferBufferToInfeedInternal()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc93 [&](const ShapeIndex& index, MaybeOwningDeviceMemory* device_memory) { in RunAndCheck() argument
94 *device_memory = se::OwningDeviceMemory(input_buffers.element(index), in RunAndCheck()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.cc55 se::DeviceMemoryBase* device_memory() { return &device_memory_; } in device_memory() function in xla::__anon907aaf0f0111::CpuInfeedBuffer
168 /*host_src=*/source, /*size=*/size, queued_buffer->device_memory()); in TransferBufferToInfeedInternal()
Dcpu_executable.cc241 [&](const ShapeIndex& index, se::DeviceMemoryBase* device_memory) { in CreateResultShapedBuffer() argument
268 *device_memory = buffer.Release(); in CreateResultShapedBuffer()
/external/crosvm/kvm/src/
Dlib.rs294 device_memory: HashMap<u32, MemoryMapping>, field
326 device_memory: HashMap::new(), in new()
347 (self.device_memory.len() in set_user_memory_region()
427 self.device_memory.insert(slot, mem); in add_device_memory()
436 if self.device_memory.contains_key(&slot) { in remove_device_memory()
442 Ok(self.device_memory.remove(&slot).unwrap()) in remove_device_memory()
519 match self.device_memory.get(&slot) { in get_dirty_log()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.h124 *device_memory);
/external/perfetto/protos/perfetto/trace/gpu/
Dvulkan_memory_event.proto89 optional fixed64 device_memory = 17; field

12