ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Static Public Member Functions
loco::TypeInference Struct Reference

#include <TypeInference.h>

Static Public Member Functions

static bool known (const Node *)
 
static DataType get (const Node *)
 
static void erase (Node *)
 

Detailed Description

Definition at line 101 of file TypeInference.h.

Member Function Documentation

◆ erase()

void loco::TypeInference::erase ( Node *  node)
static

Definition at line 93 of file TypeInference.cpp.

93{ return node->annot<DataTypeAnnotation>(nullptr); }

References loco::AnnotatedItem< Annotation >::annot().

Referenced by loco::dtype_erase().

◆ get()

DataType loco::TypeInference::get ( const Node *  node)
static

Definition at line 87 of file TypeInference.cpp.

88{
89 assert(known(node));
90 return node->annot<DataTypeAnnotation>()->dtype();
91}
compare_tensors_all.dtype
dtype
Definition compare_tensors_all.py:162
loco::TypeInference::known
static bool known(const Node *)
Definition TypeInference.cpp:85

References loco::AnnotatedItem< Annotation >::annot(), and known().

Referenced by loco::dtype_get().

◆ known()

bool loco::TypeInference::known ( const Node *  node)
static

Definition at line 85 of file TypeInference.cpp.

85{ return node->annot<DataTypeAnnotation>() != nullptr; }

References loco::AnnotatedItem< Annotation >::annot().

Referenced by loco::dtype_known(), and get().

The documentation for this struct was generated from the following files: