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

Method addInput

include/NvInfer.h:6904–6907  ·  view source on GitHub ↗

\brief Add an input tensor to the network. The name of the input tensor is used to find the index into the buffer array for an engine built from the network. The volume must be less than 2^31 elements. For networks with an implicit batch dimension, this volume includes the batch dimension with its length set to the maximum batch size. For networks with all explicit dimensions and with wildcard d

Source from the content-addressed store, hash-verified

6902 //! \return The new tensor or nullptr if there is an error.
6903 //!
6904 ITensor* addInput(char const* name, DataType type, Dims dimensions) noexcept
6905 {
6906 return mImpl->addInput(name, type, dimensions);
6907 }
6908
6909 //!
6910 //! \brief Mark a tensor as a network output.

Callers

nothing calls this directly

Calls 1

addInputMethod · 0.45

Tested by

no test coverage detected