|
|
@ -27,7 +27,7 @@ from .multitype_ops.add_impl import hyper_add
|
|
|
|
from .multitype_ops.ones_like_impl import ones_like
|
|
|
|
from .multitype_ops.ones_like_impl import ones_like
|
|
|
|
from .multitype_ops.zeros_like_impl import zeros_like
|
|
|
|
from .multitype_ops.zeros_like_impl import zeros_like
|
|
|
|
from .random_ops import normal, laplace, uniform, gamma, poisson, multinomial
|
|
|
|
from .random_ops import normal, laplace, uniform, gamma, poisson, multinomial
|
|
|
|
from .math_ops import count_nonzero, tensor_dot, batch_dot
|
|
|
|
from .math_ops import count_nonzero, tensor_dot, dot, batch_dot
|
|
|
|
from .array_ops import repeat_elements, sequence_mask
|
|
|
|
from .array_ops import repeat_elements, sequence_mask
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -53,6 +53,7 @@ __all__ = [
|
|
|
|
'clip_by_global_norm',
|
|
|
|
'clip_by_global_norm',
|
|
|
|
'count_nonzero',
|
|
|
|
'count_nonzero',
|
|
|
|
'tensor_dot',
|
|
|
|
'tensor_dot',
|
|
|
|
|
|
|
|
'dot',
|
|
|
|
'batch_dot',
|
|
|
|
'batch_dot',
|
|
|
|
'repeat_elements',
|
|
|
|
'repeat_elements',
|
|
|
|
'sequence_mask']
|
|
|
|
'sequence_mask']
|
|
|
|