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

Get the shape of each node as a node annotation. More...

#include <ShapeInference.h>

Static Public Member Functions

static ShapeDescription get (loco::Node *node)
 

Detailed Description

Get the shape of each node as a node annotation.

HOW TO USE

ShapeInference::get(g->nodes()->at(..));

Definition at line 34 of file ShapeInference.h.

Member Function Documentation

◆ get()

ShapeDescription exo::ShapeInference::get ( loco::Node *  node)
static

Definition at line 35 of file ShapeInference.cpp.

36{
37 // TODO Adjust indentation level
38 {
39 assert(loco::shape_known(node));
40 return to_shape_description(loco::shape_get(node));
41 }
42}
exo::to_shape_description
ShapeDescription to_shape_description(const loco::TensorShape &shape)
Definition ExporterUtils.cpp:26
loco::shape_known
bool shape_known(const Node *node)
Definition ShapeInference.h:60
loco::shape_get
NodeShape shape_get(const Node *node)
Definition ShapeInference.h:61

References loco::shape_get(), loco::shape_known(), and exo::to_shape_description().

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