ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
moco::onnx::ConstantGraphBuilder Class Reference

GraphBuilder for Constant node. More...

#include <Constant.h>

Collaboration diagram for moco::onnx::ConstantGraphBuilder:

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 Constant node.

Definition at line 51 of file Constant.h.

Member Function Documentation

◆ build()

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

Implements moco::onnx::GraphBuilder.

Definition at line 36 of file Constant.cpp.

38{
39 if (opset_version >= 9)
40 Constant_V9().build(node, context);
41 else if (opset_version >= 1)
42 Constant_V1().build(node, context);
43 else
44 throw std::runtime_error("Invalid ONNX IR version");
45}

References moco::onnx::Constant_V1::build(), and moco::onnx::Constant_V9::build().

◆ validate()

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

Reimplemented from moco::onnx::GraphBuilder.

Definition at line 26 of file Constant.cpp.

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

References moco::onnx::Constant_V1::validate(), and moco::onnx::Constant_V9::validate().


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