From 797b5ea3454e4973853de47d0083d0cf07723f0e Mon Sep 17 00:00:00 2001 From: Siju Samuel Date: Thu, 25 Oct 2018 08:07:32 +0530 Subject: [PATCH] Review comment, added docstring --- include/tvm/relay/type.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/include/tvm/relay/type.h b/include/tvm/relay/type.h index 6612cfaea88a9..e2a0c2a2a7ed4 100644 --- a/include/tvm/relay/type.h +++ b/include/tvm/relay/type.h @@ -82,7 +82,9 @@ class TensorTypeNode : public BaseTensorTypeNode { v->Visit("span", &span); } - /*! \brief Return product of elements in the shape */ + /*! \brief Return product of elements in the shape. + * \return (d1 * d_2 ... * d_n) if shape is (d_1, d_2, ..., d_n) and 1 if shape size is zero. + */ TVM_DLL IndexExpr Size() const; TVM_DLL static TensorType make(Array shape, DataType dtype);