From: 이한종/동작제어Lab(SR)/Engineer/삼성전자 Date: Wed, 8 Aug 2018 22:16:39 +0000 (+0900) Subject: [neurun] Graph : Introduce operation Conv2D (#2222) X-Git-Tag: 0.2~314 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=f5cb257c2b851db73339d94da127f4cbe55d6184;p=platform%2Fcore%2Fml%2Fnnfw.git [neurun] Graph : Introduce operation Conv2D (#2222) Introduce `neurun::graph::operation::Conv2D` which is a node in Graph IR. Signed-off-by: Hanjoung Lee --- diff --git a/runtimes/neurun/src/graph/operation/Conv2D.cc b/runtimes/neurun/src/graph/operation/Conv2D.cc new file mode 100644 index 0000000..5f17931 --- /dev/null +++ b/runtimes/neurun/src/graph/operation/Conv2D.cc @@ -0,0 +1,20 @@ +#include "Conv2D.h" + +namespace neurun +{ +namespace graph +{ +namespace operation +{ +namespace Conv2D +{ +namespace Implicit +{ + +// NO IMPLEMENTATION YET + +} // namespace Implicit +} // namespace Conv2D +} // namespace operation +} // namespace graph +} // namespace neurun diff --git a/runtimes/neurun/src/graph/operation/Conv2D.h b/runtimes/neurun/src/graph/operation/Conv2D.h new file mode 100644 index 0000000..22d0d99 --- /dev/null +++ b/runtimes/neurun/src/graph/operation/Conv2D.h @@ -0,0 +1,43 @@ +#ifndef __NEURUN_GRAPH_OPERATION_CONV2D_H__ +#define __NEURUN_GRAPH_OPERATION_CONV2D_H__ + +#include + +#include "graph/operation/Node.h" +#include "internal/op/Conv2D.h" + +namespace neurun +{ +namespace graph +{ +namespace operation +{ +namespace Conv2D +{ +namespace Implicit +{ + +class Node : public graph::operation::Node +{ +public: + Node(std::unique_ptr<::internal::tflite::op::Conv2D::implicit::Node> &&op) : _op{std::move(op)} {} + +public: + virtual operand::IndexSet inputs() const override + { + return {_op->param().ifm_index, _op->param().ker_index, _op->param().bias_index}; + } + virtual operand::IndexSet outputs() const override { return {_op->param().ofm_index}; } + virtual const ::internal::tflite::op::Node *op() const override { return _op.get(); } + +private: + std::unique_ptr<::internal::tflite::op::Conv2D::implicit::Node> _op; +}; + +} // namespace Implicit +} // namespace Conv2D +} // namespace coperation +} // namespace graph +} // namespace neurun + +#endif // __NEURUN_GRAPH_OPERATION_CONV2D_H__