MCPcopy Create free account

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

Functions423 in github.com/CaoWGG/TensorRT-CenterNet

⨍ Functions 423◇ Types & classes 77
↓ 2 callersMethodmax_batch_size
onnx-tensorrt/onnx_trt_backend.cpp:240
↓ 2 callersFunctionmove_tensor_dimension
onnx-tensorrt/builtin_op_importers.cpp:95
↓ 2 callersMethodparse
onnx-tensorrt/ModelImporter.cpp:502
↓ 2 callersMethodparseWithWeightDescriptors
onnx-tensorrt/ModelImporter.cpp:476
↓ 2 callersFunctionprintSubGraphs
onnx-tensorrt/getSupportedAPITest.cpp:42
↓ 2 callersMethodprintTime
include/utils.h:44
↓ 2 callersFunctionsafeCudaMalloc
include/utils.h:134
↓ 2 callersFunctionslice_clip_index
Takes idx from [MIN_INT, MAX_INT] to [0, ax_size] (for Slice op)
onnx-tensorrt/builtin_op_importers.cpp:40
↓ 2 callersFunctionstr_join
src/argparse.cpp:44
↓ 2 callersMethodsupportsModel
onnx-tensorrt/ModelImporter.cpp:316
↓ 2 callersFunctiontoposort
onnx-tensorrt/toposort.hpp:76
↓ 2 callersFunctiontranspose_tensor
Note: perm should not include the batch dim
onnx-tensorrt/builtin_op_importers.cpp:67
↓ 1 callersFunctionCheckShape
onnx-tensorrt/onnx_trt_backend.cpp:23
↓ 1 callersMethodCheckState
onnx-tensorrt/onnx_trt_backend.cpp:128
↓ 1 callersMethodImportModel
onnx-tensorrt/onnx_trt_backend.cpp:200
↓ 1 callersMethodInitIO
onnx-tensorrt/onnx_trt_backend.cpp:338
↓ 1 callersMethodLogger
include/utils.h:92
↓ 1 callersMethodRun
onnx-tensorrt/onnx_trt_backend.cpp:425
↓ 1 callersMethod_infer_shape
(ctx, input, weight)
readme/dcn/functions/deform_conv.py:174
↓ 1 callersMethod_output_size
(input, weight, padding, dilation, stride)
readme/dcn/functions/deform_conv.py:92
↓ 1 callersMethod_serialize_deserialize
(self, trt_engine)
onnx-tensorrt/onnx_tensorrt/backend.py:125
↓ 1 callersMethod_set_device
(self, device)
onnx-tensorrt/onnx_tensorrt/backend.py:121
↓ 1 callersMethodaddOpset
onnx-tensorrt/ImporterContext.hpp:91
↓ 1 callersFunctionapplyLegacyBinaryOpBroadcasting
Explicit broadcasting for ONNX opset < 7 This function adds extra dimensions to the end of rhs's shape in order to line up the dimensions based on the
onnx-tensorrt/builtin_op_importers.cpp:207
↓ 1 callersFunctionbasename
src/argparse.cpp:114
↓ 1 callersFunctionblas_handle
onnx-tensorrt/DCNv2.cpp:15
↓ 1 callersFunctioncheck_broadcast_attrs
onnx-tensorrt/builtin_op_importers.cpp:327
↓ 1 callersFunctioncheck_for_input
onnx-tensorrt/onnx2trt_utils.hpp:123
↓ 1 callersMethodclearOpsets
onnx-tensorrt/ImporterContext.hpp:90
↓ 1 callersMethodconfigure
onnx-tensorrt/onnx2trt_common.hpp:50
↓ 1 callersMethodconfigureWithFormat
onnx-tensorrt/plugin.cpp:58
↓ 1 callersFunctionconvertINT64
onnx-tensorrt/ShapedWeights.cpp:29
↓ 1 callersFunctionconvert_dims
onnx-tensorrt/onnx2trt_utils.hpp:172
↓ 1 callersFunctionconvert_input_dtype
onnx-tensorrt/onnx2trt_utils.hpp:154
↓ 1 callersFunctionconvert_trt2cudnn_dtype
onnx-tensorrt/InstanceNormalization.cpp:142
↓ 1 callersFunctionconvert_weight_descriptor
onnx-tensorrt/onnx2trt_utils.hpp:190
↓ 1 callersFunctioncount_trailing_ones
(vals)
onnx-tensorrt/onnx_tensorrt/backend.py:41
↓ 1 callersMethoddeserialize
onnx-tensorrt/ResizeNearest.hpp:35
↓ 1 callersMethoddeserialize
onnx-tensorrt/FancyActivation.hpp:55
↓ 1 callersMethoddeserialize
onnx-tensorrt/DCNv2.hpp:38
↓ 1 callersMethoddeserialize
onnx-tensorrt/InstanceNormalization.hpp:43
↓ 1 callersMethoddeserialize
onnx-tensorrt/Split.hpp:39
↓ 1 callersMethoddevice_id
onnx-tensorrt/onnx_trt_backend.cpp:197
↓ 1 callersFunctiondiv_ceil
onnx-tensorrt/onnx2trt_utils.hpp:318
↓ 1 callersMethodformat_help
src/argparse.cpp:388
↓ 1 callersMethodformat_option_help
src/argparse.cpp:167
↓ 1 callersMethodgetDataType
onnx-tensorrt/plugin.hpp:46
↓ 1 callersFunctiongetElementSize
include/utils.h:121
↓ 1 callersMethodgetMaxBatchSize
onnx-tensorrt/plugin.hpp:45
↓ 1 callersMethodgetOutputDimensions
onnx-tensorrt/DCNv2.cpp:100
↓ 1 callersMethodgetUserInput
onnx-tensorrt/ImporterContext.hpp:73
↓ 1 callersMethodgetUserOutput
onnx-tensorrt/ImporterContext.hpp:80
↓ 1 callersMethodget_async
(self, batch_size, stream)
onnx-tensorrt/onnx_tensorrt/tensorrt_engine.py:85
↓ 1 callersFunctionget_dtype_size
onnx-tensorrt/trt_utils.hpp:35
↓ 1 callersFunctionget_post_order
onnx-tensorrt/toposort.hpp:38
↓ 1 callersFunctionget_scale_mode
onnx-tensorrt/onnx2trt_utils.hpp:374
↓ 1 callersFunctionimg_preprocess
(image, target_shape=(512,512),mean = (0.485,0.456,0.406),std =(0.229,0.224,0.225))
eval_coco.py:63
↓ 1 callersFunctionimportInput
onnx-tensorrt/ModelImporter.cpp:48
↓ 1 callersFunctionimportInputs
onnx-tensorrt/ModelImporter.cpp:87
↓ 1 callersMethodimportModel
onnx-tensorrt/ModelImporter.cpp:508
↓ 1 callersMethodimportNode
onnx-tensorrt/ModelImporter.cpp:138
↓ 1 callersMethodinit_offset
(self)
readme/dcn/modules/deform_conv.py:74
↓ 1 callersMethodinit_offset
(self)
readme/dcn/modules/deform_conv.py:146
↓ 1 callersFunctioninsert_dim
onnx-tensorrt/trt_utils.hpp:62
↓ 1 callersFunctionis_CHW
TODO: Move this to a common header
onnx-tensorrt/InstanceNormalization.cpp:41
↓ 1 callersFunctionis_transpose_required
Returns false if the transpose does not require any data movement (i.e., it's equivalent to a reshape)
onnx-tensorrt/builtin_op_importers.cpp:50
↓ 1 callersMethodlog
onnx-tensorrt/common.hpp:63
↓ 1 callersFunctionndarray_to_image
(img)
eval_coco.py:57
↓ 1 callersMethodparseFromFile
onnx-tensorrt/ModelImporter.cpp:226
↓ 1 callersFunctionprint_usage
onnx-tensorrt/getSupportedAPITest.cpp:35
↓ 1 callersFunctionremove_first_dim
onnx-tensorrt/trt_utils.hpp:114
↓ 1 callersFunctionremove_raw_data_strings
Removes raw data from the text representation of an ONNX model
onnx-tensorrt/onnx_utils.hpp:30
↓ 1 callersFunctionremove_repeated_data_strings
Removes float_data, int32_data etc. from the text representation of an ONNX model
onnx-tensorrt/onnx_utils.hpp:47
↓ 1 callersMethodreset_parameters
(self)
readme/dcn/modules/deform_conv.py:47
↓ 1 callersMethodreset_parameters
(self)
readme/dcn/modules/deform_conv.py:116
↓ 1 callersMethodreset_tensor
onnx-tensorrt/TensorOrWeights.hpp:68
↓ 1 callersMethodsaveEngine
src/ctdetNet.cpp:167
↓ 1 callersMethodsetNode
onnx-tensorrt/Status.hpp:97
↓ 1 callersFunctionsqueeze_trailing_dims
onnx-tensorrt/trt_utils.hpp:125
↓ 1 callersFunctionstr_inc
src/argparse.cpp:97
↓ 1 callersMethodsupportsOperator
onnx-tensorrt/ModelImporter.cpp:472
↓ 1 callersFunctiontop_k
(pred,K=100)
eval_coco.py:91
↓ 1 callersFunctionupdate_padded_values
onnx-tensorrt/onnx2trt_utils.hpp:386
↓ 1 callersFunctionvolume
include/utils.h:116
MethodCheckAndBindTensor
onnx-tensorrt/onnx_trt_backend.cpp:304
MethodClearDeviceBuffers
onnx-tensorrt/onnx_trt_backend.cpp:297
MethodCudaDeviceGuard
onnx-tensorrt/onnx_trt_backend.cpp:157
MethodDCNv2Plugin
onnx-tensorrt/DCNv2.hpp:93
MethodDCNv2Plugin
onnx-tensorrt/DCNv2.cpp:34
FunctionDEFINE_BUILTIN_OP_IMPORTER
onnx-tensorrt/builtin_op_importers.cpp:486
MethodDims>
onnx-tensorrt/OnnxAttrs.cpp:49
MethodDimsHW>
onnx-tensorrt/OnnxAttrs.cpp:57
FunctionEnumMax<ErrorCode>
onnx-tensorrt/NvOnnxParser.h:61
MethodFancyActivationPlugin
onnx-tensorrt/FancyActivation.hpp:73
MethodGraphRep
onnx-tensorrt/onnx_trt_backend.cpp:256
MethodImporterContext
onnx-tensorrt/ImporterContext.hpp:41
MethodInitEngine
src/ctdetNet.cpp:121
MethodInstanceNormalizationPlugin
onnx-tensorrt/InstanceNormalization.hpp:67
MethodInstanceNormalizationPlugin
onnx-tensorrt/InstanceNormalization.cpp:63
MethodModelImporter
onnx-tensorrt/ModelImporter.hpp:47
← previousnext →101–200 of 423, ranked by callers