ONE - On-device Neural Engine
Loading...
Searching...
No Matches
luci::IndexTensorOutputs Class Reference

Set of Tensor Index of outputs of operators including graph input nodes. More...

#include <GraphBuilderContext.h>

Public Member Functions

void enroll (TensorIndex idx)
 
bool find (TensorIndex idx)
 

Detailed Description

Set of Tensor Index of outputs of operators including graph input nodes.

Definition at line 55 of file GraphBuilderContext.h.

Member Function Documentation

◆ enroll()

void luci::IndexTensorOutputs::enroll ( TensorIndex  idx)

Definition at line 48 of file GraphBuilderContext.cpp.

49{
50 auto iter = _set.find(idx);
51 if (iter != _set.end())
52 {
53 LOGGER(l);
54 INFO(l) << "[luci] TensorOutputs SKIP (" << idx << ") existing" << std::endl;
55 return;
56 }
57 _set.insert(idx);
58}
#define LOGGER(name)
Definition Log.h:65
#define INFO(name)
Definition Log.h:68

References INFO, and LOGGER.

◆ find()

bool luci::IndexTensorOutputs::find ( TensorIndex  idx)

Definition at line 60 of file GraphBuilderContext.cpp.

60{ return (_set.find(idx) != _set.end()); }

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