TensorRT 10.16.0
nvinfer1::IMoELayer Member List

This is the complete list of members for nvinfer1::IMoELayer, including all inherited members.

getActivationType() const noexceptnvinfer1::IMoELayerinline
getDynQOutputScaleType() const noexceptnvinfer1::IMoELayerinline
getInput(int32_t index) const noexceptnvinfer1::ILayerinline
getMetadata() const noexceptnvinfer1::ILayerinline
getName() const noexceptnvinfer1::ILayerinline
getNbInputs() const noexceptnvinfer1::ILayerinline
getNbOutputs() const noexceptnvinfer1::ILayerinline
getNbRanks() const noexceptnvinfer1::ILayerinline
getOutput(int32_t index) const noexceptnvinfer1::ILayerinline
getOutputType(int32_t index) const noexceptnvinfer1::ILayerinline
getPrecision() const noexceptnvinfer1::ILayerinline
getQuantizationBlockShape() const noexceptnvinfer1::IMoELayerinline
getQuantizationToType() const noexceptnvinfer1::IMoELayerinline
getSwigluParamAlpha() const noexceptnvinfer1::IMoELayerinline
getSwigluParamBeta() const noexceptnvinfer1::IMoELayerinline
getSwigluParamLimit() const noexceptnvinfer1::IMoELayerinline
getType() const noexceptnvinfer1::ILayerinline
INoCopy()=defaultnvinfer1::INoCopyprotected
INoCopy(INoCopy const &other)=deletenvinfer1::INoCopyprotected
INoCopy(INoCopy &&other)=deletenvinfer1::INoCopyprotected
mImplnvinfer1::IMoELayerprotected
mLayernvinfer1::ILayerprotected
operator=(INoCopy const &other)=deletenvinfer1::INoCopyprotected
operator=(INoCopy &&other)=deletenvinfer1::INoCopyprotected
outputTypeIsSet(int32_t index) const noexceptnvinfer1::ILayerinline
precisionIsSet() const noexceptnvinfer1::ILayerinline
resetOutputType(int32_t index) noexceptnvinfer1::ILayerinline
resetPrecision() noexceptnvinfer1::ILayerinline
setActivationType(MoEActType activationType) noexceptnvinfer1::IMoELayerinline
setDynQOutputScaleType(DataType type) noexceptnvinfer1::IMoELayerinline
setGatedBiases(ITensor &fcGateBiases, ITensor &fcUpBiases, ITensor &fcDownBiases) noexceptnvinfer1::IMoELayerinline
setGatedWeights(ITensor &fcGateWeights, ITensor &fcUpWeights, ITensor &fcDownWeights, MoEActType activationType) noexceptnvinfer1::IMoELayerinline
setInput(int32_t index, ITensor &tensor) noexceptnvinfer1::IMoELayerinline
setInput(int32_t index, ITensor &tensor) noexceptnvinfer1::IMoELayerinline
setMetadata(char const *metadata) noexceptnvinfer1::ILayerinline
setName(char const *name) noexceptnvinfer1::ILayerinline
setNbRanks(int32_t nbRanks) noexceptnvinfer1::ILayerinline
setOutputType(int32_t index, DataType dataType) noexceptnvinfer1::ILayerinline
setPrecision(DataType dataType) noexceptnvinfer1::ILayerinline
setQuantizationBlockShape(Dims const &blockShape) noexceptnvinfer1::IMoELayerinline
setQuantizationDynamicDblQ(ITensor &fcDownActivationDblQScale, DataType dataType, Dims const &blockShape, DataType dynQOutputScaleType) noexceptnvinfer1::IMoELayerinline
setQuantizationStatic(ITensor &fcDownActivationScale, DataType dataType) noexceptnvinfer1::IMoELayerinline
setQuantizationToType(DataType type) noexceptnvinfer1::IMoELayerinline
setSwigluParamAlpha(float alpha) noexceptnvinfer1::IMoELayerinline
setSwigluParamBeta(float beta) noexceptnvinfer1::IMoELayerinline
setSwigluParamLimit(float limit) noexceptnvinfer1::IMoELayerinline
setSwigluParams(float limit, float alpha, float beta) noexceptnvinfer1::IMoELayerinline
~ILayer() noexcept=defaultnvinfer1::ILayerprotectedvirtual
~IMoELayer() noexcept=defaultnvinfer1::IMoELayerprotectedvirtual
~INoCopy()=defaultnvinfer1::INoCopyprotectedvirtual

  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