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

Method addScatter

include/NvInfer.h:8096–8099  ·  view source on GitHub ↗

\brief Add a Scatter layer to the network with specified mode and axis=0. \param data The input tensor to be updated with additional values. \param indices indices of the elements to be updated. \param updates values to be used for updates. \param mode scatter mode. \see IScatterLayer \p indices tensor data type must be DataType::kINT32. \p updates tensor data type must be the same as \p data

Source from the content-addressed store, hash-verified

8094 //! \return The new Scatter layer, or nullptr if it could not be created.
8095 //!
8096 IScatterLayer* addScatter(ITensor& data, ITensor& indices, ITensor& updates, ScatterMode mode) noexcept
8097 {
8098 return mImpl->addScatter(data, indices, updates, mode);
8099 }
8100
8101 //!
8102 //! \brief Add a quantization layer to the network.

Callers 1

pyGraph.cppFile · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected