Loading...
Searching...
No Matches
matmul_cudnn.h File Reference

Go to the source code of this file.

Classes

struct  opennn::matmul::cudnn::Problem
 
struct  opennn::matmul::cudnn::CandidateInfo
 

Namespaces

namespace  opennn
 
namespace  opennn::matmul
 
namespace  opennn::matmul::cudnn
 

Functions

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