Searched refs:ScopedDeviceMemory (Results 1 – 6 of 6) sorted by relevance
45 class ScopedDeviceMemory {50 ScopedDeviceMemory() : device_ordinal_(-1), allocator_(nullptr) {} in ScopedDeviceMemory() function60 ScopedDeviceMemory(DeviceMemoryBase mem, int device_ordinal, in ScopedDeviceMemory() function70 ScopedDeviceMemory(StreamExecutor *parent, DeviceMemoryBase value);76 ScopedDeviceMemory(StreamExecutor *parent,83 ScopedDeviceMemory(ScopedDeviceMemory &&other) in ScopedDeviceMemory() function90 ~ScopedDeviceMemory() { TF_CHECK_OK(Free()); } in ~ScopedDeviceMemory()95 ScopedDeviceMemory &operator=(ScopedDeviceMemory &&other) {149 SE_DISALLOW_COPY_AND_ASSIGN(ScopedDeviceMemory);153 using OwningDeviceMemory = ScopedDeviceMemory<uint8>;[all …]
128 ScopedDeviceMemory<T> AllocateOwnedArray(uint64 element_count) { in AllocateOwnedArray()129 return ScopedDeviceMemory<T>(this, AllocateArray<T>(element_count)); in AllocateOwnedArray()141 ScopedDeviceMemory<T> AllocateOwnedScalar() { in AllocateOwnedScalar()152 ScopedDeviceMemory<T> AllocateOwnedZeroed() { in AllocateOwnedZeroed()153 return ScopedDeviceMemory<T>(this, AllocateZeroed<T>()); in AllocateOwnedZeroed()828 ScopedDeviceMemory<ElemT>::ScopedDeviceMemory(StreamExecutor *parent, in ScopedDeviceMemory() function835 ScopedDeviceMemory<ElemT>::ScopedDeviceMemory( in ScopedDeviceMemory() function837 : ScopedDeviceMemory(parent, parent->AllocateArray<ElemT>(values.size())) { in ScopedDeviceMemory()
60 se::ScopedDeviceMemory<uint8> device_memory_;
45 se::ScopedDeviceMemory<ElementType> lhs_buffer = in CompareEqualBuffers()47 se::ScopedDeviceMemory<ElementType> rhs_buffer = in CompareEqualBuffers()
569 se::ScopedDeviceMemory<uint64> out_param = in DeviceCompare()
321 ScopedDeviceMemory<uint64> out_param = in CheckRedzones()