File tree Expand file tree Collapse file tree 2 files changed +2
-1
lines changed
core/conversion/converters/impl
tests/core/conversion/converters Expand file tree Collapse file tree 2 files changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -21,6 +21,7 @@ auto topk_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().patter
2121 auto k = args[1 ].unwrapToInt ();
2222 auto dim = args[2 ].unwrapToInt ();
2323 auto largest = args[3 ].unwrapToBool ();
24+ LOG_DEBUG (" Note: sorted argument is not used in TensorRT for aten::topk, results will depend on the value of largest" );
2425 // auto sorted = args[4].unwrapToBool(); # Currently unused
2526
2627 auto selfDim = util::toVec (self->getDimensions ());
Original file line number Diff line number Diff line change @@ -102,7 +102,7 @@ test_suite(
102102 ":test_stack" ,
103103 ":test_lstm_cell" ,
104104 ":test_unsqueeze" ,
105- ":test_squeeze"
105+ ":test_squeeze" ,
106106 ":test_topk" ,
107107 ]
108108)
You can’t perform that action at this time.
0 commit comments