MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / setWeights

Method setWeights

include/NvInfer.h:4754–4757  ·  view source on GitHub ↗

\brief Set the weights for the layer. If weights.type is DataType::kINT32, the output is a tensor of 32-bit indices. Otherwise the output is a tensor of real values and the output type will be follow TensorRT's normal precision rules. \see getWeights()

Source from the content-addressed store, hash-verified

4752 //! \see getWeights()
4753 //!
4754 void setWeights(Weights weights) noexcept
4755 {
4756 mImpl->setWeights(weights);
4757 }
4758
4759 //!
4760 //! \brief Get the weights for the layer.

Callers 2

timeRefitFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected