[RELAY] Fix reshape header file (apache#7218)

The header file definition of InferNewShape was incorrect, this patch fixes it. Change-Id: Id24b8eccb52323692fe88bdda46cc49cba54588c
cjia4 · Jan 6, 2021 · 1f931e3 · 1f931e3
1 parent 040afb0
commit 1f931e3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/relay/op/tensor/transform.h b/src/relay/op/tensor/transform.h
@@ -193,9 +193,11 @@ static inline Array<Array<Layout>> ConcatenateLayout(const Attrs& attrs,
  *
  * \param data_shape The input data shape.
  * \param attrs The attributes.
+ * \param reverse Whether to reverse the indices.
  * \return Output shape.
  */
-Array<IndexExpr> InferNewShape(const Array<IndexExpr>& data_shape, const Attrs& attrs);
+Array<IndexExpr> InferNewShape(const Array<IndexExpr>& data_shape, const Attrs& attrs,
+                               bool reverse);
 
 }  // namespace relay
 }  // namespace tvm