File tree Expand file tree Collapse file tree 3 files changed +7
-2
lines changed
core/conversion/converters/impl
tests/core/conversion/converters Expand file tree Collapse file tree 3 files changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -21,8 +21,8 @@ auto topk_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().patter
21
21
auto k = args[1 ].unwrapToInt ();
22
22
auto dim = args[2 ].unwrapToInt ();
23
23
auto largest = args[3 ].unwrapToBool ();
24
- auto sorted = args[4 ].unwrapToBool ();
25
-
24
+ // auto sorted = args[4].unwrapToBool(); # Currently unused
25
+
26
26
auto selfDim = util::toVec (self->getDimensions ());
27
27
28
28
// reduceAxes The reduction dimensions. The bit in position i of bitmask reduceAxes corresponds to explicit dimension i of the result.
Original file line number Diff line number Diff line change @@ -67,6 +67,10 @@ converter_test(
67
67
name = "test_stack"
68
68
)
69
69
70
+ converter_test (
71
+ name = "test_topk"
72
+ )
73
+
70
74
converter_test (
71
75
name = "test_lstm_cell"
72
76
)
@@ -99,5 +103,6 @@ test_suite(
99
103
":test_lstm_cell" ,
100
104
":test_unsqueeze" ,
101
105
":test_squeeze"
106
+ ":test_topk" ,
102
107
]
103
108
)
File renamed without changes.
You can’t perform that action at this time.
0 commit comments