|
|
|
@@ -21,7 +21,7 @@ Examples: |
|
|
|
>>> import mindspore.ops as ops |
|
|
|
""" |
|
|
|
|
|
|
|
from .primitive import Primitive, PrimitiveWithInfer, prim_attr_register |
|
|
|
from .primitive import Primitive, PrimitiveWithInfer, PrimitiveWithCheck, prim_attr_register |
|
|
|
from .vm_impl_registry import get_vm_impl_fn, vm_impl_registry |
|
|
|
from .op_info_register import op_info_register, AkgGpuRegOp, AkgAscendRegOp, AiCPURegOp, TBERegOp, DataType |
|
|
|
from .primitive import constexpr |
|
|
|
@@ -32,7 +32,7 @@ from .operations import * |
|
|
|
from .functional import * |
|
|
|
|
|
|
|
__primitive__ = [ |
|
|
|
"prim_attr_register", "Primitive", "PrimitiveWithInfer", "signature" |
|
|
|
"prim_attr_register", "Primitive", "PrimitiveWithInfer", "PrimitiveWithCheck", "signature" |
|
|
|
] |
|
|
|
|
|
|
|
__all__ = ["get_vm_impl_fn", "vm_impl_registry", |
|
|
|
|