ONE - On-device Neural Engine
•All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
Static Public Member Functions
loco::ShapeInference Struct Reference

#include <ShapeInference.h>

Static Public Member Functions

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

Detailed Description

Definition at line 53 of file ShapeInference.h.

Member Function Documentation

◆ erase()

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

Definition at line 102 of file ShapeInference.cpp.

102{ node->annot<ShapeAnnotation>(nullptr); }

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

Referenced by loco::shape_erase().

◆ get()

NodeShape loco::ShapeInference::get ( const Node *  node)
static

Definition at line 96 of file ShapeInference.cpp.

97{
98 assert(known(node));
99 return node->annot<ShapeAnnotation>()->shape();
100}
loco::ShapeInference::known
static bool known(const Node *)
Definition ShapeInference.cpp:94

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

Referenced by loco::shape_get().

◆ known()

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

Definition at line 94 of file ShapeInference.cpp.

94{ return node->annot<ShapeAnnotation>() != nullptr; }

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

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

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