opennn::opennn::Backend Class Reference

#include <neural_network.h>

Public Member Functions
ThreadPoolDevice *	get_thread_pool_device ()
void	set_threads_number (int num_threads)

Static Public Member Functions
static Backend &	instance ()
static cublasHandle_t	get_cublas_handle ()
static cublasLtHandle_t	get_cublas_lt_handle ()
static cudnnHandle_t	get_cudnn_handle ()
static cudaStream_t	get_compute_stream ()
static cudnnOpTensorDescriptor_t	get_operator_sum_descriptor ()

Member Function Documentation

get_compute_stream()

static cudaStream_t opennn::opennn::Backend::get_compute_stream ( )

inlinestatic

get_cublas_handle()

static cublasHandle_t opennn::opennn::Backend::get_cublas_handle ( )

inlinestatic

get_cublas_lt_handle()

static cublasLtHandle_t opennn::opennn::Backend::get_cublas_lt_handle ( )

inlinestatic

get_cudnn_handle()

static cudnnHandle_t opennn::opennn::Backend::get_cudnn_handle ( )

inlinestatic

get_operator_sum_descriptor()

static cudnnOpTensorDescriptor_t opennn::opennn::Backend::get_operator_sum_descriptor ( )

inlinestatic

get_thread_pool_device()

ThreadPoolDevice * opennn::opennn::Backend::get_thread_pool_device

(

)

instance()

static Backend & opennn::opennn::Backend::instance ( )

static

set_threads_number()

void opennn::opennn::Backend::set_threads_number

(

int

num_threads

)