moco::onnx::Constant_V9 Class Reference

GraphBuilder for Constant(since version 9) node. More...

#include <Constant.h>

Public Member Functions
bool	validate (const ::onnx::NodeProto &) const
void	build (const ::onnx::NodeProto &, GraphBuilderContext *) const

Detailed Description

GraphBuilder for Constant(since version 9) node.

Note

Until version 1, only FLOAT16, FLOAT, DOUBLE was supported Since version 9, all types are supported

Definition at line 41 of file Constant.h.

Member Function Documentation

build()

void moco::onnx::Constant_V9::build	(	const ::onnx::NodeProto &	node,
		GraphBuilderContext *	context
	)		const

Definition at line 35 of file Constant_V9.cpp.

{
  assert(context != nullptr);
 
  loco::Graph *graph = context->graph();
  SymbolTable *nodes = context->nodes();
 
  // Create a "ConstGen" node for Constant
  auto const_node = graph->nodes()->create<loco::ConstGen>();
  auto tensor_attribute = node.attribute().Get(0).t();
  const_node->dtype(as_loco_datatype(tensor_attribute.data_type()));
  const_node->rank(tensor_attribute.dims_size());
  // TODO Support other data types
  assert(const_node->dtype() == loco::DataType::FLOAT32);
  const_node->size<loco::DataType::FLOAT32>(tensor_attribute.float_data_size());
 
  for (uint32_t i = 0; i < const_node->rank(); ++i)
  {
    const_node->dim(i) = tensor_attribute.dims(i);
  }
 
  // TODO Support other data types
  for (int i = 0; i < tensor_attribute.float_data_size(); ++i)
  {
    const_node->at<loco::DataType::FLOAT32>(i) = tensor_attribute.float_data(i);
  }
 
  nodes->enroll(node.name(), const_node);
  nodes->enroll(node.output(0), const_node);
}

References moco::onnx::as_loco_datatype(), loco::ConstGen::at(), moco::onnx::SymbolTable::enroll(), moco::onnx::GraphBuilderContext::graph(), and moco::onnx::GraphBuilderContext::nodes().

Referenced by moco::onnx::ConstantGraphBuilder::build().

validate()

bool moco::onnx::Constant_V9::validate

(

const ::onnx::NodeProto &

node

)

const

Definition at line 27 of file Constant_V9.cpp.

{
  if (node.attribute_size() == 0 || !node.attribute(0).has_t())
    return false;
 
  return true;
}

Referenced by moco::onnx::ConstantGraphBuilder::validate().

---

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

• compiler/oneco/src/Op/Constant.h
• compiler/oneco/src/Op/Constant_V9.cpp