diff --git a/mace/python/tools/converter_tool/caffe_converter.py b/mace/python/tools/converter_tool/caffe_converter.py index 5855725d8a552aec4bd5d0de3a9955ccb4743d07..d9cc35aafbaed1295c025d8fad024e283430741a 100644 --- a/mace/python/tools/converter_tool/caffe_converter.py +++ b/mace/python/tools/converter_tool/caffe_converter.py @@ -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 diff --git a/mace/python/tools/converter_tool/shape_inference.py b/mace/python/tools/converter_tool/shape_inference.py index 3861d64884d209adc12086e9455d9eb61a6d36d1..463a511775f91a99404fa17825c4c7ca9ccac0ab 100644 --- a/mace/python/tools/converter_tool/shape_inference.py +++ b/mace/python/tools/converter_tool/shape_inference.py @@ -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,