__init__.py
 1  # ONNX Runtime Custom Ops Python bindings
 2  from .custom_ops import (
 3      assert_custom_op_is_active,
 4      create_gelu_test_model,
 5      create_session,
 6      get_library_path,
 7      get_registered_library_path,
 8      register_custom_ops,
 9      run_inference,
10  )
11  
12  __all__ = [
13      "register_custom_ops",
14      "create_gelu_test_model",
15      "create_session",
16      "assert_custom_op_is_active",
17      "get_library_path",
18      "get_registered_library_path",
19      "run_inference",
20  ]