ONE - On-device Neural Engine
Loading...
Searching...
No Matches
moco::onnx::IdentityGraphBuilder Class Reference

GraphBuilder for Identity node. More...

#include <Identity.h>

Collaboration diagram for moco::onnx::IdentityGraphBuilder:

Public Member Functions

bool validate (OpsetVersion, const ::onnx::NodeProto &) const
 
void build (OpsetVersion, const ::onnx::NodeProto &, GraphBuilderContext *) const
 
- Public Member Functions inherited from moco::onnx::GraphBuilder
virtual ~GraphBuilder ()
 

Additional Inherited Members

- Public Types inherited from moco::onnx::GraphBuilder
using OpsetVersion = int64_t
 

Detailed Description

GraphBuilder for Identity node.

Definition at line 39 of file Identity.h.

Member Function Documentation

◆ build()

void moco::onnx::IdentityGraphBuilder::build ( OpsetVersion  opset_version,
const ::onnx::NodeProto &  node,
GraphBuilderContext context 
) const
virtual

Implements moco::onnx::GraphBuilder.

Definition at line 34 of file Identity.cpp.

36{
37 if (opset_version >= 1)
38 Identity_V1().build(node, context);
39 else
40 throw std::runtime_error("Invalid ONNX IR version");
41}

References moco::onnx::Identity_V1::build().

◆ validate()

bool moco::onnx::IdentityGraphBuilder::validate ( OpsetVersion  opset_version,
const ::onnx::NodeProto &  node 
) const
virtual

Reimplemented from moco::onnx::GraphBuilder.

Definition at line 26 of file Identity.cpp.

27{
28 if (opset_version >= 1)
29 return Identity_V1().validate(node);
30 else
31 throw std::runtime_error("Invalid ONNX IR version");
32}

References moco::onnx::Identity_V1::validate().


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