Home
last modified time | relevance | path

Searched refs:TemporaryDeviceMemoryBase (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dtemporary_device_memory.cc22 TemporaryDeviceMemoryBase::~TemporaryDeviceMemoryBase() { in ~TemporaryDeviceMemoryBase()
28 DeviceMemoryBase* TemporaryDeviceMemoryBase::mutable_device_memory() { in mutable_device_memory()
34 const DeviceMemoryBase& TemporaryDeviceMemoryBase::device_memory() const { in device_memory()
40 void TemporaryDeviceMemoryBase::Finalize() { in Finalize()
47 bool TemporaryDeviceMemoryBase::IsFinalized() const { in IsFinalized()
52 bool TemporaryDeviceMemoryBase::IsAllocated() const { in IsAllocated()
57 TemporaryDeviceMemoryBase::TemporaryDeviceMemoryBase( in TemporaryDeviceMemoryBase() function in stream_executor::TemporaryDeviceMemoryBase
Dtemporary_device_memory.h55 class TemporaryDeviceMemoryBase {
59 ~TemporaryDeviceMemoryBase();
93 TemporaryDeviceMemoryBase(Stream* parent, DeviceMemoryBase device_memory,
109 class TemporaryDeviceMemory : public TemporaryDeviceMemoryBase {
115 TemporaryDeviceMemoryBase::mutable_device_memory()); in mutable_device_memory()
122 TemporaryDeviceMemoryBase::device_memory()); in device_memory()
128 sizeof(TemporaryDeviceMemory) == sizeof(TemporaryDeviceMemoryBase), in StaticSlicingAssertionDummy()
Dtemporary_memory_manager.cc92 port::StatusOr<std::unique_ptr<TemporaryDeviceMemoryBase>>
120 std::unique_ptr<TemporaryDeviceMemoryBase> result( in AllocateArrayBase()
121 new TemporaryDeviceMemoryBase(stream_, device_memory, generation)); in AllocateArrayBase()
Dtemporary_memory_manager.h107 port::StatusOr<std::unique_ptr<TemporaryDeviceMemoryBase>> AllocateArrayBase(
138 port::StatusOr<std::unique_ptr<TemporaryDeviceMemoryBase>> temporary_memory = in AllocateArray()