MCPcopy Create free account

hub / github.com/TrojanXu/onnxparser-trt-plugin-sample / functions

Functions47 in github.com/TrojanXu/onnxparser-trt-plugin-sample

⨍ Functions 47◇ Types & classes 7
↓ 3 callersMethodsetPluginNamespace
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:215
↓ 1 callersMethod__str__
(self)
common.py:146
↓ 1 callersFunctionbuild_engine_onnx
(model_file)
test_plugin_result.py:139
↓ 1 callersFunctionexport_onnx_model
(onnx_model_file)
test_plugin_result.py:101
↓ 1 callersFunctionget_data_path
(data_dir)
common.py:96
↓ 1 callersFunctionlocate_files
Locates the specified files in the specified data directories. If a file exists in multiple data directories, the first directory is used.
common.py:110
↓ 1 callersFunctionmodify_onnx
(onnx_model_file)
test_plugin_result.py:116
FunctionGiB
(val)
common.py:68
MethodGridSamplerPlugin
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:61
MethodGridSamplerPlugin
It doesn't make sense to make GridSamplerPlugin without arguments, so we delete default constructor.
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.h:50
MethodGridSamplerPluginCreator
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:266
Method__init__
(self)
test_plugin_result.py:94
Method__init__
(self, host_mem, device_mem)
common.py:142
Method__repr__
(self)
common.py:149
Functionadd_help
(description)
common.py:72
Functionallocate_buffers
(engine, is_explicit_batch=False, context_batch_size=1)
common.py:153
MethodattachToContext
Attach the plugin object to an execution context and grant the plugin the access to some context resource.
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:258
Methodclone
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:206
MethodconfigurePlugin
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:235
MethodcreatePlugin
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:290
MethoddeserializePlugin
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:322
Methoddestroy
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:200
MethoddetachFromContext
Detach the plugin object from its execution context.
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:264
Functiondo_inference
(context, bindings, inputs, outputs, stream, batch_size=1)
common.py:181
Functiondo_inference_v2
(context, bindings, inputs, outputs, stream)
common.py:195
Methodenqueue
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:148
Functionfind_sample_data
Parses sample arguments. Args: description (str): Description of the sample. subfolder (str): The subfolder containing data
common.py:77
Methodforward
(self, input, grid)
test_plugin_result.py:97
MethodgetFieldNames
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:285
MethodgetNbOutputs
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:118
MethodgetOutputDataType
Return the DataType of the plugin output at the requested index.
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:226
MethodgetOutputDimensions
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:123
MethodgetPluginName
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:275
MethodgetPluginNamespace
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:220
MethodgetPluginType
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:108
MethodgetPluginVersion
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:113
MethodgetSerializationSize
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:165
MethodgetWorkspaceSize
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:142
Functiongrid_sampler
(g, input, grid, mode, padding_mode, align_corners)
test_plugin_result.py:64
Methodinitialize
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:137
FunctionreadFromBuffer
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:54
Methodserialize
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:170
MethodsupportsFormatCombination
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:186
Methodterminate
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:198
FunctionwriteToBuffer
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:46
Method~GridSamplerPlugin
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:106
Method~GridSamplerPluginCreator
TensorRT/plugin/gridSamplerPlugin/gridSamplerPlugin.cpp:273