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: