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

Method setDimensions

include/NvInfer.h:222–225  ·  view source on GitHub ↗

\brief Set the dimensions of a tensor. For a network input, the dimensions are assigned by the application. For a network output, the dimensions are computed based on the layer parameters and the inputs to the layer. If a tensor size or a parameter is modified in the network, the dimensions of all dependent tensors will be recomputed. This call is only legal for network input tensors, since the

Source from the content-addressed store, hash-verified

220 //! \see getDimensions()
221 //!
222 void setDimensions(Dims dimensions) noexcept
223 {
224 mImpl->setDimensions(dimensions);
225 }
226
227 //!
228 //! \brief Get the dimensions of a tensor.

Callers 7

pyCore.cppFile · 0.45
buildPredictionEngineMethod · 0.45
setupNetworkAndConfigFunction · 0.45
setDimensionsMethod · 0.45
setDimensionsMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected