未验证 提交 31a8b805 编写于 作者: W whs 提交者: GitHub

Add walkers for some activations (#28)

上级 5312f46e
......@@ -271,12 +271,24 @@ class bilinear_interp(activation):
super(bilinear_interp, self).__init__(op, pruned_params, visited)
@PRUNE_WORKER.register
class nearest_interp(activation):
def __init__(self, op, pruned_params, visited):
super(nearest_interp, self).__init__(op, pruned_params, visited)
@PRUNE_WORKER.register
class relu(activation):
def __init__(self, op, pruned_params, visited):
super(relu, self).__init__(op, pruned_params, visited)
@PRUNE_WORKER.register
class leaky_relu(activation):
def __init__(self, op, pruned_params, visited):
super(leaky_relu, self).__init__(op, pruned_params, visited)
@PRUNE_WORKER.register
class floor(activation):
def __init__(self, op, pruned_params, visited):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册