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

Method setInput

include/NvInfer.h:631–634  ·  view source on GitHub ↗

\brief Replace an input of this layer with a specific tensor. \param index the index of the input to modify. \param tensor the new input tensor Except for IFillLayer, ILoopOutputLayer, INMSLayer, IResizeLayer, IShuffleLayer, and ISliceLayer, this method cannot change the number of inputs to a layer. The index argument must be less than the value of getNbInputs(). See comments for overloads of s

Source from the content-addressed store, hash-verified

629 //! See comments for overloads of setInput() for layers with special behavior.
630 //!
631 void setInput(int32_t index, ITensor& tensor) noexcept
632 {
633 return mLayer->setInput(index, tensor);
634 }
635
636 //!
637 //! \brief Set the computational precision of this layer

Callers 1

addLSTMCellMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected