Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
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
224
template <typename T>
225
DEVICE_FUNCTION void stg(T* gmem, int32_t (&src)[1])
226
{
227
reinterpret_cast<int32_t*>(gmem)[0] = src[0];
228
}
229
230
template <typename T>
231
DEVICE_FUNCTION void stgStream(T* gmem, int32_t (&src)[1])
Callers
1
stgStream
Function · 0.85
Calls
1
fromFloat
Function · 0.85
Tested by
no test coverage detected