提交 953e4150 编写于 作者: S SunAhong1993

add DepthwiseConvolution

上级 36ae71a4
...@@ -39,6 +39,8 @@ class CaffeOpMapper(OpMapper): ...@@ -39,6 +39,8 @@ class CaffeOpMapper(OpMapper):
print("Total nodes: {}".format(len(self.graph.topo_sort))) print("Total nodes: {}".format(len(self.graph.topo_sort)))
for node_name in self.graph.topo_sort: for node_name in self.graph.topo_sort:
node = self.graph.get_node(node_name) node = self.graph.get_node(node_name)
if node.layer_type == 'DepthwiseConvolution':
node.layer_type = 'ConvolutionDepthwise'
op = node.layer_type op = node.layer_type
if hasattr(self, op): if hasattr(self, op):
self.set_node_shape(node) self.set_node_shape(node)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册