microsoft/onnxruntime-extensions

Public

f74770feed077546874ed7e66d1aba9e2509fea9

Find a branch or tag

HTTPS

onnxruntime_extensions/_ortcustomops.pyi

36lines · 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	`def enable_custom_op(enabled: bool) -> bool:`
27	`...`
28
29	`def add_custom_op(opdef: PyCustomOpDef) -> None:`
30	`...`
31
32	`def hash_64(s: str, num_buckets: int, fast: int) -> int:`
33	`...`
34
35	`def default_opset_domain() -> str:`
36	`...`
37