MCPcopy Create free account

hub / github.com/LitLeo/TensorRT_Tutorial / functions

Functions601 in github.com/LitLeo/TensorRT_Tutorial

↓ 13 callersMethodlog
\brief Implementation of the nvinfer1::ILogger::log() virtual method Note samples should not be calling this function directly; it will eventually go
resource_for_billibilli/debug_plugin/plugin/logging.h:239
↓ 11 callersFunctionbuiltin_isnan
resource_for_billibilli/debug_plugin/plugin/half.h:581
↓ 9 callersFunctionbuiltin_signbit
resource_for_billibilli/debug_plugin/plugin/half.h:598
↓ 9 callersFunctionlog
Natural logorithm. \param arg function argument \return logarithm of \a arg to base e template<typename T> typename enable<expr,T>::type log(T arg) {
resource_for_billibilli/debug_plugin/plugin/half.h:3110
↓ 8 callersMethodaddLinear
(self, x, weight, bias, layer_name=None, precision=None)
shenlan_homework_bert/trt_helper.py:127
↓ 7 callersMethodLogger
resource_for_billibilli/debug_plugin/plugin/logging.h:204
↓ 7 callersFunctionbuiltin_isinf
resource_for_billibilli/debug_plugin/plugin/half.h:564
↓ 7 callersFunctionfabs
Absolute value. \param arg operand \return absolute value of \a arg template<typename T> typename enable<T,T>::type fabs(T arg) { return unary_special
resource_for_billibilli/debug_plugin/plugin/half.h:2870
↓ 6 callersFunctionexp
Exponential function. \param arg function argument \return e raised to \a arg template<typename T> typename enable<expr,T>::type exp(T arg) { return f
resource_for_billibilli/debug_plugin/plugin/half.h:3071
↓ 6 callersFunctionisnan
Check for NaN. \param arg number to check \retval true for NaNs \retval false else template<typename T> typename enable<bool,T>::type isnan(T arg) { r
resource_for_billibilli/debug_plugin/plugin/half.h:3835
↓ 6 callersMethodlayer_post_process
Tool function. set precision, set_layer_name and check_trt_layer
shenlan_homework_bert/trt_helper.py:53
↓ 6 callersFunctionldexp
Multiply by power of two. \param arg number to modify \param exp power of two to multiply with \return \a arg multplied by 2 raised to \a exp template
resource_for_billibilli/debug_plugin/plugin/half.h:3636
↓ 6 callersMethodsetReportableSeverity
\brief Method for controlling the verbosity of logging output \param severity The logger will only emit messages that have severity of this level or
resource_for_billibilli/debug_plugin/plugin/logging.h:249
↓ 5 callersMethodgetReportableSeverity
resource_for_billibilli/debug_plugin/plugin/logging.h:365
↓ 5 callersFunctionlgamma
Natural logarithm of gamma function. \param arg function argument \return natural logarith of gamma function for \a arg template<typename T> typename
resource_for_billibilli/debug_plugin/plugin/half.h:3455
↓ 5 callersFunctionsqrt
Square root. \param arg function argument \return square root of \a arg template<typename T> typename enable<expr,T>::type sqrt(T arg) { return functi
resource_for_billibilli/debug_plugin/plugin/half.h:3166
↓ 4 callersFunctionabs
Absolute value. \param arg operand \return absolute value of \a arg template<typename T> typename enable<T,T>::type abs(T arg) { return unary_speciali
resource_for_billibilli/debug_plugin/plugin/half.h:2857
↓ 4 callersMethodaddAdd
(self, a, b, layer_name=None, precision=None)
shenlan_homework_bert/trt_helper.py:161
↓ 4 callersMethodaddLayerNorm
(self, x, gamma, beta, layer_name=None, precision=None)
shenlan_homework_bert/trt_helper.py:122
↓ 4 callersMethodaddShuffle
( self, x: trt.ITensor, first_transpose: trt.Permutation, reshape_dims: trt.Di
shenlan_homework_bert/trt_helper.py:195
↓ 4 callersFunctionfmod
Remainder of division. \param x first operand \param y second operand \return remainder of floating point division. template<typename T,typename U> ty
resource_for_billibilli/debug_plugin/plugin/half.h:2884
↓ 4 callersMethodget_batch
(self, names)
Hackathon2022/calibrator.py:62
↓ 4 callersMethodinfer
(self, inputs: list)
shenlan_homework_bert/trt_helper.py:236
↓ 4 callersFunctionscalbln
Multiply by power of two. \param arg number to modify \param exp power of two to multiply with \return \a arg multplied by 2 raised to \a exp template
resource_for_billibilli/debug_plugin/plugin/half.h:3682
↓ 3 callersMethodPluginField
resource_for_billibilli/debug_plugin/include/NvInferRuntimeCommon.h:836
↓ 3 callersMethodaddEmbedding
(self, indices, weight, layer_name=None, precision=None)
shenlan_homework_bert/trt_helper.py:81
↓ 3 callersMethodaddInput
(self, name, dtype, shape)
shenlan_homework_bert/trt_helper.py:63
↓ 3 callersFunctionerf
Error function. \param arg function argument \return error function value of \a arg template<typename T> typename enable<expr,T>::type erf(T arg) { re
resource_for_billibilli/debug_plugin/plugin/half.h:3429
↓ 3 callersMethodget_batch
(self, names)
shenlan_homework_bert/calibrator.py:72
↓ 3 callersFunctionmodf
Extract integer and fractional parts. \param arg number to decompress \param iptr address to store integer part at \return fractional part template<ty
resource_for_billibilli/debug_plugin/plugin/half.h:3651
↓ 3 callersFunctionpow
Power function. \param base first argument \param exp second argument \return \a base raised to \a exp template<typename T,typename U> typename enable
resource_for_billibilli/debug_plugin/plugin/half.h:3217
↓ 3 callersFunctionsin
Sine function. \param arg function argument \return sine value of \a arg template<typename T> typename enable<expr,T>::type sin(T arg) { return functi
resource_for_billibilli/debug_plugin/plugin/half.h:3242
↓ 3 callersMethodsqrt
Square root implementation. \param arg function argument \return function value stored in single-preicision
resource_for_billibilli/debug_plugin/plugin/half.h:1839
↓ 2 callersMethodaddGELU
(self, x, layer_name=None, precision=None)
shenlan_homework_bert/trt_helper.py:95
↓ 2 callersMethodaddMatMul
(self, a: trt.ITensor, b: trt.ITensor, layer_name: Optional[str] = None)
shenlan_homework_bert/trt_helper.py:178
↓ 2 callersFunctionbuild_engine
(workspace_size, config, weights_dict, vocab_file, calibrationCacheFile, calib_num)
shenlan_homework_bert/builder.py:293
↓ 2 callersFunctioncreateDebugPluginInernal
Debug************************************/
resource_for_billibilli/debug_plugin/plugin/infer_plugin_api.cc:66
↓ 2 callersFunctioninitArray
cublas&cudnn_int8_demo/cublasGemmEx/gemmInt8_rect.cpp:17
↓ 2 callersFunctionrint
Nearest integer using half's internal rounding mode. \param arg half expression to round \return nearest integer using default rounding mode template<
resource_for_billibilli/debug_plugin/plugin/half.h:3563
↓ 2 callersMethodserialize
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/LayerNormPlugin.h:69
↓ 1 callersFunctionDimsCHW
\brief Construct a DimsCHW given channel count, height and width. \param channels The channel count. \param height The height of the data. \param wid
resource_for_billibilli/debug_plugin/include/NvInfer.h:205
↓ 1 callersFunctionDimsNCHW
\brief Construct a DimsNCHW given batch size, channel count, height and width. \param batchSize The batch size (commonly denoted N). \param channels
resource_for_billibilli/debug_plugin/include/NvInfer.h:335
↓ 1 callersFunctionLOG_ERROR
\brief produces a LogStreamConsumer object that can be used to log messages of severity kERROR Example usage: LOG_ERROR(logger) << "hello world" <<
resource_for_billibilli/debug_plugin/plugin/logging.h:483
↓ 1 callersFunctionLOG_FATAL
\brief produces a LogStreamConsumer object that can be used to log messages of severity kINTERNAL_ERROR ("fatal" severity) Example usage: LOG_FATAL(
resource_for_billibilli/debug_plugin/plugin/logging.h:496
↓ 1 callersFunctionLOG_INFO
\brief produces a LogStreamConsumer object that can be used to log messages of severity kINFO Example usage: LOG_INFO(logger) << "hello world" << st
resource_for_billibilli/debug_plugin/plugin/logging.h:459
↓ 1 callersFunctionLOG_VERBOSE
\brief produces a LogStreamConsumer object that can be used to log messages of severity kVERBOSE Example usage: LOG_VERBOSE(logger) << "hello world"
resource_for_billibilli/debug_plugin/plugin/logging.h:447
↓ 1 callersFunctionLOG_WARN
\brief produces a LogStreamConsumer object that can be used to log messages of severity kWARNING Example usage: LOG_WARN(logger) << "hello world" <<
resource_for_billibilli/debug_plugin/plugin/logging.h:471
↓ 1 callersFunctionacos
Arc cosine function. \param arg function argument \return arc cosine value of \a arg template<typename T> typename enable<expr,T>::type acos(T arg) {
resource_for_billibilli/debug_plugin/plugin/half.h:3294
↓ 1 callersFunctionacosh
Hyperbolic area cosine. \param arg function argument \return area cosine value of \a arg template<typename T> typename enable<expr,T>::type acosh(T ar
resource_for_billibilli/debug_plugin/plugin/half.h:3399
↓ 1 callersMethodaddScale
scale
shenlan_homework_bert/trt_helper.py:166
↓ 1 callersMethodaddSoftmax
(self, x: trt.ITensor, dim: int = -1, layer_name=None, precision=None)
shenlan_homework_bert/trt_helper.py:143
↓ 1 callersFunctionasin
Arc sine. \param arg function argument \return arc sine value of \a arg template<typename T> typename enable<expr,T>::type asin(T arg) { return functi
resource_for_billibilli/debug_plugin/plugin/half.h:3281
↓ 1 callersFunctionasinh
Hyperbolic area sine. \param arg function argument \return area sine value of \a arg template<typename T> typename enable<expr,T>::type asinh(T arg) {
resource_for_billibilli/debug_plugin/plugin/half.h:3386
↓ 1 callersFunctionatan
Arc tangent function. \param arg function argument \return arc tangent value of \a arg template<typename T> typename enable<expr,T>::type atan(T arg)
resource_for_billibilli/debug_plugin/plugin/half.h:3307
↓ 1 callersFunctionatan2
Arc tangent function. \param x first argument \param y second argument \return arc tangent value template<typename T,typename U> typename enable<expr,
resource_for_billibilli/debug_plugin/plugin/half.h:3322
↓ 1 callersFunctionatanh
Hyperbolic area tangent. \param arg function argument \return area tangent value of \a arg template<typename T> typename enable<expr,T>::type atanh(T
resource_for_billibilli/debug_plugin/plugin/half.h:3412
↓ 1 callersFunctionattention_layer
(network_helper, prefix, config, weights_dict, input_tensor, imask)
shenlan_homework_bert/builder.py:130
↓ 1 callersFunctionbert_model
Create the bert model
shenlan_homework_bert/builder.py:190
↓ 1 callersFunctioncbrt
Cubic root. \param arg function argument \return cubic root of \a arg template<typename T> typename enable<expr,T>::type cbrt(T arg) { return function
resource_for_billibilli/debug_plugin/plugin/half.h:3179
↓ 1 callersFunctionceildiv
resource_for_billibilli/debug_plugin/plugin/plugin_common.h:79
↓ 1 callersFunctioncheck
(a, b, weak = False)
Hackathon2022/LayerNormPlugin/testLayerNormPlugin.py:31
↓ 1 callersFunctioncheck
(a, b, weak = False)
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/testLayerNormPlugin.py:31
↓ 1 callersMethodcheck_trt_layer
Tool function. check trt layer,
shenlan_homework_bert/trt_helper.py:39
↓ 1 callersFunctioncos
Cosine function. \param arg function argument \return cosine value of \a arg template<typename T> typename enable<expr,T>::type cos(T arg) { return fu
resource_for_billibilli/debug_plugin/plugin/half.h:3255
↓ 1 callersFunctioncosh
Hyperbolic cosine. \param arg function argument \return hyperbolic cosine value of \a arg template<typename T> typename enable<expr,T>::type cosh(T ar
resource_for_billibilli/debug_plugin/plugin/half.h:3360
↓ 1 callersMethodcreatePlugin
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:192
↓ 1 callersFunctioncreatePluginInternal
resource_for_billibilli/debug_plugin/plugin/infer_plugin_api.cc:32
↓ 1 callersFunctiondecoder_surgeon
(src_onnx, dst_onnx)
Hackathon2022/decoder2trt.py:14
↓ 1 callersFunctiondeserialize_value
resource_for_billibilli/debug_plugin/plugin/serialize.hpp:116
↓ 1 callersFunctionemb_layernorm
(network_helper, config, weights_dict, builder_config, sequence_lengths, batch_sizes)
shenlan_homework_bert/builder.py:269
↓ 1 callersFunctionencoder_surgeon
(src_onnx, dst_onnx)
Hackathon2022/encoder2trt.py:136
↓ 1 callersFunctionencoder_surgeon_layer_norm
(graph)
Hackathon2022/encoder2trt.py:16
↓ 1 callersFunctionencoder_surgeon_skip_layer_norm
(graph)
Hackathon2022/encoder2trt.py:82
↓ 1 callersFunctionerfc
Complementary error function. \param arg function argument \return 1 minus error function value of \a arg template<typename T> typename enable<expr,T>
resource_for_billibilli/debug_plugin/plugin/half.h:3442
↓ 1 callersFunctionexp2
Binary exponential. \param arg function argument \return 2 raised to \a arg template<typename T> typename enable<expr,T>::type exp2(T arg) { return fu
resource_for_billibilli/debug_plugin/plugin/half.h:3097
↓ 1 callersFunctionexpm1
Exponential minus one. \param arg function argument \return e raised to \a arg subtracted by 1 template<typename T> typename enable<expr,T>::type expm
resource_for_billibilli/debug_plugin/plugin/half.h:3084
↓ 1 callersFunctionfdim
Positive difference. \param x first operand \param y second operand \return \a x - \a y or 0 if difference negative template<typename T,typename U> ty
resource_for_billibilli/debug_plugin/plugin/half.h:3039
↓ 1 callersFunctionfma
Fused multiply add. \param x first operand \param y second operand \param z third operand \return ( \a x * \a y ) + \a z rounded as one operation. tem
resource_for_billibilli/debug_plugin/plugin/half.h:2955
↓ 1 callersFunctionfmax
Maximum of half expressions. \param x first operand \param y second operand \return maximum of operands template<typename T,typename U> typename resul
resource_for_billibilli/debug_plugin/plugin/half.h:2994
↓ 1 callersFunctionfmin
Minimum of half expressions. \param x first operand \param y second operand \return minimum of operands template<typename T,typename U> typename resul
resource_for_billibilli/debug_plugin/plugin/half.h:3017
↓ 1 callersMethodfree
(self)
shenlan_homework_bert/calibrator.py:62
↓ 1 callersFunctionfrexp
Decompress floating point number. \param arg number to decompress \param exp address to store exponent at \return significant in range [0.5, 1) templa
resource_for_billibilli/debug_plugin/plugin/half.h:3621
↓ 1 callersFunctiongetLayerNormPlugin
()
Hackathon2022/LayerNormPlugin/testLayerNormPlugin.py:52
↓ 1 callersFunctiongetLayerNormPlugin
()
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/testLayerNormPlugin.py:52
↓ 1 callersMethodgetPluginName
Hackathon2022/LayerNormPlugin/LayerNormPlugin.h:213
↓ 1 callersFunctionget_pre_node
(graph, node)
Hackathon2022/encoder2trt.py:76
↓ 1 callersFunctionhalf2float_impl
Convert half-precision to IEEE single-precision. Credit for this goes to [Jeroen van der Zijp](ftp://ftp.fox-toolkit.org/pub/fasthalffloatconversion.p
resource_for_billibilli/debug_plugin/plugin/half.h:929
↓ 1 callersFunctionhypot
Hypotenuse function. \param x first argument \param y second argument \return square root of sum of squares without internal over- or underflows templ
resource_for_billibilli/debug_plugin/plugin/half.h:3194
↓ 1 callersFunctionisgreater
Comparison for greater than. \param x first operand \param y second operand \retval true if \a x greater than \a y \retval false else template<typenam
resource_for_billibilli/debug_plugin/plugin/half.h:3883
↓ 1 callersFunctionisgreaterequal
Comparison for greater equal. \param x first operand \param y second operand \retval true if \a x greater equal \a y \retval false else template<typen
resource_for_billibilli/debug_plugin/plugin/half.h:3907
↓ 1 callersFunctionisinf
Check for infinity. \param arg number to check \retval true for positive or negative infinity \retval false else template<typename T> typename enable<
resource_for_billibilli/debug_plugin/plugin/half.h:3821
↓ 1 callersFunctionisless
Comparison for less than. \param x first operand \param y second operand \retval true if \a x less than \a y \retval false else template<typename T,ty
resource_for_billibilli/debug_plugin/plugin/half.h:3932
↓ 1 callersFunctionislessequal
Comparison for less equal. \param x first operand \param y second operand \retval true if \a x less equal \a y \retval false else template<typename T,
resource_for_billibilli/debug_plugin/plugin/half.h:3956
↓ 1 callersFunctionlayerNormCPU
(bufferH)
Hackathon2022/LayerNormPlugin/testLayerNormPlugin.py:37
↓ 1 callersFunctionlayerNormCPU
(bufferH)
shenlan_homework_bert/基础款LayerNormPlugin/基础款LayerNormPlugin/testLayerNormPlugin.py:37
↓ 1 callersFunctionload_onnx_weights_and_quant
Load the weights from the onnx checkpoint
shenlan_homework_bert/builder.py:242
↓ 1 callersFunctionlog10
Common logorithm. \param arg function argument \return logarithm of \a arg to base 10 template<typename T> typename enable<expr,T>::type log10(T arg)
resource_for_billibilli/debug_plugin/plugin/half.h:3123
↓ 1 callersFunctionlog1p
Natural logorithm. \param arg function argument \return logarithm of \a arg plus 1 to base e template<typename T> typename enable<expr,T>::type log1p(
resource_for_billibilli/debug_plugin/plugin/half.h:3136
↓ 1 callersFunctionlog2
Binary logorithm. \param arg function argument \return logarithm of \a arg to base 2 template<typename T> typename enable<expr,T>::type log2(T arg) {
resource_for_billibilli/debug_plugin/plugin/half.h:3149
↓ 1 callersFunctionmain
()
Hackathon2022/calibrator.py:121
next →1–100 of 601, ranked by callers