| /external/tensorflow/tensorflow/stream_executor/ |
| D | temporary_memory_manager.cc | 31 DeviceMemoryBase device_memory = it->first; in ForceDeallocateAll() local
37 const DeviceMemoryBase& device_memory, uint64 generation, bool must_exist) { in MarkFinalized()
55 DeviceMemoryBase device_memory = it->first; in DeallocateFinalizedTemporaries() local
66 bool TemporaryMemoryManager::IsFinalized(const DeviceMemoryBase& device_memory, in IsFinalized()
82 bool TemporaryMemoryManager::HasAllocated(const DeviceMemoryBase& device_memory, in HasAllocated()
96 DeviceMemoryBase device_memory = in AllocateArrayBase() local
|
| 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()
|
| D | temporary_device_memory.h | 119 const DeviceMemory<T>& device_memory() const { in device_memory() function
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tracked_device_buffer.h | 168 absl::InlinedVector<se::DeviceMemoryBase, 1>& device_memory() { in device_memory() function
171 const absl::InlinedVector<se::DeviceMemoryBase, 1>& device_memory() const { in device_memory() function
|
| D | tracked_device_buffer.cc | 165 absl::Span<se::DeviceMemoryBase const> device_memory, in TrackedDeviceBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | infeed_manager.h | 57 se::DeviceMemoryBase* device_memory() { return device_memory_.ptr(); } in device_memory() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | allocation_tracker.h | 80 se::OwningDeviceMemory device_memory; member
|
| D | allocation_tracker.cc | 207 se::DeviceMemoryBase device_memory, int device_ordinal) { in AddAllocationOrIncrementRefCount()
220 Status AllocationTracker::DecrementRefCount(se::DeviceMemoryBase device_memory, in DecrementRefCount()
|
| D | transfer_manager.cc | 302 se::DeviceMemoryBase device_memory = device_buffer.buffer(index); in WriteTupleIndexTablesAsync() local
328 se::DeviceMemoryBase device_memory = device_buffer.buffer({}); in WriteRootTupleIndexTable() local
347 se::DeviceMemoryBase device_memory = in WriteRootTupleIndexTable() local
|
| D | generic_transfer_manager.cc | 111 se::DeviceMemoryBase device_memory = device_buffer.buffer(index); in TransferLiteralToDeviceAsync() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.h | 71 ExecutionContext(TRTBaseAllocator* allocator, void* device_memory, in ExecutionContext()
|
| D | trt_engine_utils.cc | 51 void* device_memory = nullptr; in Create() local
|
| /external/crosvm/rutabaga_gfx/src/rutabaga_gralloc/ |
| D | vulkano_gralloc.rs | 350 let device_memory = in allocate_memory() localVariable
392 let device_memory = in import_and_map() localVariable
|
| /external/perfetto/protos/perfetto/trace/gpu/ |
| D | vulkan_memory_event.proto | 99 optional fixed64 device_memory = 17; field
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.cc | 117 MaybeOwningDeviceMemory* device_memory = in AllocateOutputMemoryWithInputReuse() local
|
| /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::__anon31ee37eb0111::CpuInfeedBuffer
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 112 [&](const ShapeIndex& index, MaybeOwningDeviceMemory* device_memory) { in RunAndCheck()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 313 DeviceMemoryBase SliceDeviceMemory(const DeviceMemoryBase& device_memory, in SliceDeviceMemory()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 1938 DeviceMemory<MAPPED_T> *device_memory, in ReorganizeMemory()
1990 DeviceMemory<typename RocBlasTypeConversionHelper<T>::mapped_type> in AllocateStridedBuffer()
|
| /external/perfetto/protos/perfetto/trace/ |
| D | perfetto_trace.proto | 5966 optional fixed64 device_memory = 17; field
|