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

  /external/tensorflow/tensorflow/stream_executor/
stream_executor_pimpl.h 125 // As AllocateArray(), but returns a ScopedDeviceMemory<T>.
127 ScopedDeviceMemory<T> AllocateOwnedArray(uint64 element_count) {
128 return ScopedDeviceMemory<T>(this, AllocateArray<T>(element_count));
138 // As AllocateScalar(), but returns a ScopedDeviceMemory<T>.
140 ScopedDeviceMemory<T> AllocateOwnedScalar() {
149 // As AllocateZeroed(), but returns a ScopedDeviceMemory<T>.
151 ScopedDeviceMemory<T> AllocateOwnedZeroed() {
152 return ScopedDeviceMemory<T>(this, AllocateZeroed<T>());
168 // As AllocateSubBuffer(), but returns a ScopedDeviceMemory<T>.
170 ScopedDeviceMemory<T> AllocateOwnedSubBuffer(DeviceMemory<T> *parent
    [all...]
device_memory.h 187 // memory, if any. ScopedDeviceMemory is thread-compatible. It is also
190 class ScopedDeviceMemory {
195 ScopedDeviceMemory();
202 ScopedDeviceMemory(StreamExecutor *parent, DeviceMemoryBase value);
205 ScopedDeviceMemory(StreamExecutor *parent,
212 ScopedDeviceMemory(ScopedDeviceMemory &&other) noexcept:
213 ScopedDeviceMemory(other.parent_, other.Release()) {}
217 ~ScopedDeviceMemory();
222 ScopedDeviceMemory& operator=(ScopedDeviceMemory &&other)
    [all...]

Completed in 168 milliseconds