Searched refs:device_memory_ (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | temporary_device_memory.cc | 23 parent_->temporary_memory_manager()->MarkFinalized(device_memory_, in ~TemporaryDeviceMemoryBase()
31 return &device_memory_; in mutable_device_memory()
37 return device_memory_; in device_memory()
42 parent_->temporary_memory_manager()->MarkFinalized(device_memory_, in Finalize()
49 device_memory_, allocation_generation_); in IsFinalized()
54 device_memory_, allocation_generation_); in IsAllocated()
60 : device_memory_(device_memory), in TemporaryDeviceMemoryBase()
|
| D | temporary_device_memory.h | 97 DeviceMemoryBase device_memory_; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | infeed_manager.h | 50 : device_memory_(executor, executor->AllocateArray<uint8>(length)), in InfeedBuffer()
52 CHECK(!device_memory_->is_null()); in InfeedBuffer()
57 se::DeviceMemoryBase* device_memory() { return device_memory_.ptr(); } in device_memory()
60 se::ScopedDeviceMemory<uint8> device_memory_;
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.h | 74 device_memory_(device_memory), in ExecutionContext()
85 device_memory_(other.device_memory_), in ExecutionContext()
88 other.device_memory_ = nullptr; in ExecutionContext()
107 void* device_memory_; variable
|
| D | trt_engine_utils.cc | 39 if (device_memory_) { in ~ExecutionContext()
41 << (char*)device_memory_ << "is not freed"; in ~ExecutionContext()
42 memory_allocator_->free(device_memory_); in ~ExecutionContext()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tracked_device_buffer.h | 169 return device_memory_; in device_memory()
172 return device_memory_; in device_memory()
184 void ReleaseDeviceMemory() { device_memory_.clear(); } in ReleaseDeviceMemory()
221 absl::InlinedVector<se::DeviceMemoryBase, 1> device_memory_; variable
|
| D | tracked_device_buffer.cc | 125 for (const se::DeviceMemoryBase& buf : device_memory_) { in AsShapedBuffer()
140 for (const se::DeviceMemoryBase& buf : device_memory_) { in AddToInputAsImmutable()
153 for (const se::DeviceMemoryBase& buf : device_memory_) { in AddToInputAsDonated()
170 device_memory_(device_memory.begin(), device_memory.end()), in TrackedDeviceBuffer()
178 for (const se::DeviceMemoryBase& buffer : device_memory_) { in ~TrackedDeviceBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 48 device_memory_(buffer_, length_) {} in CpuInfeedBuffer()
55 se::DeviceMemoryBase* device_memory() { return &device_memory_; } in device_memory()
60 se::DeviceMemoryBase device_memory_; member in xla::__anon31ee37eb0111::CpuInfeedBuffer
|