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

Method getOutputDataType

plugin/fcPlugin/fcPlugin.cpp:627–634  ·  view source on GitHub ↗

IPluginV2Ext Methods

Source from the content-addressed store, hash-verified

625
626// IPluginV2Ext Methods
627DataType FCPluginDynamic::getOutputDataType(int32_t index, DataType const* inputTypes, int32_t nbInputs) const noexcept
628{
629 PLUGIN_ASSERT(index == 0);
630 PLUGIN_ASSERT(nbInputs == 1);
631 PLUGIN_ASSERT(inputTypes != nullptr);
632 PLUGIN_ASSERT(inputTypes[0] == DataType::kFLOAT || inputTypes[0] == DataType::kHALF);
633 return inputTypes[0];
634}
635
636// IPluginV2 Methods
637char const* FCPluginDynamic::getPluginType() const noexcept

Callers 1

pyPlugin.cppFile · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected