Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/CaoWGG/TensorRT-CenterNet
/ functions
Functions
423 in github.com/CaoWGG/TensorRT-CenterNet
⨍
Functions
423
◇
Types & classes
77
Method
getSerializationSize
onnx-tensorrt/plugin.hpp:118
Method
getSerializationSize
onnx-tensorrt/Split.hpp:44
Method
getTensorRTVersion
onnx-tensorrt/onnx2trt_common.hpp:42
Method
getWorkspaceSize
onnx-tensorrt/plugin.cpp:107
Method
getWorkspaceSize
onnx-tensorrt/plugin.hpp:49
Method
getWorkspaceSize
onnx-tensorrt/DCNv2.cpp:117
Method
getWorkspaceSize
onnx-tensorrt/InstanceNormalization.cpp:183
Function
get_dtype_size
onnx-tensorrt/onnx2trt_utils.hpp:80
Method
get_usage
src/argparse.cpp:432
Method
get_version
src/argparse.cpp:446
Method
handle_long_opt
src/argparse.cpp:251
Method
handle_short_opt
src/argparse.cpp:209
Method
host_buffer
(self)
onnx-tensorrt/onnx_tensorrt/tensorrt_engine.py:76
Function
initNet
src/python_api.cpp:8
Method
initialize
onnx-tensorrt/plugin.cpp:110
Method
initialize
onnx-tensorrt/plugin.hpp:51
Method
initialize
onnx-tensorrt/DCNv2.cpp:58
Method
initialize
onnx-tensorrt/InstanceNormalization.cpp:92
Method
int8EntroyCalibrator
src/entroyCalibrator.cpp:13
Method
interspersed_args
include/argparse.h:125
Method
is_error
onnx-tensorrt/Status.hpp:110
Method
is_set_by_user
include/argparse.h:56
Method
is_success
onnx-tensorrt/Status.hpp:96
Function
main
example/runDet.cpp:14
Function
main
example/buildEngine.cpp:12
Function
main
onnx-tensorrt/main.cpp:58
Function
main
onnx-tensorrt/getSupportedAPITest.cpp:73
Function
modulated_deform_conv_cuda_backward
readme/dcn/src/deform_conv_cuda.cpp:566
Function
modulated_deform_conv_cuda_forward
readme/dcn/src/deform_conv_cuda.cpp:486
Method
nargs
include/argparse.h:234
Function
ndarrayToImage
src/python_api.cpp:49
Function
onnxGetBackendCompatibility
onnx-tensorrt/onnx_trt_backend.cpp:784
Function
onnxGetBackendIDs
onnx-tensorrt/onnx_trt_backend.cpp:472
Function
onnxGetBackendInfo
onnx-tensorrt/onnx_trt_backend.cpp:541
Function
onnxGetEventState
onnx-tensorrt/onnx_trt_backend.cpp:886
Function
onnxInitBackend
NB: Passing arguments to backend is tricky. And we need more documentation for it I didn't put any arguments here for now. TODO: submit arguments for
onnx-tensorrt/onnx_trt_backend.cpp:815
Function
onnxInitEvent
onnx-tensorrt/onnx_trt_backend.cpp:844
Function
onnxInitGraph
onnx-tensorrt/onnx_trt_backend.cpp:913
Function
onnxReleaseBackend
onnx-tensorrt/onnx_trt_backend.cpp:832
Function
onnxReleaseBackendID
onnx-tensorrt/onnx_trt_backend.cpp:500
Function
onnxReleaseEvent
onnx-tensorrt/onnx_trt_backend.cpp:901
Function
onnxReleaseGraph
onnx-tensorrt/onnx_trt_backend.cpp:1002
Function
onnxRunGraph
onnx-tensorrt/onnx_trt_backend.cpp:972
Function
onnxSetGraphIO
NB: in the context of TRT, this step will setup the input/output bindings for ICudaEngine
onnx-tensorrt/onnx_trt_backend.cpp:954
Function
onnxSignalEvent
onnx-tensorrt/onnx_trt_backend.cpp:864
Function
onnxWaitEvent
onnx-tensorrt/onnx_trt_backend.cpp:875
Function
operator!=
onnx-tensorrt/trt_utils.hpp:160
Method
operator()
src/argparse.cpp:30
Method
operator()
onnx-tensorrt/onnx2trt_common.hpp:71
Method
operator()
onnx-tensorrt/common.hpp:40
Function
operator-
onnx-tensorrt/trt_utils.hpp:144
Function
operator==
Note: These are used for checking beg_padding == end_padding
onnx-tensorrt/trt_utils.hpp:149
Function
predict
src/python_api.cpp:20
Method
printTime
include/ctdetNet.h:56
Method
print_help
src/argparse.cpp:415
Method
print_usage
src/argparse.cpp:437
Method
print_version
src/argparse.cpp:449
Method
process_opt
src/argparse.cpp:335
Function
randomColor
src/utils.cpp:132
Method
readCalibrationCache
src/entroyCalibrator.cpp:68
Function
registerBuiltinOpImporter
onnx-tensorrt/builtin_op_importers.cpp:398
Function
registerBuiltinPlugin
onnx-tensorrt/builtin_plugins.cpp:41
Method
registerPlugin
onnx-tensorrt/PluginFactory.hpp:48
Method
reportLayerTime
include/utils.h:80
Method
run
(self, inputs)
onnx-tensorrt/onnx_tensorrt/tensorrt_engine.py:127
Method
run_model
Build and run an engine from the given model. model -- An ONNX model as a deserialized protobuf, or a string or file- object
onnx-tensorrt/onnx_tensorrt/backend.py:220
Method
run_no_dma
(self, batch_size)
onnx-tensorrt/onnx_tensorrt/tensorrt_engine.py:167
Method
run_node
Build and run an engine from the given node. node -- An ONNX node as a deserialized protobuf. Note: This function is intended for te
onnx-tensorrt/onnx_tensorrt/backend.py:230
Method
serialize
onnx-tensorrt/ResizeNearest.hpp:43
Method
serialize
onnx-tensorrt/FancyActivation.hpp:66
Method
serialize
onnx-tensorrt/DCNv2.hpp:66
Method
serialize
onnx-tensorrt/InstanceNormalization.hpp:56
Method
serialize
onnx-tensorrt/plugin.hpp:112
Method
serialize
onnx-tensorrt/Split.hpp:48
Method
serialize
onnx-tensorrt/serialize.hpp:54
Method
serialize
onnx-tensorrt/serialize.hpp:71
Method
serialize
onnx-tensorrt/serialize.hpp:92
Method
serializeBase
onnx-tensorrt/plugin.cpp:31
Method
serialized_size
onnx-tensorrt/serialize.hpp:51
Method
serialized_size
onnx-tensorrt/serialize.hpp:68
Method
serialized_size
onnx-tensorrt/serialize.hpp:89
Function
setDevice
src/python_api.cpp:14
Method
setUserInput
onnx-tensorrt/ImporterContext.hpp:65
Method
setUserOutput
onnx-tensorrt/ImporterContext.hpp:69
Method
set_usage
src/argparse.cpp:419
Function
show_img
(pred,image)
eval_coco.py:79
Function
squeeze_hw
(x)
onnx-tensorrt/onnx_tensorrt/tensorrt_engine.py:91
Method
str_wrap
src/argparse.cpp:28
Method
stream
onnx-tensorrt/onnx_trt_backend.cpp:198
Method
string>
onnx-tensorrt/OnnxAttrs.cpp:38
Method
success
onnx-tensorrt/Status.hpp:82
Method
supportsFormat
onnx-tensorrt/plugin.cpp:52
Method
supportsFormat
onnx-tensorrt/DCNv2.cpp:96
Method
supportsFormat
onnx-tensorrt/InstanceNormalization.cpp:135
Method
supports_device
(cls, device_str)
onnx-tensorrt/onnx_tensorrt/backend.py:249
Method
symbolic
(g, input, offset, mask, weight, bias,stride,padding,dilation,groups,deformable_groups)
readme/dcn/functions/deform_conv.py:111
Method
terminate
onnx-tensorrt/plugin.cpp:112
Method
terminate
onnx-tensorrt/plugin.hpp:52
Method
terminate
onnx-tensorrt/DCNv2.cpp:82
Method
terminate
onnx-tensorrt/InstanceNormalization.cpp:118
← previous
next →
301–400 of 423, ranked by callers