test_examples Namespace Reference

Classes
class	Artifact
class	Example

Functions
def	load_commands_from_readme (readme)
def	infer_model (path)
def	test_examples (example_dir, artifacts)
def	load_code_blocks_from_readme (readme)
def	test_api_examples (example)
def	test_cli_examples (example)
def	test_cli_inspect_examples (example)

Variables
	ROOT_DIR = os.path.realpath(os.path.join(os.path.dirname(__file__), os.path.pardir))
	EXAMPLES_ROOT = os.path.join(ROOT_DIR, "examples")
list	EXAMPLES
list	API_EXAMPLES
	reason
	ids
list	CLI_EXAMPLES
list	CLI_INSPECT_EXAMPLES

Function Documentation

load_commands_from_readme()

def test_examples.load_commands_from_readme

(

readme

)

Here is the caller graph for this function:

infer_model()

def test_examples.infer_model

(

path

)

Here is the caller graph for this function:

test_examples()

def test_examples.test_examples	(		example_dir,
			artifacts
	)

Here is the call graph for this function:

load_code_blocks_from_readme()

def test_examples.load_code_blocks_from_readme

(

readme

)

Here is the caller graph for this function:

test_api_examples()

def test_examples.test_api_examples

(

example

)

test_cli_examples()

def test_examples.test_cli_examples

(

example

)

test_cli_inspect_examples()

def test_examples.test_cli_inspect_examples

(

example

)

Variable Documentation

ROOT_DIR

test_examples.ROOT_DIR = os.path.realpath(os.path.join(os.path.dirname(__file__), os.path.pardir))

EXAMPLES_ROOT

test_examples.EXAMPLES_ROOT = os.path.join(ROOT_DIR, "examples")

EXAMPLES

list test_examples.EXAMPLES

Initial value:

=  [
    ("01_creating_a_model", [Artifact("test_globallppool.onnx")]),
    ("02_creating_a_model_with_initializer", [Artifact("test_conv.onnx")]),
    ("03_isolating_a_subgraph", [Artifact("model.onnx"), Artifact("subgraph.onnx")]),
    ("04_modifying_a_model", [Artifact("model.onnx"), Artifact("modified.onnx")]),
    ("05_folding_constants", [Artifact("model.onnx"), Artifact("folded.onnx")]),
    ("06_removing_nodes", [Artifact("model.onnx", infer=False), Artifact("removed.onnx")]),
    ("07_creating_a_model_with_the_layer_api", [Artifact("model.onnx")]),
    ("08_replacing_a_subgraph", [Artifact("model.onnx"), Artifact("replaced.onnx")]),
    ("09_shape_operations_with_the_layer_api", [Artifact("model.onnx")]),
]

API_EXAMPLES

test_examples.API_EXAMPLES

Initial value:

=  [
    Example(["api", "01_comparing_frameworks"]),
    Example(["api", "02_using_real_data"]),
    Example(["api", "03_interoperating_with_tensorrt"]),
    Example(["api", "04_int8_calibration_in_tensorrt"], artifact_names=["identity-calib.cache"]),
    Example(["api", "05_using_tensorrt_network_api"]),
]

reason

test_examples.reason

ids

test_examples.ids

CLI_EXAMPLES

test_examples.CLI_EXAMPLES

Initial value:

=  [
    # Run
    Example(["cli", "run", "01_comparing_frameworks"]),
    Example(["cli", "run", "02_comparing_across_runs"], artifact_names=["system_a_results.pkl"]),
    Example(["cli", "run", "03_generating_a_comparison_script"], artifact_names=["compare_trt_onnxrt.py"]),
    # Surgeon
    Example(["cli", "surgeon", "01_isolating_subgraphs"], artifact_names=["subgraph.onnx"]),
]

CLI_INSPECT_EXAMPLES

test_examples.CLI_INSPECT_EXAMPLES

Initial value:

=  [
    Example(["cli", "inspect", "01_inspecting_a_tensorrt_network"]),
    Example(["cli", "inspect", "02_inspecting_a_tensorrt_engine"], artifact_names=["dynamic_identity.engine"]),
    Example(["cli", "inspect", "03_inspecting_an_onnx_model"]),
    Example(["cli", "inspect", "04_inspecting_a_tensorflow_graph"]),
    Example(["cli", "inspect", "05_inspecting_run_results"], artifact_names=["onnxrt_results.pkl"]),
    Example(["cli", "inspect", "06_inspecting_input_data"], artifact_names=["inputs.pkl"]),
]