tqchen commented on code in PR #15652:
URL: https://github.com/apache/tvm/pull/15652#discussion_r1312977545
##########
include/tvm/runtime/container/shape_tuple.h:
##########
@@ -42,6 +43,9 @@ class ShapeTupleObj : public Object {
/*! \brief The size of the shape tuple object. */
uint64_t size;
+ /*! \brief Get number of elements in the shape */
+ index_type Numel() const;
Review Comment:
trying to dig a bit about related APIs. How about Product? the product of
the shape values
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
