MCPcopy Create free account

hub / github.com/NVIDIA/TensorRT / functions

Functions8,179 in github.com/NVIDIA/TensorRT

↓ 2 callersFunctioncheckSafeEngine
samples/common/sampleEngines.cpp:1705
↓ 2 callersFunctioncheck_decoded
(obj)
tools/Polygraphy/polygraphy/json/serde.py:333
↓ 2 callersFunctioncheck_model
(model_name)
demo/DeBERTa/deberta_onnx_modify.py:175
↓ 2 callersFunctionclean_for_display
Prepare the dataframe for display
tools/experimental/trt-engine-explorer/trex/df_preprocessing.py:257
↓ 2 callersMethodcleanup
Cleans up any saved models or loaded models from memory.
demo/HuggingFace/NNDF/models.py:253
↓ 2 callersMethodclearFlag
\brief clear a single build mode flag. clears the builder mode flag from the enabled flags. \see setFlags()
include/NvInfer.h:9262
↓ 2 callersMethodcombine
\brief Combine input timing cache into local instance. This function allows combining entries in the input timing cache to local cache object. \para
include/NvInfer.h:8914
↓ 2 callersFunctioncompute_max
(buffer)
tools/Polygraphy/polygraphy/comparator/util.py:61
↓ 2 callersFunctionconvert_keras_model_to_onnx
Convert in-memory Keras model into ONNX format. Works directly with CreateAssetsFolder object path. Args: keras_model (tf.keras.Model
tools/tensorflow-quantization/tensorflow_quantization/utils.py:140
↓ 2 callersFunctionconvert_to_ascii
(text)
demo/Tacotron2/tacotron2/text/cleaners.py:80
↓ 2 callersMethodcreate
plugin/fcPlugin/fcPlugin.h:270
↓ 2 callersFunctioncreateRefitter
samples/common/sampleEntrypoints.h:89
↓ 2 callersFunctioncreateSafeInferRuntime
samples/common/sampleEngines.cpp:1652
↓ 2 callersMethodcreate_network
Parse the ONNX graph and create the corresponding TensorRT network definition. :param onnx_path: The path to the ONNX graph to load.
samples/python/efficientdet/build_engine.py:138
↓ 2 callersFunctioncustomMatmulRun
plugin/fcPlugin/fcPlugin.cpp:73
↓ 2 callersFunctioncustom_fc
(config, network, input_tensor, out_dims, W)
demo/BERT/builder.py:191
↓ 2 callersMethoddestroy
plugin/fcPlugin/fcPlugin.cpp:684
↓ 2 callersMethoddestroy
\brief Destroy this object. \deprecated Deprecated in TensorRT 8.0. Superseded by `delete`. \warning Calling destroy on a managed pointer will resul
include/NvInfer.h:10016
↓ 2 callersFunctiondetectionForwardPostNMSSize
plugin/common/nmsHelper.cpp:63
↓ 2 callersFunctiondetectionForwardPreNMSSize
plugin/common/nmsHelper.cpp:57
↓ 2 callersMethoddeviceToHost
samples/common/sampleDevice.h:459
↓ 2 callersFunctiondiv
(self, a, b, **kwargs)
tools/Polygraphy/tests/models/make_models.py:41
↓ 2 callersFunctiondivUp
samples/common/safeCommon.h:91
↓ 2 callersFunctiondraw_engine
(engine_json_fname: str)
tools/experimental/trt-engine-explorer/utils/draw_engine.py:34
↓ 2 callersFunctiondump
samples/common/sampleReporting.cpp:533
↓ 2 callersMethodenable_clip
Enable clip stage
tools/pytorch-quantization/pytorch_quantization/nn/modules/tensor_quantizer.py:164
↓ 2 callersMethodenable_quant
(self)
tools/pytorch-quantization/pytorch_quantization/nn/modules/tensor_quantizer.py:187
↓ 2 callersMethodencode_image
(self, init_image)
demo/Diffusion/stable_diffusion_pipeline.py:441
↓ 2 callersMethodendCapture
samples/common/sampleDevice.h:192
↓ 2 callersMethodenqueue
plugin/fcPlugin/fcPlugin.cpp:568
↓ 2 callersMethodenqueue
samples/python/onnx_custom_plugin/plugin/customHardmaxPlugin.cpp:152
↓ 2 callersMethodentering
(self, marker)
tools/Polygraphy/tests/test_examples.py:99
↓ 2 callersFunctionevaluate_acc
(model, validation_data)
tools/tensorflow-quantization/examples/efficientnet/run_qat_workflow.py:59
↓ 2 callersMethodexecute
(self, args: argparse.Namespace)
demo/HuggingFace/run.py:63
↓ 2 callersMethodexecute_inference
( self, metadata: NetworkMetadata, onnx_fpaths: Dict[str, NetworkModel], infer
demo/HuggingFace/T5/onnxrt.py:121
↓ 2 callersMethodexecute_inference
( self, metadata: NetworkMetadata, onnx_fpaths: Dict[str, NetworkModel], infer
demo/HuggingFace/T5/trt.py:505
↓ 2 callersMethodexecute_inference
( self, metadata: NetworkMetadata, network_fpaths: NetworkModels, inference_in
demo/HuggingFace/T5/frameworks.py:175
↓ 2 callersMethodexecute_inference
( self, metadata: NetworkMetadata, onnx_fpaths: Dict[str, NetworkModel], infer
demo/HuggingFace/BART/onnxrt.py:117
↓ 2 callersMethodexecute_inference
( self, metadata: NetworkMetadata, onnx_fpaths: Dict[str, NetworkModel], infer
demo/HuggingFace/BART/trt.py:684
↓ 2 callersMethodexecute_inference
( self, metadata: NetworkMetadata, network_fpaths: NetworkModels, inference_in
demo/HuggingFace/BART/frameworks.py:200
↓ 2 callersMethodexecute_inference
( self, metadata: NetworkMetadata, onnx_fpaths: Dict[str, NetworkModel], infer
demo/HuggingFace/GPT2/trt.py:374
↓ 2 callersMethodexecute_inference
( self, metadata: NetworkMetadata, network_fpaths: NetworkModels, inference_in
demo/HuggingFace/GPT2/frameworks.py:162
↓ 2 callersMethodexiting
(self, marker)
tools/Polygraphy/tests/test_examples.py:102
↓ 2 callersMethodexport_graph
(self, graph, do_type_check=False)
tools/Polygraphy/polygraphy/tools/surgeon/subtool/base.py:39
↓ 2 callersMethodexport_node
(node: Node, do_type_check: bool)
tools/onnx-graphsurgeon/onnx_graphsurgeon/exporters/onnx_exporter.py:62
↓ 2 callersFunctionexport_onnx
Exports an onnx-graphsurgeon Graph to an ONNX model. Args: graph (Graph): The graph to export do_type_check (bool): Whether
tools/onnx-graphsurgeon/onnx_graphsurgeon/exporters/onnx_exporter.py:130
↓ 2 callersFunctionfail
()
tools/Polygraphy/polygraphy/mod/importer.py:126
↓ 2 callersFunctionfake_context
(name)
tools/Polygraphy/tests/backend/trt/test_algorithm_selector.py:34
↓ 2 callersFunctionfake_tensor_quant
tools/pytorch-quantization/src/tensor_quant.cpp:33
↓ 2 callersFunctionfiler_layer
(layer, layer_type)
tools/experimental/trt-engine-explorer/trex/compare_engines.py:475
↓ 2 callersFunctionfind_best_thresh
(preds, scores, na_probs, qid_to_has_ans)
demo/BERT/squad/evaluate-v2.0.py:222
↓ 2 callersFunctionfind_method
(symbol, method)
tools/Polygraphy/polygraphy/mod/exporter.py:102
↓ 2 callersMethodforward
(self, input)
tools/pytorch-quantization/pytorch_quantization/nn/modules/quant_instancenorm.py:45
↓ 2 callersFunctionfreq_to_int
(clk_freq: str, max_clk_freq: int)
tools/experimental/trt-engine-explorer/utils/process_engine.py:167
↓ 2 callersMethodfrom_inference_args
(args)
demo/HuggingFace/NNDF/interface.py:73
↓ 2 callersMethodfrom_name
(name)
tools/Polygraphy/tests/test_examples.py:39
↓ 2 callersMethodfrom_yaml
Create descriptor from yaml str
tools/pytorch-quantization/pytorch_quantization/tensor_quant.py:211
↓ 2 callersFunctionfull_inference
( gpt2, input_ids, tokenizer, timing_profile, max_length, min_length = 0, use_cuda
demo/HuggingFace/GPT2/measurements.py:51
↓ 2 callersMethodgenerate
( self, input_ids, min_length: int = None, max_length: int = None, num
demo/HuggingFace/BART/trt.py:604
↓ 2 callersMethodgenerate_and_download_framework
( self, metadata: NetworkMetadata, workspace: NNFolderWorkspace )
demo/HuggingFace/BART/frameworks.py:68
↓ 2 callersMethodgenerate_roi_name
Generates the roi input name for the Resize node.
samples/python/yolov3_onnx/yolov3_to_onnx.py:221
↓ 2 callersFunctionget
Gets a command-line argument if it exists, otherwise returns a default value. Args: args: The command-line arguments. attr (
tools/Polygraphy/polygraphy/tools/args/util/util.py:100
↓ 2 callersMethodgetAlpha
\brief Get the LRN alpha value. \see setAlpha()
include/NvInfer.h:2121
↓ 2 callersFunctiongetAndDelRepeatedOption
samples/common/sampleOptions.cpp:273
↓ 2 callersMethodgetBeta
\brief Get the LRN beta value. \see setBeta()
include/NvInfer.h:2142
↓ 2 callersMethodgetConstantValue
If isConstant(), returns value of the constant. If !isConstant(), return std::numeric_limits<int32_t>::min().
include/NvInferRuntime.h:254
↓ 2 callersFunctiongetCudaDriverVersion
samples/common/sampleDevice.h:538
↓ 2 callersFunctiongetCudaRuntimeVersion
samples/common/sampleDevice.h:545
↓ 2 callersMethodgetDynamicRangeMax
\brief Get maximum of dynamic range. \return Maximum of dynamic range, or quiet NaN if range was not set.
include/NvInfer.h:394
↓ 2 callersMethodgetDynamicRangeMin
\brief Get minimum of dynamic range. \return Minimum of dynamic range, or quiet NaN if range was not set.
include/NvInfer.h:384
↓ 2 callersMethodgetEnginePath
(self, model_name, engine_dir)
demo/Diffusion/stable_diffusion_pipeline.py:181
↓ 2 callersMethodgetFlag
\brief Returns true if the build mode flag is set \see getFlags() \return True if flag is set, false if unset.
include/NvInfer.h:9286
↓ 2 callersMethodgetKernelSizeNd
\brief Get the multi-dimension kernel size of the convolution. \see setKernelSizeNd()
include/NvInfer.h:1383
↓ 2 callersFunctiongetLayerInformation
samples/common/sampleInference.cpp:1272
↓ 2 callersFunctiongetLayerWeightsRolePair
samples/common/sampleEngines.cpp:1285
↓ 2 callersMethodgetMaxBatchSize
< set the Max Batch Size
samples/common/sampleConfig.h:179
↓ 2 callersFunctiongetMemorySize
quickstart/common/util.cpp:32
↓ 2 callersMethodgetMissingWeights
\brief Get names of missing weights. For example, if some Weights have been set, but the engine was optimized in a way that combines weights, any uns
include/NvInferRuntime.h:1179
↓ 2 callersMethodgetNbDLACores
\brief Return the number of DLA engines available to this builder.
include/NvInfer.h:10036
↓ 2 callersMethodgetNbGroups
\brief Get the number of groups of the convolution. \see setNbGroups()
include/NvInfer.h:1203
↓ 2 callersMethodgetNbOutputMaps
\brief Get the number of output maps for the convolution. \see setNbOutputMaps()
include/NvInfer.h:1115
↓ 2 callersMethodgetNbOutputs
plugin/batchedNMSPlugin/batchedNMSPlugin.cpp:120
↓ 2 callersMethodgetNbOutputs
\brief Get the number of outputs of a layer.
include/NvInfer.h:603
↓ 2 callersMethodgetPadding
\brief Get the padding for pooling. Default: 0 \see setPadding() \deprecated Superseded by getPaddingNd. Deprecated prior to TensorRT 8.0 and will
include/NvInfer.h:1844
↓ 2 callersMethodgetPaddingMode
\brief Get the padding mode. Default: kEXPLICIT_ROUND_DOWN \see setPaddingMode()
include/NvInfer.h:1980
↓ 2 callersMethodgetPaddingNd
\brief Get the multi-dimension padding for pooling. If the padding is asymmetric, the pre-padding is returned. \see setPaddingNd()
include/NvInfer.h:2059
↓ 2 callersFunctiongetParamWordType
plugin/skipLayerNormPlugin/skipLayerNormPlugin.cpp:50
↓ 2 callersFunctiongetPluginLogger
plugin/common/vfcCommon.cpp:65
↓ 2 callersMethodgetPluginName
plugin/fcPlugin/fcPlugin.cpp:724
↓ 2 callersMethodgetReshapeDimensions
\brief Get the reshaped dimensions. \return The reshaped dimensions. If a second input is present and non-null, or setReshapeDimensions has not yet
include/NvInfer.h:4070
↓ 2 callersMethodgetScale
\brief Get the scale value. \see setScale()
include/NvInfer.h:2280
↓ 2 callersMethodgetScales
\brief Copies resize scales to scales[0, ..., nbScales-1], where nbScales is the number of scales that were set. \param size The number of scales to
include/NvInfer.h:5079
↓ 2 callersMethodgetShift
\brief Get the shift value. \see setShift()
include/NvInfer.h:2260
↓ 2 callersMethodgetStrideNd
\brief Get the multi-dimension stride for pooling. \see setStrideNd()
include/NvInfer.h:2028
↓ 2 callersMethodgetTacticSources
\brief Get tactic sources. Get the tactic sources currently set in the engine build configuration. \see setTacticSources() \return tactic sources
include/NvInfer.h:9630
↓ 2 callersMethodgetTensorBytesPerComponent
\brief Return the number of bytes per component of an element, or -1 if the provided name does not map to an input or output tensor. The vector compo
include/NvInferRuntime.h:1906
↓ 2 callersMethodgetTensorDataType
\brief Determine the required data type for a buffer from its tensor name. \param tensorName The name of an input or output tensor. \return The type
include/NvInferRuntime.h:1697
↓ 2 callersMethodgetTensorFormatDesc
\brief Return the human readable description of the tensor format, or empty string if the provided name does not map to an input or output tensor. Th
include/NvInferRuntime.h:2070
↓ 2 callersMethodgetTensorStrides
\brief Return the strides of the buffer for the given tensor name. The strides are in units of elements, not components or bytes. For example, for Te
include/NvInferRuntime.h:2747
↓ 2 callersMethodgetTensorsWithDynamicRange
\brief Get names of all tensors that have refittable dynamic ranges. \param size The number of items that can be safely written to a non-null tensorN
include/NvInferRuntime.h:1104
← previousnext →1,001–1,100 of 8,179, ranked by callers