File tree Expand file tree Collapse file tree 2 files changed +8
-3
lines changed Expand file tree Collapse file tree 2 files changed +8
-3
lines changed Original file line number Diff line number Diff line change @@ -39,9 +39,11 @@ struct SortAttrs : public tvm::AttrsNode<SortAttrs> {
3939 TVM_ATTR_FIELD (axis).set_default (-1 ).describe (
4040 " Axis along which the sort is computed."
4141 " The default the last axis is used." );
42- TVM_ATTR_FIELD (descending).set_default (false ).describe (
43- " Whether to sort in descending order."
44- " If it is not specified, it defaults to the ascending order." );
42+ TVM_ATTR_FIELD (descending)
43+ .set_default (false )
44+ .describe (
45+ " Whether to sort in descending order."
46+ " If it is not specified, it defaults to the ascending order." );
4547 }
4648}; // struct SortAttrs
4749} // namespace relax
Original file line number Diff line number Diff line change 2626
2727#include < tvm/relax/attrs/sort.h>
2828
29+ #include < algorithm>
30+ #include < utility>
31+
2932#include " ../op_common.h"
3033
3134namespace tvm {
You can’t perform that action at this time.
0 commit comments