提交 d98386d6 编写于 作者: 李寅

Merge branch 'feature_wuch' into 'master'

fix caffe graph converter

See merge request !436
......@@ -429,6 +429,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.
先完成此消息的编辑!
想要评论请 注册