microsoft/onnxruntime-extensions

Public

2cf9bab611e9ad563822dee69c44e23bd017fadc

Find a branch or tag

HTTPS

onnxruntime_extensions/_extensions_pydll.pyi

44lines · modecode

unknown

1	`# Copyright (c) Microsoft Corporation. All rights reserved.`
2	`# Licensed under the MIT License. See License.txt in the project root for`
3	`# license information.`
4	`###############################################################################`
5	`from typing import Callable`
6
7
8	`class PyCustomOpDef:`
9	`undefined: int = ...`
10	`dt_float: int = ...`
11	`dt_uint8: int = ...`
12	`dt_int8: int = ...`
13	`dt_uint16: int = ...`
14	`dt_int16: int = ...`
15	`dt_int32: int = ...`
16	`dt_int64: int = ...`
17	`dt_string: int = ...`
18	`dt_bool: int = ...`
19	`dt_float16: int = ...`
20	`dt_double: int = ...`
21	`dt_uint32: int = ...`
22	`dt_uint64: int = ...`
23	`dt_complex64: int = ...`
24	`dt_complex128: int = ...`
25	`dt_bfloat16: int = ...`
26	`def install_hooker(self, invocation_handler: Callable) -> None:`
27	`...`
28	`...`
29
30
31	`def enable_py_op(enabled: bool) -> bool:`
32	`...`
33
34
35	`def add_custom_op(opdef: PyCustomOpDef) -> None:`
36	`...`
37
38
39	`def hash_64(s: str, num_buckets: int, fast: int) -> int:`
40	`...`
41
42
43	`def default_opset_domain() -> str:`
44	`...`
45