OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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