ONE - On-device Neural Engine
Loading...
Searching...
No Matches
locomotiv::NodeExecution Class Reference

Helper class for Session, responsible to process one node calculation. More...

#include <NodeExecution.h>

Public Member Functions

void run (loco::Node *node)
 Run calculation for one unspecified Node.
 

Static Public Member Functions

static NodeExecutionget ()
 

Detailed Description

Helper class for Session, responsible to process one node calculation.

Definition at line 45 of file NodeExecution.h.

Member Function Documentation

◆ get()

static NodeExecution & locomotiv::NodeExecution::get ( )
inlinestatic

Definition at line 51 of file NodeExecution.h.

52 {
53 static NodeExecution me;
54 return me;
55 }

Referenced by locomotiv::Session::infer().

◆ run()

void locomotiv::NodeExecution::run ( loco::Node node)

Run calculation for one unspecified Node.

Definition at line 54 of file NodeExecution.cpp.

55{
56 erase_annot_data(node);
57
58#define NODE(Name) \
59 if (as<loco::Name>(node)) \
60 { \
61 execute(as<loco::Name>(node)); \
62 return; \
63 }
64#include "Node.lst"
65#undef NODE
66
67 throw std::runtime_error("Not supported loco::Node type");
68}

Referenced by locomotiv::Session::infer(), and package.infer.session::inference().


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