Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LitLeo/TensorRT_Tutorial
/ functions
Functions
601 in github.com/LitLeo/TensorRT_Tutorial
⨍
Functions
601
◇
Types & classes
229
Method
addReLU
(self, layer, x, layer_name=None, precision=None)
shenlan_homework_bert/trt_helper.py:132
Function
alignTo
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:84
Method
asin
Arc sine implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1910
Method
asinh
Hyperbolic area sine implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1967
Method
atan
Arc tangent implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1926
Method
atan2
Arc tangent implementation. \param x first argument \param y second argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1935
Method
atanh
Hyperbolic area tangent implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1994
Method
attachToContext
\brief Attach the plugin object to an execution context and grant the plugin the access to some context resource. \param cudnn The cudnn context hand
resource_for_billibilli/debug_plugin/include/NvInferRuntimeCommon.h:636
Function
c
\brief Get the channel count. \return The channel count.
resource_for_billibilli/debug_plugin/include/NvInfer.h:227
Method
cast
resource_for_billibilli/debug_plugin/plugin/half.h:2622
Method
cast
resource_for_billibilli/debug_plugin/plugin/half.h:2644
Method
cast
resource_for_billibilli/debug_plugin/plugin/half.h:2666
Method
cast
resource_for_billibilli/debug_plugin/plugin/half.h:2684
Method
cast_impl
resource_for_billibilli/debug_plugin/plugin/half.h:2628
Method
cast_impl
resource_for_billibilli/debug_plugin/plugin/half.h:2650
Method
cast_impl
resource_for_billibilli/debug_plugin/plugin/half.h:2672
Method
cbrt
Cubic root implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1847
Function
ceil
Nearest integer not less than half value. \param arg half to round \return nearest integer not less than \a arg template<typename T> typename enable<h
resource_for_billibilli/debug_plugin/plugin/half.h:3485
Method
ceil
Ceiling implementation. \param arg value to round \return rounded value
resource_for_billibilli/debug_plugin/plugin/half.h:2086
Function
checkDeviceData
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:147
Method
clone
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:60
Method
clone
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:76
Method
clone
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:60
Method
clone
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:74
Method
configurePlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:111
Method
configurePlugin
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:126
Method
configurePlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:110
Method
configurePlugin
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:119
Function
convertAndCopyToDevice
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:200
Function
copysign
Take sign. \param x value to change sign for \param y value to take sign from \return value equal to \a x in magnitude and to \a y in sign template<ty
resource_for_billibilli/debug_plugin/plugin/half.h:3764
Method
copysign
Sign implementation \param x first operand \param y second operand \return composed value
resource_for_billibilli/debug_plugin/plugin/half.h:2321
Method
cos
Cosine implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1894
Method
cosh
Hyperbolic cosine implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1951
Function
createDebugDynamicPlugin
resource_for_billibilli/debug_plugin/plugin/infer_plugin_api.cc:84
Function
createDebugPlugin
resource_for_billibilli/debug_plugin/plugin/infer_plugin_api.cc:78
Function
createInferBuilder
\brief Create an instance of an IBuilder class. This class is the logging class for the builder. unnamed namespace avoids linkage surprises when lin
resource_for_billibilli/debug_plugin/include/NvInfer.h:7427
Function
createInferRefitter
\brief Create an instance of an IRefitter class. This class is the logging class for the refitter.
resource_for_billibilli/debug_plugin/include/NvInferRuntime.h:2000
Function
createInferRuntime
\brief Create an instance of an IRuntime class. This class is the logging class for the runtime.
resource_for_billibilli/debug_plugin/include/NvInferRuntime.h:1990
Function
createParser
resource_for_billibilli/debug_plugin/include/NvOnnxParser.h:238
Method
createPlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:177
Method
createPlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:176
Method
createPlugin
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:185
Function
custom_fc
(config, network, input_tensor, out_dims, W)
shenlan_homework_bert/builder.py:74
Method
defineTest
\brief Define a test for logging \param[in] name The name of the test. This should be a string starting with "TensorRT" and containing dot-separated
resource_for_billibilli/debug_plugin/plugin/logging.h:292
Function
deserNvDimsToHost
deserialize NvInfer Dims from buffer
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:127
Function
deserToDev
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:90
Function
deserToHost
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:110
Method
deserialize
resource_for_billibilli/debug_plugin/plugin/serialize.hpp:51
Method
deserialize
resource_for_billibilli/debug_plugin/plugin/serialize.hpp:68
Method
deserialize
resource_for_billibilli/debug_plugin/plugin/serialize.hpp:91
Method
deserializeCudaEngine
\brief Deserialize an engine from a stream when plugin factory is not used. \param blob The memory that holds the serialized engine. \param size The
resource_for_billibilli/debug_plugin/include/NvInferRuntime.h:815
Method
deserializePlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:183
Method
deserializePlugin
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:198
Method
deserializePlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:182
Method
deserializePlugin
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:191
Method
destroy
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:153
Method
destroy
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:168
Method
destroy
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:152
Method
destroy
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:161
Method
detachFromContext
\brief Detach the plugin object from its execution context. This function is called automatically for each plugin when a execution context is destroy
resource_for_billibilli/debug_plugin/include/NvInferRuntimeCommon.h:644
Function
dims2String
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:289
Method
divides
Division implementation. \param x first operand \param y second operand \return Half-precision quotient stored in single-precision
resource_for_billibilli/debug_plugin/plugin/half.h:1616
Method
erf
Error function implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:2006
Method
erfc
Complementary implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:2018
Method
exp
Exponential implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1767
Method
exp2
Binary exponential implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1787
Method
expm1
Exponential implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1775
Method
fabs
Absolute value implementation. \param arg function argument \return absolute value
resource_for_billibilli/debug_plugin/plugin/half.h:2524
Method
fabs
resource_for_billibilli/debug_plugin/plugin/half.h:2536
Method
fdim
Positive difference implementation. \param x first operand \param y second operand \return Positive difference stored in single-precision
resource_for_billibilli/debug_plugin/plugin/half.h:1734
Function
fieldTypeToDataType
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:256
Function
float2half
resource_for_billibilli/debug_plugin/plugin/half.h:859
Function
float2half_impl
resource_for_billibilli/debug_plugin/plugin/half.h:617
Function
floor
Nearest integer not greater than half value. \param arg half to round \return nearest integer not greater than \a arg template<typename T> typename en
resource_for_billibilli/debug_plugin/plugin/half.h:3498
Method
floor
Floor implementation. \param arg value to round \return rounded value
resource_for_billibilli/debug_plugin/plugin/half.h:2078
Method
fma
Fused multiply-add implementation. \param x first operand \param y second operand \param z third operand \return \a x * \a y + \a z stored in single-p
resource_for_billibilli/debug_plugin/plugin/half.h:1748
Method
fmax
Maximum implementation. \param x first operand \param y second operand \return maximum value
resource_for_billibilli/debug_plugin/plugin/half.h:2569
Method
fmax
resource_for_billibilli/debug_plugin/plugin/half.h:2594
Method
fmin
Minimum implementation. \param x first operand \param y second operand \return minimum value
resource_for_billibilli/debug_plugin/plugin/half.h:2552
Method
fmin
resource_for_billibilli/debug_plugin/plugin/half.h:2585
Method
fmod
Modulo implementation. \param x first operand \param y second operand \return Half-precision division remainder stored in single-precision
resource_for_billibilli/debug_plugin/plugin/half.h:1648
Method
forward
(self, xs, xs_lens)
Hackathon2022/encoder_small.py:17
Function
fpclassify
Classify floating point value. \param arg number to classify \retval FP_ZERO for positive and negative zero \retval FP_SUBNORMAL for subnormal numbers
resource_for_billibilli/debug_plugin/plugin/half.h:3793
Method
fpclassify
Classification implementation. \param arg value to classify \retval true if infinite number \retval false else
resource_for_billibilli/debug_plugin/plugin/half.h:2330
Method
free
(self)
Hackathon2022/calibrator.py:53
Method
frexp
Decompression implementation. \param arg number to decompress \param exp address to store exponent at \return normalized significant
resource_for_billibilli/debug_plugin/plugin/half.h:2153
Method
genCmdlineString
\brief generate a command line string from the given (argc, argv) values
resource_for_billibilli/debug_plugin/plugin/logging.h:422
Function
generate_calibration_cache
BERT demo needs a separate engine building path to generate calibration cache. This is because we need to configure SLN and MHA plugins in FP
shenlan_homework_bert/builder.py:346
Method
getEngineCapability
resource_for_billibilli/debug_plugin/include/NvInfer.h:7327
Method
getFieldNames
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:208
Method
getFieldNames
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:223
Method
getFieldNames
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:207
Method
getFieldNames
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:216
Method
getNbOutputs
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:66
Method
getNbOutputs
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:82
Method
getNbOutputs
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:66
Method
getNbOutputs
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:80
Method
getOutputDataType
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:105
Method
getOutputDataType
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:120
Method
getOutputDataType
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:104
← previous
next →
201–300 of 601, ranked by callers