microsoft/onnxruntime-extensions

Public

1f0c76cefaa9359b9fc74ff4b9ed93214486205a

Find a branch or tag

HTTPS

onnxruntime_extensions/_ortcustomops.pyi

40lines · 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
6	`class PyCustomOpDef:`
7	`undefined: int = ...`
8	`dt_float: int = ...`
9	`dt_uint8: int = ...`
10	`dt_int8: int = ...`
11	`dt_uint16: int = ...`
12	`dt_int16: int = ...`
13	`dt_int32: int = ...`
14	`dt_int64: int = ...`
15	`dt_string: int = ...`
16	`dt_bool: int = ...`
17	`dt_float16: int = ...`
18	`dt_double: int = ...`
19	`dt_uint32: int = ...`
20	`dt_uint64: int = ...`
21	`dt_complex64: int = ...`
22	`dt_complex128: int = ...`
23	`dt_bfloat16: int = ...`
24	`...`
25
26
27	`def enable_py_op(enabled: bool) -> bool:`
28	`...`
29
30
31	`def add_custom_op(opdef: PyCustomOpDef) -> None:`
32	`...`
33
34
35	`def hash_64(s: str, num_buckets: int, fast: int) -> int:`
36	`...`
37
38
39	`def default_opset_domain() -> str:`
40	`...`