ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Types | Public Member Functions
ann::Operation Class Reference

#include <Operation.h>

Public Types

enum class  Code { ANN_OPERATION }
 

Public Member Functions

 Operation (const Code &code, std::initializer_list< OperandID > inputs, std::initializer_list< OperandID > outputs)
 
const Code & code (void) const
 
const std::vector< OperandID > & inputs (void) const
 
const std::vector< OperandID > & outputs (void) const
 

Detailed Description

Definition at line 28 of file Operation.h.

Member Enumeration Documentation

◆ Code

enum class ann::Operation::Code
strong
Enumerator
ANN_OPERATION 

Definition at line 31 of file Operation.h.

32 {
33#define ANN_OPERATION(TAG, VALUE) TAG,
34#include "Operation.def"
35#undef ANN_OPERATION
36 };

Constructor & Destructor Documentation

◆ Operation()

ann::Operation::Operation ( const Code &  code,
std::initializer_list< OperandID >  inputs,
std::initializer_list< OperandID >  outputs 
)
inline

Definition at line 39 of file Operation.h.

41 : _code{code}, _inputs{inputs}, _outputs{outputs}
42 {
43 // DO NOTHING
44 }
ann::Operation::inputs
const std::vector< OperandID > & inputs(void) const
Definition Operation.h:48
ann::Operation::code
const Code & code(void) const
Definition Operation.h:47
ann::Operation::outputs
const std::vector< OperandID > & outputs(void) const
Definition Operation.h:49

Member Function Documentation

◆ code()

const Code & ann::Operation::code ( void  ) const
inline

Definition at line 47 of file Operation.h.

47{ return _code; }

◆ inputs()

const std::vector< OperandID > & ann::Operation::inputs ( void  ) const
inline

Definition at line 48 of file Operation.h.

48{ return _inputs; }

Referenced by package.experimental.train.dataloader.DataLoader::__init__(), validate_onnx2circle.OnnxRunner::feed_random_inputs(), package.common.basesession.BaseSession::set_inputs(), and package.experimental.train.dataloader.DataLoader::split().

◆ outputs()

const std::vector< OperandID > & ann::Operation::outputs ( void  ) const
inline

Definition at line 49 of file Operation.h.

49{ return _outputs; }

Referenced by package.common.basesession.BaseSession::_set_outputs(), and validate_onnx2circle.OnnxRunner::get_outputs().

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