提交 476de095 编写于 作者: Y yejianwu

fix slice converter for caffe

上级 e62ea9b1
......@@ -561,7 +561,7 @@ class CaffeConverter(base_converter.ConverterInterface):
def convert_slice(self, caffe_op):
op = self.convert_general_op(caffe_op)
op.type = MaceOp.Slice.name
op.type = MaceOp.Split.name
if caffe_op.layer.HasField('slice_param'):
param = caffe_op.layer.slice_param
......
......@@ -40,7 +40,7 @@ class ShapeInference(object):
MaceOp.Activation.name: self.infer_shape_general,
MaceOp.Pooling.name: self.infer_shape_conv_pool_shape,
MaceOp.Concat.name: self.infer_shape_concat,
MaceOp.Slice.name: self.infer_shape_slice,
MaceOp.Split.name: self.infer_shape_slice,
MaceOp.Softmax.name: self.infer_shape_general,
MaceOp.FullyConnected.name: self.infer_shape_fully_connected,
MaceOp.Crop.name: self.infer_shape_crop,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册