masahi commented on a change in pull request #7391:
URL: https://github.com/apache/tvm/pull/7391#discussion_r572703773
##########
File path: include/tvm/relay/attrs/transform.h
##########
@@ -442,9 +442,13 @@ struct MatrixSetDiagAttrs : public
tvm::AttrsNode<MatrixSetDiagAttrs> {
struct CumsumAttrs : public tvm::AttrsNode<CumsumAttrs> {
Integer axis;
DataType dtype;
+ Integer exclusive;
TVM_DECLARE_ATTRS(CumsumAttrs, "relay.attrs.CumsumAttrs") {
TVM_ATTR_FIELD(axis).describe("The axis to sum
over").set_default(NullValue<Integer>());
TVM_ATTR_FIELD(dtype).describe("Output data
type").set_default(NullValue<DataType>());
+ TVM_ATTR_FIELD(exclusive)
+ .describe("The top element is not included")
Review comment:
top -> first, since we don't support reverse and "top" always means
"first".
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
[email protected]
