MCPcopy Create free account

hub / github.com/CaoWGG/TensorRT-CenterNet / functions

Functions423 in github.com/CaoWGG/TensorRT-CenterNet

⨍ Functions 423◇ Types & classes 77
↓ 213 callersMethodat
onnx-tensorrt/OnnxAttrs.hpp:42
↓ 68 callersMethodis_tensor
onnx-tensorrt/TensorOrWeights.hpp:44
↓ 52 callersMethodnetwork
onnx-tensorrt/ImporterContext.hpp:44
↓ 41 callersMethodcount
onnx-tensorrt/ShapedWeights.cpp:46
↓ 29 callersFunctiondeserialize_value
onnx-tensorrt/serialize.hpp:123
↓ 29 callersFunctionserialize_value
onnx-tensorrt/serialize.hpp:118
↓ 28 callersMethodget
onnx-tensorrt/OnnxAttrs.hpp:49
↓ 28 callersMethodis_weights
onnx-tensorrt/TensorOrWeights.hpp:45
↓ 26 callersFunctionserialized_size
onnx-tensorrt/serialize.hpp:113
↓ 24 callersMethodshape
onnx-tensorrt/TensorOrWeights.hpp:62
↓ 23 callersFunctionreshape_tensor
onnx-tensorrt/builtin_op_importers.cpp:114
↓ 21 callersFunctionget_shape_size
onnx-tensorrt/trt_utils.hpp:50
↓ 19 callersMethodnode
onnx-tensorrt/Status.hpp:94
↓ 18 callersMethodempty
onnx-tensorrt/ShapedWeights.cpp:58
↓ 18 callersFunctioninfer_object
onnx-tensorrt/common.hpp:48
↓ 17 callersFunctionsetUIntInfo
onnx-tensorrt/onnx_trt_backend.cpp:512
↓ 15 callersFunctionOnnxifiTryCatch
onnx-tensorrt/onnx_trt_backend.cpp:457
↓ 15 callersMethodaddPlugin
onnx-tensorrt/ImporterContext.hpp:53
↓ 14 callersMethodcreateTempWeights
onnx-tensorrt/ImporterContext.hpp:46
↓ 13 callersFunctionapply_unary_function
onnx-tensorrt/trt_utils.hpp:185
↓ 13 callersMethodget<float>
onnx-tensorrt/OnnxAttrs.cpp:27
↓ 12 callersMethoddestroy
onnx-tensorrt/plugin.cpp:142
↓ 11 callersMethodget<int>
onnx-tensorrt/OnnxAttrs.cpp:30
↓ 11 callersMethodis_error
onnx-tensorrt/Status.hpp:95
↓ 9 callersFunctioncombineTensorsElementwise
onnx-tensorrt/builtin_op_importers.cpp:252
↓ 8 callersFunctionconvert_axis
Convert an ONNX axis into a TRT axis
onnx-tensorrt/onnx2trt_utils.hpp:323
↓ 7 callersMethodis_set_by_user
src/argparse.cpp:472
↓ 7 callersFunctiononnx_ir_version_string
onnx-tensorrt/common.hpp:102
↓ 7 callersFunctionreduceTensor
onnx-tensorrt/builtin_op_importers.cpp:1420
↓ 7 callersFunctionsetStringInfo
onnx-tensorrt/onnx_trt_backend.cpp:526
↓ 6 callersMethodfile
onnx-tensorrt/Status.hpp:91
↓ 6 callersMethodget
include/argparse.h:58
↓ 5 callersFunctioncreateParser
onnx-tensorrt/NvOnnxParser.h:200
↓ 5 callersFunctioncudaSetDevice
(device_idx)
onnx-tensorrt/onnx_tensorrt/backend.py:35
↓ 5 callersMethodgetOpsetVersion
onnx-tensorrt/ImporterContext.hpp:94
↓ 5 callersMethodserialize
onnx-tensorrt/plugin.cpp:79
↓ 5 callersFunctionstr_replace
src/argparse.cpp:58
↓ 4 callersMethodbuildCudaEngine
onnx-tensorrt/onnx_trt_backend.cpp:236
↓ 4 callersMethodcode
onnx-tensorrt/Status.hpp:89
↓ 4 callersFunctioncols
src/argparse.cpp:105
↓ 4 callersMethoddesc
onnx-tensorrt/Status.hpp:90
↓ 4 callersMethodfunc
onnx-tensorrt/Status.hpp:93
↓ 4 callersFunctionget_DimsHW_from_CHW
onnx-tensorrt/trt_utils.hpp:164
↓ 4 callersFunctionget_kernel_params
onnx-tensorrt/onnx2trt_utils.cpp:27
↓ 4 callersFunctionidentity
onnx-tensorrt/trt_utils.hpp:169
↓ 4 callersMethodline
onnx-tensorrt/Status.hpp:92
↓ 4 callersFunctionprint_usage
onnx-tensorrt/main.cpp:41
↓ 4 callersFunctionremove_dim
onnx-tensorrt/trt_utils.hpp:78
↓ 4 callersFunctionstr_format
src/argparse.cpp:63
↓ 4 callersFunctionstr_join_trans
src/argparse.cpp:34
↓ 3 callersFunctionGetTensorFootprint
onnx-tensorrt/onnx_trt_backend.cpp:54
↓ 3 callersFunctionParseFromFile_WAR
onnx-tensorrt/common.hpp:82
↓ 3 callersMethod__init__
(self, in_channels, out_channels, kernel_size,
readme/dcn/modules/deform_conv.py:12
↓ 3 callersMethodcheck_type
///// class Option { //////////
src/argparse.cpp:481
↓ 3 callersFunctionconvert_dtype
onnx-tensorrt/onnx2trt_utils.hpp:135
↓ 3 callersMethoddoInference
src/ctdetNet.cpp:146
↓ 3 callersMethodenqueue
onnx-tensorrt/DCNv2.cpp:121
↓ 3 callersFunctionimportScaleOp
onnx-tensorrt/builtin_op_importers.cpp:346
↓ 3 callersFunctionis_CHW
onnx-tensorrt/DCNv2.cpp:27
↓ 3 callersFunctionnp2onnx_dtype
(np_dtype)
onnx-tensorrt/onnx_tensorrt/backend.py:172
↓ 3 callersFunctionpostProcess
src/utils.cpp:51
↓ 3 callersMethodprepare
Build an engine from the given model. model -- An ONNX model as a deserialized protobuf, or a string or file- object contain
onnx-tensorrt/onnx_tensorrt/backend.py:211
↓ 3 callersFunctionprepareImage
src/utils.cpp:9
↓ 3 callersFunctionpretty_print_onnx_to_string
onnx-tensorrt/onnx_utils.hpp:69
↓ 3 callersFunctionprint_version
onnx-tensorrt/common.hpp:111
↓ 3 callersMethodrun
Execute the prepared engine and return the outputs as a named tuple. inputs -- Input tensor(s) as a Numpy array or list of Numpy arrays.
onnx-tensorrt/onnx_tensorrt/backend.py:142
↓ 3 callersFunctionset_dims_CHW
onnx-tensorrt/trt_utils.hpp:134
↓ 3 callersFunctionshape_check
readme/dcn/src/deform_conv_cuda.cpp:61
↓ 3 callersMethodsize_bytes
onnx-tensorrt/ShapedWeights.cpp:69
↓ 3 callersMethodstream
onnx-tensorrt/onnx_trt_backend.cpp:276
↓ 3 callersFunctiontransposeWeights
onnx-tensorrt/ShapedWeights.cpp:111
↓ 2 callersFunctionParseFromTextFile
onnx-tensorrt/common.hpp:94
↓ 2 callersMethodSignal
onnx-tensorrt/onnx_trt_backend.cpp:107
↓ 2 callersMethodWait
onnx-tensorrt/onnx_trt_backend.cpp:121
↓ 2 callersMethod__init__
(self, spatial_scale, out_size, out_channels,
readme/dcn/modules/deform_pool.py:8
↓ 2 callersFunction_tensorrt_version
()
onnx-tensorrt/onnx_tensorrt/tensorrt_engine.py:29
↓ 2 callersFunction_tensorrt_version
()
onnx-tensorrt/onnx_tensorrt/backend.py:49
↓ 2 callersFunctionaddScale
onnx-tensorrt/builtin_op_importers.cpp:142
↓ 2 callersFunctionargMinMaxHelper
Helper for ArgMax/ArgMin
onnx-tensorrt/builtin_op_importers.cpp:441
↓ 2 callersMethodcallback
include/argparse.h:239
↓ 2 callersFunctionconvert_onnx_weights
onnx-tensorrt/onnx2trt_utils.hpp:236
↓ 2 callersMethodcount
onnx-tensorrt/OnnxAttrs.hpp:41
↓ 2 callersFunctioncreateNvOnnxParserPluginFactory_INTERNAL
onnx-tensorrt/NvOnnxParserRuntime.cpp:26
↓ 2 callersFunctioncreateNvOnnxParser_INTERNAL
onnx-tensorrt/NvOnnxParser.cpp:26
↓ 2 callersFunctioncreatePluginFactory
onnx-tensorrt/NvOnnxParserRuntime.h:53
↓ 2 callersFunctiondeserialize_onnx_model
onnx-tensorrt/ModelImporter.cpp:188
↓ 2 callersFunctiondrawImg
src/utils.cpp:137
↓ 2 callersFunctionexpand_dims
Adds unitary dimensions on the left
onnx-tensorrt/trt_utils.hpp:100
↓ 2 callersFunctionflatten_tensor
onnx-tensorrt/builtin_op_importers.cpp:129
↓ 2 callersMethodgetError
onnx-tensorrt/ModelImporter.hpp:73
↓ 2 callersMethodgetNbErrors
virtual void registerOpImporter(std::string op, NodeImporter const &node_importer) override { // Note: This allows existing importers to be replaced _
onnx-tensorrt/ModelImporter.hpp:72
↓ 2 callersMethodgetNbOutputs
onnx-tensorrt/plugin.cpp:71
↓ 2 callersMethodgetPluginType
onnx-tensorrt/plugin.hpp:125
↓ 2 callersFunctionget_conv_output_size
onnx-tensorrt/onnx2trt_utils.hpp:339
↓ 2 callersFunctionget_dtype_name
onnx-tensorrt/onnx2trt_utils.hpp:102
↓ 2 callersFunctionget_infer_dim
Helper function to help extract the index of a potential -1 dimension in the reshape node
onnx-tensorrt/onnx2trt_utils.hpp:349
↓ 2 callersFunctionhalf_to_float_fast
This is derived from: https://fgiesen.wordpress.com/2012/03/28/half-to-float-done-quic/
onnx-tensorrt/InstanceNormalization.cpp:49
↓ 2 callersMethodis_set
include/argparse.h:55
↓ 2 callersMethodlog
include/utils.h:97
↓ 2 callersFunctionmake_node_test_model
(node, inputs, use_weights=True)
onnx-tensorrt/onnx_tensorrt/backend.py:186
next →1–100 of 423, ranked by callers