Searched refs:ScopedDeviceMemory (Results 1 – 3 of 3) sorted by relevance
190 class ScopedDeviceMemory {195 ScopedDeviceMemory();202 ScopedDeviceMemory(StreamExecutor *parent, DeviceMemoryBase value);205 ScopedDeviceMemory(StreamExecutor *parent,212 ScopedDeviceMemory(ScopedDeviceMemory &&other) noexcept: in ScopedDeviceMemory() function213 ScopedDeviceMemory(other.parent_, other.Release()) {} in ScopedDeviceMemory()217 ~ScopedDeviceMemory();222 ScopedDeviceMemory& operator=(ScopedDeviceMemory &&other) {269 SE_DISALLOW_COPY_AND_ASSIGN(ScopedDeviceMemory);
127 ScopedDeviceMemory<T> AllocateOwnedArray(uint64 element_count) { in AllocateOwnedArray()128 return ScopedDeviceMemory<T>(this, AllocateArray<T>(element_count)); in AllocateOwnedArray()140 ScopedDeviceMemory<T> AllocateOwnedScalar() { in AllocateOwnedScalar()151 ScopedDeviceMemory<T> AllocateOwnedZeroed() { in AllocateOwnedZeroed()152 return ScopedDeviceMemory<T>(this, AllocateZeroed<T>()); in AllocateOwnedZeroed()170 ScopedDeviceMemory<T> AllocateOwnedSubBuffer(DeviceMemory<T> *parent, in AllocateOwnedSubBuffer()173 return ScopedDeviceMemory<T>( in AllocateOwnedSubBuffer()782 ScopedDeviceMemory<ElemT>::ScopedDeviceMemory() in ScopedDeviceMemory() function786 ScopedDeviceMemory<ElemT>::ScopedDeviceMemory(StreamExecutor *parent, in ScopedDeviceMemory() function791 ScopedDeviceMemory<ElemT>::ScopedDeviceMemory( in ScopedDeviceMemory() function[all …]
59 se::ScopedDeviceMemory<uint8> device_memory_;