Loading...
Searching...
No Matches
opennn::cudnn_frontend::GraphSlot Struct Reference

#include <opennn/core/cuda/cudnn_frontend_utilities.h>

Public Member Functions

 operator bool () const noexcept
 
graph::Graph & operator* () const noexcept
 
void build (shared_ptr< graph::Graph > built, const string &tag, const bool fp32_graph=false)
 
void build_attention (shared_ptr< graph::Graph > built, const string &tag, bool allow_autotune=true, int64_t workspace_cap=0)
 

Public Attributes

shared_ptr< graph::Graph > graph
 
int64_t workspace_bytes = 0
 
bool autotune_pending = false
 

Member Function Documentation

◆ build()

void opennn::cudnn_frontend::GraphSlot::build ( shared_ptr< graph::Graph > built,
const string & tag,
const bool fp32_graph = false )
inline

◆ build_attention()

void opennn::cudnn_frontend::GraphSlot::build_attention ( shared_ptr< graph::Graph > built,
const string & tag,
bool allow_autotune = true,
int64_t workspace_cap = 0 )
inline

◆ operator bool()

opennn::cudnn_frontend::GraphSlot::operator bool ( ) const
inlineexplicitnoexcept

◆ operator*()

graph::Graph & opennn::cudnn_frontend::GraphSlot::operator* ( ) const
inlinenoexcept

Member Data Documentation

◆ autotune_pending

bool opennn::cudnn_frontend::GraphSlot::autotune_pending = false

◆ graph

shared_ptr<graph::Graph> opennn::cudnn_frontend::GraphSlot::graph

◆ workspace_bytes

int64_t opennn::cudnn_frontend::GraphSlot::workspace_bytes = 0