提交 4525fc54 编写于 作者: W wuchenghui

fix caffe graph converter

上级 32fa5711
......@@ -423,6 +423,7 @@ class CaffeConverter(object):
if use_winograd:
self.convert_winograd_conv_filter_cpu(op, op_def)
elif self.device == 'gpu':
self.add_tensor(weight_tensor_name, weight_data)
buffer_type = "DW_CONV2D_FILTER" \
if is_depthwise else "CONV2D_FILTER"
output_name = self.add_buffer_to_image(weight_tensor_name,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册