Home
last modified time | relevance | path

Searched refs:CUDARuntimeWrapper (Results 1 – 3 of 3) sorted by relevance

/dports/misc/mnn/MNN-1.2.0/source/backend/cuda/core/
H A DCUDABackend.cpp42 CUDARuntimeWrapper::CUDARuntimeWrapper(BackendConfig::PrecisionMode precision, BackendConfig::Power… in CUDARuntimeWrapper() function in MNN::CUDA::CUDARuntimeWrapper
58 CUDARuntimeWrapper::~CUDARuntimeWrapper() { in ~CUDARuntimeWrapper()
61 float CUDARuntimeWrapper::onGetMemoryInMB() { in onGetMemoryInMB()
66 Backend* CUDARuntimeWrapper::onCreate(const BackendConfig* config) const { in onCreate()
70 void CUDARuntimeWrapper::onGabageCollect(int level) { in onGabageCollect()
H A DCUDABackend.hpp22 class MNN_PUBLIC CUDARuntimeWrapper : public Runtime { class
24 CUDARuntimeWrapper(BackendConfig::PrecisionMode precision, BackendConfig::PowerMode power);
25 virtual ~CUDARuntimeWrapper();
/dports/misc/mnn/MNN-1.2.0/source/backend/cuda/
H A DRegister.cpp21 auto backend = new CUDARuntimeWrapper(precision, power); in onCreate()