MCPcopy Create free account

hub / github.com/LitLeo/TensorRT_Tutorial / functions

Functions601 in github.com/LitLeo/TensorRT_Tutorial

MethodaddReLU
(self, layer, x, layer_name=None, precision=None)
shenlan_homework_bert/trt_helper.py:132
FunctionalignTo
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:84
Methodasin
Arc sine implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1910
Methodasinh
Hyperbolic area sine implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1967
Methodatan
Arc tangent implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1926
Methodatan2
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
Methodatanh
Hyperbolic area tangent implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1994
MethodattachToContext
\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
Functionc
\brief Get the channel count. \return The channel count.
resource_for_billibilli/debug_plugin/include/NvInfer.h:227
Methodcast
resource_for_billibilli/debug_plugin/plugin/half.h:2622
Methodcast
resource_for_billibilli/debug_plugin/plugin/half.h:2644
Methodcast
resource_for_billibilli/debug_plugin/plugin/half.h:2666
Methodcast
resource_for_billibilli/debug_plugin/plugin/half.h:2684
Methodcast_impl
resource_for_billibilli/debug_plugin/plugin/half.h:2628
Methodcast_impl
resource_for_billibilli/debug_plugin/plugin/half.h:2650
Methodcast_impl
resource_for_billibilli/debug_plugin/plugin/half.h:2672
Methodcbrt
Cubic root implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1847
Functionceil
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
Methodceil
Ceiling implementation. \param arg value to round \return rounded value
resource_for_billibilli/debug_plugin/plugin/half.h:2086
FunctioncheckDeviceData
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:147
Methodclone
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:60
Methodclone
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:76
Methodclone
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:60
Methodclone
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:74
MethodconfigurePlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:111
MethodconfigurePlugin
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:126
MethodconfigurePlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:110
MethodconfigurePlugin
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:119
FunctionconvertAndCopyToDevice
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:200
Functioncopysign
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
Methodcopysign
Sign implementation \param x first operand \param y second operand \return composed value
resource_for_billibilli/debug_plugin/plugin/half.h:2321
Methodcos
Cosine implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1894
Methodcosh
Hyperbolic cosine implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1951
FunctioncreateDebugDynamicPlugin
resource_for_billibilli/debug_plugin/plugin/infer_plugin_api.cc:84
FunctioncreateDebugPlugin
resource_for_billibilli/debug_plugin/plugin/infer_plugin_api.cc:78
FunctioncreateInferBuilder
\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
FunctioncreateInferRefitter
\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
FunctioncreateInferRuntime
\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
FunctioncreateParser
resource_for_billibilli/debug_plugin/include/NvOnnxParser.h:238
MethodcreatePlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:177
MethodcreatePlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:176
MethodcreatePlugin
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:185
Functioncustom_fc
(config, network, input_tensor, out_dims, W)
shenlan_homework_bert/builder.py:74
MethoddefineTest
\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
FunctiondeserNvDimsToHost
deserialize NvInfer Dims from buffer
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:127
FunctiondeserToDev
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:90
FunctiondeserToHost
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:110
Methoddeserialize
resource_for_billibilli/debug_plugin/plugin/serialize.hpp:51
Methoddeserialize
resource_for_billibilli/debug_plugin/plugin/serialize.hpp:68
Methoddeserialize
resource_for_billibilli/debug_plugin/plugin/serialize.hpp:91
MethoddeserializeCudaEngine
\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
MethoddeserializePlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:183
MethoddeserializePlugin
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:198
MethoddeserializePlugin
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:182
MethoddeserializePlugin
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:191
Methoddestroy
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:153
Methoddestroy
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:168
Methoddestroy
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:152
Methoddestroy
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:161
MethoddetachFromContext
\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
Functiondims2String
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:289
Methoddivides
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
Methoderf
Error function implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:2006
Methoderfc
Complementary implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:2018
Methodexp
Exponential implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1767
Methodexp2
Binary exponential implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1787
Methodexpm1
Exponential implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1775
Methodfabs
Absolute value implementation. \param arg function argument \return absolute value
resource_for_billibilli/debug_plugin/plugin/half.h:2524
Methodfabs
resource_for_billibilli/debug_plugin/plugin/half.h:2536
Methodfdim
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
FunctionfieldTypeToDataType
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:256
Functionfloat2half
resource_for_billibilli/debug_plugin/plugin/half.h:859
Functionfloat2half_impl
resource_for_billibilli/debug_plugin/plugin/half.h:617
Functionfloor
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
Methodfloor
Floor implementation. \param arg value to round \return rounded value
resource_for_billibilli/debug_plugin/plugin/half.h:2078
Methodfma
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
Methodfmax
Maximum implementation. \param x first operand \param y second operand \return maximum value
resource_for_billibilli/debug_plugin/plugin/half.h:2569
Methodfmax
resource_for_billibilli/debug_plugin/plugin/half.h:2594
Methodfmin
Minimum implementation. \param x first operand \param y second operand \return minimum value
resource_for_billibilli/debug_plugin/plugin/half.h:2552
Methodfmin
resource_for_billibilli/debug_plugin/plugin/half.h:2585
Methodfmod
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
Methodforward
(self, xs, xs_lens)
Hackathon2022/encoder_small.py:17
Functionfpclassify
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
Methodfpclassify
Classification implementation. \param arg value to classify \retval true if infinite number \retval false else
resource_for_billibilli/debug_plugin/plugin/half.h:2330
Methodfree
(self)
Hackathon2022/calibrator.py:53
Methodfrexp
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
MethodgenCmdlineString
\brief generate a command line string from the given (argc, argv) values
resource_for_billibilli/debug_plugin/plugin/logging.h:422
Functiongenerate_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
MethodgetEngineCapability
resource_for_billibilli/debug_plugin/include/NvInfer.h:7327
MethodgetFieldNames
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:208
MethodgetFieldNames
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:223
MethodgetFieldNames
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:207
MethodgetFieldNames
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:216
MethodgetNbOutputs
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:66
MethodgetNbOutputs
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:82
MethodgetNbOutputs
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:66
MethodgetNbOutputs
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:80
MethodgetOutputDataType
Hackathon2022/LayerNormPlugin/SkipLayerNormV2Plugin.h:105
MethodgetOutputDataType
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:120
MethodgetOutputDataType
Hackathon2022/LayerNormPlugin/SkipLayerNormV1Plugin.h:104
← previousnext →201–300 of 601, ranked by callers