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

Method addQuantize

include/NvInfer.h:8115–8118  ·  view source on GitHub ↗

\brief Add a quantization layer to the network. \param input The input tensor to be quantized. \param scale A tensor with the scale value. \see IQuantizeLayer \p input tensor data type must be DataType::kFLOAT. \p scale tensor data type must be DataType::kFLOAT. The subgraph which terminates with the \p scale tensor must be a build-time constant. \return The new quantization layer, or nullptr

Source from the content-addressed store, hash-verified

8113 //! \return The new quantization layer, or nullptr if it could not be created.
8114 //!
8115 IQuantizeLayer* addQuantize(ITensor& input, ITensor& scale) noexcept
8116 {
8117 return mImpl->addQuantize(input, scale);
8118 }
8119
8120 //!
8121 //! \brief Add an If-conditional layer to the network.

Callers 1

pyGraph.cppFile · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected