Index

A

assertAlmostEqual() (onnx_diagnostic.ext_test_case.ExtTestCase method) assertEmpty() (onnx_diagnostic.ext_test_case.ExtTestCase method) assertEqual() (onnx_diagnostic.ext_test_case.ExtTestCase method) assertEqualArray() (onnx_diagnostic.ext_test_case.ExtTestCase method) assertEqualArrays() (onnx_diagnostic.ext_test_case.ExtTestCase method) assertExists() (onnx_diagnostic.ext_test_case.ExtTestCase method)

assertGreaterOrEqual() (onnx_diagnostic.ext_test_case.ExtTestCase method) assertIn() (onnx_diagnostic.ext_test_case.ExtTestCase method) assertNotEmpty() (onnx_diagnostic.ext_test_case.ExtTestCase method) assertRaise() (onnx_diagnostic.ext_test_case.ExtTestCase method) assertSetContained() (onnx_diagnostic.ext_test_case.ExtTestCase method) assertStartsWith() (onnx_diagnostic.ext_test_case.ExtTestCase method)

C

capture() (onnx_diagnostic.ext_test_case.ExtTestCase method) check_cuda_availability() (in module onnx_diagnostic.args)

check_model_ort() (in module onnx_diagnostic.torch_test_helper) convert_endian() (in module onnx_diagnostic.helpers)

D

dtype_to_tensor_dtype() (in module onnx_diagnostic.helpers) dummy_llm() (in module onnx_diagnostic.torch_test_helper)

dump_dort_onnx() (in module onnx_diagnostic.ext_test_case) dump_onnx() (onnx_diagnostic.ext_test_case.ExtTestCase method)

E

ExtTestCase (class in onnx_diagnostic.ext_test_case)

F

flatten_object() (in module onnx_diagnostic.helpers)

from_array_extended() (in module onnx_diagnostic.helpers) from_array_ml_dtypes() (in module onnx_diagnostic.helpers)

G

get_dump_file() (onnx_diagnostic.ext_test_case.ExtTestCase method) get_figure() (in module onnx_diagnostic.ext_test_case)

get_onnx_signature() (in module onnx_diagnostic.helpers) get_parsed_args() (in module onnx_diagnostic.args) get_tiny_llm() (in module onnx_diagnostic.torch_models.llms)

H

has_cuda() (in module onnx_diagnostic.ext_test_case) has_executorch() (in module onnx_diagnostic.ext_test_case) has_onnxruntime_training() (in module onnx_diagnostic.ext_test_case)

has_onnxscript() (in module onnx_diagnostic.ext_test_case) has_torch() (in module onnx_diagnostic.ext_test_case) has_transformers() (in module onnx_diagnostic.ext_test_case) hide_stdout() (in module onnx_diagnostic.ext_test_case)

I

ignore_warnings() (in module onnx_diagnostic.ext_test_case) InferenceSessionForNumpy (class in onnx_diagnostic.ort_session)

InferenceSessionForTorch (class in onnx_diagnostic.ort_session) investigate_onnxruntime_issue() (in module onnx_diagnostic.ort_session) is_azure() (in module onnx_diagnostic.ext_test_case)

L

long_test() (in module onnx_diagnostic.ext_test_case)

M

make_dynamic_cache() (in module onnx_diagnostic.cache_helpers) make_hash() (in module onnx_diagnostic.helpers) max_diff() (in module onnx_diagnostic.helpers) measure_time() (in module onnx_diagnostic.ext_test_case) module onnx_diagnostic onnx_diagnostic.args onnx_diagnostic.cache_helpers onnx_diagnostic.ext_test_case onnx_diagnostic.helpers onnx_diagnostic.onnx_tools onnx_diagnostic.ort_session onnx_diagnostic.torch_export_patches onnx_diagnostic.torch_models onnx_diagnostic.torch_models.llms onnx_diagnostic.torch_test_helper

N

never_test() (in module onnx_diagnostic.ext_test_case)

np_dtype_to_tensor_dtype() (in module onnx_diagnostic.helpers)

O

onnx_diagnostic module onnx_diagnostic.args module onnx_diagnostic.cache_helpers module onnx_diagnostic.ext_test_case module onnx_diagnostic.helpers module onnx_diagnostic.onnx_tools module onnx_diagnostic.ort_session module

onnx_diagnostic.torch_export_patches module onnx_diagnostic.torch_models module onnx_diagnostic.torch_models.llms module onnx_diagnostic.torch_test_helper module onnx_dtype_name() (in module onnx_diagnostic.helpers) onnx_dtype_to_torch_dtype() (in module onnx_diagnostic.helpers) onnx_find() (in module onnx_diagnostic.onnx_tools) onnx_lighten() (in module onnx_diagnostic.onnx_tools) onnx_unlighten() (in module onnx_diagnostic.onnx_tools)

P

pretty_onnx() (in module onnx_diagnostic.helpers)

print_model() (onnx_diagnostic.ext_test_case.ExtTestCase method) print_onnx() (onnx_diagnostic.ext_test_case.ExtTestCase method)

R

rename_dynamic_dimensions() (in module onnx_diagnostic.helpers) rename_dynamic_expression() (in module onnx_diagnostic.helpers) requires_cuda() (in module onnx_diagnostic.ext_test_case) requires_diffusers() (in module onnx_diagnostic.ext_test_case) requires_executorch() (in module onnx_diagnostic.ext_test_case) requires_monai() (in module onnx_diagnostic.ext_test_case) requires_numpy() (in module onnx_diagnostic.ext_test_case) requires_onnx() (in module onnx_diagnostic.ext_test_case) requires_onnx_array_api() (in module onnx_diagnostic.ext_test_case) requires_onnxruntime() (in module onnx_diagnostic.ext_test_case) requires_onnxruntime_training() (in module onnx_diagnostic.ext_test_case)

requires_onnxscript() (in module onnx_diagnostic.ext_test_case) requires_pyinstrument() (in module onnx_diagnostic.ext_test_case) requires_sklearn() (in module onnx_diagnostic.ext_test_case) requires_torch() (in module onnx_diagnostic.ext_test_case) requires_transformers() (in module onnx_diagnostic.ext_test_case) requires_vocos() (in module onnx_diagnostic.ext_test_case) requires_zoo() (in module onnx_diagnostic.ext_test_case) run() (onnx_diagnostic.ort_session.InferenceSessionForNumpy method) (onnx_diagnostic.ort_session.InferenceSessionForTorch method) run_dlpack() (onnx_diagnostic.ort_session.InferenceSessionForTorch method) run_training_api() (onnx_diagnostic.ort_session.InferenceSessionForTorch method)

S

setUpClass() (onnx_diagnostic.ext_test_case.ExtTestCase class method) size_type() (in module onnx_diagnostic.helpers) skipif_ci_apple() (in module onnx_diagnostic.ext_test_case) skipif_ci_linux() (in module onnx_diagnostic.ext_test_case) skipif_ci_windows() (in module onnx_diagnostic.ext_test_case) skipif_not_onnxrt() (in module onnx_diagnostic.ext_test_case)

skipif_transformers() (in module onnx_diagnostic.ext_test_case) statistics_on_file() (in module onnx_diagnostic.ext_test_case) statistics_on_folder() (in module onnx_diagnostic.ext_test_case) string_diff() (in module onnx_diagnostic.helpers) string_sig() (in module onnx_diagnostic.helpers) string_signature() (in module onnx_diagnostic.helpers) string_type() (in module onnx_diagnostic.helpers)

T

tearDownClass() (onnx_diagnostic.ext_test_case.ExtTestCase class method) tensor_dtype_to_np_dtype() (in module onnx_diagnostic.helpers) to_numpy() (in module onnx_diagnostic.torch_test_helper)

todo() (onnx_diagnostic.ext_test_case.ExtTestCase class method) torch_dtype_to_onnx_dtype() (in module onnx_diagnostic.helpers) tryCall() (onnx_diagnostic.ext_test_case.ExtTestCase method) type_info() (in module onnx_diagnostic.helpers)

U

unit_test_going() (in module onnx_diagnostic.ext_test_case)

W

with_path_append() (in module onnx_diagnostic.ext_test_case)