forked from ccf-ai-infra/Intro-ops
19 lines
554 B
Python
19 lines
554 B
Python
from .backend import Backend, normalize_backend
|
|
from .ops import (
|
|
copy, copy_, prepare_copy,
|
|
vector_add, vector_add_, prepare_vector_add,
|
|
reduce_sum, reduce_sum_, prepare_reduce_sum,
|
|
relu, relu_, prepare_relu,
|
|
softmax, softmax_, prepare_softmax,
|
|
)
|
|
|
|
__all__ = [
|
|
"Backend",
|
|
"normalize_backend",
|
|
"copy", "copy_", "prepare_copy",
|
|
"vector_add", "vector_add_", "prepare_vector_add",
|
|
"reduce_sum", "reduce_sum_", "prepare_reduce_sum",
|
|
"relu", "relu_", "prepare_relu",
|
|
"softmax", "softmax_", "prepare_softmax",
|
|
]
|