TensorRT  8.0.3
nvinfer1::apiv::VGatherLayer Class Referenceabstract
Inheritance diagram for nvinfer1::apiv::VGatherLayer:
nvinfer1::apiv::VRoot

Public Member Functions

virtual void setGatherAxis (int32_t axis) noexcept=0
 
virtual int32_t getGatherAxis () const noexcept=0
 
virtual void setNbElementWiseDims (int32_t k) noexcept=0
 
virtual int32_t getNbElementWiseDims () const noexcept=0
 

The documentation for this class was generated from the following file:

  Copyright © 2024 NVIDIA Corporation
  Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact