ops.py 348 字节
Newer Older
Y
Yu Yang 已提交
1 2 3 4
from ..registry import register_layer
__all__ = [
    'mean', 'mul', 'dropout', 'reshape', 'sigmoid', 'scale', 'transpose',
    'sigmoid_cross_entropy_with_logits', 'elementwise_add', 'elementwise_div',
Y
yangyaming 已提交
5
    'elementwise_sub', 'elementwise_mul', 'clip', 'abs', 'sequence_softmax'
Y
Yu Yang 已提交
6 7 8 9
]

for _OP in set(__all__):
    globals()[_OP] = register_layer(_OP)