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

Function stg

plugin/instanceNormalizationPlugin/instanceNormCommon.h:225–228  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

223
224template <typename T>
225DEVICE_FUNCTION void stg(T* gmem, int32_t (&src)[1])
226{
227 reinterpret_cast<int32_t*>(gmem)[0] = src[0];
228}
229
230template <typename T>
231DEVICE_FUNCTION void stgStream(T* gmem, int32_t (&src)[1])

Callers 1

stgStreamFunction · 0.85

Calls 1

fromFloatFunction · 0.85

Tested by

no test coverage detected