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

Method getScales

include/NvInfer.h:5079–5082  ·  view source on GitHub ↗

\brief Copies resize scales to scales[0, ..., nbScales-1], where nbScales is the number of scales that were set. \param size The number of scales to get. If size != nbScales, no scales will be copied. \param scales Pointer to where to copy the scales. Scales will be copied only if size == nbScales and scales != nullptr. In case the size is not known consider using size = 0 and scales = nullptr.

Source from the content-addressed store, hash-verified

5077 //! Return -1 in case no scales were set or resize layer is used in dynamic mode.
5078 //!
5079 int32_t getScales(int32_t size, float* scales) const noexcept
5080 {
5081 return mImpl->getScales(size, scales);
5082 }
5083
5084 //!
5085 //! \brief Set resize mode for an input tensor.

Callers 1

pyGraph.cppFile · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected