HomeSort by relevance Sort by last modified time
    Searched refs:ScopedDeviceMemory (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
device_memory.h 178 // memory, if any. ScopedDeviceMemory is thread-compatible. It is also
181 class ScopedDeviceMemory {
186 ScopedDeviceMemory();
193 ScopedDeviceMemory(StreamExecutor *parent, DeviceMemoryBase value);
196 ScopedDeviceMemory(StreamExecutor *parent,
203 ScopedDeviceMemory(ScopedDeviceMemory &&other) noexcept:
204 ScopedDeviceMemory(other.parent_, other.Release()) {}
208 ~ScopedDeviceMemory();
213 ScopedDeviceMemory& operator=(ScopedDeviceMemory &&other)
    [all...]
stream_executor_pimpl.h 115 // As AllocateArray(), but returns a ScopedDeviceMemory<T>.
117 ScopedDeviceMemory<T> AllocateOwnedArray(uint64 element_count) {
118 return ScopedDeviceMemory<T>(this, AllocateArray<T>(element_count));
128 // As AllocateScalar(), but returns a ScopedDeviceMemory<T>.
130 ScopedDeviceMemory<T> AllocateOwnedScalar() {
139 // As AllocateZeroed(), but returns a ScopedDeviceMemory<T>.
141 ScopedDeviceMemory<T> AllocateOwnedZeroed() {
142 return ScopedDeviceMemory<T>(this, AllocateZeroed<T>());
158 // As AllocateSubBuffer(), but returns a ScopedDeviceMemory<T>.
160 ScopedDeviceMemory<T> AllocateOwnedSubBuffer(DeviceMemory<T> *parent
    [all...]

Completed in 85 milliseconds