Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
at::HIPHooksInterface Struct Reference

Public Member Functions

virtual std::unique_ptr< THHState, void(*)(THHState *)> initHIP () const
 
virtual std::unique_ptr< GeneratorinitHIPGenerator (Context *) const
 
virtual bool hasHIP () const
 
virtual int64_t current_device () const
 
virtual AllocatorgetPinnedMemoryAllocator () const
 
virtual void registerHIPTypes (Context *) const
 
virtual int getNumGPUs () const
 

Detailed Description

Definition at line 27 of file HIPHooksInterface.h.


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