MCPcopy Create free account

hub / github.com/NVIDIA/TensorRT / endpoints

Endpoints43 in github.com/NVIDIA/TensorRT

⨍ Functions 8,179◇ Types & classes 1,602↳ Endpoints 43
Routeget_input_metadata_impl
util.check_called_by("get_input_metadata")
tools/Polygraphy/polygraphy/backend/trt/runner.py:None
Routeget_input_metadata_impl
util.check_called_by("get_input_metadata")
tools/Polygraphy/polygraphy/backend/tf/runner.py:None
Routeget_input_metadata_impl
util.check_called_by("get_input_metadata")
tools/Polygraphy/polygraphy/backend/onnxrt/runner.py:None
Routeget_input_metadata_impl
util.check_called_by("get_input_metadata")
tools/Polygraphy/polygraphy/backend/pyt/runner.py:None
Routeget_input_metadata_impl
util.check_called_by("get_input_metadata")
tools/Polygraphy/polygraphy/backend/pluginref/runner.py:None
Routeget_input_metadata_impl
util.check_called_by("get_input_metadata")
tools/Polygraphy/examples/dev/02_extending_polygraphy_run/extension_module/polygraphy_reshape_destroyer/backend/runner.py:None
Routeon_svg_button_clicked
output.capture()
tools/experimental/trt-engine-explorer/trex/report_card.py:None
Routetest_add
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_append
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_calibrator_checks_input_metadata
pytest.mark.parametrize( "expected_meta,meta,should_pass", [ ( Ten
tools/Polygraphy/tests/backend/trt/test_calibrator.py:None
Routetest_calibrator_invalid_input_fails
pytest.mark.parametrize( "names", [ (["fake-input", "x"]), (["fake-inp
tools/Polygraphy/tests/backend/trt/test_calibrator.py:None
Routetest_cast_elision_multi_use_cast
pytest.mark.parametrize("use_as_graph_output", [True, False], ids=["graph", ""])
tools/onnx-graphsurgeon/tests/ir/test_graph.py:None
Routetest_clear
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_custom_outputs
pytest.mark.parametrize("output", ["identity_out_0", "identity_out_2"])
tools/Polygraphy/tests/backend/onnx/test_loader.py:None
Routetest_del_index
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_equality
pytest.mark.parametrize( "left, right, expected", [ ( Algorithm(
tools/Polygraphy/tests/backend/trt/test_algorithm_selector.py:None
Routetest_error_on_wrong_name_feed_dict
pytest.mark.parametrize( "names, err", [ (["fake-input", "x"], "Extra inputs in"),
tools/Polygraphy/tests/backend/trt/test_runner.py:None
Routetest_error_on_wrong_name_feed_dict
pytest.mark.parametrize( "names, err", [ (["fake-input", "Input:0"], "Extra inputs
tools/Polygraphy/tests/backend/tf/test_runner.py:None
Routetest_error_on_wrong_name_feed_dict
pytest.mark.parametrize( "names, err", [ (["fake-input", "x"], "Extra inputs in"),
tools/Polygraphy/tests/backend/onnxrt/test_runner.py:None
Routetest_error_on_wrong_name_feed_dict
pytest.mark.parametrize( "names, err", [ (["fake-input", "x"], "Extra inputs in"),
tools/Polygraphy/tests/backend/pluginref/test_runner.py:None
Routetest_error_stat_per_output
pytest.mark.parametrize( "args, expected", [ (["mean", "output0:median", "output1:
tools/Polygraphy/tests/tools/args/comparator/test_compare.py:None
Routetest_export_graph
pytest.mark.parametrize( "model", [ identity_model(), lstm_model(),
tools/onnx-graphsurgeon/tests/test_exporters.py:None
Routetest_extend
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_iadd
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_ignore_fail_code
pytest.mark.parametrize( "opts,expected_output", [ ([], "Passed: 0/5 | Pass Rate:
tools/Polygraphy/tests/tools/test_debug.py:None
Routetest_import_graph
pytest.mark.parametrize( "model", [ identity_model(), lstm_model(),
tools/onnx-graphsurgeon/tests/test_importers.py:None
Routetest_independent_path
pytest.mark.parametrize("remove_unused_graph_inputs", [True, False])
tools/onnx-graphsurgeon/tests/ir/test_graph.py:None
Routetest_index_tolerance
pytest.mark.parametrize( "out0,out1,index_tolerance,expected", [ ([0, 1, 2, 3], [0
tools/Polygraphy/tests/comparator/test_compare.py:None
Routetest_infer_chw_format
pytest.mark.parametrize("hwc_input", [True, False], ids=["hwc_input", "chw_input"])
tools/Polygraphy/tests/backend/trt/test_runner.py:None
Routetest_infer_overhead
pytest.mark.parametrize("copy_outputs", [True, False], ids=["output_dtoh", "no_output_copy"])
tools/Polygraphy/tests/backend/trt/test_runner.py:None
Routetest_insert
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_invalid_error_stat
pytest.mark.parametrize( "args", [ ["not-a-stat"], ["output0:fake"],
tools/Polygraphy/tests/tools/args/comparator/test_compare.py:None
Routetest_out_of_range_index
pytest.mark.skip( reason="When quantization index out of range does not give error but still wraps \ add l
tools/tensorflow-quantization/tests/quantize_wrappers_test.py:None
Routetest_per_output_error_stat
pytest.mark.parametrize( "check_error_stat", [ {"output0": "mean", "output1": "max
tools/Polygraphy/tests/comparator/test_compare.py:None
Routetest_pop
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_pop_index
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_postprocess_network
pytest.mark.parametrize("func_name", ["postprocess", "custom_func"])
tools/Polygraphy/tests/tools/args/backend/trt/test_loader.py:None
Routetest_remove
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_serde
pytest.mark.parametrize( "obj", [ TensorInfo(trt.TensorFormat.LINEAR, trt.float32,
tools/Polygraphy/tests/util/test_serde.py:None
Routetest_setitem
pytest.mark.parametrize("field_names", [("inputs", "outputs"), ("outputs", "inputs")])
tools/onnx-graphsurgeon/tests/test_ir.py:None
Routetest_str_from_config
pytest.mark.parametrize( "create_config, expected", # NOTE: We set workspace sizes here so we
tools/Polygraphy/tests/backend/trt/test_util.py:None
Routetest_trtexec_export_output_params
pytest.mark.parametrize("trtexec_export_output_params", TRTEXEC_EXPORT_OUTPUT_PARAMS)
tools/polygraphy-extension-trtexec/tests/test_runner_args.py:None
Routetest_version
pytest.mark.parametrize( "ver0, ver1, expected", [ ("1.0.0", "2.0.0", False), ("1.0.0"
tools/Polygraphy/tests/mod/test_util.py:None