HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Public Member Functions | List of all members
onnxruntime::Stream Class Reference

#include <stream_handles.h>

Public Member Functions

 Stream (StreamHandle h, const OrtDevice &d)
 
virtual ~Stream ()=default
 
virtual std::unique_ptr
< synchronize::Notification > 		CreateNotification (size_t)
 
virtual void Flush ()
 
virtual Status CleanUpOnRunEnd ()
 
StreamHandle GetHandle () const
 
const OrtDevice & GetDevice () const
 
uint64_t GetCurrentTimestamp () const
 
uint64_t GetLastSyncTimestampWithTargetStream (Stream *target_stream) const
 
void CloneCurrentStreamSyncTable (std::unordered_map< Stream *, uint64_t > &output) const
 
uint64_t BumpTimeStampAndReturn ()
 
void UpdateStreamClock (const std::unordered_map< Stream *, uint64_t > &clock)
 
virtual void * GetResource (int, int) const
 

Detailed Description

Definition at line 26 of file stream_handles.h.

Constructor & Destructor Documentation

onnxruntime::Stream::Stream ( StreamHandle  h,
const OrtDevice &  d 
)
inline

Definition at line 28 of file stream_handles.h.

virtual onnxruntime::Stream::~Stream ( )
virtualdefault

Member Function Documentation

uint64_t onnxruntime::Stream::BumpTimeStampAndReturn ( )
inline

Definition at line 85 of file stream_handles.h.

virtual Status onnxruntime::Stream::CleanUpOnRunEnd ( )
inlinevirtual

Definition at line 39 of file stream_handles.h.

void onnxruntime::Stream::CloneCurrentStreamSyncTable ( std::unordered_map< Stream *, uint64_t > &  output) const
inline

Definition at line 76 of file stream_handles.h.

virtual std::unique_ptr<synchronize::Notification> onnxruntime::Stream::CreateNotification ( size_t  )
inlinevirtual

Definition at line 31 of file stream_handles.h.

virtual void onnxruntime::Stream::Flush ( )
inlinevirtual

Definition at line 35 of file stream_handles.h.

uint64_t onnxruntime::Stream::GetCurrentTimestamp ( ) const
inline

Definition at line 61 of file stream_handles.h.

const OrtDevice& onnxruntime::Stream::GetDevice ( ) const
inline

Definition at line 43 of file stream_handles.h.

StreamHandle onnxruntime::Stream::GetHandle ( ) const
inline

Definition at line 41 of file stream_handles.h.

uint64_t onnxruntime::Stream::GetLastSyncTimestampWithTargetStream ( Stream *  target_stream) const
inline

Definition at line 67 of file stream_handles.h.

virtual void* onnxruntime::Stream::GetResource ( int  ,
int   
) const
inlinevirtual

Definition at line 99 of file stream_handles.h.

void onnxruntime::Stream::UpdateStreamClock ( const std::unordered_map< Stream *, uint64_t > &  clock)
inline

Definition at line 90 of file stream_handles.h.

The documentation for this class was generated from the following file: