Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields | Friends
at::Context Class Reference

Public Member Functions

TypeExtendedInterfacegetNonVariableTypeRaw (Backend p, ScalarType s)
 
TypeExtendedInterfacegetNonVariableTypeOpt (Backend p, ScalarType s)
 
TypeExtendedInterfacegetNonVariableType (Backend p, ScalarType s)
 
TypeExtendedInterfacegetVariableType (Backend p, ScalarType s)
 
TypeExtendedInterfacegetType (Backend p, ScalarType s, bool is_variable)
 
LegacyTHDispatchergetLegacyTHDispatcher (Backend p, ScalarType s)
 
void registerType (Backend b, ScalarType s, Type *t)
 
void registerLegacyTHDispatcher (Backend b, ScalarType s, LegacyTHDispatcher *t)
 
GeneratordefaultGenerator (DeviceType device_type)
 
bool hasOpenMP () const
 
bool hasMKL () const
 
bool hasLAPACK () const
 
bool hasMAGMA () const
 
bool hasCUDA () const
 
bool hasHIP () const
 
bool hasXLA () const
 
THCState * lazyInitCUDA ()
 
THHState * lazyInitHIP ()
 
void lazyInitComplex ()
 
THCState * getTHCState ()
 
THHState * getTHHState ()
 
size_t freshTypeID ()
 
bool setFlushDenormal (bool on)
 
bool userEnabledCuDNN () const
 
void setUserEnabledCuDNN (bool e)
 
bool benchmarkCuDNN () const
 
void setBenchmarkCuDNN (bool)
 
bool deterministicCuDNN () const
 
void setDeterministicCuDNN (bool)
 

Data Fields

std::unique_ptr< Generatorgenerator_registry [static_cast< int >(DeviceType::COMPILE_TIME_MAX_DEVICE_TYPES)]
 

Friends

struct Type
 

Detailed Description

Definition at line 27 of file Context.h.


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