forked from ccf-ai-infra/Intro-ops
36 lines
729 B
Python
36 lines
729 B
Python
from .loader import load_library
|
|
from .tensor_view import TensorView, tensor_view, dtype_to_oprt, current_stream_ptr, OPRT_MAX_DIMS
|
|
from .bindings import (
|
|
CFunctions,
|
|
Descriptor,
|
|
OperatorRuntimeError,
|
|
Status,
|
|
bind_elementwise,
|
|
bind_unary,
|
|
bind_relu,
|
|
bind_binary,
|
|
bind_reduce_like,
|
|
check_status,
|
|
)
|
|
from .prepared import PreparedOp
|
|
|
|
__all__ = [
|
|
"load_library",
|
|
"TensorView",
|
|
"tensor_view",
|
|
"dtype_to_oprt",
|
|
"current_stream_ptr",
|
|
"OPRT_MAX_DIMS",
|
|
"CFunctions",
|
|
"Descriptor",
|
|
"OperatorRuntimeError",
|
|
"Status",
|
|
"bind_elementwise",
|
|
"bind_unary",
|
|
"bind_relu",
|
|
"bind_binary",
|
|
"bind_reduce_like",
|
|
"check_status",
|
|
"PreparedOp",
|
|
]
|