Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
at::CUDAHooksInterface Struct Reference
Inheritance diagram for at::CUDAHooksInterface:
at::cuda::detail::CUDAHooks

Public Member Functions

virtual std::unique_ptr< THCState, void(*)(THCState *)> initCUDA () const
 
virtual std::unique_ptr< Generator > initCUDAGenerator (Context *) const
 
virtual bool hasCUDA () const
 
virtual bool hasMAGMA () const
 
virtual bool hasCuDNN () const
 
virtual int64_t current_device () const
 
virtual Allocator * getPinnedMemoryAllocator () const
 
virtual void registerCUDATypes (Context *) const
 
virtual bool compiledWithCuDNN () const
 
virtual bool compiledWithMIOpen () const
 
virtual bool supportsDilatedConvolutionWithCuDNN () const
 
virtual long versionCuDNN () const
 
virtual double batchnormMinEpsilonCuDNN () const
 
virtual int64_t cuFFTGetPlanCacheMaxSize () const
 
virtual void cuFFTSetPlanCacheMaxSize (int64_t max_size) const
 
virtual int64_t cuFFTGetPlanCacheSize () const
 
virtual void cuFFTClearPlanCache () const
 
virtual int getNumGPUs () const
 

Detailed Description

Definition at line 51 of file CUDAHooksInterface.h.

The documentation for this struct was generated from the following file:
Generated on Thu Mar 21 2019 13:06:25 for Caffe2 - C++ API by   doxygen 1.8.11