[loco] Add comment to TensorConstantPad (#8022)
author채성우/On-Device Lab(SR)/Engineer/삼성전자 <sw4670.chae@samsung.com>
Tue, 15 Oct 2019 03:38:26 +0000 (12:38 +0900)
committer박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Tue, 15 Oct 2019 03:38:26 +0000 (12:38 +0900)
* [loco] Add comment to TensorConstantPad

This commit adds comment to TensorConstantPad IR in loco.

Signed-off-by: seongwoo <sw4670.chae@samsung.com>
* make comment more clear.

* apply comment.

* modify comment.

* modify comment.

compiler/loco/include/loco/IR/Nodes.h

index e844e6e..fede8aa 100644 (file)
@@ -811,6 +811,25 @@ public:
 
 using FeatureBiasAdd = BiasAdd<Domain::Feature>;
 
+/**
+ * @brief Pads a tensor with constant value
+ *
+ * Pads a input tensor according to the padding with constant value.
+ *
+ * The dimension of each axis n of the output is
+ * output.dim(n) = padding.front(n) + input.dim(n) + padding.back(n)
+ *
+ * For example, input tensor of shape [1, 2] with
+ *
+ * padding.front(0) = 1;
+ * padding.back(0) = 2;
+ *
+ * padding.front(1) = 3;
+ * padding.back(1) = 4;
+ *
+ * will be a output tensor of shape
+ * [padding.front(0) + 1 + padding.back(0), padding.front(1) + 2 + padding.back(1)] = [4,9].
+ */
 class TensorConstantPad final
     : public CanonicalNodeDef<CanonicalOpcode::TensorConstantPad, FixedArity<2>::Mixin>
 {