nvinfer1::apiv::VTopKLayer Class Referenceabstract
Inheritance diagram for nvinfer1::apiv::VTopKLayer:
Public Member Functions | |
| virtual void | setOperation (TopKOperation op) noexcept=0 |
| virtual TopKOperation | getOperation () const noexcept=0 |
| virtual void | setK (int32_t k) noexcept=0 |
| virtual int32_t | getK () const noexcept=0 |
| virtual void | setReduceAxes (uint32_t reduceAxes) noexcept=0 |
| virtual uint32_t | getReduceAxes () const noexcept=0 |
The documentation for this class was generated from the following file:
