Loading...
Searching...
No Matches
opennn::matmul::cudnn Namespace Reference

Classes

struct  CandidateInfo
 
struct  Problem
 

Functions

Plan * create (const Problem &, int64_t plan_index=-1) noexcept
 
void destroy (Plan *) noexcept
 
int candidate_count (const Plan *) noexcept
 
CandidateInfo candidate (const Plan *, int) noexcept
 
bool run (Plan *, int candidate, const void *a, const void *b, const void *bias, void *d, void *workspace) noexcept
 
bool verbose () noexcept
 

Function Documentation

◆ candidate()

CandidateInfo opennn::matmul::cudnn::candidate ( const Plan * ,
int  )
noexcept

◆ candidate_count()

int opennn::matmul::cudnn::candidate_count ( const Plan * )
noexcept

◆ create()

Plan * opennn::matmul::cudnn::create ( const Problem & ,
int64_t plan_index = -1 )
noexcept

◆ destroy()

void opennn::matmul::cudnn::destroy ( Plan * )
noexcept

◆ run()

bool opennn::matmul::cudnn::run ( Plan * ,
int candidate,
const void * a,
const void * b,
const void * bias,
void * d,
void * workspace )
noexcept

◆ verbose()

bool opennn::matmul::cudnn::verbose ( )
noexcept